Update app.py

app.py

@@ -5,7 +5,7 @@ import io
 
 def process_image(image, outline_size):
     if image is None:
-        return None
+        return None, None  # Return None for both image and file
 
     # Convert image to bytes for rembg
     img_byte_arr = io.BytesIO()
@@ -16,42 +16,40 @@ def process_image(image, outline_size):
     processed_bytes = rembg.remove(img_byte_arr.getvalue())
     processed_img = Image.open(io.BytesIO(processed_bytes)).convert("RGBA")
 
-    # Add white outline if selected
+    # Add white outline
     if outline_size > 0:
-        # Create a larger background image with white color
         background = Image.new(
             "RGBA",
             (
                 processed_img.width + 2 * outline_size,
                 processed_img.height + 2 * outline_size,
             ),
-            (255, 255, 255, 0),  # White with alpha=0 for transparency
+            (0, 0, 0, 0),  # Transparent background
         )
-        # Paste the processed image onto the background
         background.paste(processed_img, (outline_size, outline_size), processed_img)
-        
-        # Draw a white outline around the image
-        pixels = background.load()
+
+        # Draw white outline
         for x in range(background.width):
             for y in range(background.height):
-                if x < outline_size or y < outline_size or x >= background.width - outline_size or y >= background.height - outline_size:
-                    pixels[x, y] = (255, 255, 255, 255)  # White with alpha=255 for visibility
-        
+                if (x < outline_size or y < outline_size or x >= background.width - outline_size or y >= background.height - outline_size):
+                     if background.getpixel((x, y))[3] == 0:  #Check if pixel is originally transparent
+                        background.putpixel((x, y), (255, 255, 255, 255)) #Set to white if so
         final_img = background
+
     else:
         final_img = processed_img
 
-    # Save the result as bytes for Gradio
+    # Save as PNG to a BytesIO object
     output_buffer = io.BytesIO()
     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, 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 a white outline.")
+    gr.Markdown("Upload an image to remove the background and add a white outline.")
 
     with gr.Row():
         with gr.Column():
@@ -63,13 +61,24 @@ with gr.Blocks(title="Sticker Maker") as interface:
 
         with gr.Column():
             output_image = gr.Image(label="Preview", type="pil", interactive=False)
-            download_file = gr.File(label="Download Sticker as PNG")
+            download_link = gr.HTML("<a style='display: none;' id='download-link'>Download Sticker</a>")
+
+    def update_outputs(image, outline_size):
+        processed_image, image_bytes = process_image(image, outline_size)
+        if processed_image is None:
+            return None, "<a style='display: none;' id='download-link'>Download Sticker</a>"
+
+        # Create a data URL for the image
+        data_url = f"data:image/png;base64,{image_bytes.getvalue().hex()}"
+
+        # Create the download link
+        download_html = f"<a href='{data_url}' download='sticker.png'>Download Sticker</a>"
+        return processed_image, download_html
 
-    # Process image and allow direct download from preview pane
     create_btn.click(
-        fn=process_image,
+        fn=update_outputs,
         inputs=[image_upload, outline_size],
-        outputs=[output_image, download_file],
+        outputs=[output_image, download_link],
     )
 
 if __name__ == "__main__":