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:
@@ -44,7 +45,7 @@ def process_image(image, outline_size, outline_color, add_outline):
     final_img.save(output_buffer, format="PNG")
     output_buffer.seek(0)
 
-    return final_img
+    return final_img, output_buffer
 
 # Gradio Interface
 with gr.Blocks(title="Sticker Maker") as interface:
@@ -62,7 +63,7 @@ with gr.Blocks(title="Sticker Maker") as interface:
 
         with gr.Column():
             output_image = gr.Image(
-                label="Preview and Download", type="pil", interactive=False
+                label="Preview", type="pil", interactive=False
             )
 
     # Show/hide outline options based on checkbox
@@ -73,19 +74,18 @@ with gr.Blocks(title="Sticker Maker") as interface:
     )
 
     # Process image and show result in the preview pane
-    def process_and_return(image, outline_size, outline_color, add_outline):
-        processed_image = process_image(image, outline_size, outline_color, add_outline)
-        return processed_image  # Return processed image for preview
+    def process_and_download(image, outline_size, outline_color, add_outline):
+        processed_image, output_buffer = process_image(image, outline_size, outline_color, add_outline)
+        # Return the processed image for preview and the PNG file for download
+        return processed_image, output_buffer
 
+    # Add button to process image and provide preview
     create_btn = gr.Button("Create Sticker", variant="primary")
     create_btn.click(
-        fn=process_and_return,
+        fn=process_and_download,
         inputs=[image_upload, outline_size, outline_color, add_outline],
-        outputs=output_image,
+        outputs=[output_image, gr.File(label="Download as PNG")],
     )
 
-    # Enable PNG download directly from the preview pane
-    output_image.download(label="Download Sticker as PNG")
-
 if __name__ == "__main__":
     interface.launch(debug=True)