Update main.py

main.py

@@ -9,31 +9,6 @@ from starlette.middleware.cors import CORSMiddleware
 
 app = FastAPI()
 
-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")
 # Use a pipeline as a high-level helper
@@ -153,3 +128,13 @@ async def perform_document_qa_jsonp(
         return JSONResponse(content=response_data, callback=callback)
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, callback=callback)
+
+# Set up CORS middleware
+origins = ["*"]  # or specify your list of allowed origins
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)