#!/usr/bin/env bash

set -ex

name=$NAME
if [[ $name == "ljspeech-vits" ]]; then
  wget -q https://coqui.gateway.scarf.sh/v0.6.1_models/tts_models--en--ljspeech--vits.zip
  unzip tts_models--en--ljspeech--vits.zip
  mv -v tts_models--en--ljspeech--vits/model_file.pth ./
  mv -v tts_models--en--ljspeech--vits/config.json ./
elif [[ $name == "ljspeech-vits-neon" ]]; then
  wget -q https://coqui.gateway.scarf.sh/v0.8.0_models/tts_models--en--ljspeech--vits.zip
  unzip tts_models--en--ljspeech--vits.zip
  mv -v tts_models--en--ljspeech--vits/model_file.pth.tar ./model_file.pth
  mv -v tts_models--en--ljspeech--vits/config.json ./
elif [[ $name == "vctk-vits" ]]; then
  wget -q https://coqui.gateway.scarf.sh/v0.6.1_models/tts_models--en--vctk--vits.zip
  unzip tts_models--en--vctk--vits.zip
  mv -v tts_models--en--vctk--vits/config.json ./
  cp -v tts_models--en--vctk--vits/speaker_ids.json ./
  mv -v tts_models--en--vctk--vits/model_file.pth ./
  sudo mkdir -p /home/julian/.local/share/tts/tts_models--en--vctk--vits
  sudo cp -v speaker_ids.json /home/julian/.local/share/tts/tts_models--en--vctk--vits
elif [[ $name == "jenny-jenny" ]]; then
  wget -q https://coqui.gateway.scarf.sh/v0.14.0_models/tts_models--en--jenny--jenny.zip
  unzip tts_models--en--jenny--jenny.zip
  cp -v tts_models--en--jenny--jenny/config.json ./
  cp -v tts_models--en--jenny--jenny/model.pth ./model_file.pth
fi

wget -qq https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/espeak-ng-data.tar.bz2
tar xf espeak-ng-data.tar.bz2
rm espeak-ng-data.tar.bz2

pip install -q TTS onnx onnxruntime
sudo apt-get install -q -y espeak-ng espeak

python3 ./vits-coqui-en.py
ls -lh *.onnx