Update app.py

app.py

@@ -4,6 +4,8 @@ from PIL import Image
 import io
 import os
 
+OUTPUT_FILE = "output.png"
+
 # Background removal function
 def remove_background(image):
     input_bytes = io.BytesIO()
@@ -12,11 +14,9 @@ def remove_background(image):
 
     output_bytes = remove(input_bytes)
     output_img = Image.open(io.BytesIO(output_bytes))
+    output_img.save(OUTPUT_FILE, format="PNG")
 
-    output_path = "output.png"
-    output_img.save(output_path, format="PNG")
-
-    return output_path, "Background removed successfully!"
+    return OUTPUT_FILE, "Background removed successfully!"
 
 # Example images
 example_images = [
@@ -43,13 +43,10 @@ with gr.Blocks(title="Background Remover") as demo:
         with gr.Column():
             output_image = gr.Image(label="Output", type="filepath")
             output_text = gr.Textbox(label="Status", lines=1)
-            download_area = gr.Column(visible=False)
-            with download_area:
-                gr.Markdown("### Download Output")
-                download_output = gr.File(label="", file_types=[".png"])
+            download_html = gr.HTML(visible=False)
 
     with gr.Row():
-        gr.Markdown("###  Example Images")
+        gr.Markdown("### Example Images")
     with gr.Row():
         gr.Examples(
             examples=example_images,
@@ -57,12 +54,25 @@ with gr.Blocks(title="Background Remover") as demo:
             label="Click to try an example",
         )
 
-    # Button logic
+    # Logic
     def process(img):
         output_path, msg = remove_background(img)
-        return output_path, msg, gr.update(visible=True), output_path
+        download_button = f"""
+        <a href="file/{output_path}" download style="
+            display:inline-block;
+            padding: 10px 20px;
+            background-color: #4CAF50;
+            color: white;
+            text-align: center;
+            text-decoration: none;
+            border-radius: 5px;
+            font-weight: bold;
+            margin-top: 10px;
+        ">⬇ Download Output</a>
+        """
+        return output_path, msg, gr.update(visible=True, value=download_button)
 
-    submit_btn.click(fn=process, inputs=image_input, outputs=[output_image, output_text, download_area, download_output])
-    clear_btn.click(fn=lambda: (None, "", gr.update(visible=False), None), outputs=[image_input, output_text, download_area, download_output])
+    submit_btn.click(fn=process, inputs=image_input, outputs=[output_image, output_text, download_html])
+    clear_btn.click(fn=lambda: (None, "", gr.update(visible=False, value="")), outputs=[image_input, output_text, download_html])
 
 demo.launch()