# Whole-infra bring-up for watermark-remover.
#
#   docker compose up --build -d            # core HTTP service only
#   docker compose --profile harness up --build -d   # + markllm / markdiffusion harnesses
#   docker compose --profile heavy up --build -d     # + ctrlregen / synthid (local builds)
#
# The skill and any web app talk to the wr-core service at http://127.0.0.1:8765
# (loopback-only host mapping). The heavy/harness images are one-shot CLIs:
# `up` starts them with `--help` to confirm the image, and real jobs run via
# `docker compose run`, e.g.:
#   docker compose run --rm wr-ctrlregen /data/shot.png -o /data/out.png
#
# ctrlregen and synthid bake in upstream code that is not publicly
# redistributable (all-rights-reserved / non-commercial Research License), so
# they build from source locally and are never pushed to GHCR.

name: watermark-remover

services:
  wr-core:
    build:
      context: .
      dockerfile: Dockerfile
    image: ghcr.io/pymodel/watermark-remover:latest
    ports:
      - "127.0.0.1:8765:8765"
    environment:
      # Empty by default (no auth). Set to require `Authorization: Bearer <key>`.
      WATERMARKS_SERVER_API_KEY: ${WATERMARKS_SERVER_API_KEY:-}
      # Optional Gemini API key. The vendor SynthID-text detector is
      # currently disabled and this key enables nothing (see .env.example).
      WATERMARKS_GEMINI_API_KEY: ${WATERMARKS_GEMINI_API_KEY:-}
      WATERMARKS_GEMINI_MODEL: ${WATERMARKS_GEMINI_MODEL:-}
      # Optional SynthID image scorer sidecar (heavy profile).
      WATERMARKS_SYNTHID_SCORER_URL: ${WATERMARKS_SYNTHID_SCORER_URL:-}
      WATERMARKS_SYNTHID_SCORER_API_KEY: ${WATERMARKS_SYNTHID_SCORER_API_KEY:-}
    read_only: true
    tmpfs:
      - /tmp
    init: true
    user: "10001:10001"
    restart: unless-stopped

  wr-markllm:
    profiles: [harness]
    build:
      context: .
      dockerfile: Dockerfile.markllm
    image: ghcr.io/pymodel/watermark-remover:markllm-latest
    # One-shot CLI: `up` just confirms the image; run real jobs with
    # `docker compose run --rm wr-markllm detect /data/wm.txt --scheme kgw`.
    command: ["--help"]
    read_only: true
    tmpfs:
      - /tmp
    init: true
    user: "10001:10001"
    environment:
      HF_TOKEN: ${HF_TOKEN:-}
      HF_HOME: /home/markllm/.cache/huggingface
    volumes:
      - markllm-cache:/home/markllm/.cache/huggingface

  wr-markdiffusion:
    profiles: [harness]
    build:
      context: .
      dockerfile: Dockerfile.markdiffusion
    image: ghcr.io/pymodel/watermark-remover:markdiffusion-latest
    command: ["--help"]
    read_only: true
    tmpfs:
      - /tmp
    init: true
    user: "10001:10001"
    environment:
      HF_TOKEN: ${HF_TOKEN:-}
      HF_HOME: /home/markdiffusion/.cache/huggingface
    volumes:
      - markdiffusion-cache:/home/markdiffusion/.cache/huggingface

  wr-ctrlregen:
    profiles: [heavy]
    build:
      context: .
      dockerfile: Dockerfile.ctrlregen
    # Local-only tag: upstream `noai-watermark` ships no LICENSE (all-rights-
    # reserved), so this image is never published.
    image: watermark-remover-ctrlregen:local
    command: ["--help"]
    read_only: true
    tmpfs:
      - /tmp
    init: true
    user: "10001:10001"
    environment:
      HF_TOKEN: ${HF_TOKEN:-}
      NOAI_WATERMARK_DIR: /opt/noai-watermark
    volumes:
      - ctrlregen-cache:/home/remover/.cache/huggingface

  wr-synthid:
    profiles: [heavy]
    build:
      context: .
      dockerfile: Dockerfile.synthid
    # Local-only tag: upstream `reverse-SynthID` is under a non-commercial
    # Research License, so this image is never published.
    image: watermark-remover-synthid-scorer:local
    command: ["--help"]
    read_only: true
    tmpfs:
      - /tmp
    init: true
    user: "10001:10001"
    volumes:
      - synthid-cache:/home/scorer/.cache/huggingface

  # HTTP SynthID scorer sidecar: lets wr-core score images before/after
  # cleaning without bundling the non-commercial reverse-SynthID code in the
  # published core image. Reach it via WATERMARKS_SYNTHID_SCORER_URL on
  # wr-core (see .env.example) and require the same bearer key on both sides.
  wr-synthid-score:
    profiles: [heavy]
    build:
      context: .
      dockerfile: Dockerfile.synthid
    image: watermark-remover-synthid-scorer:local
    entrypoint: ["python3"]
    command: ["/app/synthid_score_server.py", "--host", "0.0.0.0", "--port", "8766"]
    read_only: true
    tmpfs:
      - /tmp
    init: true
    user: "10001:10001"
    environment:
      WATERMARKS_SYNTHID_SCORER_API_KEY: ${WATERMARKS_SYNTHID_SCORER_API_KEY:-}
      REVERSE_SYNTHID_DIR: /opt/reverse-synthid
    volumes:
      - synthid-cache:/home/scorer/.cache/huggingface

volumes:
  markllm-cache:
  markdiffusion-cache:
  ctrlregen-cache:
  synthid-cache: