Update main.py

main.py

@@ -4,23 +4,36 @@ from fastapi.responses import JSONResponse
 from transformers import pipeline
 from PIL import Image
 from io import BytesIO
-from starlette.middleware.cors import CORSMiddleware # Turns out it doesn't matter where you import this from
-from .my_router import router
-
-
-def create_app() -> CORSMiddleware:
-    fastapi_app = FastAPI()
-    fastapi_app.include_router(router)
-    return CORSMiddleware(
-        fastapi_app,
-        allow_origins=["*"],
-        allow_credentials=True,
-        allow_methods=["*"],
-        allow_headers=["*"],
-    )
-
-
-app = create_app()
+from starlette.middleware import Middleware
+from starlette.middleware.cors import CORSMiddleware
+
+app = FastAPI()
+
+# Salt to your taste
+ALLOWED_ORIGINS = [
+    "http://localhost",
+    "http://localhost:8080",
+    "http://127.0.0.1:5500",
+    "http://127.0.0.1:5500/DataExtractAI.html",
+]
+
+# Handle CORS preflight requests
+@app.options('/{rest_of_path:path}')
+async def preflight_handler(request: Request, rest_of_path: str) -> Response:
+    response = Response()
+    response.headers['Access-Control-Allow-Origin'] = ', '.join(ALLOWED_ORIGINS)
+    response.headers['Access-Control-Allow-Methods'] = 'POST, GET, DELETE, OPTIONS'
+    response.headers['Access-Control-Allow-Headers'] = 'Authorization, Content-Type'
+    return response
+
+# Set CORS headers
+@app.middleware("http")
+async def add_CORS_header(request: Request, call_next):
+    response = await call_next(request)
+    response.headers['Access-Control-Allow-Origin'] = ', '.join(ALLOWED_ORIGINS)
+    response.headers['Access-Control-Allow-Methods'] = 'POST, GET, DELETE, OPTIONS'
+    response.headers['Access-Control-Allow-Headers'] = 'Authorization, Content-Type'
+    return response
 
 # Use a pipeline as a high-level helper
 nlp_qa = pipeline("document-question-answering", model="impira/layoutlm-document-qa")