Update routing

main.py

@@ -1,6 +1,7 @@
 import os
 import requests
-from fastapi import FastAPI, UploadFile, File, Form, Query, Request
+from fastapi import FastAPI, UploadFile, File, Form, Query, Request, Body
+from fastapi.responses import Response
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.concurrency import run_in_threadpool # <--- FIX: Added Threadpool
 import uvicorn
@@ -54,7 +55,14 @@ async def proxy_process(request: Request, file: UploadFile = File(...), password
     file_bytes = await file.read()
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
     files = {"file": (file.filename, file_bytes, file.content_type)}
-    data = {"password": password, "username": username}
+    form = await request.form()
+    data = {}
+    for key, value in form.multi_items():
+        if key == "file" or hasattr(value, "filename"):
+            continue
+        data[key] = str(value)
+    data.setdefault("password", password)
+    data.setdefault("username", username)
 
     # FIX: Define the blocking request
     def make_post():
@@ -70,5 +78,58 @@ async def proxy_process(request: Request, file: UploadFile = File(...), password
     except Exception as e:
         return {"success": False, "message": f"Proxy Error (Internal): {str(e)}"}
 
+
+@app.post("/proxy_batch_stage")
+async def proxy_batch_stage(payload: dict = Body(...)):
+    username = str(payload.get("username", ""))
+    base_url = DEV_URL if username.strip().lower() == "devtest" else PROD_URL
+    target_url = f"{base_url}/batch_stage"
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+
+    def make_post():
+        return requests.post(target_url, headers=headers, json=payload, timeout=120)
+
+    try:
+        response = await run_in_threadpool(make_post)
+        response.raise_for_status()
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        return {"success": False, "message": f"Proxy Error (Hugging Face): {str(e)}", "rows": []}
+    except Exception as e:
+        return {"success": False, "message": f"Proxy Error (Internal): {str(e)}", "rows": []}
+
+
+@app.get("/proxy_preview/{session_id}/{row_id}/{preview_type}")
+async def proxy_preview(session_id: str, row_id: str, preview_type: str, username: str = Query(""), size: str = Query("thumb")):
+    base_url = DEV_URL if username.strip().lower() == "devtest" else PROD_URL
+    target_url = f"{base_url}/preview/{session_id}/{row_id}/{preview_type}"
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+
+    def make_get():
+        return requests.get(target_url, headers=headers, params={"size": size}, timeout=30)
+
+    response = await run_in_threadpool(make_get)
+    if response.status_code >= 400:
+        return Response(content=response.content, status_code=response.status_code, media_type=response.headers.get("content-type", "text/plain"))
+    return Response(content=response.content, media_type=response.headers.get("content-type", "image/jpeg"))
+
+
+@app.post("/proxy_preview_session_clear")
+async def proxy_preview_session_clear(payload: dict = Body(...)):
+    username = str(payload.get("username", ""))
+    base_url = DEV_URL if username.strip().lower() == "devtest" else PROD_URL
+    target_url = f"{base_url}/preview_session/clear"
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+
+    def make_post():
+        return requests.post(target_url, headers=headers, json=payload, timeout=20)
+
+    try:
+        response = await run_in_threadpool(make_post)
+        response.raise_for_status()
+        return response.json()
+    except Exception as e:
+        return {"success": False, "message": f"Proxy Error: {str(e)}"}
+
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 7860)))
+    uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 7860)))