Spaces:

tejani
/

TryApi

Running

App Files Files Community

tejani commited on Jul 15

Commit

dba2f84

verified ·

1 Parent(s): e522daa

Update app.py

Browse files

Files changed (1) hide show

app.py +72 -6

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
-from fastapi import FastAPI, HTTPException, UploadFile, File
 from pydantic import BaseModel
 import requests
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
-import os
 import uuid
 from pathlib import Path
 import logging
@@ -28,6 +28,10 @@ STATIC_DIR = Path("/home/user/app/static")
 STATIC_DIR.mkdir(parents=True, exist_ok=True)
 app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
 # Define the request model
 class TryOnRequest(BaseModel):
     garmentDesc: str
@@ -60,12 +64,16 @@ async def save_file_and_get_url(file: UploadFile) -> str:
             logger.error(f"File {file_path} was not saved correctly")
             raise HTTPException(status_code=500, detail="Failed to save file")
-        public_url = f"https://tejani-tryapi.hf.space/static/{unique_filename}"
         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}")
             else:
@@ -96,7 +104,7 @@ async def try_on(
         headers = {
             "accept": "*/*",
-            "f": "sdasdwsKaVgUoxa5j1jzcFtziPx",
         }
         data = {
@@ -139,4 +147,62 @@ async def list_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)}")

+from fastapi import FastAPI, HTTPException, UploadFile, File, Response
 from pydantic import BaseModel
 import requests
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
+from fastapi.responses import HTMLResponse
 import uuid
 from pathlib import Path
 import logging
 STATIC_DIR.mkdir(parents=True, exist_ok=True)
 app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
+# Alternative directory (uncomment to use /tmp/gradio)
+# STATIC_DIR = Path("/tmp/gradio")
+# STATIC_DIR.mkdir(parents=True, exist_ok=True)
 # Define the request model
 class TryOnRequest(BaseModel):
     garmentDesc: str
             logger.error(f"File {file_path} was not saved correctly")
             raise HTTPException(status_code=500, detail="Failed to save file")
+        # Generate public URL with hardcoded space_id
+        space_id = "tejani-tryapi"
+        public_url = f"https://{space_id}.hf.space/static/{unique_filename}"
+        # Alternative URL for /tmp/gradio (uncomment if using /tmp/gradio)
+        # public_url = f"https://{space_id}.hf.space/file={str(file_path)}"
         logger.info(f"Generated public URL: {public_url}")
         # Test URL accessibility
         try:
+            response = requests.head(public_url, timeout=15)
             if response.status_code != 200:
                 logger.warning(f"Public URL {public_url} returned status {response.status_code}")
             else:
         headers = {
             "accept": "*/*",
+            "f": "sdfdsfsKaVgUoxa5j1jzcFtziPx",
         }
         data = {
         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)}")
+# Debug endpoint to manually serve a file
+@app.get("/get-file/{filename}")
+async def get_file(filename: str):
+    try:
+        file_path = STATIC_DIR / filename
+        if not file_path.exists():
+            logger.error(f"File {file_path} not found")
+            raise HTTPException(status_code=404, detail="File not found")
+        with file_path.open("rb") as f:
+            content = f.read()
+        return Response(content=content, media_type="image/jpeg")
+    except Exception as e:
+        logger.error(f"Error serving file {filename}: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error serving file: {str(e)}")
+# Frontend for testing file uploads
+@app.get("/test-upload", response_class=HTMLResponse)
+async def test_upload_form():
+    html_content = """
+    <html>
+        <head>
+            <title>Test File Upload</title>
+        </head>
+        <body>
+            <h1>Test File Upload</h1>
+            <form action="/try-on" method="post" enctype="multipart/form-data">
+                <label for="human_img">Human Image (JPG/PNG):</label><br>
+                <input type="file" id="human_img" name="human_img" accept=".jpg,.jpeg,.png" required><br><br>
+                <label for="garment">Garment Image (JPG/PNG):</label><br>
+                <input type="file" id="garment" name="garment" accept=".jpg,.jpeg,.png" required><br><br>
+                <label for="garment_desc">Garment Description:</label><br>
+                <input type="text" id="garment_desc" name="garment_desc" value=""><br><br>
+                <label for="category">Category:</label><br>
+                <input type="text" id="category" name="category" value="upper_body"><br><br>
+                <input type="submit" value="Upload and Try On">
+            </form>
+            <h2>Debug Tools</h2>
+            <p><a href="/list-files">List Stored Files</a></p>
+            <p><a href="/clean-files">Clean Stored unnamed: Files</a></p>
+        </body>
+    </html>
+    """
+    return HTMLResponse(content=html_content)
+# Endpoint to clean stored files
+@app.get("/clean-files")
+async def clean_files():
+    try:
+        for f in STATIC_DIR.glob("*"):
+            if f.is_file():
+                f.unlink()
+        logger.info(f"Cleaned all files in {STATIC_DIR}")
+        return {"message": "Files cleaned"}
+    except Exception as e:
+        logger.error(f"Error cleaning files: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error cleaning files: {str(e)}")