Update app.py

app.py

@@ -40,17 +40,24 @@ def warp_design(design, mask, warp_scale):
     # Resize design to match mask dimensions
     design_resized = cv2.resize(design, (w, h))
 
+    # Convert mask to grayscale if it's not already
     if len(mask.shape) == 3 and mask.shape[2] == 3:
         mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
 
-    # Ensure mask is uint8
+    # Ensure mask is uint8 (values between 0-255)
     if mask.dtype != np.uint8:
         mask = (mask * 255).astype(np.uint8)
 
-    # Convert design_resized to 3-channel if it's grayscale
+    # Ensure mask and design are the same size
+    mask = cv2.resize(mask, (w, h), interpolation=cv2.INTER_NEAREST)
+
+    # Ensure design_resized is 3-channel
     if len(design_resized.shape) == 2:
         design_resized = cv2.cvtColor(design_resized, cv2.COLOR_GRAY2BGR)
 
+    # Debugging output
+    print(f"Design shape: {design_resized.shape}, Mask shape: {mask.shape}, Mask dtype: {mask.dtype}")
+
     return cv2.bitwise_and(design_resized, design_resized, mask=mask)
 
 def blend_images(base, overlay, mask):