Create main.py

Author: codewithdark-git

app/main.py

@@ -0,0 +1,62 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+from fastapi.middleware.trustedhost import TrustedHostMiddleware
+
+from app.routers import health, prediction
+from app.config import get_settings
+
+settings = get_settings()
+
+app = FastAPI(
+    title=settings.PROJECT_NAME,
+    description="AI-Powered Healthcare Diagnosis System with Multi-Disease Detection and LangChain Agents",
+    version="1.0.0",
+    docs_url="/docs" if settings.ENVIRONMENT != "production" else None,
+    redoc_url="/redoc" if settings.ENVIRONMENT != "production" else None,
+)
+
+# Configure CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=settings.CORS_ORIGINS,
+    allow_credentials=True,
+    allow_methods=["GET", "POST"],
+    allow_headers=["*"],
+)
+
+# Add trusted host middleware
+app.add_middleware(
+    TrustedHostMiddleware,
+    allowed_hosts=settings.ALLOWED_HOSTS,
+)
+
+# Include routers
+app.include_router(health.router, prefix=f"{settings.API_V1_STR}/health", tags=["health"])
+app.include_router(prediction.router, prefix=f"{settings.API_V1_STR}/predict", tags=["prediction"])
+
+@app.get("/", tags=["root"])
+async def root():
+    """
+    Root endpoint returning API information and health status.
+    """
+    return {
+        "status": "healthy",
+        "version": "1.0.0",
+        "environment": settings.ENVIRONMENT,
+        "documentation": "/docs" if settings.ENVIRONMENT != "production" else None
+    }
+
+@app.exception_handler(HTTPException)
+async def http_exception_handler(request, exc):
+    return JSONResponse(
+        status_code=exc.status_code,
+        content={"detail": exc.detail},
+    )
+
+@app.exception_handler(Exception)
+async def general_exception_handler(request, exc):
+    return JSONResponse(
+        status_code=500,
+        content={"detail": "Internal server error"},
+    )