File size: 2,865 Bytes
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 |
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
from tempfile import NamedTemporaryFile
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('URL')
@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)
try:
# Save human image
with open(human_path, "wb") as f:
shutil.copyfileobj(human_img.file, f)
# Save garment image
with open(garment_path, "wb") as f:
shutil.copyfileobj(garment_img.file, f)
# Generate public URLs (assuming the Space is hosted at a public URL)
# Replace YOUR_SPACE_URL with the actual Hugging Face Space URL
base_url = "https://tejani-tryapi.hf.space" # e.g., https://your-username-your-space.hf.space
human_url = f"{base_url}/static/{human_filename}"
garment_url = f"{base_url}/static/{garment_filename}"
# Call the ChangeClothesAI API
response = requests.post(CHANGE_CLOTHES_API, headers=headers, data=data)
# Check if the API call was successful
if response.status_code != 200:
raise HTTPException(status_code=response.status_code, detail="Error calling ChangeClothesAI API")
# Delete temporary files
os.remove(human_path)
os.remove(garment_path)
return response.json()
except Exception as e:
# Clean up in case of any errors
if os.path.exists(human_path):
os.remove(human_path)
if os.path.exists(garment_path):
os.remove(garment_path)
raise HTTPException(status_code=500, detail=str(e))
finally:
human_img.file.close()
garment_img.file.close()
except Exception as e:
raise HTTPException(status_code=500, detail=f"Error processing files: {str(e)}")
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860) |