#!/usr/bin/env bash

lang=$LANG
echo "lang: $LANG"

name=$NAME
if [[ $name == 'tugao' && $lang == "pt_PT" ]]; then
  name=tugão
  export NAME=tugão
fi

echo "name: $name"

type=$TYPE
echo "type: $type"

pip install iso639-lang

set -ex

# for en_US-lessac-medium.onnx
# export LANG=en_US
# export TYPE=lessac
# export NAME=medium

# if lang is en_US, then code is en
code=${lang:0:2}

if [[ $name == gyro && $lang == fa_IR && $type == medium ]]; then
  wget -qq https://huggingface.co/gyroing/Persian-Piper-Model-gyro/resolve/main/fa_IR-gyro-meduim.onnx
  wget -qq https://huggingface.co/gyroing/Persian-Piper-Model-gyro/resolve/main/fa_IR-gyro-meduim.onnx.json
  mv fa_IR-gyro-meduim.onnx fa_IR-gyro-medium.onnx
  mv fa_IR-gyro-meduim.onnx.json fa_IR-gyro-medium.onnx.json
cat >MODEL_CARD <<EOF
# Introduction

This model is converted from
https://huggingface.co/gyroing/Persian-Piper-Model-gyro/tree/main
EOF

else
  wget -qq https://huggingface.co/rhasspy/piper-voices/resolve/main/$code/$lang/$name/$type/$lang-$name-$type.onnx
  wget -qq https://huggingface.co/rhasspy/piper-voices/resolve/main/$code/$lang/$name/$type/$lang-$name-$type.onnx.json
  wget -qq https://huggingface.co/rhasspy/piper-voices/resolve/main/$code/$lang/$name/$type/MODEL_CARD
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 piper-phonemize onnx onnxruntime==1.16.0

python3 ./vits-piper.py