services:
  voice-studio:
    build: .
    image: mike-ai/vevo2-voice-studio:0.1
    container_name: mike-ai-voice-studio
    restart: "no"
    labels:
      com.mike-ai.voice-worker: vevo2
    environment:
      NVIDIA_VISIBLE_DEVICES: ${VOICE_GPU_UUID:?set VOICE_GPU_UUID to the RTX 5080 UUID}
      NVIDIA_DRIVER_CAPABILITIES: compute,utility
      VOICE_DATA_DIR: /data
    ports:
      - "127.0.0.1:8008:8008"
    volumes:
      - /data/voice/vevo2/huggingface/hub/Vevo2:/opt/amphion/ckpts/Vevo2
      - /data/voice/vevo2/huggingface:/models/huggingface
      - /data/voice/vevo2/whisper:/root/.cache/whisper
      - /data/voice/studio:/data
    healthcheck:
      test: ["CMD-SHELL", "curl -fsS http://127.0.0.1:8008/health | grep -q '\"status\":\"ok\"'"]
      interval: 5s
      timeout: 3s
      start_period: 600s
      retries: 3
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ["${VOICE_GPU_UUID:?set VOICE_GPU_UUID to the RTX 5080 UUID}"]
              capabilities: [gpu]
    networks:
      - frontend

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