Update app.py

app.py

@@ -1,48 +1,42 @@
 import gradio as gr
 from rembg import remove
 from PIL import Image
-import io
-import base64
+import io, base64, os
+
+PORT = int(os.environ.get("PORT", 7860))
 
 def remove_bg_auto_download(image):
-    # Remove background
     output = remove(image)
-    
-    # Save output to bytes
+
     buf = io.BytesIO()
     output.save(buf, format="PNG")
     buf.seek(0)
-    
-    # Encode as base64 for auto-download link
+
     b64 = base64.b64encode(buf.read()).decode()
-    download_link = f'data:image/png;base64,{b64}'
-    return download_link
+    return f"""
+    <script>
+      const a = document.createElement('a');
+      a.href = "data:image/png;base64,{b64}";
+      a.download = "no-bg.png";
+      document.body.appendChild(a);
+      a.click();
+      a.remove();
+    </script>
+    """
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Drag & Drop your image → Background removed & auto-download starts")
-    image_input = gr.Image(type="pil", label="Upload Image")
-    
-    # Custom HTML to trigger auto-download
-    html_output = gr.HTML()
-    
-    def update_download(image):
-        if image is None:
-            return ""
-        link = remove_bg_auto_download(image)
-        filename = "no-bg.png"
-        # Auto-download using JS
-        html_code = f"""
-        <script>
-        var a = document.createElement('a');
-        a.href = '{link}';
-        a.download = '{filename}';
-        document.body.appendChild(a);
-        a.click();
-        a.remove();
-        </script>
-        """
-        return html_code
-    
-    image_input.change(fn=update_download, inputs=image_input, outputs=html_output)
+    gr.Markdown("## Drag & Drop → Auto background removal & download")
+    image = gr.Image(type="pil", label="Drop image")
+    html = gr.HTML()
+
+    image.change(
+        fn=remove_bg_auto_download,
+        inputs=image,
+        outputs=html
+    )
 
-demo.launch()
+demo.launch(
+    server_name="0.0.0.0",
+    server_port=PORT,
+    show_error=True
+)