Update app.py

app.py

@@ -3,7 +3,6 @@ from PIL import Image
 import rembg
 import io
 import os
-import tempfile
 
 def process_image(image, outline_size, outline_color, add_outline):
     if image is None:
@@ -38,12 +37,11 @@ def process_image(image, outline_size, outline_color, add_outline):
     else:
         final_img = processed_img
     
-    # Use a temporary file with .png extension to ensure correct format
-    temp_dir = tempfile.gettempdir()
-    output_path = os.path.join(temp_dir, "sticker.png")
-    final_img.save(output_path, format="PNG")
+    # Save as PNG
+    output_file = "sticker.png"
+    final_img.save(output_file, format="PNG")
     
-    return output_path
+    return final_img
 
 # Gradio Interface
 with gr.Blocks(title="Sticker Maker") as interface:
@@ -57,10 +55,11 @@ with gr.Blocks(title="Sticker Maker") as interface:
             outline_size = gr.Slider(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")
+            download_btn = gr.Button("Download Sticker", visible=False)
         
         with gr.Column():
-            # Single output component for preview and download
-            output_image = gr.Image(label="Transparent Background", type="filepath", container=False)
+            # Output image is visible from the start
+            output_image = gr.Image(label="Preview with Transparent Background", type="pil")
     
     # Make outline options visible only when add_outline is checked
     add_outline.change(
@@ -69,11 +68,26 @@ with gr.Blocks(title="Sticker Maker") as interface:
         outputs=[outline_size, outline_color]
     )
     
+    # Process image and show result
     create_btn.click(
         fn=process_image,
         inputs=[image_upload, outline_size, outline_color, add_outline],
         outputs=output_image
+    ).then(
+        fn=lambda: gr.update(visible=True),
+        inputs=None,
+        outputs=download_btn
+    )
+    
+    # Handle manual download
+    def download_sticker():
+        return "sticker.png"
+    
+    download_btn.click(
+        fn=download_sticker,
+        inputs=None,
+        outputs=gr.File(label="Download")
     )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(debug=True)