fix: use /tmp for uploads/results and update template for download routes (Hugging Face Spaces compatibility)

app/routers/inference.py

@@ -140,6 +140,7 @@ def upload_image(request: Request, file: UploadFile = File(...)):
     with open(upload_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
 
+
     # Run both inferences
     try:
         damage_result = run_yolo_inference(DAMAGE_MODEL_PATH, upload_path)
@@ -149,9 +150,10 @@ def upload_image(request: Request, file: UploadFile = File(...)):
         damage_img_path = os.path.join(RESULTS_DIR, f"{session_id}_damage.png")
         parts_img_path = os.path.join(RESULTS_DIR, f"{session_id}_parts.png")
         json_path = os.path.join(RESULTS_DIR, f"{session_id}_result.json")
-        damage_img_url = f"/static/results/{session_id}_damage.png"
-        parts_img_url = f"/static/results/{session_id}_parts.png"
-        json_url = f"/static/results/{session_id}_result.json"
+        # Use custom download endpoints for Hugging Face Spaces
+        damage_img_url = f"/download/result/{session_id}_damage.png"
+        parts_img_url = f"/download/result/{session_id}_parts.png"
+        json_url = f"/download/result/{session_id}_result.json"
 
         # Defensive: set to None by default
         damage_img = None
@@ -212,6 +214,21 @@ def upload_image(request: Request, file: UploadFile = File(...)):
         {
             "request": request,
             "result": result,
-            "original_image": f"/static/uploads/{session_id}.{ext}"
+            "original_image": f"/download/upload/{session_id}.{ext}"
         }
     )
+
+# --- Add routes to serve files from /tmp/uploads and /tmp/results ---
+@router.get("/download/upload/{filename}")
+def download_uploaded_file(filename: str):
+    file_path = os.path.join(UPLOAD_DIR, filename)
+    if not os.path.exists(file_path):
+        return JSONResponse(status_code=404, content={"error": "File not found"})
+    return FileResponse(file_path, filename=filename)
+
+@router.get("/download/result/{filename}")
+def download_result_file(filename: str):
+    file_path = os.path.join(RESULTS_DIR, filename)
+    if not os.path.exists(file_path):
+        return JSONResponse(status_code=404, content={"error": "File not found"})
+    return FileResponse(file_path, filename=filename)