Update app.py

app.py

@@ -1,33 +1,257 @@
+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
+import os
+import stat
 
-url = "https://changeclothesai.online/api/try-on/edge"
-
-cookies = {}
-
-headers = {
-    "accept": "*/*",
-    "accept-language": "en-GB,en-US;q=0.9,en;q=0.8,gu;q=0.7",
-    "f": "ab6023da0b46850091054b4ce804ecf4",
-    "origin": "https://changeclothesai.online",
-    "priority": "u=1, i",
-    "referer": "https://changeclothesai.online/",
-    "sec-ch-ua": '"Not)A;Brand";v="8", "Chromium";v="138", "Google Chrome";v="138"',
-    "sec-ch-ua-mobile": "?1",
-    "sec-ch-ua-platform": '"Android"',
-    "sec-fetch-dest": "empty",
-    "sec-fetch-mode": "cors",
-    "sec-fetch-site": "same-origin",
-    "user-agent": "Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Mobile Safari/537.36"
-}
-
-data = {
-    "humanImg": "https://jallenjia-change-clothes-ai.hf.space/file=/tmp/gradio/ba40a133e16016041aec2390aa6899fede197af6/Jensen.jpeg",
-    "garment": "https://jallenjia-change-clothes-ai.hf.space/file=/tmp/gradio/7cb2f03c3ae9feb724a711d71a5fc58c1570b3c3/Girl%20outfits%205.jpg",
-    "garmentDesc": "",
-    "category": "upper_body"
-}
-
-response = requests.post(url, headers=headers, cookies=cookies, data=data)
-
-print(response.status_code)
-print(response.text)
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+app = FastAPI()
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Adjust for production
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Mount static directory for serving files
+STATIC_DIR = Path("/home/user/app/static")
+STATIC_DIR.mkdir(parents=True, exist_ok=True)
+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")
+
+# Define the request model
+class TryOnRequest(BaseModel):
+    garmentDesc: str
+    category: str
+
+# Allowed file extensions
+ALLOWED_EXTENSIONS = {".jpg", ".jpeg", ".png"}
+
+# Helper function to save file and generate public URL
+async def save_file_and_get_url(file: UploadFile) -> str:
+    try:
+        # Validate file extension
+        file_extension = f".{file.filename.split('.')[-1].lower()}"
+        if file_extension not in ALLOWED_EXTENSIONS:
+            logger.error(f"Invalid file extension for {file.filename}: {file_extension}")
+            raise HTTPException(status_code=400, detail=f"File extension {file_extension} not allowed. Use JPG or PNG.")
+
+        # Generate unique filename
+        unique_filename = f"{uuid.uuid4()}{file_extension}"
+        file_path = STATIC_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 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}"
+        logger.info(f"Generated public URL: {public_url}")
+        
+        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)}")
+
+# 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/"
+        
+        headers = {
+            "accept": "*/*",
+            "f": "sdfdsfsKaVgUoxa5j1jzcFtziPx",
+        }
+        
+        data = {
+            "humanImg": human_img_url,
+            "garment": garment_url,
+            "garmentDesc": garment_desc,
+            "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)
+        logger.info(f"API response status: {response.status_code}, content: {response.text}")
+        response.raise_for_status()
+        
+        return {
+            "status_code": response.status_code,
+            "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:
+        logger.error(f"Error forwarding request: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error forwarding request: {str(e)}")
+    except Exception as 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
+@app.get("/")
+async def root():
+    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 STATIC_DIR.glob("*") if f.is_file()]
+        logger.info(f"Files in {STATIC_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)}")
+
+# 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)}")
+
+# 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_content = """
+    <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; }
+                pre { background: #f4f4f4; padding: 10px; max-height: 300px; overflow: auto; }
+            </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, <1MB):</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, <1MB):</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>
+            <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)