Update app.py

app.py

@@ -5,6 +5,11 @@ from fastapi.middleware.cors import CORSMiddleware
 import os
 import uuid
 from pathlib import Path
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
@@ -22,25 +27,47 @@ class TryOnRequest(BaseModel):
     garmentDesc: str
     category: str
 
-# Base directory for temporary file storage
+# Base directory for file storage (use /tmp/gradio for Hugging Face Spaces)
 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
-    
-    # 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://tejani-tryapi.hf.space/file={str(file_path)}"
-    return public_url
+    try:
+        # Generate unique filename
+        file_extension = file.filename.split(".")[-1]
+        unique_filename = f"{uuid.uuid4()}.{file_extension}"
+        file_path = UPLOAD_DIR / unique_filename
+        
+        # Save file
+        logger.info(f"Saving file to {file_path}")
+        with file_path.open("wb") as buffer:
+            content = await file.read()
+            buffer.write(content)
+        
+        # Verify file exists
+        if not file_path.exists():
+            logger.error(f"File {file_path} was not saved correctly")
+            raise HTTPException(status_code=500, detail="Failed to save file")
+        
+        # Generate public URL
+        # Use SPACE_ID environment variable or fallback to placeholder
+        space_id = os.getenv("SPACE_ID", "your-username-space-name")
+        public_url = f"https://tejani-tryapi.hf.space/file={str(file_path)}"
+        logger.info(f"Generated public URL: {public_url}")
+        
+        # Test URL accessibility
+        try:
+            response = requests.head(public_url, timeout=5)
+            if response.status_code != 200:
+                logger.warning(f"Public URL {public_url} returned status {response.status_code}")
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Failed to access public URL {public_url}: {str(e)}")
+        
+        return public_url
+    except Exception as e:
+        logger.error(f"Error in save_file_and_get_url: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error processing file: {str(e)}")
 
 # Updated endpoint to handle file uploads and proxy the request
 @app.post("/try-on")
@@ -60,7 +87,7 @@ async def try_on(
         
         headers = {
             "accept": "*/*",
-            "f": "sdfdsdfKaVgUoxa5j1jzcFtziPx",
+            "f": "sdfdsfsKaVgUoxa5j1jzcFtziPx",
         }
         
         data = {
@@ -70,6 +97,8 @@ async def try_on(
             "category": category
         }
         
+        logger.info(f"Forwarding request to {url} with data: {data}")
+        
         # Forward request to the original API
         response = requests.post(url, headers=headers, cookies={}, data=data)
         response.raise_for_status()
@@ -81,11 +110,24 @@ async def try_on(
             "garment_url": garment_url
         }
     except requests.exceptions.RequestException as e:
+        logger.error(f"Error forwarding request: {str(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)}")
+        logger.error(f"Error in try_on endpoint: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error processing request: {str(e)}")
 
 # Health check endpoint for Hugging Face Spaces
 @app.get("/")
 async def root():
-    return {"message": "FastAPI proxy for try-on API with file upload is running"}
+    return {"message": "FastAPI proxy for try-on API with file upload is running"}
+
+# Debug endpoint to list stored files
+@app.get("/list-files")
+async def list_files():
+    try:
+        files = [str(f) for f in UPLOAD_DIR.glob("*") if f.is_file()]
+        logger.info(f"Files in {UPLOAD_DIR}: {files}")
+        return {"files": files}
+    except Exception as e:
+        logger.error(f"Error listing files: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error listing files: {str(e)}")