Spaces:

hackerbhai
/

RemBG_API

Paused

App Files Files Community

Arena Agent commited on 15 days ago

Commit

91f053c

1 Parent(s): a250d07

Build advanced background removal studio

Browse files

Files changed (4) hide show

Dockerfile +10 -6
README.md +16 -12
app.py +248 -422
requirements.txt +3 -1

Dockerfile CHANGED Viewed

@@ -2,20 +2,24 @@ FROM python:3.11-slim
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
-    PLAYWRIGHT_BROWSERS_PATH=/ms-playwright
 WORKDIR /app
 RUN apt-get update \
-    && apt-get install -y --no-install-recommends ca-certificates \
-    && rm -rf /var/lib/apt/lists/*
 COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip \
-    && pip install --no-cache-dir -r requirements.txt \
-    && python -m playwright install --with-deps chromium
 COPY app.py .
 EXPOSE 7860
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--threads", "4", "--timeout", "90", "app:app"]

 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
+    U2NET_HOME=/models
 WORKDIR /app
 RUN apt-get update \
+    && apt-get install -y --no-install-recommends \
+       ca-certificates \
+       libgl1 \
+       libglib2.0-0 \
+       libgomp1 \
+    && rm -rf /var/lib/apt/lists/* \
+    && mkdir -p /models
 COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip \
+    && pip install --no-cache-dir -r requirements.txt
 COPY app.py .
 EXPOSE 7860
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--threads", "2", "--timeout", "180", "app:app"]

README.md CHANGED Viewed

@@ -1,24 +1,28 @@
 ---
-title: Web Test Lab
-emoji: 🧪
-colorFrom: indigo
 colorTo: blue
 sdk: docker
 app_port: 7860
 pinned: false
 ---
-# Web Test Lab
-A safe screenshot-based website testing dashboard powered by Playwright and Chromium.
 ## Features
-- Test public HTTP and HTTPS websites
-- Desktop, tablet, and mobile viewport profiles
-- HTTP status, page title, load time, final URL, and screenshot
-- Search page at `/search`
-- Browser test page at `/`
-- Health check at `/health`
-The service returns screenshots and search results; it does not provide a remote interactive browser. Localhost and private-network addresses are blocked. Use it only on websites you own or are authorized to test.

 ---
+title: Background Studio
+emoji: ✂️
+colorFrom: purple
 colorTo: blue
 sdk: docker
 app_port: 7860
 pinned: false
 ---
+# Background Studio
+An advanced background-removal web app powered by `rembg`.
 ## Features
+- Drag-and-drop image upload
+- Quality, balanced, fast, people, and anime models
+- Optional alpha matting for hair, fur, and fine edges
+- Before/after comparison slider
+- Transparent PNG, WEBP, or JPG output
+- Custom JPG background color
+- 12 MB upload limit and basic rate limiting
+- Optional `API_KEY` Space Secret for API protection
+Open `/` for the web interface or call `POST /remove-bg` with a multipart `image` field.
+For a public Space, add `API_KEY` under **Settings → Variables and secrets** if you want the processing endpoint protected. Do not commit `.env` files or credentials.

app.py CHANGED Viewed

@@ -1,40 +1,33 @@
-import base64
-import ipaddress
-import socket
 import time
 from threading import Lock
-from time import perf_counter
-from urllib.parse import parse_qs, urlencode, urlparse
-from flask import Flask, jsonify, request
-from playwright.sync_api import TimeoutError as PlaywrightTimeoutError
-from playwright.sync_api import sync_playwright
 app = Flask(__name__)
-app.config["MAX_CONTENT_LENGTH"] = 32 * 1024
-TEST_TIMEOUT_MS = 20_000
-RATE_WINDOW_SECONDS = 60
-RATE_LIMIT = 12
 RATE_LOCK = Lock()
 RATE_BUCKETS = {}
-BROWSER_LOCK = Lock()
-PLAYWRIGHT = None
-BROWSER = None
-DEVICES = {
-    "desktop": {"viewport": {"width": 1440, "height": 900}},
-    "tablet": {
-        "viewport": {"width": 834, "height": 1112},
-        "is_mobile": True,
-        "has_touch": True,
-    },
-    "mobile": {
-        "viewport": {"width": 390, "height": 844},
-        "is_mobile": True,
-        "has_touch": True,
-    },
-}
 INDEX_HTML = r"""
 <!doctype html>
@@ -42,277 +35,176 @@ INDEX_HTML = r"""
 <head>
   <meta charset="utf-8">
   <meta name="viewport" content="width=device-width, initial-scale=1">
-  <title>Web Test Lab</title>
   <style>
-    :root {
-      color-scheme: dark;
-      --bg: #090d18;
-      --panel: rgba(20, 27, 49, .82);
-      --panel-strong: #151d35;
-      --line: rgba(148, 163, 184, .18);
-      --text: #f8fafc;
-      --muted: #93a4c3;
-      --blue: #6d8cff;
-      --cyan: #39d9c4;
-      --danger: #ff7b91;
-      --shadow: 0 24px 70px rgba(0, 0, 0, .35);
-    }
-    * { box-sizing: border-box; }
-    body {
-      margin: 0;
-      min-height: 100vh;
-      color: var(--text);
-      font: 15px/1.55 Inter, ui-sans-serif, system-ui, -apple-system, BlinkMacSystemFont, "Segoe UI", sans-serif;
-      background:
-        radial-gradient(circle at 15% 0%, rgba(109, 140, 255, .24), transparent 33rem),
-        radial-gradient(circle at 95% 15%, rgba(57, 217, 196, .13), transparent 28rem),
-        var(--bg);
-    }
-    a { color: #9bb5ff; }
-    .shell { width: min(1160px, calc(100% - 36px)); margin: 0 auto; padding: 34px 0 54px; }
-    .topbar { display: flex; align-items: center; justify-content: space-between; gap: 20px; margin-bottom: 46px; }
-    .brand { display: flex; align-items: center; gap: 12px; font-weight: 800; letter-spacing: .04em; }
-    .brand-mark { display: grid; place-items: center; width: 38px; height: 38px; border-radius: 12px; color: #07111e; background: linear-gradient(135deg, var(--cyan), var(--blue)); box-shadow: 0 8px 26px rgba(57, 217, 196, .22); }
-    .brand small { display: block; color: var(--muted); font-weight: 500; letter-spacing: 0; }
-    .engine { display: inline-flex; align-items: center; gap: 8px; color: #b6c4de; font-size: 13px; }
-    .dot { width: 8px; height: 8px; border-radius: 50%; background: var(--cyan); box-shadow: 0 0 14px var(--cyan); }
-    .hero { max-width: 760px; margin-bottom: 30px; }
-    .eyebrow { margin: 0 0 12px; color: var(--cyan); font-size: 12px; font-weight: 800; letter-spacing: .14em; text-transform: uppercase; }
-    h1 { max-width: 680px; margin: 0; font-size: clamp(2.4rem, 6vw, 4.8rem); line-height: .98; letter-spacing: -.065em; }
-    .hero p { max-width: 620px; margin: 20px 0 0; color: var(--muted); font-size: 17px; }
-    .workspace { display: grid; grid-template-columns: 360px minmax(0, 1fr); gap: 18px; align-items: start; }
-    .card { border: 1px solid var(--line); border-radius: 22px; background: var(--panel); box-shadow: var(--shadow); backdrop-filter: blur(18px); }
-    .controls { padding: 22px; position: sticky; top: 18px; }
-    .card-title { display: flex; justify-content: space-between; align-items: center; gap: 12px; margin-bottom: 20px; font-weight: 750; }
-    .badge { padding: 5px 9px; border: 1px solid rgba(57, 217, 196, .26); border-radius: 999px; color: var(--cyan); background: rgba(57, 217, 196, .08); font-size: 11px; font-weight: 700; }
-    label { display: block; margin: 18px 0 8px; color: #c8d4ea; font-size: 12px; font-weight: 700; letter-spacing: .06em; text-transform: uppercase; }
-    input[type=url] { width: 100%; padding: 13px 14px; border: 1px solid var(--line); border-radius: 12px; outline: 0; color: var(--text); background: rgba(5, 10, 22, .64); font: inherit; }
-    input[type=url]:focus { border-color: var(--blue); box-shadow: 0 0 0 4px rgba(109, 140, 255, .14); }
-    .presets { display: flex; flex-wrap: wrap; gap: 8px; margin-top: 10px; }
-    .preset, .device { border: 1px solid var(--line); border-radius: 9px; color: var(--muted); background: transparent; cursor: pointer; font: inherit; }
-    .preset { padding: 6px 9px; font-size: 12px; }
-    .preset:hover, .device:hover { border-color: var(--blue); color: var(--text); }
-    .devices { display: grid; grid-template-columns: repeat(3, 1fr); gap: 8px; }
-    .device { padding: 10px 5px; }
-    .device.active { border-color: var(--blue); color: white; background: rgba(109, 140, 255, .16); }
-    .run { width: 100%; margin-top: 24px; padding: 14px 16px; border: 0; border-radius: 12px; color: #07111e; background: linear-gradient(100deg, var(--cyan), #85a0ff); box-shadow: 0 12px 26px rgba(66, 137, 255, .2); cursor: pointer; font: inherit; font-weight: 850; }
-    .run:disabled { opacity: .55; cursor: wait; }
-    .notice { margin: 18px 0 0; color: var(--muted); font-size: 12px; }
-    .results { padding: 22px; min-height: 480px; }
-    .result-head { display: flex; align-items: center; justify-content: space-between; gap: 16px; margin-bottom: 18px; }
-    .result-head h2 { margin: 0; font-size: 18px; }
-    #status { color: var(--muted); font-size: 13px; }
-    #status.error { color: var(--danger); }
-    #status.success { color: var(--cyan); }
-    .metrics { display: grid; grid-template-columns: repeat(3, 1fr); gap: 10px; margin-bottom: 18px; }
-    .metric { min-width: 0; padding: 14px; border: 1px solid var(--line); border-radius: 14px; background: rgba(7, 12, 27, .42); }
-    .metric-label { color: var(--muted); font-size: 11px; text-transform: uppercase; letter-spacing: .07em; }
-    .metric-value { overflow: hidden; margin-top: 4px; font-size: 17px; font-weight: 750; text-overflow: ellipsis; white-space: nowrap; }
-    .preview { display: grid; place-items: center; min-height: 330px; overflow: hidden; border: 1px solid var(--line); border-radius: 16px; background: #0a1020; }
-    .preview.empty::before { content: "Your browser screenshot will appear here"; color: #60708f; }
-    #screenshot { display: block; width: 100%; max-height: 610px; object-fit: contain; }
-    .result-foot { display: flex; justify-content: space-between; gap: 12px; margin-top: 14px; color: var(--muted); font-size: 12px; }
-    .download { display: none; color: #b9c8ff; font-weight: 700; text-decoration: none; }
-    .download.show { display: inline; }
-    @media (max-width: 820px) { .workspace { grid-template-columns: 1fr; } .controls { position: static; } .topbar { margin-bottom: 32px; } }
-    @media (max-width: 520px) { .shell { width: min(100% - 24px, 1160px); padding-top: 20px; } .engine { display: none; } .metrics { grid-template-columns: 1fr; } .result-foot { flex-direction: column; } }
   </style>
 </head>
 <body>
   <main class="shell">
     <header class="topbar">
-      <div class="brand"><div class="brand-mark">✦</div><div>WEB TEST LAB<small>Browser-powered checks</small></div></div>
-      <div class="engine"><a href="/search">Search web ↗</a><span class="dot"></span> Chromium ready</div>
     </header>
     <section class="hero">
-      <p class="eyebrow">Live website testing</p>
-      <h1>See how your site behaves before users do.</h1>
-      <p>Enter a public URL, choose a device profile, and run a real headless Chromium check with a visual screenshot. Need to find something? <a href="/search">Search the web</a>.</p>
     </section>
-    <section class="workspace">
       <aside class="card controls">
-        <div class="card-title"><span>Test setup</span><span class="badge">SECURE PREVIEW</span></div>
-        <form id="test-form">
-          <label for="url">Website URL</label>
-          <input id="url" type="url" value="https://example.com" placeholder="https://your-site.com" required>
-          <div class="presets">
-            <button class="preset" type="button" data-url="https://example.com">Example</button>
-            <button class="preset" type="button" data-url="https://huggingface.co">Hugging Face</button>
           </div>
-          <label>Viewport</label>
-          <div class="devices">
-            <button class="device active" type="button" data-device="desktop">Desktop</button>
-            <button class="device" type="button" data-device="tablet">Tablet</button>
-            <button class="device" type="button" data-device="mobile">Mobile</button>
           </div>
-          <button class="run" id="run" type="submit">Run browser test →</button>
         </form>
-        <p class="notice">Only public HTTP and HTTPS websites are allowed. Use this tool only on sites you own or are authorized to test.</p>
       </aside>
-      <section class="card results">
-        <div class="result-head"><h2>Test result</h2><span id="status">Ready when you are</span></div>
-        <div class="metrics">
-          <div class="metric"><div class="metric-label">HTTP status</div><div class="metric-value" id="http-status">—</div></div>
-          <div class="metric"><div class="metric-label">Load time</div><div class="metric-value" id="load-time">—</div></div>
-          <div class="metric"><div class="metric-label">Page title</div><div class="metric-value" id="page-title">—</div></div>
         </div>
-        <div class="preview empty" id="preview"><img id="screenshot" alt="Website screenshot" hidden></div>
-        <div class="result-foot"><span id="final-url">No test run yet</span><a class="download" id="download" download="website-screenshot.png">Download screenshot</a></div>
       </section>
     </section>
   </main>
   <script>
-    const form = document.getElementById('test-form');
-    const urlInput = document.getElementById('url');
-    const runButton = document.getElementById('run');
     const status = document.getElementById('status');
-    const preview = document.getElementById('preview');
-    const screenshot = document.getElementById('screenshot');
     const download = document.getElementById('download');
-    const httpStatus = document.getElementById('http-status');
-    const loadTime = document.getElementById('load-time');
-    const pageTitle = document.getElementById('page-title');
-    const finalUrl = document.getElementById('final-url');
-    let device = 'desktop';
-    document.querySelectorAll('.device').forEach((button) => button.addEventListener('click', () => {
-      device = button.dataset.device;
-      document.querySelectorAll('.device').forEach((item) => item.classList.toggle('active', item === button));
-    }));
-    document.querySelectorAll('.preset').forEach((button) => button.addEventListener('click', () => { urlInput.value = button.dataset.url; }));
-    form.addEventListener('submit', async (event) => {
-      event.preventDefault();
-      runButton.disabled = true;
-      status.className = '';
-      status.textContent = 'Opening Chromium…';
-      preview.classList.add('empty');
-      screenshot.hidden = true;
-      download.classList.remove('show');
-      try {
-        const response = await fetch('/api/test', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ url: urlInput.value, device }) });
-        const data = await response.json();
-        if (!response.ok) throw new Error(data.error || `Request failed: ${response.status}`);
-        httpStatus.textContent = data.status ?? '—';
-        loadTime.textContent = `${data.load_time_ms} ms`;
-        pageTitle.textContent = data.title || '(untitled)';
-        finalUrl.textContent = data.url;
-        screenshot.src = `data:image/png;base64,${data.screenshot}`;
-        screenshot.hidden = false;
-        preview.classList.remove('empty');
-        download.href = screenshot.src;
-        download.classList.add('show');
-        status.className = data.ok ? 'success' : 'error';
-        status.textContent = data.ok ? 'Test completed' : 'Page returned an error status';
-      } catch (error) {
-        status.className = 'error';
-        status.textContent = error.message;
-        httpStatus.textContent = '—';
-        loadTime.textContent = '—';
-        pageTitle.textContent = '—';
-        finalUrl.textContent = 'Test did not complete';
-      } finally {
-        runButton.disabled = false;
-      }
-    });
-  </script>
-</body>
-</html>
-"""
-SEARCH_HTML = r"""
-<!doctype html>
-<html lang="en">
-<head>
-  <meta charset="utf-8">
-  <meta name="viewport" content="width=device-width, initial-scale=1">
-  <title>Web Search · Web Test Lab</title>
-  <style>
-    :root { color-scheme: dark; --bg:#080c17; --panel:#121a30; --line:rgba(148,163,184,.18); --text:#f8fafc; --muted:#92a2bf; --blue:#7b96ff; --cyan:#43ddc6; --pink:#ff8eaa; }
-    * { box-sizing:border-box; }
-    body { margin:0; min-height:100vh; color:var(--text); font:15px/1.55 Inter,ui-sans-serif,system-ui,sans-serif; background:radial-gradient(circle at 20% 0%,rgba(123,150,255,.24),transparent 34rem),radial-gradient(circle at 90% 15%,rgba(67,221,198,.13),transparent 30rem),var(--bg); }
-    a { color:#abc0ff; }
-    .wrap { width:min(940px,calc(100% - 32px)); margin:0 auto; padding:28px 0 56px; }
-    nav { display:flex; align-items:center; justify-content:space-between; gap:20px; margin-bottom:68px; }
-    .brand { display:flex; align-items:center; gap:10px; color:var(--text); font-weight:850; letter-spacing:.05em; text-decoration:none; }
-    .mark { display:grid; place-items:center; width:35px; height:35px; border-radius:11px; color:#07111e; background:linear-gradient(135deg,var(--cyan),var(--blue)); }
-    .engine { color:var(--muted); font-size:13px; }
-    .hero { text-align:center; }
-    .eyebrow { margin:0 0 14px; color:var(--cyan); font-size:12px; font-weight:800; letter-spacing:.14em; text-transform:uppercase; }
-    h1 { margin:0; font-size:clamp(2.5rem,8vw,5.2rem); line-height:.95; letter-spacing:-.07em; }
-    .hero p { max-width:600px; margin:20px auto 0; color:var(--muted); font-size:17px; }
-    .search-box { display:flex; gap:10px; max-width:760px; margin:32px auto 0; padding:8px; border:1px solid rgba(123,150,255,.38); border-radius:16px; background:rgba(11,17,34,.8); box-shadow:0 20px 60px rgba(0,0,0,.25); }
-    input { min-width:0; flex:1; padding:13px 14px; border:0; outline:0; color:var(--text); background:transparent; font:inherit; }
-    button { padding:0 22px; border:0; border-radius:11px; color:#07111e; background:linear-gradient(100deg,var(--cyan),#8da6ff); cursor:pointer; font:inherit; font-weight:850; }
-    button:disabled { opacity:.55; cursor:wait; }
-    .chips { display:flex; justify-content:center; flex-wrap:wrap; gap:8px; margin-top:16px; }
-    .chip { padding:7px 11px; border:1px solid var(--line); border-radius:999px; color:var(--muted); background:transparent; cursor:pointer; font:inherit; font-size:12px; }
-    .chip:hover { border-color:var(--blue); color:var(--text); }
-    .meta { min-height:23px; margin:42px 0 14px; color:var(--muted); font-size:13px; }
-    .results { display:grid; gap:10px; }
-    .result { padding:18px 20px; border:1px solid var(--line); border-radius:16px; background:rgba(18,26,48,.78); transition:transform .16s,border-color .16s; }
-    .result:hover { transform:translateY(-2px); border-color:rgba(123,150,255,.55); }
-    .result-top { display:flex; gap:13px; align-items:flex-start; }
-    .num { flex:0 0 auto; display:grid; place-items:center; width:27px; height:27px; border-radius:9px; color:var(--cyan); background:rgba(67,221,198,.1); font-size:12px; font-weight:800; }
-    .title { display:block; color:#c4d1ff; font-size:17px; font-weight:750; text-decoration:none; }
-    .title:hover { text-decoration:underline; }
-    .url { overflow:hidden; margin-top:3px; color:#6fcdbd; font-size:12px; text-overflow:ellipsis; white-space:nowrap; }
-    .snippet { margin:10px 0 0 40px; color:var(--muted); }
-    .empty { padding:42px 20px; border:1px dashed var(--line); border-radius:16px; color:var(--muted); text-align:center; }
-    footer { margin-top:42px; color:#6d7b96; font-size:12px; text-align:center; }
-    @media (max-width:560px) { .wrap{width:min(100% - 22px,940px);padding-top:18px;} nav{margin-bottom:48px;} .engine{display:none;} .search-box{margin-top:25px;} button{padding:0 15px;} .snippet{margin-left:0;} }
-  </style>
-</head>
-<body>
-  <main class="wrap">
-    <nav><a class="brand" href="/"><span class="mark">✦</span> WEB TEST LAB</a><span class="engine">Chromium-powered web search</span></nav>
-    <section class="hero">
-      <p class="eyebrow">Search the open web</p>
-      <h1>Find what you need.</h1>
-      <p>Search from a clean headless Chromium browser and open results in a new tab.</p>
-      <form class="search-box" id="search-form">
-        <input id="query" type="search" placeholder="Search the web…" autocomplete="off" required>
-        <button id="search-button" type="submit">Search</button>
-      </form>
-      <div class="chips"><button class="chip" type="button" data-query="web development news">Web development</button><button class="chip" type="button" data-query="AI tools">AI tools</button><button class="chip" type="button" data-query="Bengaluru weather">Bengaluru weather</button></div>
-    </section>
-    <div class="meta" id="meta">Search results will appear here</div>
-    <section class="results" id="results"><div class="empty">Enter a search query to begin.</div></section>
-    <footer>Search uses public web results. Use this service responsibly and follow each website's terms.</footer>
-  </main>
-  <script>
-    const form = document.getElementById('search-form');
-    const queryInput = document.getElementById('query');
-    const button = document.getElementById('search-button');
     const meta = document.getElementById('meta');
-    const results = document.getElementById('results');
-    document.querySelectorAll('.chip').forEach((chip) => chip.addEventListener('click', () => { queryInput.value = chip.dataset.query; form.requestSubmit(); }));
     form.addEventListener('submit', async (event) => {
       event.preventDefault();
-      const query = queryInput.value.trim();
-      if (!query) return;
-      button.disabled = true;
-      button.textContent = 'Searching…';
-      meta.textContent = 'Opening Chromium and searching…';
-      results.textContent = '';
       try {
-        const response = await fetch('/api/search', { method:'POST', headers:{'Content-Type':'application/json'}, body:JSON.stringify({query}) });
-        const data = await response.json();
-        if (!response.ok) throw new Error(data.error || `Search failed: ${response.status}`);
-        meta.textContent = `${data.results.length} results for “${data.query}” · ${data.elapsed_ms} ms`;
-        if (!data.results.length) { results.innerHTML = '<div class="empty">No results found. Try a different search.</div>'; return; }
-        data.results.forEach((item, index) => {
-          const card = document.createElement('article'); card.className = 'result';
-          const top = document.createElement('div'); top.className = 'result-top';
-          const num = document.createElement('span'); num.className = 'num'; num.textContent = String(index + 1);
-          const body = document.createElement('div');
-          const title = document.createElement('a'); title.className = 'title'; title.href = item.url; title.target = '_blank'; title.rel = 'noopener noreferrer'; title.textContent = item.title || item.url;
-          const url = document.createElement('div'); url.className = 'url'; url.textContent = item.url;
-          const snippet = document.createElement('p'); snippet.className = 'snippet'; snippet.textContent = item.snippet || 'Open this result in a new tab.';
-          body.append(title, url); top.append(num, body); card.append(top, snippet); results.append(card);
-        });
-      } catch (error) {
-        meta.textContent = error.message;
-        results.innerHTML = '<div class="empty">Search could not be completed. Please try again.</div>';
-      } finally { button.disabled = false; button.textContent = 'Search'; }
     });
   </script>
 </body>
@@ -322,14 +214,14 @@ SEARCH_HTML = r"""
 def _client_key():
     forwarded = request.headers.get("X-Forwarded-For", "")
-    return (forwarded.split(",")[0].strip() or request.remote_addr or "unknown")
 def _within_rate_limit():
     now = time.time()
     key = _client_key()
     with RATE_LOCK:
-        recent = [stamp for stamp in RATE_BUCKETS.get(key, []) if now - stamp < RATE_WINDOW_SECONDS]
         if len(recent) >= RATE_LIMIT:
             RATE_BUCKETS[key] = recent
             return False
@@ -338,122 +230,39 @@ def _within_rate_limit():
         return True
-def _public_url(value):
-    if not isinstance(value, str) or len(value.strip()) > 2048:
-        raise ValueError("Enter a valid public URL")
-    value = value.strip()
-    parsed = urlparse(value)
-    if parsed.scheme not in {"http", "https"} or not parsed.hostname:
-        raise ValueError("URL must start with http:// or https://")
-    if parsed.username or parsed.password:
-        raise ValueError("URLs with embedded credentials are not allowed")
-    host = parsed.hostname.lower().rstrip(".")
-    if host in {"localhost", "localhost.localdomain"} or host.endswith(".local"):
-        raise ValueError("Local network URLs are not allowed")
-    try:
-        addresses = {info[4][0] for info in socket.getaddrinfo(host, None)}
-    except socket.gaierror as exc:
-        raise ValueError("The hostname could not be resolved") from exc
-    if not addresses or any(not ipaddress.ip_address(address).is_global for address in addresses):
-        raise ValueError("Only public internet URLs are allowed")
-    return value
-def _route_guard(route):
-    target = route.request.url
-    if target.startswith(("data:", "blob:", "about:", "chrome:")):
-        route.continue_()
-        return
     try:
-        _public_url(target)
-    except ValueError:
-        route.abort()
-    else:
-        route.continue_()
-def _browser_test(target_url, device):
-    global PLAYWRIGHT, BROWSER
-    with BROWSER_LOCK:
-        if BROWSER is None or not BROWSER.is_connected():
-            if PLAYWRIGHT is not None:
-                PLAYWRIGHT.stop()
-            PLAYWRIGHT = sync_playwright().start()
-            BROWSER = PLAYWRIGHT.chromium.launch(
-                headless=True,
-                args=["--no-sandbox", "--disable-dev-shm-usage", "--disable-gpu"],
-            )
-        context = BROWSER.new_context(**DEVICES[device])
-        page = context.new_page()
-        page.route("**/*", _route_guard)
-        started = perf_counter()
-        try:
-            response = page.goto(target_url, wait_until="domcontentloaded", timeout=TEST_TIMEOUT_MS)
-            page.wait_for_timeout(500)
-            screenshot = base64.b64encode(page.screenshot(type="png", full_page=False)).decode("ascii")
-            status_code = response.status if response is not None else None
-            return {
-                "ok": status_code is not None and 200 <= status_code < 400,
-                "status": status_code,
-                "title": page.title(),
-                "url": page.url,
-                "load_time_ms": round((perf_counter() - started) * 1000),
-                "screenshot": screenshot,
-            }
-        finally:
-            context.close()
-def _web_search(query):
-    global PLAYWRIGHT, BROWSER
-    search_url = "https://html.duckduckgo.com/html/?" + urlencode({"q": query})
-    with BROWSER_LOCK:
-        if BROWSER is None or not BROWSER.is_connected():
-            if PLAYWRIGHT is not None:
-                PLAYWRIGHT.stop()
-            PLAYWRIGHT = sync_playwright().start()
-            BROWSER = PLAYWRIGHT.chromium.launch(
-                headless=True,
-                args=["--no-sandbox", "--disable-dev-shm-usage", "--disable-gpu"],
-            )
-        context = BROWSER.new_context(viewport={"width": 1280, "height": 900})
-        page = context.new_page()
-        page.route("**/*", _route_guard)
-        started = perf_counter()
-        try:
-            page.goto(search_url, wait_until="domcontentloaded", timeout=TEST_TIMEOUT_MS)
-            page.wait_for_timeout(700)
-            rows = page.locator("a.result__a, a[data-testid='result-title-a']").evaluate_all(
-                """
-                (links) => links.slice(0, 8).map((link) => {
-                  const result = link.closest('.result') || link.closest('[data-testid="result"]') || link.parentElement;
-                  const snippet = result?.querySelector('.result__snippet')?.innerText || '';
-                  return { title: link.innerText.trim(), url: link.href, snippet: snippet.trim() };
-                })
-                """
-            )
-            results = []
-            for row in rows:
-                href = row.get("url", "")
-                parsed = urlparse(href)
-                if parsed.hostname in {"duckduckgo.com", "www.duckduckgo.com"} and parsed.path.startswith("/l/"):
-                    href = parse_qs(parsed.query).get("uddg", [href])[0]
-                try:
-                    href = _public_url(href)
-                except ValueError:
-                    continue
-                results.append({
-                    "title": row.get("title") or href,
-                    "url": href,
-                    "snippet": row.get("snippet", ""),
-                })
-            return {
-                "query": query,
-                "results": results,
-                "elapsed_ms": round((perf_counter() - started) * 1000),
-            }
-        finally:
-            context.close()
 @app.get("/")
