remove-ai-watermarks 0.3.3__tar.gz

remove_ai_watermarks-0.3.3/.claude/settings.json

@@ -0,0 +1,14 @@
+{
+  "permissions": {
+    "allow": [
+      "WebSearch",
+      "WebFetch"
+    ]
+  },
+  "enabledPlugins": {
+    "pyright-lsp@claude-plugins-official": true,
+    "context7@claude-plugins-official": true,
+    "code-simplifier@claude-plugins-official": true,
+    "claude-md-management@claude-plugins-official": true
+  }
+}

remove_ai_watermarks-0.3.3/.env.example

@@ -0,0 +1,3 @@
+# HuggingFace token (required for invisible watermark removal)
+# Get yours at: https://huggingface.co/settings/tokens
+HF_TOKEN=

remove_ai_watermarks-0.3.3/.gitattributes

@@ -0,0 +1,9 @@
+# Auto detect text files and perform LF normalization
+* text=auto
+
+# Binary files
+*.png binary
+*.jpg binary
+*.jpeg binary
+*.webp binary
+*.pt binary

remove_ai_watermarks-0.3.3/.github/dependabot.yml

@@ -0,0 +1,29 @@
+version: 2
+
+updates:
+  - package-ecosystem: "uv"
+    directory: "/"
+    schedule:
+      interval: "weekly"
+    open-pull-requests-limit: 10
+    labels:
+      - "dependencies"
+      - "python"
+    groups:
+      minor-and-patch:
+        update-types:
+          - "minor"
+          - "patch"
+
+  - package-ecosystem: "github-actions"
+    directory: "/"
+    schedule:
+      interval: "weekly"
+    open-pull-requests-limit: 5
+    labels:
+      - "dependencies"
+      - "github-actions"
+    groups:
+      actions:
+        patterns:
+          - "*"

remove_ai_watermarks-0.3.3/.github/workflows/publish.yml

@@ -0,0 +1,24 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+permissions:
+  id-token: write
+  contents: read
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    environment: pypi
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: astral-sh/setup-uv@v6
+
+      - name: Build package
+        run: uv build
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

remove_ai_watermarks-0.3.3/.gitignore

@@ -0,0 +1,32 @@
+# Dependencies
+.venv/
+__pycache__/
+*.egg-info/
+dist/
+build/
+
+# Environment secrets
+.env
+
+# OS files
+.DS_Store
+Thumbs.db
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+
+# Test results
+data/results/
+
+# Reference materials
+_refs/
+
+# Downloaded model weights
+yolov8n.pt
+.coverage
+
+# Claude Code local settings
+.claude/settings.local.json

remove_ai_watermarks-0.3.3/CLAUDE.md

@@ -0,0 +1,12 @@
+# Remove-AI-Watermarks
+
+You are a **principal Python engineer** maintaining a CLI tool and library for removing visible and invisible AI watermarks from images.
+
+## How to run
+
+- `uv run remove-ai-watermarks all <image.png> -o <output.png>`
+
+## Configuration
+
+- GPU/ML modules (invisible_engine, ctrlregen, watermark_remover) are optional — guard imports with `is_available()` checks
+- Tests for ML modules are limited to availability checks (require multi-GB downloads)

remove_ai_watermarks-0.3.3/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 wiltodelta
+
+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.

remove_ai_watermarks-0.3.3/PKG-INFO

