Update app.py

app.py

@@ -10,23 +10,23 @@ import gradio as gr
 
 MAX_SEED = 999999
 
-# ✅ Hardcoded Pixelcut API key
+# ✅ Pixelcut API key
 pixelcut_api_key = "sk_7c19e4f2ad434a3ebc70fdd85ade5309"
 
 
-# Convert images to PNG format
+# 🎯 Convert image to PNG format
 def convert_to_png(image):
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
 
-# Encode images to base64 PNG format
+# 🔥 Encode image to base64 PNG format
 def encode_image(image):
     image = convert_to_png(image)
     _, buffer = cv2.imencode('.png', image)
     return base64.b64encode(buffer).decode('utf-8')
 
 
-# Main try-on function using Pixelcut API
+# 🚀 Main try-on function using Pixelcut API
 def tryon(person_img, garment_img, seed, randomize_seed):
     start_time = time.time()
 
@@ -45,54 +45,60 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     # 🌟 Setup Pixelcut API endpoint and headers
     url = "https://api.developer.pixelcut.ai/v1/remove-background"
     headers = {
-        'Content-Type': 'application/json',
-        'Accept': 'application/json',
-        'X-API-KEY': pixelcut_api_key
+        "Content-Type": "application/json",
+        "Accept": "application/json",
+        "X-API-KEY": pixelcut_api_key
     }
 
-    # 🔧 Send person image for background removal
-    person_data = {"image": encoded_person_img, "format": "png"}
-    garment_data = {"image": encoded_garment_img, "format": "png"}
+    # 📩 Setup request payload for person image
+    person_data = {
+        "image_url": f"data:image/png;base64,{encoded_person_img}",
+        "format": "png"
+    }
+
+    result_img = None
+    max_retries = 5
+    retry_delay = 5
 
     try:
-        # 🚀 Process Person Image
-        person_response = requests.post(url, headers=headers, json=person_data, timeout=60)
-        if person_response.status_code == 200:
-            person_result_url = person_response.json().get('url', '')
-        else:
-            return None, None, f"Person image error: {person_response.status_code}"
+        session = requests.Session()
+
+        # 🚀 Send person image for background removal
+        for attempt in range(max_retries):
+            person_response = session.post(url, headers=headers, json=person_data, timeout=60)
+            print("Person Response:", person_response.status_code, person_response.text)
+
+            # ✅ Success — get the result URL
+            if person_response.status_code == 200:
+                person_result_url = person_response.json().get("result_url", "")
+                print("✅ Person Success:", person_result_url)
+
+                # Fetch the processed person image
+                if person_result_url:
+                    person_img_data = requests.get(person_result_url).content
+                    person_np = np.frombuffer(person_img_data, np.uint8)
+                    result_img = cv2.imdecode(person_np, cv2.IMREAD_UNCHANGED)
+                    info = "Success"
+                    break
+
+            else:
+                # 🔥 If API returns "Too many users" or other errors
+                error_details = person_response.json().get("error", "Unknown error")
+                print(f"Attempt {attempt + 1}/{max_retries}: Error - {error_details}")
+                time.sleep(retry_delay)
+                info = f"Person image error: {person_response.status_code} - {error_details}"
 
-        # 🚀 Process Garment Image
-        garment_response = requests.post(url, headers=headers, json=garment_data, timeout=60)
-        if garment_response.status_code == 200:
-            garment_result_url = garment_response.json().get('url', '')
         else:
-            return None, None, f"Garment image error: {garment_response.status_code}"
-
-        # 🔥 Call the try-on API with both processed images
-        tryon_url = "https://api.example.com/v1/tryon"
-        tryon_payload = {
-            "person_image_url": person_result_url,
-            "garment_image_url": garment_result_url
-        }
-        tryon_response = requests.post(tryon_url, headers=headers, json=tryon_payload, timeout=60)
-
-        # ✅ Handle the try-on result
-        if tryon_response.status_code == 200:
-            result_url = tryon_response.json().get('result_url', '')
-            result_img_data = requests.get(result_url).content
-            result_np = np.frombuffer(result_img_data, np.uint8)
-            result_img = cv2.imdecode(result_np, cv2.IMREAD_UNCHANGED)
-            info = "Success"
-        else:
-            info = f"Try-On error: {tryon_response.status_code}"
+            # ❌ If retries fail
+            info = "API still overloaded — please try again later."
 
-    # 🛑 Handle timeout or API errors
+    # 🛑 Handle timeouts
     except requests.exceptions.ReadTimeout:
         print("Timeout!")
         info = "Too many users, please try again later"
         raise gr.Error("Too many users, please try again later")
 
+    # ❗ Catch any other unexpected errors
     except Exception as err:
         print(f"❌ Other error: {err}")
         info = "Error, please contact the admin"
@@ -103,6 +109,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
 
     return result_img, seed, info
 
+
 example_path = os.path.join(os.path.dirname(__file__), 'assets')
 
 garm_list = os.listdir(os.path.join(example_path,"cloth"))