File size: 4,485 Bytes
f332e2d a842a46 f332e2d a842a46 f332e2d a842a46 bb37d17 a842a46 bb37d17 a842a46 bb37d17 a842a46 bb37d17 a842a46 bb37d17 a842a46 f332e2d bb37d17 a842a46 f332e2d a842a46 f332e2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 |
from fastapi import FastAPI, File, UploadFile, HTTPException, Form
from fastapi.responses import FileResponse
from fastapi.staticfiles import StaticFiles
import requests
import os
import uuid
import shutil
import logging
import time
# Set up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
app = FastAPI()
# Mount a static directory to serve files
STATIC_DIR = "/app/static"
os.makedirs(STATIC_DIR, exist_ok=True)
app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
# API endpoint for ChangeClothesAI
CHANGE_CLOTHES_API = os.environ.get('CHANGE_CLOTHES_API', 'https://changeclothesai.online/api/try-on/edge')
@app.post("/try-on/")
async def try_on(
human_img: UploadFile = File(...),
garment_img: UploadFile = File(...),
garment_desc: str = Form("dresses"),
category: str = Form("dresses")
):
try:
# Generate unique filenames
human_filename = f"{uuid.uuid4()}_{human_img.filename}"
garment_filename = f"{uuid.uuid4()}_{garment_img.filename}"
# Save files temporarily in the static directory
human_path = os.path.join(STATIC_DIR, human_filename)
garment_path = os.path.join(STATIC_DIR, garment_filename)
logger.info(f"Saving human image to: {human_path}")
with open(human_path, "wb") as f:
shutil.copyfileobj(human_img.file, f)
logger.info(f"Human image saved: {os.path.exists(human_path)}")
logger.info(f"Saving garment image to: {garment_path}")
with open(garment_path, "wb") as f:
shutil.copyfileobj(garment_img.file, f)
logger.info(f"Garment image saved: {os.path.exists(garment_path)}")
# Generate public URLs
base_url = "https://tejani-tryapi.hf.space" # Verify this is correct
human_url = f"{base_url}/static/{human_filename}"
garment_url = f"{base_url}/static/{garment_filename}"
logger.info(f"Human URL: {human_url}")
logger.info(f"Garment URL: {garment_url}")
# Skip URL verification for debugging
# try:
# url_check = requests.head(human_url, timeout=10)
# logger.info(f"Human URL status: {url_check.status_code}")
# if url_check.status_code != 200:
# raise HTTPException(status_code=500, detail=f"Human image URL inaccessible: {human_url}")
# url_check = requests.head(garment_url, timeout=10)
# logger.info(f"Garment URL status: {url_check.status_code}")
# if url_check.status_code != 200:
# raise HTTPException(status_code=500, detail=f"Garment image URL inaccessible: {garment_url}")
# except requests.RequestException as e:
# raise HTTPException(status_code=500, detail=f"Error accessing URLs: {str(e)}")
# Prepare headers, cookies, and data for the API
headers = {
"accept": "*/*",
"f": "sdfdsfsKaVgUoxa5j1jzcFtziPx", # Match Trinket script
}
cookies = {}
data = {
"humanImg": human_url,
"garment": garment_url,
"garmentDesc": garment_desc,
"category": category,
}
# Call the ChangeClothesAI API
logger.info(f"Calling API: {CHANGE_CLOTHES_API}")
response = requests.post(CHANGE_CLOTHES_API, headers=headers, cookies=cookies, data=data, timeout=60)
logger.info(f"API Response: {response.status_code}, {response.text}")
# Check if the API call was successful
if response.status_code != 200:
raise HTTPException(status_code=response.status_code, detail=f"Error calling ChangeClothesAI API: {response.text}")
# Delete temporary files (commented out for debugging)
# os.remove(human_path)
# os.remove(garment_path)
# logger.info("Temporary files deleted")
return response.json()
except Exception as e:
# Clean up in case of errors (commented out for debugging)
# if os.path.exists(human_path):
# os.remove(human_path)
# if os.path.exists(garment_path):
# os.remove(garment_path)
logger.error(f"Error processing request: {str(e)}")
raise HTTPException(status_code=500, detail=f"Error processing files: {str(e)}")
finally:
human_img.file.close()
garment_img.file.close()
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860) |