Update app.py

app.py

@@ -16,11 +16,11 @@ def remove_background(image):
     output_path = "output.png"
     output_img.save(output_path, format="PNG")
 
-    return output_path, "Background removed successfully!", output_path
+    return output_path, "Background removed successfully!"
 
-# Example images (replace with your actual image paths in the 'examples' folder)
+# Example images
 example_images = [
-    "examples/1.jpg",
+    "examples/1.jpeg",
     "examples/2.jpg",
     "examples/3.jpg",
     "examples/4.jpg",
@@ -31,19 +31,22 @@ example_images = [
 with gr.Blocks(title="Background Remover") as demo:
     gr.Markdown("# Background Remover")
     gr.Markdown("""
-    Upload an image or use an example to remove its background.
+    Upload an image or use an example to remove its background.  
     PS: Not the best image background remover out there but good enough for a free tool.
     """)
 
     with gr.Row():
         with gr.Column():
-            image_input = gr.Image(type="pil", label="Input Image")  # FIXED: removed tool="editor"
+            image_input = gr.Image(type="pil", label="Input Image")
             submit_btn = gr.Button("Submit")
             clear_btn = gr.Button("Clear")
         with gr.Column():
             output_image = gr.Image(label="Output", type="filepath")
             output_text = gr.Textbox(label="Status", lines=1)
-            download_output = gr.File(label="Download Output")
+            download_area = gr.Column(visible=False)
+            with download_area:
+                gr.Markdown("### Download Output")
+                download_output = gr.File(label="", file_types=[".png"])
 
     with gr.Row():
         gr.Markdown("### Example Images")
@@ -54,8 +57,12 @@ with gr.Blocks(title="Background Remover") as demo:
             label="Click to try an example",
         )
 
-    # Logic binding
-    submit_btn.click(fn=remove_background, inputs=image_input, outputs=[output_image, output_text, download_output])
-    clear_btn.click(fn=lambda: (None, "", None), outputs=[image_input, output_text, download_output])
+    # Button logic
+    def process(img):
+        output_path, msg = remove_background(img)
+        return output_path, msg, gr.update(visible=True), output_path
+
+    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])
 
 demo.launch()