Remove-Background-From-Images-CPU

Sleeping

App Files Files Community

Xlnk commited on Jan 24

Commit

3292460

verified ·

1 Parent(s): 62c5a78

Update app.py

Browse files

Files changed (1) hide show

app.py +98 -82

app.py CHANGED Viewed

@@ -1,89 +1,105 @@
-import gradio as gr
 from rembg import remove
 from rembg.session_factory import new_session
 from PIL import Image
-import io, base64, os
-PORT = int(os.environ.get("PORT", 7860))
-# 🧠 FULL MODEL (not lite)
 session = new_session("u2net")
-def process(image):
-    if image is None:
-        return None, ""
-    # 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
-    )
-    # 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="""
-    body { background:#0b0f19; }
-    h1, h2, p { color:white; }
-    """
-) as demo:
-    gr.Markdown("""
-    # ✂️ Xlnk Rmbg
-    **High-Quality AI Background Removal (U²-Net)**
-    Drag & drop → preview → auto-download
-    No storage. No tracking.
-    """)
-    with gr.Row():
-        input_img = gr.Image(type="pil", label="Upload image")
-        output_img = gr.Image(type="pil", label="Preview (Transparent PNG)")
-    html = gr.HTML()
-    input_img.change(
-        fn=process,
-        inputs=input_img,
-        outputs=[output_img, html]
-    )
-    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
-)

+from flask import Flask, request, Response
 from rembg import remove
 from rembg.session_factory import new_session
 from PIL import Image
+import io
+app = Flask(__name__)
+# FULL U²-Net
 session = new_session("u2net")
+HTML = """
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Xlnk RMBG</title>
+  <style>
+    html, body {
+      height: 100%;
+      margin: 0;
+      display: flex;
+      align-items: center;
+      justify-content: center;
+      background: #0b0f19;
+      color: white;
+      font-family: sans-serif;
+    }
+    #dropZone {
+      width: 100vw;
+      height: 100vh;
+      border: 2px dashed #555;
+      display: flex;
+      align-items: center;
+      justify-content: center;
+      cursor: pointer;
+      text-align: center;
+    }
+    #dropZone.dragover {
+      background-color: #1b2030;
+    }
+    input { display: none; }
+  </style>
+</head>
+<body>
+  <div id="dropZone">
+    <form id="uploadForm" method="post" enctype="multipart/form-data">
+      <input id="fileInput" type="file" name="file" />
+      Drop image or click anywhere
+    </form>
+  </div>
+<script>
+const dropZone = document.getElementById("dropZone");
+const fileInput = document.getElementById("fileInput");
+const uploadForm = document.getElementById("uploadForm");
+dropZone.addEventListener("click", () => fileInput.click());
+fileInput.addEventListener("change", () => {
+  if(fileInput.files.length > 0) uploadForm.submit();
+});
+dropZone.addEventListener("dragover", e => {
+  e.preventDefault();
+  dropZone.classList.add("dragover");
+});
+dropZone.addEventListener("dragleave", e => {
+  dropZone.classList.remove("dragover");
+});
+dropZone.addEventListener("drop", e => {
+  e.preventDefault();
+  dropZone.classList.remove("dragover");
+  fileInput.files = e.dataTransfer.files;
+  uploadForm.submit();
+});
+</script>
+</body>
+</html>
+"""
+@app.route("/", methods=["GET", "POST"])
+def index():
+    if request.method == "POST":
+        file = request.files.get("file")
+        if not file:
+            return "No file uploaded", 400
+        input_bytes = file.read()
+        # Remove background
+        output_bytes = remove(input_bytes, session=session, alpha_matting=False)
+        # Return as downloadable PNG
+        return Response(
+            output_bytes,
+            mimetype="image/png",
+            headers={"Content-Disposition": "attachment; filename=xlnk-rmbg.png"}
+        )
+    return HTML
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)