Update app.py

app.py

@@ -1,7 +1,10 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, UploadFile, File
 from pydantic import BaseModel
 import requests
 from fastapi.middleware.cors import CORSMiddleware
+import os
+import uuid
+from pathlib import Path
 
 app = FastAPI()
 
@@ -16,40 +19,73 @@ app.add_middleware(
 
 # Define the request model
 class TryOnRequest(BaseModel):
-    humanImg: str
-    garment: str
     garmentDesc: str
     category: str
 
-# Endpoint to proxy the request to the original API
-@app.post("/try-on")
-async def try_on(request: TryOnRequest):
-    url = "https://changeclothesai.online/api/try-on/edge"
-    
-    headers = {
-        "accept": "*/*",
-        "f": "sdfdsfsKaVgUoxa5j1jzcFtziPx",
-    }
+# Base directory for temporary file storage
+UPLOAD_DIR = Path("/tmp/gradio")
+UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
+
+# Helper function to save file and generate public URL
+async def save_file_and_get_url(file: UploadFile) -> str:
+    # Generate unique filename
+    file_extension = file.filename.split(".")[-1]
+    unique_filename = f"{uuid.uuid4()}.{file_extension}"
+    file_path = UPLOAD_DIR / unique_filename
     
-    data = {
-        "humanImg": request.humanImg,
-        "garment": request.garment,
-        "garmentDesc": request.garmentDesc,
-        "category": request.category
-    }
+    # Save file
+    with file_path.open("wb") as buffer:
+        content = await file.read()
+        buffer.write(content)
     
+    # Generate public URL (Hugging Face Spaces format)
+    public_url = f"https://{os.getenv('SPACE_ID', 'your-username-space-name')}.hf.space/file={str(file_path)}"
+    return public_url
+
+# Updated endpoint to handle file uploads and proxy the request
+@app.post("/try-on")
+async def try_on(
+    human_img: UploadFile = File(...),
+    garment: UploadFile = File(...),
+    garment_desc: str = "",
+    category: str = "upper_body"
+):
     try:
+        # Save files and get public URLs
+        human_img_url = await save_file_and_get_url(human_img)
+        garment_url = await save_file_and_get_url(garment)
+        
+        # Original API endpoint
+        url = "https://changeclothesai.online/api/try-on/edge"
+        
+        headers = {
+            "accept": "*/*",
+            "f": "sdfdsdfKaVgUoxa5j1jzcFtziPx",
+        }
+        
+        data = {
+            "humanImg": human_img_url,
+            "garment": garment_url,
+            "garmentDesc": garment_desc,
+            "category": category
+        }
+        
+        # Forward request to the original API
         response = requests.post(url, headers=headers, cookies={}, data=data)
-        response.raise_for_status()  # Raise exception for bad status codes
+        response.raise_for_status()
         
         return {
             "status_code": response.status_code,
-            "response": response.json() if response.headers.get('content-type') == 'application/json' else response.text
+            "response": response.json() if response.headers.get('content-type') == 'application/json' else response.text,
+            "human_img_url": human_img_url,
+            "garment_url": garment_url
         }
     except requests.exceptions.RequestException as e:
         raise HTTPException(status_code=500, detail=f"Error forwarding request: {str(e)}")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error processing files: {str(e)}")
 
 # Health check endpoint for Hugging Face Spaces
 @app.get("/")
 async def root():
-    return {"message": "FastAPI proxy for try-on API is running"}
+    return {"message": "FastAPI proxy for try-on API with file upload is running"}