Update modelLoanAPI.py

modelLoanAPI.py

@@ -1,10 +1,10 @@
+
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import JSONResponse, FileResponse
 import pandas as pd
 import numpy as np
 from sklearn.ensemble import RandomForestClassifier, GradientBoostingRegressor
 from sklearn.preprocessing import RobustScaler, LabelEncoder
-from sklearn.metrics import accuracy_score
 from sklearn.model_selection import TimeSeriesSplit, GridSearchCV
 import matplotlib.pyplot as plt
 import json
@@ -36,7 +36,7 @@ async def worker_forecast(input_data: WorkerInput):
 
     # Load dataset
     try:
-        df = pd.read_csv('extended_worker_dataset_random_reduced.csv')
+        df = pd.read_csv('/app/extended_worker_dataset_random_reduced.csv')
     except FileNotFoundError:
         raise HTTPException(status_code=500, detail="Dataset file not found")
 
@@ -213,10 +213,13 @@ async def worker_forecast(input_data: WorkerInput):
     plot_base64 = base64.b64encode(buf_jpg.getvalue()).decode("utf-8")
     results['plot'] = f"data:image/jpeg;base64,{plot_base64}"
 
-    # Save plot to file
-    plot_filename = f"worker_{worker_id}_forecast.jpg"
-    with open(plot_filename, "wb") as f:
-        f.write(base64.b64decode(plot_base64))
+    # Save plot to file in /tmp
+    plot_filename = f"/tmp/worker_{worker_id}_forecast.jpg"
+    try:
+        with open(plot_filename, "wb") as f:
+            f.write(base64.b64decode(plot_base64))
+    except PermissionError as e:
+        raise HTTPException(status_code=500, detail=f"Failed to write plot file: {str(e)}")
 
     # Worker profile
     worker_data = df.copy()
@@ -247,8 +250,8 @@ async def worker_forecast(input_data: WorkerInput):
 
 @app.get("/worker_forecast/plot/{worker_id}")
 async def get_forecast_plot(worker_id: int):
-    plot_filename = f"worker_{worker_id}_forecast.jpg"
+    plot_filename = f"/tmp/worker_{worker_id}_forecast.jpg"
     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")