@@ -461,52 +270,69 @@ def home():
     return INDEX_HTML
-@app.get("/search")
-def search_page():
-    return SEARCH_HTML
-@app.post("/api/search")
-def search_api():
     if not _within_rate_limit():
         return jsonify({"error": "Rate limit reached. Please wait a minute."}), 429
-    payload = request.get_json(silent=True) or {}
-    query = payload.get("query", "")
-    if not isinstance(query, str) or not 2 <= len(query.strip()) <= 200:
-        return jsonify({"error": "Enter a search query between 2 and 200 characters."}), 400
-    try:
-        return jsonify(_web_search(query.strip()))
-    except PlaywrightTimeoutError:
-        return jsonify({"error": "The search took too long to complete."}), 504
-    except Exception:
-        app.logger.exception("Web search failed")
-        return jsonify({"error": "The web search could not be completed."}), 502
-@app.post("/api/test")
-def run_test():
-    if not _within_rate_limit():
-        return jsonify({"error": "Rate limit reached. Please wait a minute."}), 429
-    payload = request.get_json(silent=True) or {}
-    device = payload.get("device", "desktop")
-    if device not in DEVICES:
-        return jsonify({"error": "Unknown viewport"}), 400
     try:
-        target_url = _public_url(payload.get("url"))
-        return jsonify(_browser_test(target_url, device))
     except ValueError as exc:
         return jsonify({"error": str(exc)}), 400
