Update app.py

app.py

@@ -126,6 +126,12 @@ def main():
                 try:
                     # Convert image to base64 for API
                     buffered = BytesIO()
+                    # Convert RGBA to RGB if needed for JPEG
+                    if image.mode == 'RGBA':
+                        # Create white background
+                        rgb_image = Image.new('RGB', image.size, (255, 255, 255))
+                        rgb_image.paste(image, mask=image.split()[-1])  # Use alpha channel as mask
+                        image = rgb_image
                     image.save(buffered, format="JPEG")
                     image_base64 = base64.b64encode(buffered.getvalue()).decode()