Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from PIL import Image
 import rembg
 import io
+import tempfile
 
 def process_image(image, outline_size, outline_color, add_outline):
     if image is None:
@@ -41,6 +42,15 @@ def process_image(image, outline_size, outline_color, add_outline):
 
     return final_img
 
+# Handle download
+def download_sticker(image):
+    if image:
+        # Save image to a temporary file
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+        image.save(temp_file.name, format="PNG")
+        temp_file.close()
+        return temp_file.name  # Return the file path
+
 # Gradio Interface
 with gr.Blocks(title="Sticker Maker") as interface:
     gr.Markdown("# Sticker Maker")
@@ -76,14 +86,7 @@ with gr.Blocks(title="Sticker Maker") as interface:
         outputs=output_image,
     )
 
-    # Handle download
-    def download_sticker(image):
-        if image:
-            img_byte_arr = io.BytesIO()
-            image.save(img_byte_arr, format="PNG")
-            img_byte_arr.seek(0)
-            return img_byte_arr
-
+    # Enable download functionality
     download_btn.click(
         fn=download_sticker,
         inputs=output_image,