Update app.py

app.py

@@ -111,6 +111,7 @@ def prepare_image_and_mask(image, width, height, overlap_percentage, resize_opti
     scale_factor = min(target_size[0] / image.width, target_size[1] / image.height)
     new_width = int(image.width * scale_factor)
     new_height = int(image.height * scale_factor)
+    
     source = image.resize((new_width, new_height), Image.LANCZOS)
     
     if resize_option == "Full":
@@ -166,14 +167,17 @@ def prepare_image_and_mask(image, width, height, overlap_percentage, resize_opti
         background = Image.new("RGBA", target_size, (255, 255, 255, 255)
     else:
         background = Image.new("RGB", target_size, (255, 255, 255)
+    
+    background.paste(source, (margin_x, margin_y))
 
-    background = Image.new('RGB', target_size, (255, 255, 255))
+    # background = Image.new('RGB', target_size, (255, 255, 255))
     # Paste the source (resized image) onto the background
-    background.paste(source, (margin_x, margin_y)
+    # Paste the source image (resized) onto the background
+    # If the source has alpha, it will be preserved in the background
+                               
 
     # Create the generated mask (L mode)
     mask = Image.new("L", target_size, 255)
-    
     mask_draw = ImageDraw.Draw(mask)
     white_gaps_patch = 2