# audio2text — 双语字幕生成服务
# CPU dev / GPU prod 两套配置，按需切换 service。
#
# CPU 开发：
#   docker compose up -d            # 默认 cpu
# GPU 生产：
#   docker compose -f docker-compose.yml up -d   # 本机有 nvidia runtime 时自动用 gpu
#
# 模型缓存（/models）跨容器复用，首次启动下载、之后秒起。

services:
  audio2text-cpu:
    build:
      context: .
      args:
        VARIANT: cpu
    image: audio2text:cpu
    container_name: audio2text
    ports:
      - "8000:8000"
    volumes:
      - ./data:/data
      - ./models:/models
      - ./config.yaml:/app/config.yaml:ro
    environment:
      CONFIG_PATH: /app/config.yaml
    restart: unless-stopped
    profiles: ["cpu", ""]

  audio2text-gpu:
    build:
      context: .
      args:
        VARIANT: gpu
    image: audio2text:gpu
    container_name: audio2text
    ports:
      - "8000:8000"
    volumes:
      - ./data:/data
      - ./models:/models
      - ./config.yaml:/app/config.yaml:ro
    environment:
      CONFIG_PATH: /app/config.yaml
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    restart: unless-stopped
    profiles: ["gpu"]