@@ -0,0 +1,309 @@
+Metadata-Version: 2.4
+Name: remove-ai-watermarks
+Version: 0.3.3
+Summary: Remove visible and invisible AI watermarks from images (Gemini / Nano Banana, ChatGPT, Stable Diffusion)
+Project-URL: Repository, https://github.com/wiltodelta/remove-ai-watermarks
+License: MIT
+License-File: LICENSE
+Requires-Python: >=3.10
+Requires-Dist: click>=8.0.0
+Requires-Dist: numpy>=1.24.0
+Requires-Dist: opencv-python-headless>=4.8.0
+Requires-Dist: piexif>=1.1.3
+Requires-Dist: pillow>=10.0.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: rich>=13.0.0
+Provides-Extra: all
+Requires-Dist: accelerate>=0.25.0; extra == 'all'
+Requires-Dist: color-matcher>=0.5.0; extra == 'all'
+Requires-Dist: controlnet-aux>=0.0.9; extra == 'all'
+Requires-Dist: diffusers>=0.25.0; extra == 'all'
+Requires-Dist: pyright>=1.1.0; extra == 'all'
+Requires-Dist: pytest-cov>=4.1.0; extra == 'all'
+Requires-Dist: pytest>=8.0.0; extra == 'all'
+Requires-Dist: ruff>=0.4.0; extra == 'all'
+Requires-Dist: safetensors; extra == 'all'
+Requires-Dist: torch>=2.0.0; extra == 'all'
+Requires-Dist: transformers>=4.35.0; extra == 'all'
+Requires-Dist: ultralytics>=8.0.0; extra == 'all'
+Provides-Extra: dev
+Requires-Dist: pyright>=1.1.0; extra == 'dev'
+Requires-Dist: pytest-cov>=4.1.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.4.0; extra == 'dev'
+Provides-Extra: gpu
+Requires-Dist: accelerate>=0.25.0; extra == 'gpu'
+Requires-Dist: color-matcher>=0.5.0; extra == 'gpu'
+Requires-Dist: controlnet-aux>=0.0.9; extra == 'gpu'
+Requires-Dist: diffusers>=0.25.0; extra == 'gpu'
+Requires-Dist: safetensors; extra == 'gpu'
+Requires-Dist: torch>=2.0.0; extra == 'gpu'
+Requires-Dist: transformers>=4.35.0; extra == 'gpu'
+Requires-Dist: ultralytics>=8.0.0; extra == 'gpu'
+Description-Content-Type: text/markdown
+
+# Remove-AI-Watermarks
+
+Remove **visible** and **invisible** AI watermarks from images generated by Google Gemini (Nano Banana), ChatGPT / DALL-E, Stable Diffusion, Adobe Firefly, Midjourney, and other AI models.
+
+Strips SynthID, C2PA Content Credentials, EXIF/XMP "Made with AI" labels, and visible sparkle overlays — all in one command.
+
+## Features
+
+- **Visible watermark removal** — Gemini / Nano Banana sparkle logo via reverse alpha blending (fast, offline, deterministic)
+- **Invisible watermark removal** — SynthID, StableSignature, TreeRing via diffusion-based regeneration
+- **AI metadata stripping** — EXIF, PNG text chunks, C2PA provenance manifests, XMP DigitalSourceType
+- **"Made with AI" label removal** — removes the metadata that triggers AI labels on Instagram, Facebook, X (Twitter)
+- **Analog Humanizer** — film grain and chromatic aberration to bypass AI image classifiers
+- **Smart Face Protection** — automatic extraction and blending of human faces to prevent AI distortion
+- **Batch processing** — process entire directories
+- **Detection** — three-stage NCC watermark detection with confidence scoring
+
+> **Try it online** — don't want to install anything? Use [raiw.cc](https://raiw.cc), a free web service powered by this library.
+
+## Examples
+
+| Before (Watermarked) | After (Cleaned) |
+| --- | --- |
+| ![Before](demo_banana_before.png) | ![After](demo_banana_after.png) |
+
+## Supported models
+
+| AI model | Visible watermark | Invisible watermark | Metadata | Our approach |
+| --- | --- | --- | --- | --- |
+| **Google Gemini / Nano Banana** | ✅ Sparkle logo | ✅ SynthID | ✅ C2PA + EXIF | Alpha reversal + diffusion + metadata strip |
+| **OpenAI DALL-E 3 / ChatGPT** | — | — | ✅ C2PA manifest | Metadata strip |
+| **Stable Diffusion (AUTOMATIC1111, ComfyUI)** | — | ✅ DWT / steganographic | ✅ PNG text chunks | Diffusion regeneration + metadata strip |
+| **Adobe Firefly** | — | — | ✅ Content Credentials (C2PA) | Metadata strip |
+| **Midjourney** | — | — | ✅ EXIF + XMP (prompt, model, seed) | Metadata strip |
+| **StableSignature** (Meta) | — | ✅ In-model watermark | — | Diffusion regeneration |
+| **TreeRing** | — | ✅ Latent space watermark | — | Diffusion regeneration |
+
+> Visible watermarks (logo overlays) are currently used only by Google Gemini / Nano Banana. Other services rely on invisible watermarks and/or metadata. Our diffusion-based regeneration works against any invisible watermark in pixel or frequency domain.
+
+## How it works
+
+### Removing the Gemini / Nano Banana sparkle watermark
+
+Google Gemini (internally codenamed **Nano Banana**) adds a visible sparkle logo to generated images using alpha blending:
+
+```text
+watermarked = α × logo + (1 − α) × original
+```
+
+We reverse this with a known alpha map (extracted from Gemini / Nano Banana output on a pure-black background):
+
+```text
+original = (watermarked − α × logo) / (1 − α)
+```
+
+A three-stage NCC (Normalized Cross-Correlation) detector finds the watermark position and scale dynamically, so it works even if the image was resized or cropped. After removal, residual sparkle-edge artifacts are cleaned via gradient-masked inpainting.
+
+**Speed**: ~0.05s per image. No GPU needed.
+
+### Removing SynthID and other invisible watermarks
+
+Google embeds **SynthID** into every image generated by Gemini / Nano Banana. Other AI services use StableSignature, TreeRing, and similar schemes. These imperceptible frequency-domain patterns survive cropping, resizing, and JPEG compression.
+
+The removal pipeline:
+
+```text
+image → downscale to 768px → encode to latent space (VAE)
+      → add controlled noise (forward diffusion)
+      → denoise (reverse diffusion, ~2 steps at strength 0.02)
+      → decode back to pixels (VAE) → upscale to original resolution
+```
+
+The key insight: even minimal noise injection (strength 0.02 = 2% perturbation) breaks the watermark signal while preserving visual quality. The diffusion model acts as a learned image prior — it reconstructs the image faithfully while destroying the watermark pattern.
+
+**Face Protection**: before diffusion, YOLO detects people in the image and extracts them. After diffusion, the original faces are blended back with a soft elliptical mask to prevent AI distortion of facial features.
+
+**Analog Humanizer**: optional film grain and chromatic aberration injection that makes the output indistinguishable from a photo of a screen, defeating AI-generated image classifiers.
+
+### Stripping C2PA, EXIF, and "Made with AI" metadata
+
+AI tools embed generation metadata that social platforms use to show "Made with AI" labels:
+
+- **EXIF tags** — prompt, seed, model hash, sampler settings (Stable Diffusion, Midjourney)
+- **XMP DigitalSourceType** — `trainedAlgorithmicMedia` tag used by Instagram, Facebook, and X (Twitter) to show "Made with AI"
+- **PNG text chunks** — ComfyUI workflows, AUTOMATIC1111 parameters
+- **C2PA Content Credentials** — cryptographic provenance manifests from Google Imagen, OpenAI DALL-E, Adobe Firefly
+
+The cleaner parses each layer, removes AI-related fields, and preserves standard metadata (Author, Copyright, Title).
+
+## Installation
+
+### Recommended
+
+Install as an isolated CLI tool — no need to manage virtual environments:
+
+```bash
+# Using pipx (https://pipx.pypa.io)
+pipx install git+https://github.com/wiltodelta/remove-ai-watermarks.git
+
+# Or using uv (https://docs.astral.sh/uv)
+uv tool install git+https://github.com/wiltodelta/remove-ai-watermarks.git
+```
+
+To update to the latest version:
+
+```bash
+pipx upgrade remove-ai-watermarks
+
+# or
+uv tool upgrade remove-ai-watermarks
+```
+
+### Install from repository
+
+**Prerequisites:** Python 3.10+ and `pip` (or [`uv`](https://docs.astral.sh/uv/)).
+
+```bash
+# 1. Clone the repository
+git clone https://github.com/wiltodelta/remove-ai-watermarks.git
+cd remove-ai-watermarks
+
+# 2. Install the package in editable mode
+pip install -e .
+
+# Or, if you use uv:
+uv pip install -e .
+```
+
+After installation the `remove-ai-watermarks` command is available system-wide.
+
+> **Note**: The base install covers visible watermark removal and metadata stripping.
+> For invisible watermark removal (SynthID etc.), install GPU dependencies:
+>
+> ```bash
+> pip install -e ".[gpu]"   # or: uv pip install -e ".[gpu]"
+> ```
+
+#### Invisible watermark removal
+
+Invisible removal uses diffusion models and a GPU for reasonable speed.
+
+```bash
+# On first run, the model (~2 GB) will be downloaded automatically.
+# Device is auto-detected: CUDA (Linux/Windows) > MPS (macOS) > CPU.
+# To force a device: --device cuda / --device mps / --device cpu
+
+# Optional: set a HuggingFace token for gated/private models
+cp .env.example .env
+# Edit .env and set HF_TOKEN=hf_your_token_here
+```
+
+#### Developer setup
+
+```bash
+# Install with dev dependencies (pytest, ruff, pyright)
+pip install -e ".[dev]"
+# Or with uv:
+uv pip install -e ".[dev]"
+
+# Run tests
+pytest
+
+# Run linters
+./maintain.sh
+```
+
+## Usage
+
+### CLI
+
+```bash
+# Remove all watermarks from a single image (visible + invisible + metadata)
+remove-ai-watermarks all image.png -o clean.png
+
+# Process an entire directory
+remove-ai-watermarks batch ./images/ --mode all
+```
+
+#### Individual commands
+
+```bash
+# Visible watermark only (Gemini / Nano Banana sparkle) — fast, offline
+remove-ai-watermarks visible image.png -o clean.png
+
+# Invisible watermark only (SynthID etc.) — requires GPU
+remove-ai-watermarks invisible image.png -o clean.png --humanize 4.0
+
+# Check / strip AI metadata (C2PA, EXIF, "Made with AI" labels)
+remove-ai-watermarks metadata image.png --check
+remove-ai-watermarks metadata image.png --remove
+
+# Batch with a specific mode
+remove-ai-watermarks batch ./images/ --mode visible
+```
+
+### Python API
+
+```python
+from remove_ai_watermarks.gemini_engine import GeminiEngine
+import cv2
+
+engine = GeminiEngine()
+image = cv2.imread("watermarked.png")
+
+# Detect
+result = engine.detect_watermark(image)
+print(f"Detected: {result.detected} (confidence: {result.confidence:.1%})")
+
+# Remove
+clean = engine.remove_watermark(image)
+cv2.imwrite("clean.png", clean)
+```
+
+### Metadata stripping
+
+```python
+from remove_ai_watermarks.metadata import has_ai_metadata, remove_ai_metadata
+from pathlib import Path
+
+if has_ai_metadata(Path("image.png")):
+    remove_ai_metadata(Path("image.png"), Path("clean.png"))
+```
+
+## Requirements
+
+- Python ≥ 3.10
+- **Visible removal / metadata**: CPU only, no GPU required
+- **Invisible removal**: GPU recommended (CUDA or MPS), works on CPU (slow)
+
+## Troubleshooting
+
+**SSL certificate error** (`CERTIFICATE_VERIFY_FAILED`):
+
+```bash
+# Install certifi (the tool auto-detects it)
+pip install certifi
+
+# macOS only: run the Python certificate installer
+/Applications/Python\ 3.*/Install\ Certificates.command
+```
+
+**First run is slow** — this is expected. The tool downloads model weights (~2 GB) on first launch. Subsequent runs use cached models.
+
+## Credits
+
+- [noai-watermark](https://github.com/mertizci/noai-watermark) by mertizci — invisible watermark removal engine
+- [GeminiWatermarkTool](https://github.com/allenk/GeminiWatermarkTool) by Allen Kuo (MIT) — visible watermark removal algorithm
+- [CtrlRegen](https://github.com/yepengliu/CtrlRegen) by Liu et al. (ICLR 2025) — controllable regeneration pipeline
+- NeuralBleach (MIT) — analog humanizer technique
+
+## ⚠️ Disclaimer
+
+This tool is provided for **educational and research purposes only**.
+
+Removing AI watermarks to misrepresent AI-generated content as human-created
+may violate applicable laws, including the U.S. Digital Millennium Copyright Act
+(DMCA) and the COPIED Act. Users are solely responsible for ensuring their use
+complies with all applicable laws and platform terms of service.
+
+The authors do not condone the use of this tool for deception, fraud,
+or any activity that violates applicable laws or regulations.
+
+## License
+
+MIT