Remove-Background-From-Images-CPU

Sleeping

App Files Files Community

James040 commited on Jan 30

Commit

bd54bb5

verified ·

1 Parent(s): 1bbc284

Update app.py

Browse files

Files changed (1) hide show

app.py +63 -111

app.py CHANGED Viewed

@@ -1,109 +1,28 @@
-from flask import Flask, request, Response
-from rembg import remove
-from rembg.session_factory import new_session
 import io
 app = Flask(__name__)
-# FULL U²-Net
 session = new_session("u2net")
 HTML = """
 <!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Xlnk RMBG - AI Background Remover</title>
-      <link rel="icon" type="image/svg+xml" href="https://me.xo.je/icon.svg">
-    <meta name="description" content="Remove backgrounds from images instantly with Xlnk RMBG. Drag & drop images from anywhere, including Google Images. Auto-download transparent PNGs.">
-    <meta name="keywords" content="background removal, AI background remover, remove background online, transparent PNG, drag drop image, U2-Net, Xlnk RMBG">
-    <meta name="robots" content="index, follow">
-    <!-- Open Graph for social sharing -->
-    <meta property="og:title" content="Xlnk RMBG - AI Background Remover">
-    <meta property="og:description" content="Drag & drop images from anywhere and remove the background instantly. Full U²-Net AI engine.">
-    <meta property="og:type" content="website">
-    <meta property="og:image" content="https://huggingface.co/spaces/Xlnk/rmbg/resolve/main/preview.png">
-    <meta property="og:url" content="https://huggingface.co/spaces/Xlnk/rmbg">
-    <style>
-        html, body {
-            margin: 0;
-            height: 100%;
-            background: #0b0f19;
-            display: flex;
-            justify-content: center;
-            align-items: center;
-            color: white;
-            font-family: sans-serif;
-        }
-        #dropZone {
-            width: 100vw;
-            height: 100vh;
-            border: 2px dashed #555;
-            display: flex;
-            justify-content: center;
-            align-items: center;
-            cursor: pointer;
-            text-align: center;
-        }
-        #dropZone.dragover { background-color: #1b2030; }
-        input { display: none; }
-    </style>
-</head>
-<body>
-<div id="dropZone">Drop image from anywhere</div>
-<form id="uploadForm" method="post" enctype="multipart/form-data">
-    <input type="file" id="fileInput" name="file" />
-</form>
-<script>
-const dropZone = document.getElementById("dropZone");
-const fileInput = document.getElementById("fileInput");
-const form = document.getElementById("uploadForm");
-dropZone.addEventListener("click", () => fileInput.click());
-fileInput.addEventListener("change", () => {
-    if(fileInput.files.length > 0) form.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");
-    // Local files
-    if (e.dataTransfer.files.length > 0) {
-        fileInput.files = e.dataTransfer.files;
-        form.submit();
-    }
-    // Drag from browser (Google Images, etc)
-    else {
-        let url = e.dataTransfer.getData('text/uri-list');
-        if(url){
-            fetch(url)
-            .then(res => res.blob())
-            .then(blob => {
-                let dt = new DataTransfer();
-                dt.items.add(new File([blob], "image.png", {type: blob.type}));
-                fileInput.files = dt.files;
-                form.submit();
-            });
-        }
-    }
-});
-</script>
 </body>
 </html>
 """
@@ -111,18 +30,51 @@ dropZone.addEventListener("drop", e => {
 @app.route("/", methods=["GET", "POST"])
 def index():
     if request.method == "POST":
-        file = request.files.get("file")
-        if not file:
-            return "No file", 400
-        input_bytes = file.read()
-        output_bytes = remove(input_bytes, session=session, alpha_matting=False)
-        return Response(
-            output_bytes,
-            mimetype="image/png",
-            headers={"Content-Disposition": "attachment; filename=xlnk-rmbg.png"}
-        )
     return HTML

+import os
 import io
+import cv2
+import zipfile
+import shutil
+from flask import Flask, request, send_file, render_template_string
+from rembg import remove, new_session
+from PIL import Image
 app = Flask(__name__)
 session = new_session("u2net")
+# Simple UI for Video Upload
 HTML = """
 <!DOCTYPE html>
+<html>
+<head><title>Video Background Remover</title></head>
+<body style="background:#0b0f19; color:white; font-family:sans-serif; text-align:center; padding:50px;">
+    <h1>Video to Transparent PNG Frames</h1>
+    <p>Upload a video (MP4/MOV). We will zip the transparent frames for you.</p>
+    <form method="post" enctype="multipart/form-data">
+        <input type="file" name="video" accept="video/*">
+        <input type="submit" value="Process and Download ZIP">
+    </form>
+    <p style="color:#888; font-size:0.8em;">Note: Processing takes ~1 sec per frame on CPU.</p>
 </body>
 </html>
 """
 @app.route("/", methods=["GET", "POST"])
 def index():
     if request.method == "POST":
+        file = request.files.get("video")
+        if not file: return "No file", 400
+        # Create temporary working directories
+        temp_dir = "temp_frames"
+        if os.path.exists(temp_dir): shutil.rmtree(temp_dir)
+        os.makedirs(temp_dir)
+        # Save uploaded video temporarily
+        video_path = "input_video.mp4"
+        file.save(video_path)
+        # Process Video
+        cap = cv2.VideoCapture(video_path)
+        frame_count = 0
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret: break
+            # Convert BGR to RGB for rembg
+            frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            # Remove background
+            output = remove(frame_rgb, session=session)
+            # Save frame as PNG
+            frame_name = f"frame_{frame_count:04d}.png"
+            Image.fromarray(output).save(os.path.join(temp_dir, frame_name))
+            frame_count += 1
+        cap.release()
+        # Create ZIP file
+        zip_path = "processed_frames.zip"
+        with zipfile.ZipFile(zip_path, 'w') as zipf:
+            for root, dirs, files in os.walk(temp_dir):
+                for f in files:
+                    zipf.write(os.path.join(root, f), f)
+        # Cleanup
+        shutil.rmtree(temp_dir)
+        os.remove(video_path)
+        return send_file(zip_path, as_attachment=True)
     return HTML