Update app.py

app.py

@@ -3,18 +3,27 @@ from rembg import remove
 from PIL import Image
 
 def remove_bg(input_image: Image.Image) -> Image.Image:
-    """Remove background from a PIL image and return the result."""
+    """Remove background from a PIL image and return the result with a white background."""
+    # remove original background (transparent)
     output = remove(input_image)
-    return output
+    # ensure output has alpha channel
+    if output.mode != "RGBA":
+        output = output.convert("RGBA")
+    # create white background image
+    white_bg = Image.new("RGBA", output.size, (255, 255, 255, 255))
+    # composite the output onto white background
+    white_bg.paste(output, (0, 0), output)
+    # convert to RGB, discarding alpha channel
+    return white_bg.convert("RGB")
 
 iface = gr.Interface(
     fn=remove_bg,
     inputs=gr.Image(type="pil", label="Upload any image"),
     outputs=gr.Image(type="pil", label="Background removed"),
     title="🖼️ Background Remover",
-    description="Upload an image and get it back with the background stripped away.",
+    description="Upload an image and get it back with the background stripped away and replaced with a white canvas.",
     examples=None,
 )
-    
+
 if __name__ == "__main__":
     iface.launch(server_name="0.0.0.0", server_port=7860)