Update app.py

app.py

@@ -53,32 +53,41 @@ def tryon(person_img, cloth_img, seed, random_seed):
     if random_seed:
         seed = random.randint(0, 1000000)
 
-    # 🔵 Convert images to RGB (keeps the blue tint glitch!)
-    person_img = cv2.cvtColor(person_img, cv2.COLOR_BGR2RGB)
-    cloth_img = cv2.cvtColor(cloth_img, cv2.COLOR_BGR2RGB)
+    try:
+        # ✅ Convert images to RGB (fixing blue tint)
+        person_img = cv2.cvtColor(person_img, cv2.COLOR_BGR2RGB)
+        cloth_img = cv2.cvtColor(cloth_img, cv2.COLOR_BGR2RGB)
 
-    # ✂️ Resize both images to 256x256
-    person_img = cv2.resize(person_img, (256, 256))
-    cloth_img = cv2.resize(cloth_img, (256, 256))
+        # ✂️ Resize both images to 256x256 (keep consistent sizes)
+        person_img = cv2.resize(person_img, (256, 256))
 
-    try:
-    # 🔥 Blend garment onto person (simple overlay — still has blue tint)
+        # ✅ Resize garment to fit person size
+        g_h, g_w = person_img.shape[:2]
+        cloth_img = cv2.resize(cloth_img, (g_w, g_h))
+
+        # 🔥 Blend garment onto person (centered overlay)
         output_img = person_img.copy()
         h, w, _ = person_img.shape
-        g_h, g_w, _ = cloth_img.shape
 
-    # Position the garment roughly on the torso (manual offset)
+        # Center garment on the torso (adjust the offset if needed)
         x_offset = (w - g_w) // 2
         y_offset = int(h * 0.35)
 
-    # ✅ Ensure garment fits inside the person image
+        # ✅ Ensure garment fits 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
+        # 🛠️ Overlay garment on the person (with transparency support)
+        alpha = 0.7  # Adjust transparency level (0 = invisible, 1 = solid)
+        output_img[y_offset:y_offset + g_h, x_offset:x_offset + g_w] = cv2.addWeighted(
+            person_img[y_offset:y_offset + g_h, x_offset:x_offset + g_w],
+            1 - alpha,
+            cloth_img,
+            alpha,
+            0
+        )
 
-        return output_img, seed, "✅ Success (Blue tint + Garment added)"
+        return output_img, seed, "✅ Success (Blue tint removed + Garment added)"
 
     except Exception as e:
         print(f"❌ Error in tryon function: {e}")
@@ -86,6 +95,7 @@ def tryon(person_img, cloth_img, seed, random_seed):
 
 
 
+
 # 🔧 Paths for examples
 example_path = os.path.join(os.path.dirname(__file__), "assets")