Update app.py

app.py

@@ -61,6 +61,7 @@ def tryon(person_img, cloth_img, seed, random_seed):
     person_img = cv2.resize(person_img, (256, 256))
     cloth_img = cv2.resize(cloth_img, (256, 256))
 
+   try:
     # 🔥 Blend garment onto person (simple overlay — still has blue tint)
     output_img = person_img.copy()
     h, w, _ = person_img.shape
@@ -70,11 +71,20 @@ def tryon(person_img, cloth_img, seed, random_seed):
     x_offset = (w - g_w) // 2
     y_offset = int(h * 0.35)
 
+    # ✅ Ensure garment fits inside the person image
+    if y_offset + g_h > h or x_offset + g_w > w:
+        raise ValueError("❗ Garment image is too large for the person image!")
+
     # 🛠️ Add garment onto person (no transparency support — raw overlay)
     output_img[y_offset:y_offset + g_h, x_offset:x_offset + g_w] = cloth_img
 
     return output_img, seed, "✅ Success (Blue tint + Garment added)"
 
+    except Exception as e:
+        print(f"❌ Error in tryon function: {e}")
+        return person_img, seed, f"❌ Error: {e}"
+
+
 
 # 🔧 Paths for examples
 example_path = os.path.join(os.path.dirname(__file__), "assets")