-    except PlaywrightTimeoutError:
-        return jsonify({"error": "The page took too long to load."}), 504
     except Exception:
-        app.logger.exception("Browser test failed")
-        return jsonify({"error": "The browser test could not be completed."}), 502
-@app.get("/health")
-def health_check():
-    return jsonify({"status": "healthy", "browser": "chromium"}), 200
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)

+import hmac
+import io
+import os
 import time
 from threading import Lock
+from dotenv import load_dotenv
+from flask import Flask, jsonify, request, send_file
+from PIL import Image, ImageColor, ImageOps, UnidentifiedImageError
+from rembg import new_session, remove
+load_dotenv()
 app = Flask(__name__)
+app.config["MAX_CONTENT_LENGTH"] = 12 * 1024 * 1024  # 12 MB upload limit
+MODELS = {
+    "isnet-general-use": "Quality · ISNet General",
+    "u2net": "Balanced · U2Net",
+    "u2netp": "Fast · U2NetP",
+    "u2net_human_seg": "People · Human Segmentation",
+    "isnet-anime": "Anime · ISNet Anime",
+}
+FORMATS = {"png", "jpg", "webp"}
+SESSIONS = {}
+SESSION_LOCK = Lock()
 RATE_LOCK = Lock()
 RATE_BUCKETS = {}
