services:
  stem-separator:
    build: .
    image: mike-ai/bs-roformer-separator:0.47.0
    container_name: mike-ai-stem-separator
    labels:
      com.mike-ai.stem-separator: "bs-roformer"
    environment:
      NVIDIA_VISIBLE_DEVICES: ${SEPARATOR_GPU_UUID:?set SEPARATOR_GPU_UUID to the RTX 5080 UUID}
      MODEL_FILENAME: model_bs_roformer_ep_317_sdr_12.9755.ckpt
      MODEL_DIR: /models
      JOB_DIR: /data/jobs
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ["${SEPARATOR_GPU_UUID:?set SEPARATOR_GPU_UUID to the RTX 5080 UUID}"]
              capabilities: [gpu]
    ports:
      - "127.0.0.1:${SEPARATOR_PORT:-8007}:8080"
    volumes:
      - ${SEPARATOR_MODEL_DIR:-/data/models/audio-separator}:/models
      - ${SEPARATOR_DATA_DIR:-/data/audio/separation}:/data
    shm_size: "2gb"
    restart: "no"
    healthcheck:
      test: ["CMD-SHELL", "curl -fsS http://127.0.0.1:8080/health | grep -q '\"status\":\"ok\"'"]
      interval: 15s
      timeout: 5s
      start_period: 600s
      retries: 3
    networks: [frontend]

networks:
  frontend:
    external: true
    name: mike-ai_frontend