Update app.py

app.py

@@ -5,7 +5,12 @@ import requests
 import os
 import uuid
 import shutil
-from tempfile import NamedTemporaryFile
+import logging
+import time
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
@@ -15,7 +20,7 @@ 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')
+CHANGE_CLOTHES_API = os.environ.get('CHANGE_CLOTHES_API', 'https://changeclothesai.online/api/try-on/edge')
 
 @app.post("/try-on/")
 async def try_on(
@@ -33,63 +38,79 @@ async def try_on(
         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"
+        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}")
+
+        # Test URL accessibility
+        time.sleep(1)  # Ensure files are available
         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)
-            base_url = "https://tejani-tryapi.hf.space"
-            human_url = f"{base_url}/static/{human_filename}"
-            garment_url = f"{base_url}/static/{garment_filename}"
-
-            print(human_url)
-            print(garment_url)
-
-            # Prepare headers and data for the original API
-            headers = {
-                "accept": "*/*",
-                "f": str(uuid.uuid4()).replace("-", ""),
-            }
-
-            data = {
-                "humanImg": human_url,
-                "garment": garment_url,
-                "garmentDesc": garment_desc,
-                "category": category,
-            }
-            
-            # Call the ChangeClothesAI API
-            response = requests.post(CHANGE_CLOTHES_API, headers=headers, data=data, timeout=60)
-
-            # 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()
+            url_check = requests.head(human_url, timeout=5)
+            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=5)
+            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": str(uuid.uuid4()).replace("-", ""),
+        }
+        cookies = {}  # Add empty cookies to match the first script
+        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
+        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
+        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)