services:
  kokoro-fastapi:
    image: ghcr.io/remsky/kokoro-fastapi-gpu:v0.2.4
    container_name: kokoro-fastapi
    restart: unless-stopped
    volumes:
      - ${MODELS_DIR}/models:/app/api/src/models
      - ${MODELS_DIR}/voices:/app/api/src/voices
    ports:
      - "8880:8880"
    environment:
      - USE_GPU=true
      - PYTHONUNBUFFERED=1
      - REPO_ID=hexgrad/Kokoro-82M
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8880/health"]
      interval: 20s
      timeout: 5s
      retries: 5
      start_period: 90s
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ['0']
              capabilities: [gpu]
             
  kokoro-wyoming:
    build: kokoro-wyoming
    container_name: kokoro-wyoming
    depends_on:
      kokoro-fastapi:
        condition: service_healthy
    links:
      - kokoro-fastapi
    ports:
      - "10200:10200"
    restart: unless-stopped
    tty: true
    stdin_open: true
    environment:
      - API_HOST=http://kokoro-fastapi  # Set TTS service URL
      - API_PORT=8880  # Set TTS service PORT
      - VOICE_SPEED=${VOICE_SPEED}
    labels:
      com.centurylinklabs.watchtower.enable: false