Fix scheduling

main.py

@@ -201,6 +201,27 @@ async def proxy_batch_stage(payload: dict = Body(...)):
         return {"success": False, "message": f"Proxy Error (Internal): {str(e)}", "rows": []}
 
 
+@app.post("/proxy_flush_queue")
+async def proxy_flush_queue(payload: dict = Body(...)):
+    username = str(payload.get("username", ""))
+    if username.strip().lower() != "devtest":
+        return {"success": False, "message": "Queue flush is only available for devtest."}
+    target_url = f"{DEV_URL}/flush_queue"
+    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 requests.exceptions.RequestException as e:
+        return proxy_error_payload("Proxy Error (Hugging Face)", e)
+    except Exception as e:
+        return {"success": False, "message": f"Proxy Error (Internal): {str(e)}"}
+
+
 @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