Update app.py

app.py

@@ -77,17 +77,23 @@ def remove_background():
         # Remove background using withoutBG
         logger.info("🔄 Removing background...")
         result = model.remove_background(img)
-        logger.info("✅ Background removed successfully!")
+        logger.info(f"✅ Background removed successfully! Mode: {result.mode}, Size: {result.size}")
         
-        # Ensure output is RGB with white background
-        if result.mode == 'RGBA':
-            # Create white background
-            background = Image.new('RGB', result.size, (255, 255, 255))
-            # Paste the image with alpha as mask
-            background.paste(result, mask=result.split()[3])
-            result = background
-        elif result.mode != 'RGB':
-            result = result.convert('RGB')
+        # Convert to RGBA first if not already
+        if result.mode != 'RGBA':
+            result = result.convert('RGBA')
+            logger.info(f"🔄 Converted to RGBA mode")
+        
+        # Create white background and composite
+        logger.info("🎨 Creating white background composite...")
+        white_bg = Image.new('RGBA', result.size, (255, 255, 255, 255))
+        
+        # Composite the image onto white background
+        output = Image.alpha_composite(white_bg, result)
+        
+        # Convert to RGB (remove alpha channel)
+        result = output.convert('RGB')
+        logger.info(f"✅ Final image mode: {result.mode}")
         
         # Convert to PNG bytes
         output_buffer = io.BytesIO()