Add RGBA PNG download button

app.py

@@ -4,7 +4,7 @@ from gradio_imageslider import ImageSlider
 import torch
 from diffusers import DiffusionPipeline
 from PIL import Image
-import os
+import tempfile
 
 # Load the model
 print("Loading Fibo-Edit-RMBG model...")
@@ -25,7 +25,7 @@ def process_image(image, num_steps=10, guidance_scale=1.0):
     Process an image with Fibo-Edit-RMBG
     """
     if image is None:
-        return None, "Please upload an image"
+        return None, None, "Please upload an image"
 
     # Fixed instruction for background removal
     instruction = "\"{'edit_instruction':'Generate a detailed grayscale alpha matte. Map the opaque foreground to white and the background to black. Produce soft, anti-aliased grayscale gradients at the edges of the subject to represent fine details and transparency.'}\""
@@ -52,10 +52,13 @@ def process_image(image, num_steps=10, guidance_scale=1.0):
         # Use the mask as the alpha channel
         original_rgba.putalpha(mask_gray)
 
-        return (original_rgba, image), "✅ Image processed successfully!"
+        tmp = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
+        original_rgba.save(tmp.name, format="PNG")
+
+        return (original_rgba, image), tmp.name, "✅ Image processed successfully!"
 
     except Exception as e:
-        return None, f"❌ Error: {str(e)}"
+        return None, None, f"❌ Error: {str(e)}"
 
 # Create Gradio interface
 with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal") as demo:
@@ -106,6 +109,8 @@ with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal") as demo:
                 height=400
             )
 
+            download_file = gr.File(label="Download PNG (with transparency)")
+
             status_text = gr.Textbox(
                 label="Status",
                 interactive=False,
@@ -116,7 +121,7 @@ with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal") as demo:
     process_btn.click(
         fn=process_image,
         inputs=[input_image, num_steps, guidance_scale],
-        outputs=[output_image, status_text]
+        outputs=[output_image, download_file, status_text]
     )
 
     gr.Markdown("""