Update app.py

app.py

@@ -85,86 +85,38 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     max_retries = 5
     retry_delay = 3  # Faster retries now!
 
-    try:
-        session = requests.Session()
-
-        # 🛠️ Ensure both images are resized to 256x256
-        person_img = cv2.resize(person_img, (256, 256))
-        garment_img = cv2.resize(garment_img, (256, 256))
-
-        # 🎯 Handle transparency if the garment has an alpha channel
-        if garment_img.shape[-1] == 4:
-            alpha = garment_img[:, :, 3] / 255.0
-            for c in range(0, 3):
-                person_img[:, :, c] = (1 - alpha) * person_img[:, :, c] + alpha * garment_img[:, :, c]
-
-        # 🔧 Optional: Center the garment on the torso area
-        h, w, _ = person_img.shape
-        g_h, g_w, _ = garment_img.shape
-        x_offset = (w - g_w) // 2
-        y_offset = int(h * 0.35)  # Adjust this value for higher/lower garment placement
-        person_img[y_offset:y_offset + g_h, x_offset:x_offset + g_w] = garment_img
-
-        # 🔁 Retry loop — smarter handling of "please try again later"
-        for attempt in range(max_retries):
-            # 🌐 Debugging payload, URL, and headers
-            print("📩 Sending data:", seed)
-            print("🌐 URL:", url)
-            print("🛠️ Headers:", headers)
-
-            response = session.post(url, headers=headers, json={"person_data": seed}, timeout=60)
-            print("Response code:", response.status_code)
-
-            # ✅ Success — process the result
-            if response.status_code == 200:
-                result_url = response.json().get('result_url', '')
-                print("✅ Success:", result_url)
-
-                # Fetch the final image
-                if 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"
-                    # 🔵 Blue tint fix: Ensure correct color space (BGR to RGB)
-                    if result_img is not None and len(result_img.shape) == 3:
-                        result_img = cv2.cvtColor(result_img, cv2.COLOR_BGR2RGB)
-
-                    info = "✅ Success"
-                    break
-
-            else:
-                # 🔥 Handle "please try again later" or other errors
-                error_response = response.json().get('error', '')
-
-                if "please try again later" in error_response.lower():
-                    print(f"Attempt {attempt + 1}/{max_retries}: API said 'please try again later'. Retrying in {retry_delay} seconds...")
-                    time.sleep(retry_delay)
-                    retry_delay += 2  # Exponential backoff — retries longer each time
-                else:
-                    info = f"❌ Error: {response.status_code} - {response.text}"
-                    break
-
-        else:
-            # ❌ If all retries fail
-            info = "API still says 'please try again later' — waited too long."
-
-    # 🛑 Handle timeouts specifically
+   try:
+    session = requests.Session()
+
+    # 🛠️ Ensure both images are resized to 256x256
+    person_img = cv2.resize(person_img, (256, 256))
+    garment_img = cv2.resize(garment_img, (256, 256))
+
+    # 🎯 Handle transparency if the garment has an alpha channel
+    if garment_img.shape[-1] == 4:
+        alpha = garment_img[:, :, 3] / 255.0
+        for c in range(0, 3):
+            person_img[:, :, c] = (1 - alpha) * person_img[:, :, c] + alpha * garment_img[:, :, c]
+
+    # 🔧 Optional: Center the garment on the torso area
+    h, w, _ = person_img.shape
+    g_h, g_w, _ = garment_img.shape
+
+    x_offset = (w - g_w) // 2
+    y_offset = int(h * 0.35)  # Adjust this value for higher/lower garment placement
+    person_img[y_offset:y_offset + g_h, x_offset:x_offset + g_w] = garment_img
+
+    # ✅ Now the rest of your code continues here (retry loop, API call, etc.)
+
     except requests.exceptions.ReadTimeout:
         print("Timeout!")
         info = "⚡ Timeout — please try again later"
         raise gr.Error("Too many users, please try again later")
 
-    # ❗ Handle any other unexpected errors
     except Exception as err:
-        print(f"❌ Detailed error: {repr(err)}")  # This now prints the complete error traceback
+        print(f"❌ Other error: {err}")
         info = "Error, please contact the admin"
 
-    end_time = time.time()
-    print(f"⏳ Time used: {end_time - start_time:.2f} seconds")
-
-    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"))