added white background

copaint/gradio_ui.py

@@ -26,15 +26,23 @@ fromTensortoPIL = torchvision.transforms.ToPILImage()
 
 from PIL import Image
 
-def remove_transparency(image: Image.Image):
+def load_with_transparency(image_file):
+    # Open manually to preserve RGBA
+    image = image_file #Image.open(image_file.name).convert("RGBA")
+    
+    # Example: return image size and check mode
+    print( f"Image mode: {image.mode}, size: {image.size}")
+
+    logger.info(f"TRANSPARENCY MODE: {image.mode}")
     # Convert transparency to white background
     if image.mode in ("RGBA", "LA"):
         background = Image.new("RGB", image.size, (255, 255, 255))  # white background
         background.paste(image, mask=image.split()[-1])  # paste using alpha channel as mask
         image = background
+        logger.info(f"TRANSPARENCY REMOVED WITH WHITE BACKGROUND")
     else:
+        logger.info(f"TRANSPARENCY REMOVED")
         image = image.convert("RGB")  # just to be safe
-
     return image  # or continue processing
 
 def add_grid_to_image(image, h_cells, w_cells):
@@ -189,11 +197,12 @@ def build_gradio_ui():
 
             # Upload Design Template
             with gr.Column(scale=2):
-                input_image = gr.Image(type="pil", label="Upload Your Design", preprocess=handle_upload)
-            image_input.upload(
-                fn=remove_transparency,
-                inputs=image_input,
-                outputs=image_input
+                input_image = gr.Image(type="pil", image_mode="RGBA", label="Upload Your Design")
+                
+            input_image.upload(
+                fn=load_with_transparency,
+                inputs=input_image,
+                outputs=input_image
             )
             with gr.Column(scale=1):
                 # Grid