+RATE_LIMIT = 8
+RATE_WINDOW = 60
 INDEX_HTML = r"""
 <!doctype html>
 <head>
   <meta charset="utf-8">
   <meta name="viewport" content="width=device-width, initial-scale=1">
+  <title>Background Studio</title>
   <style>
+    :root { color-scheme: dark; --bg:#090d18; --panel:#121a2e; --panel2:#18233d; --line:rgba(148,163,184,.2); --text:#f7f9ff; --muted:#95a5c2; --cyan:#43ddc6; --blue:#8298ff; --pink:#ff8eaa; --shadow:0 26px 80px rgba(0,0,0,.32); }
+    * { box-sizing:border-box; }
+    body { min-height:100vh; margin:0; color:var(--text); font:15px/1.55 Inter,ui-sans-serif,system-ui,-apple-system,"Segoe UI",sans-serif; background:radial-gradient(circle at 12% 0%,rgba(130,152,255,.24),transparent 34rem),radial-gradient(circle at 94% 17%,rgba(67,221,198,.13),transparent 30rem),var(--bg); }
+    button,input,select { font:inherit; }
+    button { cursor:pointer; }
+    .shell { width:min(1180px,calc(100% - 34px)); margin:0 auto; padding:30px 0 58px; }
+    .topbar { display:flex; align-items:center; justify-content:space-between; gap:20px; }
+    .brand { display:flex; align-items:center; gap:11px; font-weight:850; letter-spacing:.05em; }
+    .brand-mark { display:grid; place-items:center; width:39px; height:39px; border-radius:13px; color:#06111b; background:linear-gradient(135deg,var(--cyan),var(--blue)); box-shadow:0 9px 28px rgba(67,221,198,.2); }
+    .brand small { display:block; color:var(--muted); font-weight:500; letter-spacing:0; }
+    .system { display:inline-flex; align-items:center; gap:8px; color:#b7c6df; font-size:13px; }
+    .dot { width:8px; height:8px; border-radius:50%; background:var(--cyan); box-shadow:0 0 13px var(--cyan); }
+    .hero { max-width:770px; margin:72px 0 34px; }
+    .eyebrow { margin:0 0 13px; color:var(--cyan); font-size:12px; font-weight:850; letter-spacing:.15em; text-transform:uppercase; }
+    h1 { margin:0; font-size:clamp(2.7rem,7vw,5.5rem); line-height:.93; letter-spacing:-.075em; }
+    .hero p { max-width:650px; margin:22px 0 0; color:var(--muted); font-size:18px; }
+    .studio { display:grid; grid-template-columns:380px minmax(0,1fr); gap:18px; align-items:start; }
+    .card { border:1px solid var(--line); border-radius:23px; background:rgba(18,26,46,.82); box-shadow:var(--shadow); backdrop-filter:blur(16px); }
+    .controls { padding:22px; position:sticky; top:18px; }
+    .card-head { display:flex; align-items:center; justify-content:space-between; gap:12px; margin-bottom:20px; font-weight:800; }
+    .pill { padding:5px 9px; border:1px solid rgba(67,221,198,.25); border-radius:999px; color:var(--cyan); background:rgba(67,221,198,.08); font-size:11px; font-weight:800; }
+    .dropzone { display:grid; place-items:center; min-height:166px; padding:22px; border:1px dashed rgba(130,152,255,.55); border-radius:17px; color:var(--muted); background:rgba(8,13,28,.45); text-align:center; transition:.18s; }
+    .dropzone.drag { border-color:var(--cyan); background:rgba(67,221,198,.08); transform:scale(1.01); }
+    .upload-icon { display:grid; place-items:center; width:44px; height:44px; margin-bottom:10px; border-radius:14px; color:var(--cyan); background:rgba(67,221,198,.1); font-size:21px; }
+    .dropzone strong { display:block; color:var(--text); }
+    .dropzone span { display:block; margin-top:4px; font-size:12px; }
+    .file-name { overflow:hidden; max-width:100%; margin-top:10px; color:#b8c8ff; text-overflow:ellipsis; white-space:nowrap; }
+    .browse { display:inline-block; margin-top:13px; padding:8px 12px; border:1px solid var(--line); border-radius:9px; color:var(--text); background:rgba(130,152,255,.12); cursor:pointer; font-size:12px; font-weight:750; }
+    #file { display:none; }
+    label { display:block; margin:18px 0 8px; color:#c9d5ec; font-size:11px; font-weight:800; letter-spacing:.09em; text-transform:uppercase; }
+    select,input[type=password],input[type=color] { width:100%; padding:11px 12px; border:1px solid var(--line); border-radius:11px; outline:0; color:var(--text); background:rgba(7,12,25,.62); }
+    select:focus,input:focus { border-color:var(--blue); box-shadow:0 0 0 4px rgba(130,152,255,.13); }
+    input[type=color] { height:42px; padding:5px; }
+    .row { display:grid; grid-template-columns:1fr 1fr; gap:10px; }
+    .check { display:flex; align-items:center; gap:9px; margin-top:17px; color:#c5d2e9; font-size:13px; }
+    .check input { width:16px; height:16px; accent-color:var(--cyan); }
+    details { margin-top:14px; color:var(--muted); font-size:12px; }
+    summary { cursor:pointer; color:#b8c8ff; font-weight:750; }
+    .advanced { display:grid; grid-template-columns:1fr 1fr 1fr; gap:9px; margin-top:10px; }
+    .advanced input { padding:8px; font-size:12px; }
+    .advanced label { margin:0 0 5px; font-size:10px; }
+    .run { width:100%; margin-top:22px; padding:14px 16px; border:0; border-radius:12px; color:#06111b; background:linear-gradient(100deg,var(--cyan),#91a7ff); box-shadow:0 12px 30px rgba(67,221,198,.15); font-weight:850; }
+    .run:disabled { opacity:.55; cursor:wait; }
+    .status { min-height:21px; margin:13px 0 0; color:var(--muted); font-size:12px; }
+    .status.success { color:var(--cyan); }
+    .status.error { color:var(--pink); }
+    .result { min-width:0; padding:22px; }
+    .result-head { display:flex; align-items:center; justify-content:space-between; gap:14px; margin-bottom:18px; }
+    .result-head h2 { margin:0; font-size:19px; }
+    .result-head span { color:var(--muted); font-size:12px; }
+    .empty { display:grid; place-items:center; min-height:500px; border:1px dashed var(--line); border-radius:17px; color:#687995; text-align:center; }
+    .empty-icon { margin-bottom:9px; font-size:32px; opacity:.7; }
+    .compare-wrap { display:none; }
+    .compare-wrap.show { display:block; }
+    .compare { position:relative; overflow:hidden; min-height:430px; border:1px solid var(--line); border-radius:17px; background-color:#edf2fa; background-image:linear-gradient(45deg,#dfe6f2 25%,transparent 25%),linear-gradient(-45deg,#dfe6f2 25%,transparent 25%),linear-gradient(45deg,transparent 75%,#dfe6f2 75%),linear-gradient(-45deg,transparent 75%,#dfe6f2 75%); background-position:0 0,0 11px,11px -11px,-11px 0; background-size:22px 22px; }
+    .compare img { position:absolute; inset:0; width:100%; height:100%; object-fit:contain; }
+    .after-clip { position:absolute; inset:0; clip-path:inset(0 50% 0 0); }
+    .divider { position:absolute; top:0; bottom:0; left:50%; width:2px; background:white; box-shadow:0 0 8px rgba(0,0,0,.4); pointer-events:none; }
+    .tag { position:absolute; top:13px; z-index:2; padding:5px 8px; border-radius:7px; color:white; background:rgba(5,10,22,.65); font-size:11px; font-weight:800; letter-spacing:.06em; text-transform:uppercase; }
+    .tag.before { left:13px; }
+    .tag.after { right:13px; }
+    .range { width:100%; margin:18px 0 0; accent-color:var(--blue); cursor:ew-resize; }
+    .result-actions { display:flex; align-items:center; justify-content:space-between; gap:12px; margin-top:15px; color:var(--muted); font-size:12px; }
+    .download { display:none; padding:8px 11px; border:1px solid rgba(67,221,198,.28); border-radius:9px; color:var(--cyan); text-decoration:none; font-weight:800; }
+    .download.show { display:inline-block; }
+    .hint { margin-top:16px; color:#6e7e9b; font-size:12px; }
+    @media (max-width:850px) { .studio{grid-template-columns:1fr;} .controls{position:static;} }
+    @media (max-width:540px) { .shell{width:min(100% - 22px,1180px);padding-top:20px;} .system{display:none;} .hero{margin-top:55px;} .row,.advanced{grid-template-columns:1fr;} .compare{min-height:300px;} }
   </style>
 </head>
 <body>
   <main class="shell">
     <header class="topbar">
+      <div class="brand"><span class="brand-mark">✦</span><div>BACKGROUND STUDIO<small>Advanced cutout workspace</small></div></div>
+      <div class="system"><span class="dot"></span> CPU processing ready</div>
     </header>
     <section class="hero">
+      <p class="eyebrow">Clean edges. Better cutouts.</p>
+      <h1>Make the subject stand out.</h1>
+      <p>Upload an image, choose the right segmentation model, fine-tune the edges, and download a transparent result.</p>
     </section>
+    <section class="studio">
       <aside class="card controls">
+        <div class="card-head"><span>Removal settings</span><span class="pill">ADVANCED</span></div>
+        <form id="form">
+          <div class="dropzone" id="dropzone">
+            <div class="upload-icon">↑</div>
+            <strong>Drop an image here</strong>
+            <span>PNG, JPG, WEBP · up to 12 MB</span>
+            <label class="browse" for="file">Choose image</label>
+            <input id="file" type="file" accept="image/png,image/jpeg,image/webp">
+            <div class="file-name" id="file-name">No file selected</div>
           </div>
+          <label for="model">Segmentation model</label>
+          <select id="model">
+            <option value="isnet-general-use">Quality · ISNet General</option>
+            <option value="u2net">Balanced · U2Net</option>
+            <option value="u2netp">Fast · U2NetP</option>
+            <option value="u2net_human_seg">People · Human Segmentation</option>
+            <option value="isnet-anime">Anime · ISNet Anime</option>
+          </select>
+          <div class="row">
+            <div><label for="format">Output</label><select id="format"><option value="png">PNG · transparent</option><option value="webp">WEBP · smaller</option><option value="jpg">JPG · solid background</option></select></div>
+            <div><label for="background">JPG background</label><input id="background" type="color" value="#ffffff" title="Used when JPG output is selected"></div>
           </div>
+          <label class="check"><input id="matting" type="checkbox"> Enable alpha matting for finer edges</label>
+          <details><summary>Fine-tune alpha matting</summary><div class="advanced"><div><label for="foreground">Foreground</label><input id="foreground" type="number" min="0" max="255" value="240"></div><div><label for="threshold">Background</label><input id="threshold" type="number" min="0" max="255" value="10"></div><div><label for="erode">Erode</label><input id="erode" type="number" min="0" max="40" value="10"></div></div></details>
+          <label for="api-key">API key <span style="color:var(--muted);font-weight:500;text-transform:none;letter-spacing:0">(optional)</span></label>
+          <input id="api-key" type="password" autocomplete="off" placeholder="Only needed if the Space secret is enabled">
+          <button class="run" id="run" type="submit">Remove background →</button>
+          <div class="status" id="status">Choose an image to begin.</div>
         </form>
       </aside>
+      <section class="card result">
+        <div class="result-head"><h2>Preview</h2><span id="meta">Before and after comparison</span></div>
+        <div class="empty" id="empty"><div><div class="empty-icon">◌</div>Upload an image and your result will appear here.</div></div>
+        <div class="compare-wrap" id="compare-wrap">
+          <div class="compare" id="compare"><span class="tag before">Original</span><span class="tag after">Cutout</span><img id="before" alt="Original image"><div class="after-clip" id="after-clip"><img id="after" alt="Background removed image"></div><div class="divider" id="divider"></div></div>
+          <input class="range" id="range" type="range" min="0" max="100" value="50" aria-label="Compare original and cutout">
+          <div class="result-actions"><span id="details">Transparent PNG ready</span><a class="download" id="download" download="background-removed.png">Download result ↓</a></div>
         </div>
+        <p class="hint">Tip: use alpha matting when hair, fur, glass, or fine object edges need extra refinement.</p>
       </section>
     </section>
   </main>
   <script>
+    const form = document.getElementById('form');
+    const fileInput = document.getElementById('file');
+    const dropzone = document.getElementById('dropzone');
+    const fileName = document.getElementById('file-name');
+    const run = document.getElementById('run');
     const status = document.getElementById('status');
+    const empty = document.getElementById('empty');
+    const wrap = document.getElementById('compare-wrap');
+    const before = document.getElementById('before');
+    const after = document.getElementById('after');
+    const clip = document.getElementById('after-clip');
+    const divider = document.getElementById('divider');
+    const range = document.getElementById('range');
     const download = document.getElementById('download');
+    const details = document.getElementById('details');
     const meta = document.getElementById('meta');
+    let originalUrl = null;
+    let resultUrl = null;
+    function setFile(file) { if (!file) return; fileInput.files = (() => { const dt = new DataTransfer(); dt.items.add(file); return dt.files; })(); fileName.textContent = file.name; status.textContent = 'Ready to process.'; status.className = 'status'; if (originalUrl) URL.revokeObjectURL(originalUrl); originalUrl = URL.createObjectURL(file); before.src = originalUrl; }
+    fileInput.addEventListener('change', () => setFile(fileInput.files[0]));
+    ['dragenter','dragover'].forEach((event) => dropzone.addEventListener(event, (e) => { e.preventDefault(); dropzone.classList.add('drag'); }));
+    ['dragleave','drop'].forEach((event) => dropzone.addEventListener(event, (e) => { e.preventDefault(); dropzone.classList.remove('drag'); }));
+    dropzone.addEventListener('drop', (e) => setFile(e.dataTransfer.files[0]));
+    function updateCompare() { const value = Number(range.value); clip.style.clipPath = `inset(0 ${100 - value}% 0 0)`; divider.style.left = `${value}%`; }
+    range.addEventListener('input', updateCompare);
     form.addEventListener('submit', async (event) => {
       event.preventDefault();
+      const file = fileInput.files[0];
+      if (!file) { status.textContent = 'Choose an image first.'; status.className = 'status error'; return; }
+      run.disabled = true; status.className = 'status'; status.textContent = 'Processing… the first model run may take longer.'; meta.textContent = 'Working on your cutout';
+      const outputFormat = document.getElementById('format').value;
+      const body = new FormData(); body.append('image', file); body.append('model', document.getElementById('model').value); body.append('format', outputFormat); body.append('background', outputFormat === 'jpg' ? document.getElementById('background').value : 'transparent'); body.append('alpha_matting', document.getElementById('matting').checked ? 'true' : 'false'); body.append('foreground_threshold', document.getElementById('foreground').value); body.append('background_threshold', document.getElementById('threshold').value); body.append('erode_size', document.getElementById('erode').value);
+      const apiKey = document.getElementById('api-key').value.trim();
       try {
+        const headers = apiKey ? { 'X-API-Key': apiKey } : {};
+        const response = await fetch('/remove-bg', { method:'POST', headers, body });
+        if (!response.ok) { const data = await response.json().catch(() => ({})); throw new Error(data.error || `Request failed: ${response.status}`); }
+        const blob = await response.blob();
+        if (resultUrl) URL.revokeObjectURL(resultUrl); resultUrl = URL.createObjectURL(blob); after.src = resultUrl; download.href = resultUrl; download.classList.add('show'); empty.style.display = 'none'; wrap.classList.add('show'); range.value = 50; updateCompare();
+        const model = document.getElementById('model').value; const format = outputFormat.toUpperCase(); details.textContent = `${model} · ${format} ready`; meta.textContent = 'Drag the slider to compare'; status.className = 'status success'; status.textContent = 'Background removed successfully.';
+      } catch (error) { status.className = 'status error'; status.textContent = error.message; }
+      finally { run.disabled = false; }
     });
   </script>
 </body>
 def _client_key():
     forwarded = request.headers.get("X-Forwarded-For", "")
+    return forwarded.split(",")[0].strip() or request.remote_addr or "unknown"
 def _within_rate_limit():
     now = time.time()
     key = _client_key()
     with RATE_LOCK:
+        recent = [stamp for stamp in RATE_BUCKETS.get(key, []) if now - stamp < RATE_WINDOW]
         if len(recent) >= RATE_LIMIT:
             RATE_BUCKETS[key] = recent
             return False
         return True
+def _check_api_key():
+    configured = os.getenv("API_KEY", "").strip()
+    if not configured:
+        return True
+    supplied = request.headers.get("X-API-Key", "")
+    return bool(supplied) and hmac.compare_digest(supplied, configured)
+def _int_field(name, default, low, high):
     try:
+        value = int(request.form.get(name, default))
+    except (TypeError, ValueError):
+        raise ValueError(f"{name} must be a number")
+    return max(low, min(high, value))
+def _get_session(model):
+    if model not in MODELS:
+        raise ValueError("Unknown segmentation model")
+    with SESSION_LOCK:
+        if model not in SESSIONS:
+            SESSIONS[model] = new_session(model)
+        return SESSIONS[model]
+def _composite(image, color):
+    try:
+        rgb = ImageColor.getrgb(color)
+    except ValueError as exc:
+        raise ValueError("Invalid background color") from exc
+    background = Image.new("RGBA", image.size, rgb + (255,))
+    background.alpha_composite(image.convert("RGBA"))
+    return background
 @app.get("/")
     return INDEX_HTML
+@app.get("/health")
+def health():
+    return jsonify({"status": "healthy", "service": "background-studio"}), 200
+@app.post("/remove-bg")
+def remove_background():
     if not _within_rate_limit():
         return jsonify({"error": "Rate limit reached. Please wait a minute."}), 429
+    if not _check_api_key():
+        return jsonify({"error": "Invalid or missing API key"}), 401
+    if "image" not in request.files:
+        return jsonify({"error": "Choose an image file"}), 400
+    uploaded = request.files["image"]
+    if not uploaded.filename:
+        return jsonify({"error": "No image selected"}), 400
+    model = request.form.get("model", "isnet-general-use")
+    output_format = request.form.get("format", "png").lower()
+    background = request.form.get("background", "#ffffff")
+    alpha_matting = request.form.get("alpha_matting", "false").lower() == "true"
+    if output_format not in FORMATS:
+        return jsonify({"error": "Unsupported output format"}), 400
     try:
+        image = ImageOps.exif_transpose(Image.open(uploaded.stream))
+        image.load()
+        if image.width * image.height > 25_000_000:
+            return jsonify({"error": "Image dimensions are too large"}), 400
+        image = image.convert("RGBA")
+        session = _get_session(model)
+        output = remove(
+            image,
+            session=session,
+            alpha_matting=alpha_matting,
+            alpha_matting_foreground_threshold=_int_field("foreground_threshold", 240, 0, 255),
+            alpha_matting_background_threshold=_int_field("background_threshold", 10, 0, 255),
+            alpha_matting_erode_size=_int_field("erode_size", 10, 0, 40),
+        ).convert("RGBA")
+        if output_format == "jpg" or background.lower() != "transparent":
+            output = _composite(output, background if background.lower() != "transparent" else "#ffffff")
+        if output_format == "jpg":
+            output = output.convert("RGB")
+        result = io.BytesIO()
+        if output_format == "jpg":
+            output.save(result, format="JPEG", quality=95, optimize=True)
+        elif output_format == "webp":
+            output.save(result, format="WEBP", quality=95, method=6)
+        else:
+            output.save(result, format="PNG", optimize=True)
+        result.seek(0)
+        return send_file(result, mimetype=f"image/{'jpeg' if output_format == 'jpg' else output_format}", as_attachment=True, download_name=f"background-removed.{output_format}")
+    except (UnidentifiedImageError, OSError):
+        return jsonify({"error": "The uploaded file is not a readable image"}), 400
     except ValueError as exc:
         return jsonify({"error": str(exc)}), 400
     except Exception:
+        app.logger.exception("Background removal failed")
+        return jsonify({"error": "Background removal failed. Try a smaller image or another model."}), 500
 if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=int(os.getenv("PORT", "7860")))

requirements.txt CHANGED Viewed

@@ -1,3 +1,5 @@
 Flask>=3.0,<4
-playwright>=1.45,<2
 gunicorn>=22,<24

 Flask>=3.0,<4
+Pillow>=10,<12
+python-dotenv>=1,<2
+rembg[cpu]>=2.0.50,<3
 gunicorn>=22,<24