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:
@@ -45,6 +44,62 @@ def process_image(image, outline_size, outline_color, add_outline):
     final_img.save(output_buffer, format="PNG")
     output_buffer.seek(0)
 
+    return final_img
+
+# 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 and Download", 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_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=process_and_return,
+        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