Update app.py

app.py

@@ -10,8 +10,11 @@ import gradio as gr
 
 MAX_SEED = 999999
 
-# ✅ New Pixelcut API key
-pixelcut_api_key = "sk_3a15d5b79d5944f3995c54615f8928f8"
+# ✅ Pixelcut API Key
+pixelcut_api_key = "YOUR_PIXELCUT_API_KEY"
+
+# ✅ ImgBB API Key (for uploading images to get valid URLs)
+imgbb_api_key = "YOUR_IMGBB_API_KEY"
 
 # 🎯 Convert image to PNG format
 def convert_to_png(image):
@@ -23,15 +26,16 @@ def encode_image(image):
     _, buffer = cv2.imencode('.png', image)
     return base64.b64encode(buffer).decode('utf-8')
 
-# 🛠️ Upload images to a temporary hosting service
-def upload_image(image_data):
-    files = {"file": ("image.png", image_data, "image/png")}
-    upload_response = requests.post("https://tmpfiles.org/api/v1/upload", files=files)
-    
-    if upload_response.status_code == 200:
-        return upload_response.json().get("data", {}).get("url", None)
+# 🛠️ Upload images to ImgBB to get permanent URLs
+def upload_image_to_imgbb(image_data):
+    url = f"https://api.imgbb.com/1/upload?key={imgbb_api_key}"
+    files = {"image": base64.b64encode(image_data).decode('utf-8')}
+    response = requests.post(url, files=files)
+
+    if response.status_code == 200:
+        return response.json().get("data", {}).get("url")
     else:
-        print("❌ Image upload failed:", upload_response.text)
+        print("❌ ImgBB upload failed:", response.text)
         return None
 
 # 🚀 Main try-on function using Pixelcut API
@@ -50,12 +54,12 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     _, person_encoded = cv2.imencode('.png', convert_to_png(person_img))
     _, garment_encoded = cv2.imencode('.png', convert_to_png(garment_img))
 
-    # ✅ Upload person and garment images to get URLs
-    person_url = upload_image(person_encoded)
-    garment_url = upload_image(garment_encoded)
+    # ✅ Upload person and garment images to get valid URLs
+    person_url = upload_image_to_imgbb(person_encoded)
+    garment_url = upload_image_to_imgbb(garment_encoded)
 
     if not person_url or not garment_url:
-        return None, None, "Image upload failed"
+        return None, None, "Image upload failed — check API keys or connection"
 
     # 🌟 Setup Pixelcut API endpoint and headers
     url = "https://api.developer.pixelcut.ai/v1/remove-background"
@@ -65,16 +69,9 @@ def tryon(person_img, garment_img, seed, randomize_seed):
         'X-API-KEY': pixelcut_api_key
     }
 
-    # 🔧 Use uploaded URLs in the payload
-    person_data = {
-        "image_url": person_url,
-        "format": "png"
-    }
-
-    garment_data = {
-        "image_url": garment_url,
-        "format": "png"
-    }
+    # 🔧 Use the uploaded URLs in the payload
+    person_data = {"image_url": person_url, "format": "png"}
+    garment_data = {"image_url": garment_url, "format": "png"}
 
     result_img = None
     max_retries = 5
@@ -135,6 +132,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
 
 
 
+
 example_path = os.path.join(os.path.dirname(__file__), 'assets')
 
 garm_list = os.listdir(os.path.join(example_path,"cloth"))