Update app.py

app.py

@@ -44,12 +44,29 @@ def warp_design(design, mask, warp_scale):
 
 def blend_images(base, overlay, mask):
     """Blends the design onto the dress using seamless cloning."""
-    center = tuple(np.array(base.shape[:2]) // 2)
-    
-    # Convert mask to uint8
+    if overlay is None or mask is None:
+        raise ValueError("Overlay or mask is None, check segmentation and warping.")
+
+    base = np.array(base)  # Ensure base is a NumPy array
+    overlay = np.array(overlay)
+
+    # Ensure overlay and base are 3-channel images
+    if len(overlay.shape) == 2:
+        overlay = cv2.cvtColor(overlay, cv2.COLOR_GRAY2BGR)
+    if len(base.shape) == 2:
+        base = cv2.cvtColor(base, cv2.COLOR_GRAY2BGR)
+
+    # Ensure mask is single-channel grayscale
+    if len(mask.shape) == 3:
+        mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
+
+    # Convert mask to uint8 if necessary
     if mask.dtype != np.uint8:
-        mask = (mask * 255).astype(np.uint8)  # Normalize and convert
-    
+        mask = (mask * 255).astype(np.uint8)
+
+    # Compute center of the mask for seamless cloning
+    center = tuple(np.array(base.shape[:2]) // 2)
+
     return cv2.seamlessClone(overlay, base, mask, center, cv2.NORMAL_CLONE)
 
 def apply_design(image_path, design_path, warp_scale):