Create main.py

main.py

@@ -0,0 +1,74 @@
+import os
+import requests
+from fastapi import FastAPI, UploadFile, File, Form, Query, Request
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.concurrency import run_in_threadpool # <--- FIX: Added Threadpool
+import uvicorn
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+DEV_URL = "https://PPAL-SongLab-UGA-fruit-analyzer-dev.hf.space"
+PROD_URL = "https://PPAL-SongLab-UGA-fruit-analyzer.hf.space"
+
+@app.get("/")
+def read_root():
+    return {"status": "Render Proxy is awake!"}
+
+@app.get("/proxy_status")
+async def proxy_status(username: str = Query("")):
+    base_url = DEV_URL if username.strip().lower() == "devtest" else PROD_URL
+    status_url = f"{base_url}/queue_status"
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    
+    # FIX: Define the blocking request
+    def fetch_status():
+        return requests.get(status_url, headers=headers, timeout=5)
+        
+    try:
+        # FIX: Run it in a background thread so the server doesn't freeze
+        response = await run_in_threadpool(fetch_status)
+        response.raise_for_status()
+        return response.json()
+    except Exception as e:
+        return {"active_requests": 0, "max_concurrent": 2}
+
+
+@app.post("/proxy_process")
+async def proxy_process(request: Request, file: UploadFile = File(...), password: str = Form(""), username: str = Form("")):
+    base_url = DEV_URL if username.strip().lower() == "devtest" else PROD_URL
+    target_url = f"{base_url}/process_single"
+
+    query_str = request.url.query
+    if query_str: target_url += f"?{query_str}"
+
+    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}
+
+    # FIX: Define the blocking request
+    def make_post():
+        return requests.post(target_url, headers=headers, files=files, data=data, timeout=120)
+
+    try:
+        # FIX: Run it in a background thread so other users can check the status!
+        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)}"}
+    except Exception as e:
+        return {"success": False, "message": f"Proxy Error (Internal): {str(e)}"}
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", 7860)))