Taskfile.yml

version: '3'

vars:
  IMAGE: melotts:test
  PORT: '8888'
  BUILD_ID: ''
  GPU_DEVICE: '0'
  CONTAINER_EN: melotts_en
  CONTAINER_FULL: melotts_full
  CONTAINER_LOCAL: melotts_local
  LANG_EN: EN
  LANG_FULL: EN,EN_V2,EN_NEWEST,ES,FR,ZH,JP,KR
  TEMP_WAV: '%TEMP%\\melotts_task.wav'
  WAIT_URL: http://localhost:{{.PORT}}/tts/ping
  RELEASE_VERSION: ''
  NEXT_VERSION: ''
  RELEASE_REMOTE: origin
  RELEASE_BRANCH: main

tasks:
  gpuinfo:
    desc: Show host GPU indexes (use with GPU_DEVICE variable)
    cmds:
      - cmd /c nvidia-smi --query-gpu=index,name --format=csv,noheader

  image:
    desc: Build local MeloTTS Docker image (FULL profile)
    cmds:
      - cmd /c docker build --build-arg BUILD_ID={{.BUILD_ID}} --build-arg INIT_DOWNLOADS_PROFILE=FULL -t {{.IMAGE}} .

  imagesmall:
    desc: Build local MeloTTS Docker image with EN-only preload profile
    cmds:
      - cmd /c docker build --build-arg BUILD_ID={{.BUILD_ID}} --build-arg INIT_DOWNLOADS_PROFILE=EN_ONLY -t {{.IMAGE}} .

  imagerun:
    desc: Run existing image with full language set (default GPU_DEVICE=0)
    cmds:
      - cmd: cmd /c docker rm -f {{.CONTAINER_FULL}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker rm -f {{.CONTAINER_EN}} >nul 2>&1
        ignore_error: true
      - cmd /c docker run -d --name {{.CONTAINER_FULL}} -p {{.PORT}}:8888 --gpus all -e CUDA_VISIBLE_DEVICES={{.GPU_DEVICE}} -e TTS_LANGUAGES={{.LANG_FULL}} {{.IMAGE}}
      - cmd /c docker ps --filter "name={{.CONTAINER_FULL}}"

  imageweb:
    desc: Open default browser to the local MeloTTS UI
    cmds:
      - cmd /c start "" "http://localhost:{{.PORT}}"

  imagewait:
    desc: Wait until the MeloTTS API is ready
    cmds:
      - 'cmd /c curl.exe -sS --retry 30 --retry-delay 2 --retry-connrefused --retry-all-errors {{.WAIT_URL}} >nul'

  imagestop:
    desc: Stop running MeloTTS test containers
    cmds:
      - cmd: cmd /c docker stop {{.CONTAINER_EN}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker stop {{.CONTAINER_FULL}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker stop {{.CONTAINER_LOCAL}} >nul 2>&1
        ignore_error: true

  imagerunsmall:
    desc: Run existing image with English only (default GPU_DEVICE=0)
    cmds:
      - cmd: cmd /c docker rm -f {{.CONTAINER_EN}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker rm -f {{.CONTAINER_FULL}} >nul 2>&1
        ignore_error: true
      - cmd /c docker run -d --name {{.CONTAINER_EN}} -p {{.PORT}}:8888 --gpus all -e CUDA_VISIBLE_DEVICES={{.GPU_DEVICE}} -e TTS_LANGUAGES={{.LANG_EN}} {{.IMAGE}}
      - cmd /c docker ps --filter "name={{.CONTAINER_EN}}"

  localrun:
    desc: Run image with local melo/app.py bind-mounted for quick UI/API testing (default GPU_DEVICE=0)
    cmds:
      - cmd: cmd /c docker rm -f {{.CONTAINER_EN}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker rm -f {{.CONTAINER_FULL}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker rm -f {{.CONTAINER_LOCAL}} >nul 2>&1
        ignore_error: true
      - cmd /c docker run -d --name {{.CONTAINER_LOCAL}} -p {{.PORT}}:8888 --gpus all -e CUDA_VISIBLE_DEVICES={{.GPU_DEVICE}} -e TTS_LANGUAGES={{.LANG_EN}} -v "{{.TASKFILE_DIR}}\melo\app.py:/app/melo/app.py:ro" -v "{{.TASKFILE_DIR}}\icon.png:/app/icon.png:ro" {{.IMAGE}}
      - cmd /c docker ps --filter "name={{.CONTAINER_LOCAL}}"

  localdev:
    desc: Restart local app.py bind mount, wait for readiness, then open the UI
    cmds:
      - task: localrun
      - task: imagewait
      - task: imageweb

  localapi:
    desc: Restart local app.py bind mount, wait for readiness, then call the TTS API
    cmds:
      - task: localrun
      - task: imageapi

  locallogs:
    desc: Follow logs for the local app.py bind-mounted container
    cmds:
      - cmd /c docker logs -f {{.CONTAINER_LOCAL}}

  localrunfull:
    desc: Run image with local melo/app.py bind-mounted and full language set (default GPU_DEVICE=0)
    cmds:
      - cmd: cmd /c docker rm -f {{.CONTAINER_EN}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker rm -f {{.CONTAINER_FULL}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker rm -f {{.CONTAINER_LOCAL}} >nul 2>&1
        ignore_error: true
      - cmd /c docker run -d --name {{.CONTAINER_LOCAL}} -p {{.PORT}}:8888 --gpus all -e CUDA_VISIBLE_DEVICES={{.GPU_DEVICE}} -e TTS_LANGUAGES={{.LANG_FULL}} -v "{{.TASKFILE_DIR}}\melo\app.py:/app/melo/app.py:ro" -v "{{.TASKFILE_DIR}}\icon.png:/app/icon.png:ro" {{.IMAGE}}
      - cmd /c docker ps --filter "name={{.CONTAINER_LOCAL}}"

  imageapi:
    desc: Call TTS API, play generated temp WAV, then remove it
    cmds:
      - task: imagewait
      - 'cmd /c curl.exe -sS -X POST http://localhost:{{.PORT}}/tts/convert/tts -H "Content-Type: application/json" -d "{\"text\":\"Hello world. I wanted to test this and see if this works properly\",\"speed\":1.0,\"language\":\"EN\",\"speaker_id\":\"EN-BR\",\"sdp_ratio\":0.21,\"noise_scale\":0.61,\"noise_scale_w\":0.81}" --output {{.TEMP_WAV}}'
      - 'cmd /c if not exist {{.TEMP_WAV}} exit /b 1'
      - 'cmd /c python -c "import winsound; winsound.PlaySound(r''{{.TEMP_WAV}}'', winsound.SND_FILENAME)"'
      - 'cmd /c del /f /q {{.TEMP_WAV}}'

  release:
    desc: Release current VERSION snapshot, tag it, then bump to next snapshot
    cmds:
      - powershell -NoProfile -ExecutionPolicy Bypass -File "{{.TASKFILE_DIR}}\scripts\release.ps1" -Version "{{.RELEASE_VERSION}}" -NextVersion "{{.NEXT_VERSION}}"

  releasepush:
    desc: Push release tag first, then branch, so GitHub tag workflows build the release
    cmds:
      - powershell -NoProfile -ExecutionPolicy Bypass -File "{{.TASKFILE_DIR}}\scripts\release-push.ps1" -Version "{{.RELEASE_VERSION}}" -Remote "{{.RELEASE_REMOTE}}" -Branch "{{.RELEASE_BRANCH}}"

  nuke:
    desc: Remove local MeloTTS containers and image
    cmds:
      - cmd: cmd /c docker rm -f {{.CONTAINER_EN}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker rm -f {{.CONTAINER_FULL}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker rm -f {{.CONTAINER_LOCAL}} >nul 2>&1
        ignore_error: true
      - cmd: cmd /c docker rmi -f {{.IMAGE}} >nul 2>&1
        ignore_error: true