Upload app.py

app.py

@@ -0,0 +1,151 @@
+"""
+CodeCraftLab — FastAPI entrypoint.
+
+Startup order:
+  1. Logging configured (structlog)
+  2. Settings validated (Pydantic v2)
+  3. Database initialised
+  4. Routers mounted
+  5. Background job scheduler started
+"""
+
+from __future__ import annotations
+
+import contextlib
+from collections.abc import AsyncGenerator
+
+import structlog
+import uvicorn
+from fastapi import Depends, FastAPI, HTTPException, Request, status
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.middleware.trustedhost import TrustedHostMiddleware
+from fastapi.responses import JSONResponse
+
+from core.logging import configure_logging
+from core.settings import Settings, get_settings
+from routers import auth, datasets, inference, training
+
+# ---------------------------------------------------------------------------
+# Logging — configure before anything else imports the logger
+# ---------------------------------------------------------------------------
+configure_logging()
+log = structlog.get_logger(__name__)
+
+
+# ---------------------------------------------------------------------------
+# Lifespan
+# ---------------------------------------------------------------------------
+@contextlib.asynccontextmanager
+async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
+    settings: Settings = get_settings()
+    log.info(
+        "codecraftlab.startup",
+        env=settings.env,
+        max_concurrent_jobs=settings.max_concurrent_jobs,
+        hub_enabled=bool(settings.hf_token),
+    )
+    yield
+    log.info("codecraftlab.shutdown")
+
+
+# ---------------------------------------------------------------------------
+# App factory
+# ---------------------------------------------------------------------------
+def create_app(settings: Settings | None = None) -> FastAPI:
+    cfg = settings or get_settings()
+
+    app = FastAPI(
+        title="CodeCraftLab",
+        description=(
+            "Production fine-tuning platform for code generation models. "
+            "Manage datasets, launch training jobs, evaluate models, and serve inference."
+        ),
+        version="1.0.0",
+        docs_url="/docs" if cfg.env != "production" else None,
+        redoc_url="/redoc" if cfg.env != "production" else None,
+        lifespan=lifespan,
+    )
+
+    # ------------------------------------------------------------------
+    # Middleware
+    # ------------------------------------------------------------------
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=cfg.cors_origins,
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+    if cfg.env == "production":
+        app.add_middleware(TrustedHostMiddleware, allowed_hosts=cfg.allowed_hosts)
+
+    # ------------------------------------------------------------------
+    # Request logging middleware
+    # ------------------------------------------------------------------
+    @app.middleware("http")
+    async def log_requests(request: Request, call_next):  # type: ignore[no-untyped-def]
+        bound_log = log.bind(
+            method=request.method,
+            path=request.url.path,
+            client=request.client.host if request.client else "unknown",
+        )
+        bound_log.info("request.received")
+        response = await call_next(request)
+        bound_log.info("request.completed", status_code=response.status_code)
+        return response
+
+    # ------------------------------------------------------------------
+    # Global exception handler
+    # ------------------------------------------------------------------
+    @app.exception_handler(Exception)
+    async def unhandled_exception_handler(request: Request, exc: Exception) -> JSONResponse:
+        log.error(
+            "unhandled_exception",
+            path=request.url.path,
+            error=str(exc),
+            exc_info=True,
+        )
+        return JSONResponse(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            content={"detail": "Internal server error"},
+        )
+
+    # ------------------------------------------------------------------
+    # Routers
+    # ------------------------------------------------------------------
+    app.include_router(auth.router, prefix="/auth", tags=["Authentication"])
+    app.include_router(datasets.router, prefix="/datasets", tags=["Datasets"])
+    app.include_router(training.router, prefix="/training", tags=["Training"])
+    app.include_router(inference.router, prefix="/inference", tags=["Inference"])
+
+    # ------------------------------------------------------------------
+    # Health endpoints
+    # ------------------------------------------------------------------
+    @app.get("/health", tags=["System"])
+    async def health() -> dict[str, str]:
+        return {"status": "ok"}
+
+    @app.get("/health/ready", tags=["System"])
+    async def readiness(settings: Settings = Depends(get_settings)) -> dict[str, object]:
+        return {
+            "status": "ready",
+            "env": settings.env,
+            "hub_connected": bool(settings.hf_token),
+        }
+
+    return app
+
+
+app = create_app()
+
+
+if __name__ == "__main__":
+    settings = get_settings()
+    uvicorn.run(
+        "app:app",
+        host="0.0.0.0",
+        port=8000,
+        reload=settings.env == "development",
+        log_config=None,  # structlog handles logging
+        workers=1 if settings.env == "development" else 4,
+    )