name: teaching-feedback

x-api-base: &api-base
  image: teaching-feedback-api:0.1.0

services:
  funasr:
    image: teaching-feedback-funasr:1.0.0
    build:
      context: ./asr-service
      dockerfile: Dockerfile
      args:
        DEBIAN_MIRROR: http://mirrors.ustc.edu.cn
        PYPI_INDEX_URL: https://mirrors.ustc.edu.cn/pypi/simple
    environment:
      ASR_MODEL: paraformer-zh
      ASR_VAD_MODEL: fsmn-vad
      ASR_PUNC_MODEL: ct-punc
      ASR_DEVICE: cpu
      ASR_MAX_CONCURRENCY: "1"
      ASR_BATCH_SIZE_SECONDS: "300"
      MODELSCOPE_CACHE: /models
    volumes:
      - funasr-model-cache:/models
    restart: unless-stopped
    init: true
    healthcheck:
      test:
        - CMD
        - python
        - -c
        - "import urllib.request; urllib.request.urlopen('http://127.0.0.1:10095/health', timeout=3)"
      interval: 15s
      timeout: 5s
      retries: 80
      start_period: 10m
    shm_size: 1gb

  migrate:
    <<: *api-base
    env_file:
      - ./server/.env
    command:
      - migrate
    restart: "no"
    healthcheck:
      disable: true

  api:
    <<: *api-base
    build:
      context: ./server
      dockerfile: Dockerfile
      args:
        DEBIAN_MIRROR: http://mirrors.ustc.edu.cn
    env_file:
      - ./server/.env
    environment:
      HOST: 0.0.0.0
      PORT: "8080"
      AUDIO_STORAGE_DIR: /data/audio
      ASR_PROVIDER: local
      LOCAL_ASR_URL: http://funasr:10095
      ASR_REQUEST_TIMEOUT_SECONDS: "1800"
      ASR_WORKER_CONCURRENCY: "1"
      ASR_JOB_LEASE_SECONDS: "3600"
      DATABASE_MAX_CONNECTIONS: "5"
      ALLOW_DEVELOPMENT_USER_HEADER: "false"
    ports:
      - "39180:8080"
    volumes:
      - audio-data:/data/audio
    restart: unless-stopped
    init: true
    depends_on:
      migrate:
        condition: service_completed_successfully
      funasr:
        condition: service_healthy

volumes:
  funasr-model-cache:
    name: teaching-feedback-funasr-model-cache
  audio-data:
    name: teaching-feedback-audio