67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
from fastapi import FastAPI, Request
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.responses import FileResponse, JSONResponse
|
|
from fastapi import HTTPException
|
|
|
|
from . import database as db, config
|
|
from .logging_config import setup_logging, get_logger
|
|
from .errors import PipelineError
|
|
from .routers import settings as settings_router
|
|
from .routers import sessions as sessions_router
|
|
from .routers import speakers as speakers_router
|
|
from .routers import notes as notes_router
|
|
from .routers import jobs as jobs_router
|
|
from .routers import models as models_router
|
|
from .routers import diagnostics as diagnostics_router
|
|
|
|
setup_logging()
|
|
log = get_logger(__name__)
|
|
|
|
app = FastAPI(title="Nat20 Notes")
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
|
|
@app.exception_handler(PipelineError)
|
|
async def pipeline_error_handler(request: Request, exc: PipelineError):
|
|
log.error("Pipeline error in %s: %s", exc.stage, exc.message, exc_info=exc.cause or exc)
|
|
return JSONResponse(status_code=502, content={"stage": exc.stage, "message": exc.message})
|
|
|
|
|
|
@app.exception_handler(Exception)
|
|
async def unhandled_error_handler(request: Request, exc: Exception):
|
|
log.exception("Unhandled error on %s %s", request.method, request.url.path)
|
|
return JSONResponse(status_code=500, content={"stage": "unknown", "message": f"Unexpected error: {exc}"})
|
|
|
|
|
|
@app.on_event("startup")
|
|
def on_startup():
|
|
db.init_db()
|
|
log.info("Database initialized at %s", config.DB_PATH)
|
|
|
|
|
|
app.include_router(settings_router.router)
|
|
app.include_router(sessions_router.router)
|
|
app.include_router(speakers_router.router)
|
|
app.include_router(notes_router.router)
|
|
app.include_router(jobs_router.router)
|
|
app.include_router(models_router.router)
|
|
app.include_router(diagnostics_router.router)
|
|
|
|
|
|
@app.get("/api/sessions/{session_id}/audio")
|
|
def get_audio(session_id: str):
|
|
audio_path = config.AUDIO_DIR / f"{session_id}.wav"
|
|
if not audio_path.exists():
|
|
raise HTTPException(404, "Audio not available for this session")
|
|
return FileResponse(audio_path, media_type="audio/wav")
|
|
|
|
|
|
@app.get("/api/health")
|
|
def health():
|
|
return {"ok": True} |