Update app.py

app.py

@@ -1,60 +1,89 @@
 import gradio as gr
 from rembg import remove
 from rembg.session_factory import new_session
+from PIL import Image
+import io, base64, os
 
-# Load FULL U²-Net ONCE (prevents HF timeout)
+PORT = int(os.environ.get("PORT", 7860))
+
+# 🧠 FULL MODEL (not lite)
 session = new_session("u2net")
 
-def rmbg(file):
-    if file is None:
-        return None
+def process(image):
+    if image is None:
+        return None, ""
 
-    input_bytes = file.read()
+    # PIL → bytes
+    buf = io.BytesIO()
+    image.save(buf, format="PNG")
+    input_bytes = buf.getvalue()
 
+    # Background removal (FULL U²-Net)
     output_bytes = remove(
         input_bytes,
         session=session,
         alpha_matting=False
     )
 
-    return ("xlnk-rmbg.png", output_bytes)
+    # Bytes → PIL
+    out_img = Image.open(io.BytesIO(output_bytes)).convert("RGBA")
+
+    # PIL → base64 (auto-download)
+    out_buf = io.BytesIO()
+    out_img.save(out_buf, format="PNG")
+    b64 = base64.b64encode(out_buf.getvalue()).decode()
+
+    html = f"""
+    <script>
+      const a = document.createElement("a");
+      a.href = "data:image/png;base64,{b64}";
+      a.download = "xlnk-rmbg.png";
+      document.body.appendChild(a);
+      a.click();
+      a.remove();
+    </script>
+    """
+
+    return out_img, html
+
 
 with gr.Blocks(
+    title="Xlnk Rmbg – AI Background Remover",
     css="""
-    html, body, .gradio-container {
-        height: 100%;
-        background: #0b0f19;
-    }
-    #drop {
-        border: 2px dashed #555;
-        height: 85vh;
-        display: flex;
-        align-items: center;
-        justify-content: center;
-        font-size: 32px;
-        color: white;
-        cursor: pointer;
-    }
+    body { background:#0b0f19; }
+    h1, h2, p { color:white; }
     """
 ) as demo:
 
-    gr.Markdown(
-        "<h1 style='text-align:center;'>Xlnk rmbg</h1>"
-        "<p style='text-align:center;'>Drag & drop → auto download</p>"
-    )
+    gr.Markdown("""
+    # ✂️ Xlnk Rmbg  
+    **High-Quality AI Background Removal (U²-Net)**
 
-    upload = gr.File(
-        file_types=["image"],
-        label="",
-        elem_id="drop"
-    )
+    Drag & drop → preview → auto-download  
+    No storage. No tracking.
+    """)
 
-    output = gr.File(visible=False)
+    with gr.Row():
+        input_img = gr.Image(type="pil", label="Upload image")
+        output_img = gr.Image(type="pil", label="Preview (Transparent PNG)")
 
-    upload.change(
-        fn=rmbg,
-        inputs=upload,
-        outputs=output
+    html = gr.HTML()
+
+    input_img.change(
+        fn=process,
+        inputs=input_img,
+        outputs=[output_img, html]
     )
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+    gr.Markdown("""
+    ### 🔒 Privacy
+    - Images processed in memory
+    - Nothing saved
+    - Open-source (Rembg + U²-Net)
+    """)
+
+demo.launch(
+    server_name="0.0.0.0",
+    server_port=PORT,
+    show_error=True
+)