bgeraser 0.1.0__tar.gz

bgeraser-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Tavrej (Yonzon Studio)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

bgeraser-0.1.0/PKG-INFO

@@ -0,0 +1,131 @@
+Metadata-Version: 2.4
+Name: bgeraser
+Version: 0.1.0
+Summary: Remove image backgrounds locally — CLI, Python library and a tiny web UI. No uploads, no API keys.
+Author-email: "Tavrej (Yonzon Studio)" <tavrejstudio@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/guesttavrej/bgeraser
+Project-URL: Issues, https://github.com/guesttavrej/bgeraser/issues
+Keywords: background removal,remove background,rembg,image,transparent png,cli
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: End Users/Desktop
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Multimedia :: Graphics
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: rembg>=2.0.50
+Requires-Dist: onnxruntime>=1.16
+Requires-Dist: pillow>=10.0
+Provides-Extra: web
+Requires-Dist: flask>=3.0; extra == "web"
+Provides-Extra: dev
+Requires-Dist: pytest>=7; extra == "dev"
+Requires-Dist: flask>=3.0; extra == "dev"
+Dynamic: license-file
+
+# bgeraser
+
+**Remove image backgrounds on your own computer.** No uploads, no API keys, no watermarks, no limits.
+
+Works three ways:
+
+| | |
+|---|---|
+| **Command line** | `bgeraser photo.jpg` → `photo-nobg.png` |
+| **Python library** | `remove_background("photo.jpg")` |
+| **Local web page** | `bgeraser serve` → drag & drop in your browser |
+
+Powered by open-source [U²-Net](https://github.com/xuebinqin/U-2-Net) / ISNet models through [rembg](https://github.com/danielgatis/rembg).
+
+## Install
+
+```bash
+pip install bgeraser            # CLI + library
+pip install "bgeraser[web]"     # also the drag & drop web page
+```
+
+Needs Python 3.9+. The first run downloads the model (4 MB for the default `u2netp`; larger models download when you choose them).
+
+## Command line
+
+```bash
+# one image → transparent PNG next to it
+bgeraser photo.jpg
+
+# choose the output name
+bgeraser photo.jpg -o clean.png
+
+# solid background instead of transparent (product photos, ID photos)
+bgeraser photo.jpg --bg white
+bgeraser photo.jpg --bg "#00ff00"
+
+# a whole folder → ./products/nobg/
+bgeraser ./products
+bgeraser ./products -o ./out --recursive
+
+# better model for people
+bgeraser --model u2net_human_seg me.png
+
+# finer edges for hair / fur (slower)
+bgeraser photo.jpg --matting
+
+# list models
+bgeraser models
+```
+
+## Web page (drag & drop)
+
+```bash
+bgeraser serve
+```
+
+Then open <http://127.0.0.1:5000>. Drop, paste, or pick an image; download the PNG. Everything stays on your machine.
+
+`bgeraser serve --port 8080 --model isnet-general-use` to change port or model.
+
+## Python
+
+```python
+from bgeraser import remove_background, remove_background_bytes, process_folder
+
+remove_background("photo.jpg")                           # -> photo-nobg.png
+remove_background("photo.jpg", "out.png", bg_color="white")
+remove_background("photo.jpg", model="isnet-general-use", alpha_matting=True)
+
+png_bytes = remove_background_bytes(open("photo.jpg", "rb").read())   # in-memory
+
+process_folder("./products", "./products/out")           # batch
+```
+
+## Models
+
+| name | best for |
+|---|---|
+| `u2netp` *(default)* | fast, small, everyday use |
+| `u2net` | general, higher quality |
+| `isnet-general-use` | often the cleanest edges |
+| `u2net_human_seg` | people / portraits |
+| `isnet-anime` | anime & illustrations |
+| `u2net_cloth_seg` | clothing segmentation |
+| `silueta` | u2net quality, smaller file |
+
+## Why
+
+Online background removers are slow, watermark the result, charge per image, or make you upload customer photos to a stranger's server. This does the same job offline, for free, in one command.
+
+## Contributing
+
+Issues and pull requests welcome. Run tests with:
+
+```bash
+pip install -e ".[dev]"
+pytest
+```
+
+## License
+
+MIT © Tavrej / Yonzon Studio

bgeraser-0.1.0/README.md

@@ -0,0 +1,102 @@
+# bgeraser
+
+**Remove image backgrounds on your own computer.** No uploads, no API keys, no watermarks, no limits.
+
+Works three ways:
+
+| | |
+|---|---|
+| **Command line** | `bgeraser photo.jpg` → `photo-nobg.png` |
+| **Python library** | `remove_background("photo.jpg")` |
+| **Local web page** | `bgeraser serve` → drag & drop in your browser |
+
+Powered by open-source [U²-Net](https://github.com/xuebinqin/U-2-Net) / ISNet models through [rembg](https://github.com/danielgatis/rembg).
+
+## Install
+
+```bash
+pip install bgeraser            # CLI + library
+pip install "bgeraser[web]"     # also the drag & drop web page
+```
+
+Needs Python 3.9+. The first run downloads the model (4 MB for the default `u2netp`; larger models download when you choose them).
+
+## Command line
+
+```bash
+# one image → transparent PNG next to it
+bgeraser photo.jpg
+
+# choose the output name
+bgeraser photo.jpg -o clean.png
+
+# solid background instead of transparent (product photos, ID photos)
+bgeraser photo.jpg --bg white
+bgeraser photo.jpg --bg "#00ff00"
+
+# a whole folder → ./products/nobg/
+bgeraser ./products
+bgeraser ./products -o ./out --recursive
+
+# better model for people
+bgeraser --model u2net_human_seg me.png
+
+# finer edges for hair / fur (slower)
+bgeraser photo.jpg --matting
+
+# list models
+bgeraser models
+```
+
+## Web page (drag & drop)
+
+```bash
+bgeraser serve
+```
+
+Then open <http://127.0.0.1:5000>. Drop, paste, or pick an image; download the PNG. Everything stays on your machine.
+
+`bgeraser serve --port 8080 --model isnet-general-use` to change port or model.
+
+## Python
+
+```python
+from bgeraser import remove_background, remove_background_bytes, process_folder
+
+remove_background("photo.jpg")                           # -> photo-nobg.png
+remove_background("photo.jpg", "out.png", bg_color="white")
+remove_background("photo.jpg", model="isnet-general-use", alpha_matting=True)
+
+png_bytes = remove_background_bytes(open("photo.jpg", "rb").read())   # in-memory
+
+process_folder("./products", "./products/out")           # batch
+```
+
+## Models
+
+| name | best for |
+|---|---|
+| `u2netp` *(default)* | fast, small, everyday use |
+| `u2net` | general, higher quality |
+| `isnet-general-use` | often the cleanest edges |
+| `u2net_human_seg` | people / portraits |
+| `isnet-anime` | anime & illustrations |
+| `u2net_cloth_seg` | clothing segmentation |
+| `silueta` | u2net quality, smaller file |
+
+## Why
+
+Online background removers are slow, watermark the result, charge per image, or make you upload customer photos to a stranger's server. This does the same job offline, for free, in one command.
+
+## Contributing
+
+Issues and pull requests welcome. Run tests with:
+
+```bash
+pip install -e ".[dev]"
+pytest
+```
+
+## License
+
+MIT © Tavrej / Yonzon Studio

bgeraser-0.1.0/pyproject.toml

@@ -0,0 +1,44 @@
+[build-system]
+requires = ["setuptools>=68", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "bgeraser"
+version = "0.1.0"
+description = "Remove image backgrounds locally — CLI, Python library and a tiny web UI. No uploads, no API keys."
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.9"
+authors = [{ name = "Tavrej (Yonzon Studio)", email = "tavrejstudio@gmail.com" }]
+keywords = ["background removal", "remove background", "rembg", "image", "transparent png", "cli"]
+classifiers = [
+  "Development Status :: 3 - Alpha",
+  "Environment :: Console",
+  "Intended Audience :: Developers",
+  "Intended Audience :: End Users/Desktop",
+  "License :: OSI Approved :: MIT License",
+  "Programming Language :: Python :: 3",
+  "Topic :: Multimedia :: Graphics",
+]
+dependencies = [
+  "rembg>=2.0.50",
+  "onnxruntime>=1.16",
+  "pillow>=10.0",
+]
+
+[project.optional-dependencies]
+web = ["flask>=3.0"]
+dev = ["pytest>=7", "flask>=3.0"]
+
+[project.scripts]
+bgeraser = "bgeraser.cli:main"
+
+[project.urls]
+Homepage = "https://github.com/guesttavrej/bgeraser"
+Issues = "https://github.com/guesttavrej/bgeraser/issues"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+bgeraser = ["templates/*.html"]

bgeraser-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

bgeraser-0.1.0/src/bgeraser/__init__.py

@@ -0,0 +1,26 @@
+"""bgeraser — remove image backgrounds locally.
+
+    >>> from bgeraser import remove_background
+    >>> remove_background("photo.jpg", "photo-nobg.png")
+
+Everything runs on your machine using open-source U²-Net / ISNet models
+via rembg. Nothing is uploaded anywhere.
+"""
+
+from .core import (
+    MODELS,
+    DEFAULT_MODEL,
+    remove_background,
+    remove_background_bytes,
+    process_folder,
+)
+
+__version__ = "0.1.0"
+__all__ = [
+    "MODELS",
+    "DEFAULT_MODEL",
+    "remove_background",
+    "remove_background_bytes",
+    "process_folder",
+    "__version__",
+]

bgeraser-0.1.0/src/bgeraser/cli.py

@@ -0,0 +1,99 @@
+"""Command-line interface for bgeraser."""
+
+from __future__ import annotations
+
+import argparse
+import sys
+import time
+from pathlib import Path
+
+from . import __version__
+from .core import DEFAULT_MODEL, MODELS, process_folder, remove_background
+
+
+def _build_parser() -> argparse.ArgumentParser:
+    p = argparse.ArgumentParser(
+        prog="bgeraser",
+        description="Remove image backgrounds locally. Works on a single file or a whole folder.",
+        epilog=(
+            "examples:\n"
+            "  bgeraser photo.jpg                     -> photo-nobg.png\n"
+            "  bgeraser photo.jpg -o clean.png\n"
+            "  bgeraser photo.jpg --bg white          -> white background instead of transparent\n"
+            "  bgeraser ./products -o ./products/out  -> every image in the folder\n"
+            "  bgeraser --model u2net_human_seg me.png\n"
+            "  bgeraser serve                         -> open a local web page (drag & drop)\n"
+            "  bgeraser models                        -> list available models"
+        ),
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    p.add_argument("input", nargs="?", help="image file or folder, or 'serve' / 'models'")
+    p.add_argument("-o", "--output", help="output file (single image) or output folder")
+    p.add_argument(
+        "-m", "--model", default=DEFAULT_MODEL, choices=list(MODELS),
+        help=f"segmentation model (default: {DEFAULT_MODEL})",
+    )
+    p.add_argument("--bg", metavar="COLOR", help="replace background with a color, e.g. white, #00ff00, 255,0,0")
+    p.add_argument("--matting", action="store_true", help="alpha matting for finer edges (hair, fur). Slower")
+    p.add_argument("--post", action="store_true", help="post-process the mask for smoother results")
+    p.add_argument("-r", "--recursive", action="store_true", help="include sub-folders when input is a folder")
+    p.add_argument("--host", default="127.0.0.1", help="host for 'serve' (default 127.0.0.1)")
+    p.add_argument("--port", type=int, default=5000, help="port for 'serve' (default 5000)")
+    p.add_argument("-q", "--quiet", action="store_true", help="no progress output")
+    p.add_argument("-V", "--version", action="version", version=f"bgeraser {__version__}")
+    return p
+
+
+def main(argv: list[str] | None = None) -> int:
+    parser = _build_parser()
+    args = parser.parse_args(argv)
+
+    if not args.input:
+        parser.print_help()
+        return 1
+
+    if args.input == "models":
+        for name, desc in MODELS.items():
+            mark = " (default)" if name == DEFAULT_MODEL else ""
+            print(f"{name:18s} {desc}{mark}")
+        return 0
+
+    if args.input == "serve":
+        from .web import serve
+
+        serve(host=args.host, port=args.port, model=args.model)
+        return 0
+
+    opts = dict(
+        model=args.model,
+        bg_color=args.bg,
+        alpha_matting=args.matting,
+        post_process=args.post,
+    )
+    src = Path(args.input)
+    if not src.exists():
+        print(f"error: {src} does not exist", file=sys.stderr)
+        return 2
+
+    t0 = time.time()
+    if src.is_dir():
+        def progress(i, o):
+            if not args.quiet:
+                print(f"  {i.name} -> {o}")
+
+        written = process_folder(src, args.output, recursive=args.recursive, on_progress=progress, **opts)
+        if not written:
+            print("no images found", file=sys.stderr)
+            return 1
+        if not args.quiet:
+            print(f"done: {len(written)} image(s) in {time.time() - t0:.1f}s")
+        return 0
+
+    out = remove_background(src, args.output, **opts)
+    if not args.quiet:
+        print(f"{src.name} -> {out}  ({time.time() - t0:.1f}s)")
+    return 0
+
+
+if __name__ == "__main__":  # pragma: no cover
+    sys.exit(main())

bgeraser-0.1.0/src/bgeraser/core.py

@@ -0,0 +1,162 @@
+"""Core background-removal functions."""
+
+from __future__ import annotations
+
+import io
+from pathlib import Path
+from typing import Callable, Iterable, Optional, Union
+
+from PIL import Image, ImageColor
+
+PathLike = Union[str, Path]
+
+#: Models available through rembg, smallest/fastest first.
+MODELS = {
+    "u2netp": "Fast, small (4 MB). Good default for quick jobs.",
+    "u2net": "General purpose, higher quality (170 MB).",
+    "u2net_human_seg": "Tuned for people / portraits.",
+    "u2net_cloth_seg": "Segments clothing (upper, lower, full body).",
+    "isnet-general-use": "Newer general model, often best edges.",
+    "isnet-anime": "For anime / illustration characters.",
+    "silueta": "u2net quality at ~43 MB.",
+    "sam": "Segment Anything (needs prompts, advanced).",
+}
+DEFAULT_MODEL = "u2netp"
+
+IMAGE_EXTS = {".jpg", ".jpeg", ".png", ".webp", ".bmp", ".tif", ".tiff"}
+
+_sessions: dict = {}
+
+
+def _get_session(model: str):
+    """Load (and cache) a rembg model session."""
+    if model not in MODELS:
+        raise ValueError(f"Unknown model '{model}'. Choose from: {', '.join(MODELS)}")
+    if model not in _sessions:
+        from rembg import new_session  # imported lazily: slow import
+
+        _sessions[model] = new_session(model)
+    return _sessions[model]
+
+
+def _parse_color(color: Optional[str]):
+    """'#ffffff', 'white', '255,0,0' or None -> RGB tuple or None."""
+    if not color:
+        return None
+    color = color.strip()
+    if "," in color:
+        parts = [int(p) for p in color.split(",")]
+        return tuple(parts[:3])
+    return ImageColor.getrgb(color)
+
+
+def remove_background_image(
+    image: Image.Image,
+    *,
+    model: str = DEFAULT_MODEL,
+    bg_color: Optional[str] = None,
+    alpha_matting: bool = False,
+    post_process: bool = False,
+) -> Image.Image:
+    """Remove the background from a PIL image and return an RGBA (or RGB) image.
+
+    Args:
+        image: Source image.
+        model: One of :data:`MODELS`.
+        bg_color: Optional replacement background ("white", "#00ff00", "255,0,0").
+            If given, the result is flattened onto that color.
+        alpha_matting: Refine edges (hair, fur). Slower.
+        post_process: Apply rembg's mask post-processing (smoother mask).
+    """
+    from rembg import remove  # lazy
+
+    session = _get_session(model)
+    result = remove(
+        image,
+        session=session,
+        alpha_matting=alpha_matting,
+        post_process_mask=post_process,
+    )
+    if not isinstance(result, Image.Image):  # very old rembg returned bytes
+        result = Image.open(io.BytesIO(result))
+    result = result.convert("RGBA")
+
+    rgb = _parse_color(bg_color)
+    if rgb is not None:
+        background = Image.new("RGBA", result.size, rgb + (255,))
+        background.alpha_composite(result)
+        result = background.convert("RGB")
+    return result
+
+
+def remove_background(
+    input_path: PathLike,
+    output_path: Optional[PathLike] = None,
+    **kwargs,
+) -> Path:
+    """Remove the background of the image at ``input_path``.
+
+    Writes a PNG (transparent unless ``bg_color`` is set) and returns its path.
+    If ``output_path`` is omitted, ``<name>-nobg.png`` is written next to the input.
+    """
+    input_path = Path(input_path)
+    if output_path is None:
+        output_path = input_path.with_name(f"{input_path.stem}-nobg.png")
+    output_path = Path(output_path)
+    output_path.parent.mkdir(parents=True, exist_ok=True)
+
+    with Image.open(input_path) as img:
+        img.load()
+        out = remove_background_image(img, **kwargs)
+
+    fmt = "PNG"
+    if output_path.suffix.lower() in {".jpg", ".jpeg"}:
+        fmt = "JPEG"
+        if out.mode == "RGBA":  # JPEG cannot hold transparency
+            flat = Image.new("RGB", out.size, (255, 255, 255))
+            flat.paste(out, mask=out.split()[3])
+            out = flat
+    elif output_path.suffix.lower() == ".webp":
+        fmt = "WEBP"
+    out.save(output_path, fmt)
+    return output_path
+
+
+def remove_background_bytes(data: bytes, **kwargs) -> bytes:
+    """Same as :func:`remove_background` but in-memory. Returns PNG bytes."""
+    with Image.open(io.BytesIO(data)) as img:
+        img.load()
+        out = remove_background_image(img, **kwargs)
+    buf = io.BytesIO()
+    out.save(buf, "PNG")
+    return buf.getvalue()
+
+
+def iter_images(folder: PathLike, recursive: bool = False) -> Iterable[Path]:
+    folder = Path(folder)
+    pattern = "**/*" if recursive else "*"
+    for p in sorted(folder.glob(pattern)):
+        if p.is_file() and p.suffix.lower() in IMAGE_EXTS and not p.stem.endswith("-nobg"):
+            yield p
+
+
+def process_folder(
+    input_dir: PathLike,
+    output_dir: Optional[PathLike] = None,
+    *,
+    recursive: bool = False,
+    on_progress: Optional[Callable[[Path, Path], None]] = None,
+    **kwargs,
+) -> list[Path]:
+    """Remove backgrounds from every image in a folder. Returns output paths."""
+    input_dir = Path(input_dir)
+    output_dir = Path(output_dir) if output_dir else input_dir / "nobg"
+    written: list[Path] = []
+    for src in iter_images(input_dir, recursive=recursive):
+        rel = src.relative_to(input_dir)
+        dst = (output_dir / rel).with_suffix(".png")
+        remove_background(src, dst, **kwargs)
+        written.append(dst)
+        if on_progress:
+            on_progress(src, dst)
+    return written

bgeraser-0.1.0/src/bgeraser/templates/index.html

@@ -0,0 +1,271 @@
+<!doctype html>
+<html lang="en">
+<head>
+<meta charset="utf-8">
+<meta name="viewport" content="width=device-width, initial-scale=1">
+<title>bgeraser — remove image backgrounds, offline</title>
+<meta name="color-scheme" content="dark light">
+<link rel="icon" href="data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 32 32'%3E%3Crect width='32' height='32' rx='8' fill='%235b8cff'/%3E%3Cpath d='M9 22 L19 9 l4 3 -10 13z' fill='%23fff'/%3E%3Crect x='8' y='23' width='16' height='2' rx='1' fill='%23fff' opacity='.7'/%3E%3C/svg%3E">
+<style>
+  :root {
+    --bg:#0b0d12; --bg2:#10131a; --card:#151922; --card2:#1b2030; --line:#262c3a; --line2:#334;
+    --text:#eef0f5; --muted:#8d94a6; --dim:#5f6678;
+    --accent:#5b8cff; --accent2:#8b5cf6; --ok:#34d399; --err:#fb7185;
+    --shadow:0 20px 60px -20px rgba(0,0,0,.6);
+    --check-a:#2a3040; --check-b:#1c2130;
+  }
+  @media (prefers-color-scheme: light) {
+    :root {
+      --bg:#f6f7fb; --bg2:#eef0f6; --card:#ffffff; --card2:#f2f4f9; --line:#e3e6ee; --line2:#cfd4e0;
+      --text:#11131a; --muted:#5b6172; --dim:#9aa0b0;
+      --shadow:0 20px 50px -24px rgba(20,30,60,.25);
+      --check-a:#e2e5ee; --check-b:#f7f8fb;
+    }
+  }
+  * { box-sizing:border-box; }
+  html, body { margin:0; }
+  body {
+    font:15px/1.55 -apple-system, BlinkMacSystemFont, "Segoe UI", Inter, Roboto, Helvetica, Arial, sans-serif;
+    background:
+      radial-gradient(900px 500px at 10% -10%, rgba(91,140,255,.18), transparent 60%),
+      radial-gradient(700px 400px at 100% 0%, rgba(139,92,246,.14), transparent 60%),
+      var(--bg);
+    color:var(--text); min-height:100vh; -webkit-font-smoothing:antialiased;
+  }
+  a { color:inherit; }
+  .wrap { max-width:1080px; margin:0 auto; padding:28px 20px 60px; }
+
+  /* header */
+  header { display:flex; align-items:center; justify-content:space-between; gap:16px; margin-bottom:28px; }
+  .brand { display:flex; align-items:center; gap:10px; font-weight:700; font-size:18px; letter-spacing:-.01em; }
+  .logo { width:30px; height:30px; border-radius:9px; background:linear-gradient(135deg,var(--accent),var(--accent2)); display:grid; place-items:center; box-shadow:0 6px 18px -6px var(--accent); }
+  .logo svg { width:18px; height:18px; }
+  .pill { display:inline-flex; align-items:center; gap:6px; font-size:12px; color:var(--muted); border:1px solid var(--line); background:var(--card); padding:5px 10px; border-radius:999px; }
+  .pill i { width:7px; height:7px; border-radius:50%; background:var(--ok); box-shadow:0 0 0 3px rgba(52,211,153,.18); }
+
+  /* hero */
+  .hero { text-align:center; margin:8px 0 26px; }
+  .hero h1 { font-size:clamp(28px,4.6vw,44px); line-height:1.1; margin:0 0 10px; letter-spacing:-.03em; font-weight:800; }
+  .hero h1 span { background:linear-gradient(90deg,var(--accent),var(--accent2)); -webkit-background-clip:text; background-clip:text; color:transparent; }
+  .hero p { margin:0 auto; color:var(--muted); max-width:560px; font-size:16px; }
+
+  /* drop zone */
+  .drop {
+    position:relative; display:block; border:1.5px dashed var(--line2); border-radius:18px; padding:38px 20px;
+    text-align:center; cursor:pointer; background:var(--card); transition:border-color .15s, background .15s, transform .15s;
+    box-shadow:var(--shadow);
+  }
+  .drop:hover, .drop.over { border-color:var(--accent); background:var(--card2); }
+  .drop.over { transform:scale(1.005); }
+  .drop input { display:none; }
+  .drop .ic { width:52px; height:52px; margin:0 auto 12px; border-radius:14px; background:linear-gradient(135deg,rgba(91,140,255,.18),rgba(139,92,246,.18)); display:grid; place-items:center; }
+  .drop .ic svg { width:26px; height:26px; stroke:var(--accent); }
+  .drop strong { font-size:17px; }
+  .drop .sub { color:var(--muted); margin-top:4px; font-size:14px; }
+  .drop kbd { font:12px ui-monospace, SFMono-Regular, Menlo, monospace; border:1px solid var(--line2); border-radius:6px; padding:1px 6px; color:var(--muted); }
+
+  /* options */
+  .opts { display:flex; flex-wrap:wrap; gap:10px 18px; align-items:center; margin:18px 0 6px; }
+  .opts label { display:flex; align-items:center; gap:8px; color:var(--muted); font-size:14px; }
+  select, input[type=text] {
+    font:inherit; background:var(--card); color:var(--text); border:1px solid var(--line); border-radius:10px; padding:8px 12px; outline:none;
+  }
+  select:focus, input[type=text]:focus { border-color:var(--accent); box-shadow:0 0 0 3px rgba(91,140,255,.18); }
+  .swatches { display:flex; gap:6px; align-items:center; }
+  .sw { width:28px; height:28px; border-radius:8px; border:2px solid var(--line); cursor:pointer; padding:0; position:relative; background:var(--card); }
+  .sw.on { border-color:var(--accent); box-shadow:0 0 0 3px rgba(91,140,255,.18); }
+  .sw[data-bg=""] { background: repeating-conic-gradient(var(--check-a) 0% 25%, var(--check-b) 0% 50%) 50% / 12px 12px; }
+  .sw input[type=color] { position:absolute; inset:0; opacity:0; cursor:pointer; width:100%; height:100%; }
+  .toggle { position:relative; width:38px; height:22px; background:var(--line2); border-radius:999px; transition:.15s; flex:none; }
+  .toggle::after { content:""; position:absolute; top:3px; left:3px; width:16px; height:16px; border-radius:50%; background:#fff; transition:.15s; }
+  input:checked + .toggle { background:var(--accent); }
+  input:checked + .toggle::after { transform:translateX(16px); }
+  .opts input[type=checkbox] { display:none; }
+
+  .status { min-height:24px; margin:10px 0 14px; color:var(--muted); font-size:14px; display:flex; align-items:center; gap:10px; }
+  .status.err { color:var(--err); }
+  .status.ok { color:var(--ok); }
+  .spin { width:16px; height:16px; border:2px solid var(--line2); border-top-color:var(--accent); border-radius:50%; animation:sp .8s linear infinite; }
+  @keyframes sp { to { transform:rotate(360deg); } }
+
+  /* result */
+  .result { display:none; }
+  .result.show { display:block; animation:up .35s ease; }
+  @keyframes up { from { opacity:0; transform:translateY(10px); } }
+  .cmp {
+    position:relative; border-radius:18px; overflow:hidden; border:1px solid var(--line); box-shadow:var(--shadow);
+    background: repeating-conic-gradient(var(--check-a) 0% 25%, var(--check-b) 0% 50%) 50% / 22px 22px;
+    user-select:none; touch-action:none; width:fit-content; max-width:100%; margin:0 auto;
+  }
+  .cmp img { display:block; width:auto; max-width:100%; height:auto; max-height:70vh; }
+  .cmp .before { position:absolute; inset:0; overflow:hidden; background:var(--card); }
+  .cmp .before img { position:absolute; left:0; top:0; max-width:none; max-height:none; }
+  .cmp .handle { position:absolute; top:0; bottom:0; width:2px; background:#fff; box-shadow:0 0 0 1px rgba(0,0,0,.25); }
+  .cmp .knob { position:absolute; top:50%; left:50%; transform:translate(-50%,-50%); width:40px; height:40px; border-radius:50%; background:#fff; display:grid; place-items:center; box-shadow:0 6px 20px rgba(0,0,0,.35); color:#111; }
+  .cmp .tag { position:absolute; top:12px; font-size:11px; font-weight:700; letter-spacing:.08em; padding:4px 9px; border-radius:6px; background:rgba(0,0,0,.55); color:#fff; backdrop-filter:blur(6px); }
+  .cmp .tag.l { left:12px; } .cmp .tag.r { right:12px; }
+  .actions { display:flex; flex-wrap:wrap; gap:10px; align-items:center; margin-top:14px; }
+  .btn { display:inline-flex; align-items:center; gap:8px; font:inherit; font-weight:600; border:0; cursor:pointer; padding:11px 18px; border-radius:12px; text-decoration:none; transition:.15s; }
+  .btn.p { background:linear-gradient(135deg,var(--accent),var(--accent2)); color:#fff; box-shadow:0 10px 24px -10px var(--accent); }
+  .btn.p:hover { transform:translateY(-1px); filter:brightness(1.05); }
+  .btn.s { background:var(--card); color:var(--text); border:1px solid var(--line); }
+  .btn.s:hover { border-color:var(--line2); }
+  .btn svg { width:16px; height:16px; }
+  .meta { color:var(--dim); font-size:13px; margin-left:auto; }
+
+  /* features */
+  .feats { display:grid; grid-template-columns:repeat(3,1fr); gap:12px; margin-top:34px; }
+  .feat { background:var(--card); border:1px solid var(--line); border-radius:14px; padding:16px; }
+  .feat b { display:block; margin-bottom:4px; }
+  .feat span { color:var(--muted); font-size:14px; }
+  footer { margin-top:34px; color:var(--dim); font-size:13px; text-align:center; }
+  footer a { color:var(--muted); }
+  @media (max-width:720px) { .feats { grid-template-columns:1fr; } .pill span { display:none; } .wrap { padding:20px 16px 40px; } .drop { padding:28px 14px; } .meta { margin-left:0; width:100%; } }
+</style>
+</head>
+<body>
+<div class="wrap">
+  <header>
+    <div class="brand">
+      <div class="logo"><svg viewBox="0 0 24 24" fill="none" stroke="#fff" stroke-width="2.2" stroke-linecap="round" stroke-linejoin="round"><path d="M5 18 15 5l4 3L9 21H5z"/><path d="M4 21h16"/></svg></div>
+      bgeraser
+    </div>
+    <span class="pill"><i></i> offline<span> · nothing leaves this computer</span></span>
+  </header>
+
+  <section class="hero">
+    <h1>Remove backgrounds.<br><span>Offline. Free. Unlimited.</span></h1>
+    <p>Drop a photo and get a clean transparent PNG in seconds — no uploads, no watermarks, no sign-up.</p>
+  </section>
+
+  <label class="drop" id="drop">
+    <input type="file" id="file" accept="image/*">
+    <div class="ic"><svg viewBox="0 0 24 24" fill="none" stroke-width="1.8" stroke-linecap="round" stroke-linejoin="round"><path d="M12 16V4M7 9l5-5 5 5"/><path d="M4 16v3a1 1 0 0 0 1 1h14a1 1 0 0 0 1-1v-3"/></svg></div>
+    <strong>Drop an image here</strong> or click to choose
+    <div class="sub">JPG · PNG · WEBP · up to 25 MB — or press <kbd>Ctrl</kbd>+<kbd>V</kbd> to paste</div>
+  </label>
+
+  <div class="opts">
+    <label>Model
+      <select id="model">
+        {% for name, desc in models.items() %}
+        <option value="{{ name }}" {% if name == default_model %}selected{% endif %}>{{ name }} — {{ desc }}</option>
+        {% endfor %}
+      </select>
+    </label>
+    <label>Background
+      <span class="swatches" id="swatches">
+        <button type="button" class="sw on" data-bg="" title="Transparent"></button>
+        <button type="button" class="sw" data-bg="#ffffff" style="background:#fff" title="White"></button>
+        <button type="button" class="sw" data-bg="#000000" style="background:#000" title="Black"></button>
+        <button type="button" class="sw" data-bg="#00ff00" style="background:#00ff00" title="Green screen"></button>
+        <span class="sw" id="customSw" title="Custom color" style="background:conic-gradient(red,yellow,lime,cyan,blue,magenta,red)"><input type="color" id="customColor" value="#5b8cff"></span>
+      </span>
+    </label>
+    <label><input type="checkbox" id="matting"><span class="toggle"></span> Fine edges (hair, fur — slower)</label>
+  </div>
+
+  <div class="status" id="status">Best on photos with one clear subject — a person, product, pet or logo.</div>
+
+  <section class="result" id="result">
+    <div class="cmp" id="cmp" style="--w:50%">
+      <img id="after" alt="Result">
+      <div class="before" id="beforeWrap" style="width:50%"><img id="before" alt="Original"></div>
+      <div class="handle" id="handle" style="left:50%"><div class="knob"><svg width="18" height="18" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2.4" stroke-linecap="round"><path d="M8 7l-5 5 5 5M16 7l5 5-5 5"/></svg></div></div>
+      <span class="tag l">BEFORE</span><span class="tag r">AFTER</span>
+    </div>
+    <div class="actions">
+      <a class="btn p" id="dl" download><svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2.2" stroke-linecap="round" stroke-linejoin="round"><path d="M12 4v12M7 11l5 5 5-5"/><path d="M4 20h16"/></svg>Download PNG</a>
+      <button class="btn s" id="again" type="button">Another image</button>
+      <span class="meta" id="meta"></span>
+    </div>
+  </section>
+
+  <section class="feats">
+    <div class="feat"><b>100% private</b><span>The AI model runs on this machine. Your photos never touch the internet.</span></div>
+    <div class="feat"><b>No limits</b><span>Process one image or a thousand. No credits, no watermark, no subscription.</span></div>
+    <div class="feat"><b>Batch from the terminal</b><span><code>bgeraser ./folder</code> clears every photo in a folder in one command.</span></div>
+  </section>
+
+  <footer>Open source · MIT · <a href="https://github.com/guesttavrej/bgeraser" target="_blank" rel="noopener">github.com/guesttavrej/bgeraser</a></footer>
+</div>
+
+<script>
+const $ = id => document.getElementById(id);
+const drop = $('drop'), file = $('file'), status = $('status'), result = $('result');
+const before = $('before'), after = $('after'), beforeWrap = $('beforeWrap'), handle = $('handle'), cmp = $('cmp');
+const dl = $('dl'), meta = $('meta');
+let bg = '', currentFile = null, busy = false;
+
+/* background swatches */
+$('swatches').addEventListener('click', e => {
+  const b = e.target.closest('.sw'); if (!b || b.id === 'customSw') return;
+  document.querySelectorAll('.sw').forEach(s => s.classList.remove('on')); b.classList.add('on');
+  bg = b.dataset.bg; if (currentFile && !busy) run(currentFile);
+});
+$('customColor').addEventListener('input', e => {
+  document.querySelectorAll('.sw').forEach(s => s.classList.remove('on')); $('customSw').classList.add('on');
+  $('customSw').style.background = e.target.value; bg = e.target.value;
+});
+$('customColor').addEventListener('change', () => { if (currentFile && !busy) run(currentFile); });
+$('model').addEventListener('change', () => { if (currentFile && !busy) run(currentFile); });
+$('matting').addEventListener('change', () => { if (currentFile && !busy) run(currentFile); });
+
+/* input methods */
+['dragenter','dragover'].forEach(t => document.addEventListener(t, e => { e.preventDefault(); drop.classList.add('over'); }));
+['dragleave','drop'].forEach(t => document.addEventListener(t, e => { e.preventDefault(); if (t === 'drop' || e.target === document.documentElement) drop.classList.remove('over'); }));
+document.addEventListener('drop', e => { const f = e.dataTransfer.files[0]; if (f) run(f); });
+file.addEventListener('change', () => { if (file.files[0]) run(file.files[0]); file.value = ''; });
+window.addEventListener('paste', e => { const f = [...(e.clipboardData?.files || [])][0]; if (f) run(f); });
+$('again').addEventListener('click', () => { result.classList.remove('show'); currentFile = null; setStatus('Drop the next image.'); file.click(); });
+
+/* comparison slider */
+function setSplit(x) {
+  const r = cmp.getBoundingClientRect(); const p = Math.max(0, Math.min(1, (x - r.left) / r.width)) * 100;
+  beforeWrap.style.width = p + '%'; handle.style.left = p + '%';
+}
+let dragging = false;
+cmp.addEventListener('pointerdown', e => { dragging = true; cmp.setPointerCapture(e.pointerId); setSplit(e.clientX); });
+cmp.addEventListener('pointermove', e => { if (dragging) setSplit(e.clientX); });
+cmp.addEventListener('pointerup', () => dragging = false);
+cmp.addEventListener('pointercancel', () => dragging = false);
+function syncBefore() { before.style.width = after.clientWidth + 'px'; before.style.height = after.clientHeight + 'px'; }
+window.addEventListener('resize', syncBefore);
+
+function setStatus(msg, cls) { status.className = 'status' + (cls ? ' ' + cls : ''); status.innerHTML = msg; }
+
+async function run(f) {
+  if (!f.type.startsWith('image/')) { setStatus('That is not an image file.', 'err'); return; }
+  currentFile = f; busy = true;
+  setStatus('<span class="spin"></span> Removing background…');
+  const slow = setTimeout(() => setStatus('<span class="spin"></span> Still working — large images and bigger models take longer.'), 8000);
+  const fd = new FormData();
+  fd.append('image', f); fd.append('model', $('model').value); fd.append('bg', bg);
+  if ($('matting').checked) fd.append('matting', 'on');
+  const t0 = performance.now();
+  try {
+    const r = await fetch('/remove', { method: 'POST', body: fd });
+    if (!r.ok) { const j = await r.json().catch(() => ({})); throw new Error(j.error || r.statusText); }
+    const blob = await r.blob(); const url = URL.createObjectURL(blob);
+    before.src = URL.createObjectURL(f); after.src = url;
+    await Promise.all([
+      new Promise(res => { after.complete && after.naturalWidth ? res() : (after.onload = res); }),
+      new Promise(res => { before.complete && before.naturalWidth ? res() : (before.onload = res); }),
+    ]);
+    result.classList.add('show');
+    syncBefore(); beforeWrap.style.width = '50%'; handle.style.left = '50%';
+    result.scrollIntoView({ behavior: 'smooth', block: 'nearest' });
+    dl.href = url; dl.download = f.name.replace(/\.[^.]+$/, '') + '-nobg.png';
+    const s = ((performance.now() - t0) / 1000).toFixed(1);
+    meta.textContent = `${after.naturalWidth}×${after.naturalHeight} · ${(blob.size/1024).toFixed(0)} KB · ${s}s`;
+    setStatus(`Done in ${s}s — drag the slider to compare.`, 'ok');
+  } catch (e) {
+    const m = e.message === 'Failed to fetch'
+      ? 'Could not reach the server. Is the “bgeraser serve” window still open? Restart it and reload this page.'
+      : e.message;
+    setStatus('Error: ' + m, 'err');
+  } finally { clearTimeout(slow); busy = false; }
+}
+</script>
+</body>
+</html>

bgeraser-0.1.0/src/bgeraser/web.py

@@ -0,0 +1,69 @@
+"""Tiny local web UI: `bgeraser serve` then open http://127.0.0.1:5000."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from .core import DEFAULT_MODEL, MODELS, remove_background_bytes
+
+TEMPLATE = (Path(__file__).parent / "templates" / "index.html").read_text(encoding="utf-8")
+
+
+def create_app(model: str = DEFAULT_MODEL):
+    try:
+        from flask import Flask, Response, render_template_string, request
+    except ImportError as e:  # pragma: no cover
+        raise SystemExit("The web UI needs Flask:  pip install 'bgeraser[web]'") from e
+
+    app = Flask(__name__)
+    app.config["MAX_CONTENT_LENGTH"] = 25 * 1024 * 1024  # 25 MB
+
+    @app.get("/")
+    def index():
+        return render_template_string(TEMPLATE, models=MODELS, default_model=model)
+
+    @app.post("/remove")
+    def remove():
+        f = request.files.get("image")
+        if not f:
+            return {"error": "no image uploaded"}, 400
+        chosen = request.form.get("model") or model
+        bg = request.form.get("bg") or None
+        matting = request.form.get("matting") == "on"
+        try:
+            png = remove_background_bytes(f.read(), model=chosen, bg_color=bg, alpha_matting=matting)
+        except ValueError as e:
+            return {"error": str(e)}, 400
+        except Exception as e:  # model download failure, bad image, etc.
+            msg = f"{type(e).__name__}: {e}"
+            print(f"[bgeraser] error: {msg}", flush=True)
+            return {"error": msg}, 500
+        stem = Path(f.filename or "image").stem
+        return Response(
+            png,
+            mimetype="image/png",
+            headers={"Content-Disposition": f'inline; filename="{stem}-nobg.png"'},
+        )
+
+    @app.get("/health")
+    def health():
+        return {"ok": True, "model": model}
+
+    return app
+
+
+def serve(host: str = "127.0.0.1", port: int = 5000, model: str = DEFAULT_MODEL) -> None:
+    from .core import _get_session
+
+    print(f"loading model '{model}' (downloads on first run, please wait)...", flush=True)
+    try:
+        _get_session(model)
+    except Exception as e:
+        print(f"could not load model: {type(e).__name__}: {e}\n"
+              f"check your internet connection and run again.", flush=True)
+        raise SystemExit(1)
+    print("model ready", flush=True)
+
+    app = create_app(model)
+    print(f"bgeraser web UI -> http://{host}:{port}   (model: {model}, Ctrl+C to stop)", flush=True)
+    app.run(host=host, port=port, debug=False, threaded=True)

bgeraser-0.1.0/src/bgeraser.egg-info/PKG-INFO

@@ -0,0 +1,131 @@
+Metadata-Version: 2.4
+Name: bgeraser
+Version: 0.1.0
+Summary: Remove image backgrounds locally — CLI, Python library and a tiny web UI. No uploads, no API keys.
+Author-email: "Tavrej (Yonzon Studio)" <tavrejstudio@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/guesttavrej/bgeraser
+Project-URL: Issues, https://github.com/guesttavrej/bgeraser/issues
+Keywords: background removal,remove background,rembg,image,transparent png,cli
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: End Users/Desktop
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Multimedia :: Graphics
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: rembg>=2.0.50
+Requires-Dist: onnxruntime>=1.16
+Requires-Dist: pillow>=10.0
+Provides-Extra: web
+Requires-Dist: flask>=3.0; extra == "web"
+Provides-Extra: dev
+Requires-Dist: pytest>=7; extra == "dev"
+Requires-Dist: flask>=3.0; extra == "dev"
+Dynamic: license-file
+
+# bgeraser
+
+**Remove image backgrounds on your own computer.** No uploads, no API keys, no watermarks, no limits.
+
+Works three ways:
+
+| | |
+|---|---|
+| **Command line** | `bgeraser photo.jpg` → `photo-nobg.png` |
+| **Python library** | `remove_background("photo.jpg")` |
+| **Local web page** | `bgeraser serve` → drag & drop in your browser |
+
+Powered by open-source [U²-Net](https://github.com/xuebinqin/U-2-Net) / ISNet models through [rembg](https://github.com/danielgatis/rembg).
+
+## Install
+
+```bash
+pip install bgeraser            # CLI + library
+pip install "bgeraser[web]"     # also the drag & drop web page
+```
+
+Needs Python 3.9+. The first run downloads the model (4 MB for the default `u2netp`; larger models download when you choose them).
+
+## Command line
+
+```bash
+# one image → transparent PNG next to it
+bgeraser photo.jpg
+
+# choose the output name
+bgeraser photo.jpg -o clean.png
+
+# solid background instead of transparent (product photos, ID photos)
+bgeraser photo.jpg --bg white
+bgeraser photo.jpg --bg "#00ff00"
+
+# a whole folder → ./products/nobg/
+bgeraser ./products
+bgeraser ./products -o ./out --recursive
+
+# better model for people
+bgeraser --model u2net_human_seg me.png
+
+# finer edges for hair / fur (slower)
+bgeraser photo.jpg --matting
+
+# list models
+bgeraser models
+```
+
+## Web page (drag & drop)
+
+```bash
+bgeraser serve
+```
+
+Then open <http://127.0.0.1:5000>. Drop, paste, or pick an image; download the PNG. Everything stays on your machine.
+
+`bgeraser serve --port 8080 --model isnet-general-use` to change port or model.
+
+## Python
+
+```python
+from bgeraser import remove_background, remove_background_bytes, process_folder
+
+remove_background("photo.jpg")                           # -> photo-nobg.png
+remove_background("photo.jpg", "out.png", bg_color="white")
+remove_background("photo.jpg", model="isnet-general-use", alpha_matting=True)
+
+png_bytes = remove_background_bytes(open("photo.jpg", "rb").read())   # in-memory
+
+process_folder("./products", "./products/out")           # batch
+```
+
+## Models
+
+| name | best for |
+|---|---|
+| `u2netp` *(default)* | fast, small, everyday use |
+| `u2net` | general, higher quality |
+| `isnet-general-use` | often the cleanest edges |
+| `u2net_human_seg` | people / portraits |
+| `isnet-anime` | anime & illustrations |
+| `u2net_cloth_seg` | clothing segmentation |
+| `silueta` | u2net quality, smaller file |
+
+## Why
+
+Online background removers are slow, watermark the result, charge per image, or make you upload customer photos to a stranger's server. This does the same job offline, for free, in one command.
+
+## Contributing
+
+Issues and pull requests welcome. Run tests with:
+
+```bash
+pip install -e ".[dev]"
+pytest
+```
+
+## License
+
+MIT © Tavrej / Yonzon Studio

bgeraser-0.1.0/src/bgeraser.egg-info/SOURCES.txt

@@ -0,0 +1,15 @@
+LICENSE
+README.md
+pyproject.toml
+src/bgeraser/__init__.py
+src/bgeraser/cli.py
+src/bgeraser/core.py
+src/bgeraser/web.py
+src/bgeraser.egg-info/PKG-INFO
+src/bgeraser.egg-info/SOURCES.txt
+src/bgeraser.egg-info/dependency_links.txt
+src/bgeraser.egg-info/entry_points.txt
+src/bgeraser.egg-info/requires.txt
+src/bgeraser.egg-info/top_level.txt
+src/bgeraser/templates/index.html
+tests/test_bgeraser.py

bgeraser-0.1.0/src/bgeraser.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

bgeraser-0.1.0/src/bgeraser.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+bgeraser = bgeraser.cli:main

bgeraser-0.1.0/src/bgeraser.egg-info/requires.txt

@@ -0,0 +1,10 @@
+rembg>=2.0.50
+onnxruntime>=1.16
+pillow>=10.0
+
+[dev]
+pytest>=7
+flask>=3.0
+
+[web]
+flask>=3.0

bgeraser-0.1.0/src/bgeraser.egg-info/top_level.txt

@@ -0,0 +1 @@
+bgeraser

bgeraser-0.1.0/tests/test_bgeraser.py

@@ -0,0 +1,84 @@
+import io
+from pathlib import Path
+
+import pytest
+from PIL import Image, ImageDraw
+
+from bgeraser import remove_background, remove_background_bytes, process_folder, MODELS, DEFAULT_MODEL
+from bgeraser.cli import main
+
+
+@pytest.fixture
+def sample(tmp_path: Path) -> Path:
+    """Red circle on a white background — an easy subject for the model."""
+    img = Image.new("RGB", (160, 160), "white")
+    ImageDraw.Draw(img).ellipse((30, 30, 130, 130), fill=(220, 30, 30))
+    p = tmp_path / "circle.jpg"
+    img.save(p, quality=95)
+    return p
+
+
+def test_default_model_listed():
+    assert DEFAULT_MODEL in MODELS
+
+
+def test_remove_creates_transparent_png(sample: Path):
+    out = remove_background(sample)
+    assert out.name == "circle-nobg.png"
+    with Image.open(out) as im:
+        assert im.mode == "RGBA"
+        assert im.size == (160, 160)
+        alpha = im.split()[3]
+        assert alpha.getpixel((2, 2)) < 40          # corner should be (mostly) transparent
+        assert alpha.getpixel((80, 80)) > 200       # centre of circle should be opaque
+
+
+def test_bg_color_flattens(sample: Path, tmp_path: Path):
+    out = remove_background(sample, tmp_path / "flat.png", bg_color="#00ff00")
+    with Image.open(out) as im:
+        assert im.mode == "RGB"
+        assert im.getpixel((2, 2)) == (0, 255, 0)
+
+
+def test_bytes_roundtrip(sample: Path):
+    png = remove_background_bytes(sample.read_bytes())
+    with Image.open(io.BytesIO(png)) as im:
+        assert im.format == "PNG" and im.mode == "RGBA"
+
+
+def test_unknown_model(sample: Path):
+    with pytest.raises(ValueError):
+        remove_background(sample, model="does-not-exist")
+
+
+def test_process_folder(sample: Path, tmp_path: Path):
+    (tmp_path / "second.png").write_bytes(sample.read_bytes())
+    written = process_folder(tmp_path, tmp_path / "out")
+    assert sorted(p.name for p in written) == ["circle.png", "second.png"]
+    assert all(p.exists() for p in written)
+
+
+def test_cli_single(sample: Path, tmp_path: Path, capsys):
+    rc = main([str(sample), "-o", str(tmp_path / "cli.png"), "-q"])
+    assert rc == 0 and (tmp_path / "cli.png").exists()
+
+
+def test_cli_models(capsys):
+    assert main(["models"]) == 0
+    assert DEFAULT_MODEL in capsys.readouterr().out
+
+
+def test_cli_missing_file(tmp_path: Path):
+    assert main([str(tmp_path / "nope.jpg")]) == 2
+
+
+def test_web_app(sample: Path):
+    flask = pytest.importorskip("flask")
+    from bgeraser.web import create_app
+
+    client = create_app().test_client()
+    assert client.get("/health").json["ok"] is True
+    assert b"bgeraser" in client.get("/").data
+    r = client.post("/remove", data={"image": (io.BytesIO(sample.read_bytes()), "circle.jpg")})
+    assert r.status_code == 200 and r.mimetype == "image/png"
+    assert client.post("/remove", data={}).status_code == 400