Update app.py

app.py

@@ -31,14 +31,17 @@ def warp_design(design, mask, warp_scale):
     # Normalize mask and convert to uint8
     scaled_mask = (mask * 255 * (warp_scale / 100)).astype(np.uint8)
 
-    # Ensure the mask is single-channel and same size as design
-    if len(scaled_mask.shape) == 3:
+    # Ensure the mask is single-channel before applying cvtColor
+    if len(scaled_mask.shape) == 3 and scaled_mask.shape[2] == 3:
         scaled_mask = cv2.cvtColor(scaled_mask, cv2.COLOR_BGR2GRAY)
 
     # Resize the mask to match design_resized if needed
     if scaled_mask.shape != (h, w):
         scaled_mask = cv2.resize(scaled_mask, (w, h), interpolation=cv2.INTER_NEAREST)
 
+    # Debug: Print shapes
+    print(f"Design Resized Shape: {design_resized.shape}, Mask Shape: {scaled_mask.shape}")
+
     return cv2.bitwise_and(design_resized, design_resized, mask=scaled_mask)
 
 def blend_images(base, overlay, mask):