Update api.py

api.py

@@ -20,6 +20,7 @@ from fastapi.security.api_key import APIKeyHeader, APIKey
 from fastapi.staticfiles import StaticFiles
 from fastapi.openapi.docs import get_swagger_ui_html
 from fastapi.responses import HTMLResponse
+from fastapi.openapi.utils import get_openapi
 
 # Load environment variables
 load_dotenv()
@@ -321,15 +322,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Add security scheme
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
 # Add security scheme
 app.add_security_requirement({"ApiKeyAuth": []})
 app.openapi_schema = None  # Reset OpenAPI schema
@@ -554,4 +546,36 @@ async def custom_swagger_ui_html():
             }
         </style>
         """
-    )
+    )
+
+def custom_openapi():
+    if app.openapi_schema:
+        return app.openapi_schema
+        
+    openapi_schema = get_openapi(
+        title=app.title,
+        version=app.version,
+        description=app.description,
+        routes=app.routes,
+    )
+    
+    # Add security scheme to components
+    openapi_schema["components"] = {
+        "securitySchemes": {
+            "ApiKeyAuth": {
+                "type": "apiKey",
+                "in": "header",
+                "name": "X-API-Key",
+                "description": "API key required for authentication"
+            }
+        }
+    }
+    
+    # Add global security requirement
+    openapi_schema["security"] = [{"ApiKeyAuth": []}]
+    
+    app.openapi_schema = openapi_schema
+    return app.openapi_schema
+
+# Set the custom OpenAPI schema
+app.openapi = custom_openapi