Update modelLoanAPI.py

modelLoanAPI.py

@@ -1,6 +1,6 @@
-
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import JSONResponse, FileResponse
+from fastapi.middleware.cors import CORSMiddleware  # Import CORSMiddleware
 import pandas as pd
 import numpy as np
 from sklearn.ensemble import RandomForestClassifier, GradientBoostingRegressor
@@ -19,6 +19,15 @@ warnings.filterwarnings("ignore")
 
 app = FastAPI()
 
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins; replace with specific origins in production
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods (GET, POST, etc.)
+    allow_headers=["*"],  # Allows all headers
+)
+
 class WorkerInput(BaseModel):
     worker_id: int
 
@@ -254,4 +263,4 @@ async def get_forecast_plot(worker_id: int):
     if os.path.exists(plot_filename):
         return FileResponse(plot_filename, media_type="image/jpeg", filename=f"worker_{worker_id}_forecast.jpg")
     else:
-        raise HTTPException(status_code=404, detail=f"Plot for worker_id {worker_id} not found")
+        raise HTTPException(status_code=404, detail=f"Plot for worker_id {worker_id} not found")