Update app.py

app.py

@@ -44,7 +44,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, output_buffer  # Return image and buffer for download
+    return final_img
 
 # Gradio Interface
 with gr.Blocks(title="Sticker Maker") as interface:
@@ -59,7 +59,6 @@ with gr.Blocks(title="Sticker Maker") as interface:
                 label="Outline Thickness", minimum=0, maximum=20, value=5, step=1
             )
             outline_color = gr.ColorPicker(label="Outline Color", value="#FFFFFF")
-            create_btn = gr.Button("Create Sticker", variant="primary")
 
         with gr.Column():
             output_image = gr.Image(
@@ -73,18 +72,20 @@ with gr.Blocks(title="Sticker Maker") as interface:
         outputs=[outline_size, outline_color],
     )
 
-    # Process image and allow direct download from preview pane
+    # 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
+
+    create_btn = gr.Button("Create Sticker", variant="primary")
     create_btn.click(
-        fn=lambda *args: process_image(*args)[0],
+        fn=process_and_return,
         inputs=[image_upload, outline_size, outline_color, add_outline],
         outputs=output_image,
     )
 
-    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="Download Sticker 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)