Update app/fastapi_server.py

app/fastapi_server.py

@@ -168,7 +168,7 @@ class ModelManager:
             logger.error(f"Prediction failed: {e}")
             raise HTTPException(
                 status_code=500,
-                detail=f"Prediction failed: {str(e)}")
+                detail=f"Prediction failed: {str(e)}"
     # Background task functions
 async def log_prediction(text: str, prediction: str, confidence: float, client_ip: str, processing_time: float):
     """Log prediction details with error handling for file access"""
@@ -236,34 +236,6 @@ async def log_batch_prediction(total_texts: int, successful_predictions: int, cl
         logger.error(f"Failed to log batch prediction: {e}")
 
 
-# Custom OpenAPI
-def custom_openapi():
-    if app.openapi_schema:
-        return app.openapi_schema
-
-    openapi_schema = get_openapi(
-        title="Fake News Detection API",
-        version="2.0.0",
-        description="Production-ready API for fake news detection with comprehensive monitoring and security features",
-        routes=app.routes,
-    )
-
-    # Add security definitions
-    openapi_schema["components"]["securitySchemes"] = {
-        "Bearer": {
-            "type": "http",
-            "scheme": "bearer",
-            "bearerFormat": "JWT",
-        }
-    }
-
-    app.openapi_schema = openapi_schema
-    return app.openapi_schema
-
-
-# Set the custom OpenAPI function AFTER app is created
-app.openapi = custom_openapi
-
 if __name__ == "__main__":
     uvicorn.run(
         "fastapi_server:app",
@@ -357,6 +329,33 @@ app.add_middleware(
     allowed_hosts=["*"]  # Configure appropriately for production
 )
 
+# Custom OpenAPI setup - RIGHT AFTER app creation
+def custom_openapi():
+    if app.openapi_schema:
+        return app.openapi_schema
+
+    openapi_schema = get_openapi(
+        title="Fake News Detection API",
+        version="2.0.0",
+        description="Production-ready API for fake news detection with comprehensive monitoring and security features",
+        routes=app.routes,
+    )
+
+    # Add security definitions
+    openapi_schema["components"]["securitySchemes"] = {
+        "Bearer": {
+            "type": "http",
+            "scheme": "bearer",
+            "bearerFormat": "JWT",
+        }
+    }
+
+    app.openapi_schema = openapi_schema
+    return app.openapi_schema
+
+# Set the custom OpenAPI function
+app.openapi = custom_openapi
+
 # Request/Response models (unchanged)
 class PredictionRequest(BaseModel):
     text: str = Field(..., min_length=1, max_length=10000,
@@ -729,7 +728,6 @@ async def health_check():
             environment_info={"error": str(e)}
         )
 
-
 @app.get("/metrics")
 async def get_metrics():
     """