Update app.py

app.py

@@ -2,7 +2,6 @@ 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:
@@ -40,16 +39,12 @@ def process_image(image, outline_size, outline_color, add_outline):
     else:
         final_img = processed_img
 
-    return final_img
+    # Save the result as bytes for Gradio
+    output_buffer = io.BytesIO()
+    final_img.save(output_buffer, format="PNG")
+    output_buffer.seek(0)
 
-# 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
+    return final_img, output_buffer  # Return image and buffer for download
 
 # Gradio Interface
 with gr.Blocks(title="Sticker Maker") as interface:
@@ -70,7 +65,6 @@ with gr.Blocks(title="Sticker Maker") as interface:
             output_image = gr.Image(
                 label="Preview and Download", type="pil", interactive=False
             )
-            download_btn = gr.Button("Download Sticker", variant="secondary")
 
     # Show/hide outline options based on checkbox
     add_outline.change(
@@ -79,18 +73,17 @@ with gr.Blocks(title="Sticker Maker") as interface:
         outputs=[outline_size, outline_color],
     )
 
-    # Process image and show result
+    # Process image and allow direct download from preview pane
     create_btn.click(
-        fn=process_image,
+        fn=lambda *args: process_image(*args)[0],
         inputs=[image_upload, outline_size, outline_color, add_outline],
         outputs=output_image,
     )
 
-    # Enable download functionality
-    download_btn.click(
-        fn=download_sticker,
+    output_image.change(
+        fn=lambda img: process_image(img, outline_size=0, outline_color="#FFFFFF", add_outline=False)[1],
         inputs=output_image,
-        outputs=gr.File(label="Sticker.png"),
+        outputs=gr.File(label="Download Sticker as PNG"),
     )
 
 if __name__ == "__main__":