Update app.py

app.py

@@ -1,17 +1,16 @@
 import gradio as gr
 from PIL import Image, ImageChops, ImageFilter
 import rembg
-import io
-import zipfile
 import tempfile
 import os
+import zipfile
 
 def remove_background(image):
     """Remove background using rembg"""
-    img_byte_arr = io.BytesIO()
-    image.save(img_byte_arr, format='PNG')
-    result = rembg.remove(img_byte_arr.getvalue())
-    return Image.open(io.BytesIO(result)).convert("RGBA")
+    with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as temp:
+        image.save(temp.name)
+        result = rembg.remove(temp.name)
+    return Image.open(result).convert("RGBA")
 
 def add_outline(image, outline_width=2, outline_color=(255, 255, 255)):
     """Add white outline to transparent image"""
@@ -28,33 +27,43 @@ def add_outline(image, outline_width=2, outline_color=(255, 255, 255)):
     return final
 
 def process_images(image_files):
-    processed = []
-    for img_file in image_files:
-        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)
-    
-    if len(processed) == 1:
-        # Return single PNG
-        buf = io.BytesIO()
-        processed[0].save(buf, format='PNG')
-        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')
-                buf.seek(0)
-                zip_file.writestr(f'sticker_{i+1}.png', buf.read())
-        zip_buf.seek(0)
-        return zip_buf
+    # Create temporary directory
+    temp_dir = tempfile.mkdtemp()
+    try:
+        processed_paths = []
+        for i, img_file in enumerate(image_files):
+            # Process image
+            img = Image.open(img_file.name).convert("RGBA")
+            no_bg = remove_background(img)
+            sticker = add_outline(no_bg)
+            
+            # Save to temporary file
+            file_path = os.path.join(temp_dir, f"sticker_{i+1}.png")
+            sticker.save(file_path)
+            processed_paths.append(file_path)
+        
+        # Return single file or ZIP
+        if len(processed_paths) == 1:
+            return processed_paths[0]
+        else:
+            zip_path = os.path.join(temp_dir, "stickers.zip")
+            with zipfile.ZipFile(zip_path, 'w') as zipf:
+                for p in processed_paths:
+                    zipf.write(p, os.path.basename(p))
+            return zip_path
+    except Exception as e:
+        # Cleanup on error
+        for p in processed_paths:
+            try:
+                os.remove(p)
+            except: pass
+        raise e
 
 # Gradio Interface
-image_upload = gr.File(label="Upload Images (Single or Batch)", file_types=["image"], file_count="multiple")
+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(
@@ -63,6 +72,7 @@ interface = gr.Interface(
     outputs=output,
     title="✨ Sticker Maker",
     description="Remove background, add white outline, and download as PNG(s)",
+    allow_flagging="never"
 )
 
 if __name__ == "__main__":