Spaces:

tejani
/

TryApi

Running

App Files Files Community

tejani commited on Jul 15

Commit

b1cf5ad

verified ·

1 Parent(s): dba2f84

Update app.py

Browse files

Files changed (1) hide show

app.py +83 -19

app.py CHANGED Viewed

@@ -7,6 +7,8 @@ from fastapi.responses import HTMLResponse
 import uuid
 from pathlib import Path
 import logging
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -26,6 +28,11 @@ app.add_middleware(
 # Mount static directory for serving files
 STATIC_DIR = Path("/home/user/app/static")
 STATIC_DIR.mkdir(parents=True, exist_ok=True)
 app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
 # Alternative directory (uncomment to use /tmp/gradio)
@@ -59,12 +66,17 @@ async def save_file_and_get_url(file: UploadFile) -> str:
             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 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)
@@ -73,7 +85,7 @@ async def save_file_and_get_url(file: UploadFile) -> str:
         # 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:
@@ -165,6 +177,40 @@ async def get_file(filename: str):
         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():
@@ -172,10 +218,14 @@ async def test_upload_form():
     <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>
@@ -186,23 +236,37 @@ async def test_upload_form():
                 <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)}")

 import uuid
 from pathlib import Path
 import logging
+import os
+import stat
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 # Mount static directory for serving files
 STATIC_DIR = Path("/home/user/app/static")
 STATIC_DIR.mkdir(parents=True, exist_ok=True)
+# Set directory permissions to ensure accessibility
+try:
+    os.chmod(STATIC_DIR, stat.S_IRWXU | stat.S_IRWXG | stat.S_IROTH | stat.S_IXOTH)
+except Exception as e:
+    logger.error(f"Failed to set permissions for {STATIC_DIR}: {str(e)}")
 app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
 # Alternative directory (uncomment to use /tmp/gradio)
             content = await file.read()
             buffer.write(content)
+        # Verify file exists and set permissions
         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")
+        try:
+            os.chmod(file_path, stat.S_IRWXU | stat.S_IRWXG | stat.S_IROTH)
+            logger.info(f"Set permissions for {file_path}")
+        except Exception as e:
+            logger.error(f"Failed to set permissions for {file_path}: {str(e)}")
+        # Generate public URL
         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)
         # Test URL accessibility
         try:
+            response = requests.head(public_url, timeout=20)
             if response.status_code != 200:
                 logger.warning(f"Public URL {public_url} returned status {response.status_code}")
             else:
         logger.error(f"Error serving file {filename}: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Error serving file: {str(e)}")
+# Debug endpoint to check file existence and permissions
+@app.get("/check-file/{filename}")
+async def check_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")
+        stats = os.stat(file_path)
+        permissions = oct(stats.st_mode)[-3:]
+        logger.info(f"File {file_path} exists with permissions {permissions}")
+        return {
+            "file": str(file_path),
+            "exists": True,
+            "permissions": permissions
+        }
+    except Exception as e:
+        logger.error(f"Error checking file {filename}: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error checking file: {str(e)}")
+# 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)}")
 # Frontend for testing file uploads
 @app.get("/test-upload", response_class=HTMLResponse)
 async def test_upload_form():
     <html>
         <head>
             <title>Test File Upload</title>
+            <style>
+                body { font-family: Arial, sans-serif; margin: 20px; }
+                .response { margin-top: 20px; padding: 10px; border: 1px solid #ccc; }
+            </style>
         </head>
         <body>
             <h1>Test File Upload</h1>
+            <form action="/try-on" method="post" enctype="multipart/form-data" onsubmit="showResponse(event)">
                 <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="text" id="category" name="category" value="upper_body"><br><br>
                 <input type="submit" value="Upload and Try On">
             </form>
+            <div id="response" class="response" style="display: none;"></div>
             <h2>Debug Tools</h2>
             <p><a href="/list-files">List Stored Files</a></p>
+            <p><a href="/clean-files">Clean Stored Files</a></p>
+            <script>
+                async function showResponse(event) {
+                    event.preventDefault();
+                    const form = event.target;
+                    const formData = new FormData(form);
+                    try {
+                        const response = await fetch('/try-on', {
+                            method: 'POST',
+                            body: formData
+                        });
+                        const result = await response.json();
+                        const responseDiv = document.getElementById('response');
+                        responseDiv.style.display = 'block';
+                        responseDiv.innerHTML = `
+                            <h3>Response</h3>
+                            <p><strong>Status Code:</strong> ${result.status_code}</p>
+                            <p><strong>Human Image URL:</strong> <a href="${result.human_img_url}" target="_blank">${result.human_img_url}</a></p>
+                            <p><strong>Garment URL:</strong> <a href="${result.garment_url}" target="_blank">${result.garment_url}</a></p>
+                            <p><strong>API Response:</strong> <pre>${JSON.stringify(result.response, null, 2)}</pre></p>
+                        `;
+                    } catch (error) {
+                        document.getElementById('response').style.display = 'block';
+                        document.getElementById('response').innerHTML = `<p><strong>Error:</strong> ${error.message}</p>`;
+                    }
+                }
+            </script>
         </body>
     </html>
     """
+    return HTMLResponse(content=html_content)