Upload main.py

main.py

@@ -478,12 +478,24 @@ async def modelfit(
 
 @app.post("/modelforecast/")
 async def modelforecast(
-    df_forecast: List[Dict] = Body(...),
+    payload: Any = Body(...),
 ):
     global stored_model, stored_result_join
     if stored_model is None or stored_result_join is None:
         raise HTTPException(status_code=400, detail="Model not fitted. Call /modelfit/ first.")
     try:
+        # Accept both payload styles:
+        # 1) raw list: [...]
+        # 2) wrapped dict: {"df_forecast": [...]}
+        if isinstance(payload, dict) and "df_forecast" in payload:
+            df_forecast = payload.get("df_forecast")
+        else:
+            df_forecast = payload
+        if not isinstance(df_forecast, list):
+            raise HTTPException(
+                status_code=422,
+                detail="Payload must be a list of rows or {'df_forecast': [rows]}",
+            )
         df_forecast1 = pd.DataFrame(df_forecast)
         forecast_df = stored_model.forecast(extern_self=stored_result_join, df_forecast=df_forecast1)
         return forecast_df.to_dict(orient="records")