Added fit/forecast file individual

main.py

@@ -161,6 +161,27 @@ async def _forward_stream(path: str, files=None, data=None, user_token: str | No
     return response
 
 
+async def _forward_multipart_json(path: str, files=None, data=None, user_token: str | None = None, method: str = "POST"):
+    """POST multipart (files + data) to upstream and return JSON response."""
+    url = f"{UPSTREAM_URL}{path}"
+    headers = {
+        "Authorization": f"Bearer {HF_TOKEN}",
+        "Accept": "application/json",
+    }
+    if user_token:
+        headers["X-API-Key"] = user_token
+    timeout = httpx.Timeout(600.0)
+    async with httpx.AsyncClient(timeout=timeout) as client:
+        r = await client.request(method, url, headers=headers, files=files, data=data)
+    ct = r.headers.get("content-type", "")
+    if "application/json" in ct:
+        try:
+            return JSONResponse(status_code=r.status_code, content=r.json())
+        except Exception:
+            return JSONResponse(status_code=r.status_code, content={"error": r.text[:500]})
+    return JSONResponse(status_code=r.status_code, content={"text": r.text[:1000]})
+
+
 # for demo
 async def _forward_demo_stream(path: str, *, files: dict | None, data: dict | None, method: str = "POST"):
     if not DEMO_FORWARD_URL or not HF_TOKEN or not DATFID_DEMO_TOKEN:
@@ -373,4 +394,58 @@ async def modelforecast_ind(req: Request):
     if not user_token:
         raise HTTPException(status_code=401, detail="Missing Authorization Bearer token (dt+...)")
     body = await req.json()
-    return await _forward("/modelforecast_ind/", "POST", json_body=body, user_token=user_token)
+    return await _forward("/modelforecast_ind/", "POST", json_body=body, user_token=user_token)
+
+
+@app.post("/modelfit-file_ind/")
+async def modelfit_file_ind(
+    req: Request,
+    file: UploadFile = File(...),
+    id_col: str = Form(...),
+    time_col: str = Form(...),
+    y: str = Form(...),
+    lag_y: str = Form(""),
+    lagged_features: str = Form(""),
+    current_features: str = Form(""),
+    filter_by_significance: str = Form("false"),
+    meanvar_test: str = Form("false"),
+    signif: str = Form("0.05"),
+):
+    user_token = _extract_user_token(req)
+    if not user_token:
+        raise HTTPException(status_code=401, detail="Missing Authorization Bearer token (dt+...)")
+    raw = await file.read()
+    if len(raw) > SDK_MAX_BODY_BYTES_extended:
+        raise HTTPException(status_code=413, detail="Payload too large.")
+    files = {
+        "file": (file.filename, raw, file.content_type or "application/octet-stream"),
+    }
+    data = {
+        "id_col": id_col,
+        "time_col": time_col,
+        "y": y,
+        "lag_y": lag_y,
+        "lagged_features": lagged_features,
+        "current_features": current_features,
+        "filter_by_significance": filter_by_significance,
+        "meanvar_test": meanvar_test,
+        "signif": signif,
+    }
+    return await _forward_multipart_json("/modelfit-file_ind/", files=files, data=data, user_token=user_token, method="POST")
+
+
+@app.post("/modelforecast-file_ind/")
+async def modelforecast_file_ind(
+    req: Request,
+    df_forecast: UploadFile = File(...),
+):
+    user_token = _extract_user_token(req)
+    if not user_token:
+        raise HTTPException(status_code=401, detail="Missing Authorization Bearer token (dt+...)")
+    raw = await df_forecast.read()
+    if len(raw) > SDK_MAX_BODY_BYTES_extended:
+        raise HTTPException(status_code=413, detail="Payload too large.")
+    files = {
+        "df_forecast": (df_forecast.filename, raw, df_forecast.content_type or "application/octet-stream"),
+    }
+    return await _forward_stream("/modelforecast-file_ind/", files=files, data=None, user_token=user_token, method="POST")