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
+    return final_img, output_buffer  # Return image and buffer for download
 
 # Gradio Interface
 with gr.Blocks(title="Sticker Maker") as interface:
@@ -59,6 +59,7 @@ 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(
@@ -72,74 +73,17 @@ with gr.Blocks(title="Sticker Maker") as interface:
         outputs=[outline_size, outline_color],
     )
 
-    # 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")
+    # Process image and allow direct download from preview pane
     create_btn.click(
-        fn=process_and_return,
+        fn=lambda *args: process_image(*args)[0],
         inputs=[image_upload, outline_size, outline_color, add_outline],
         outputs=output_image,
     )
 
-    # Enable PNG download directly from the preview pane
-    output_image.download(label="Download Sticker as PNG")
-
-if __name__ == "__main__":
-    interface.launch(debug=True)
-        # Paste the processed image onto the background
-        background.paste(processed_img, (outline_size, outline_size), processed_img)
-        final_img = background
-    else:
-        final_img = processed_img
-
-    # Save the result as bytes for Gradio
-    output_buffer = io.BytesIO()
-    final_img.save(output_buffer, format="PNG")
-    output_buffer.seek(0)
-
-    return final_img, output_buffer
-
-# Gradio Interface
-with gr.Blocks(title="Sticker Maker") as interface:
-    gr.Markdown("# Sticker Maker")
-    gr.Markdown("Upload an image to remove the background and optionally add an outline.")
-
-    with gr.Row():
-        with gr.Column():
-            image_upload = gr.Image(label="Upload Image", type="pil")
-            add_outline = gr.Checkbox(label="Add Outline", value=True)
-            outline_size = gr.Slider(
-                label="Outline Thickness", minimum=0, maximum=20, value=5, step=1
-            )
-            outline_color = gr.ColorPicker(label="Outline Color", value="#FFFFFF")
-
-        with gr.Column():
-            output_image = gr.Image(
-                label="Preview", type="pil", interactive=False
-            )
-
-    # Show/hide outline options based on checkbox
-    add_outline.change(
-        fn=lambda x: gr.update(visible=x),
-        inputs=add_outline,
-        outputs=[outline_size, outline_color],
-    )
-
-    # Process image and show result in the preview pane
-    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_download,
-        inputs=[image_upload, outline_size, outline_color, add_outline],
-        outputs=[output_image, gr.File(label="Download as PNG")],
+    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"),
     )
 
 if __name__ == "__main__":