Update app.py

app.py

@@ -1,268 +1,33 @@
-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
 
-# 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/edge"
-        
-        headers = {
-            "accept": "*/*",
-            "accept-language": "en-GB,en-US;q=0.9,en;q=0.8,gu;q=0.7",
-            "f": "ab6023da0b46850091054b4ce804ecd4",
-            "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": 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)
+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": "ab6023da0b46850091054b4ce804ecd4",
+    "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://tejani-tryapi.hf.space/static/28a23b98-3363-46a2-8ce3-8d17e89a27b8.jpg",
+    "garment": "https://tejani-tryapi.hf.space/static/1e0efc82-599f-4bad-b539-c44b0059aa94.jpg",
+    "garmentDesc": "",
+    "category": "upper_body"
+}
+
+response = requests.post(url, headers=headers, cookies=cookies, data=data)
+
+print(response.status_code)
+print(response.text)