Update app.py

app.py

@@ -3,6 +3,8 @@ from PIL import Image, ImageChops, ImageFilter
 import rembg
 import io
 import zipfile
+import tempfile
+import os
 
 def remove_background(image):
     """Remove background using rembg"""
@@ -28,28 +30,31 @@ def add_outline(image, outline_width=2, outline_color=(255, 255, 255)):
 def process_images(image_files):
     processed = []
     for img_file in image_files:
-        img = Image.open(img_file).convert("RGBA")
+        img = Image.open(img_file.name).convert("RGBA")  # Use .name for Gradio file paths
         no_bg = remove_background(img)
         sticker = add_outline(no_bg)
         processed.append(sticker)
     
-    # Return single PNG or ZIP
     if len(processed) == 1:
+        # Return single PNG
         buf = io.BytesIO()
         processed[0].save(buf, format='PNG')
-        return buf.getvalue()
+        buf.seek(0)
+        return buf
     else:
+        # Return ZIP
         zip_buf = io.BytesIO()
         with zipfile.ZipFile(zip_buf, 'w') as zip_file:
             for i, sticker in enumerate(processed):
                 buf = io.BytesIO()
                 sticker.save(buf, format='PNG')
-                zip_file.writestr(f'sticker_{i+1}.png', buf.getvalue())
+                buf.seek(0)
+                zip_file.writestr(f'sticker_{i+1}.png', buf.read())
         zip_buf.seek(0)
         return zip_buf
 
 # Gradio Interface
-image_upload = gr.Files(label="Upload Images (Single or Batch)", file_types=["image"])
+image_upload = gr.File(label="Upload Images (Single or Batch)", file_types=["image"], file_count="multiple")
 output = gr.File(label="Download Sticker(s)")
 
 interface = gr.Interface(