# =============================================================================
# MinerU ROCm Docker Compose 配置
# 原生 Linux + AMD GPU 环境
# =============================================================================

services:
  # --- CLI 模式（默认）---
  mineru:
    image: mineru-rocm:7.2.1
    build:
      context: .
      dockerfile: Dockerfile
      network: host
      args:
        ARCH: gfx1201
        GIT_PROXY: http://127.0.0.1:8118
    profiles: ["cli"]
    container_name: mineru-rocm
    stdin_open: true
    tty: true
    ipc: host
    devices:
      - /dev/kfd
      - /dev/dri
    security_opt:
      - seccomp=unconfined
    group_add:
      - video
    environment:
      - MINERU_MODEL_SOURCE=${MINERU_MODEL_SOURCE:-huggingface}
      - HF_HUB_CACHE=${HF_HUB_CACHE:-/opt/models/huggingface}
      - MODELSCOPE_CACHE=${MODELSCOPE_CACHE:-/opt/models/modelscope}
    volumes:
      - ${INPUT_DIR:-./data/input}:/data/input:ro
      - ${OUTPUT_DIR:-./data/output}:/data/output
      - ${MODEL_DIR:-./data/models}:/opt/models
      - ${MIOPEN_CACHE:-./data/miopen}:/root/.cache/miopen
    command: bash
    restart: "no"

  # --- WebUI（Gradio）---
  gradio:
    image: mineru-rocm:7.2.1
    profiles: ["gradio"]
    container_name: mineru-gradio
    stdin_open: true
    tty: true
    ipc: host
    ports:
      - "10002:7860"
    devices:
      - /dev/kfd
      - /dev/dri
    security_opt:
      - seccomp=unconfined
    group_add:
      - video
    environment:
      - MINERU_MODEL_SOURCE=${MINERU_MODEL_SOURCE:-huggingface}
      - HF_HUB_CACHE=${HF_HUB_CACHE:-/opt/models/huggingface}
      - MODELSCOPE_CACHE=${MODELSCOPE_CACHE:-/opt/models/modelscope}
    volumes:
      - ${INPUT_DIR:-./data/input}:/data/input:ro
      - ${OUTPUT_DIR:-./data/output}:/data/output
      - ${MODEL_DIR:-./data/models}:/opt/models
      - ${MIOPEN_CACHE:-./data/miopen}:/root/.cache/miopen
    command: ["mineru-gradio --server-name 0.0.0.0 --server-port 7860"]

  # --- API 服务 ---
  api:
    image: mineru-rocm:7.2.1
    profiles: ["api"]
    container_name: mineru-api
    stdin_open: true
    tty: true
    ipc: host
    ports:
      - "8000:8000"
    devices:
      - /dev/kfd
      - /dev/dri
    security_opt:
      - seccomp=unconfined
    group_add:
      - video
    environment:
      - MINERU_MODEL_SOURCE=${MINERU_MODEL_SOURCE:-huggingface}
      - HF_HUB_CACHE=${HF_HUB_CACHE:-/opt/models/huggingface}
      - MODELSCOPE_CACHE=${MODELSCOPE_CACHE:-/opt/models/modelscope}
    volumes:
      - ${INPUT_DIR:-./data/input}:/data/input:ro
      - ${OUTPUT_DIR:-./data/output}:/data/output
      - ${MODEL_DIR:-./data/models}:/opt/models
      - ${MIOPEN_CACHE:-./data/miopen}:/root/.cache/miopen
    command: ["mineru-api --host 0.0.0.0 --port 8000"]