Update app.py

app.py

@@ -47,11 +47,12 @@ def fit_and_warp_design(image_a, image_b, top_left_x, top_left_y, bottom_right_x
     # Resize the design to fit within the specified bounding box
     design_width = bottom_right[0] - top_left[0]
     design_height = bottom_right[1] - top_left[1]
+    image_b = image_b.convert("RGBA")  # Ensure design image has an alpha channel
     image_b = image_b.resize((design_width, design_height))
     
     # Create a blank canvas with the same size as Image-A
     canvas = Image.new('RGBA', (displacement_map.shape[1], displacement_map.shape[0]), (0, 0, 0, 0))
-    canvas.paste(image_b, top_left, image_b)
+    canvas.paste(image_b, top_left, image_b)  # Use image_b as the mask for pasting
     canvas_np = np.array(canvas)
 
     h, w = displacement_map.shape