import os
os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/kannada/kn-IN/female_voice_0/fe_glow.zip && unzip -q fe_glow.zip -d ttsv/checkpoints/female')
os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/kannada/kn-IN/ma_fe_hifi/ma_fe_hifi.zip && unzip -q ma_fe_hifi.zip -d ttsv/checkpoints/female')
os.system('rm fe_glow.zip && rm ma_fe_hifi.zip')
os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/kannada/kn-IN/male_voice_1/ma_glow.zip && unzip -q ma_glow.zip -d ttsv/checkpoints/male')
os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/tts/kannada/kn-IN/ma_fe_hifi/ma_fe_hifi.zip && unzip -q ma_fe_hifi.zip -d ttsv/checkpoints/male')
os.system('wget -q https://storage.googleapis.com/vakyansh-open-models/translit_models.zip -P ttsv/checkpoints/ && unzip -q ttsv/checkpoints/translit_models.zip -d ttsv/checkpoints/')


for path, subdirs, files in os.walk('ttsv/checkpoints/'):
    print(subdirs)
    for name in files:
        print(os.path.join(path, name))

from ttsv.utils.inference.run_gradio import *
from argparse import Namespace

#os.system('python ttsv/utils/inference/run_gradio.py -a ttsv/checkpoints/glow/male -v ttsv/checkpoints/hifi/male -d cpu -L hi')


args = {
    'acoustic':'/home/user/app/ttsv/checkpoints/female/fe_glow,/home/user/app/ttsv/checkpoints/male/ma_glow',
    'vocoder':'/home/user/app/ttsv/checkpoints/female/ma_fe_hifi,/home/user/app/ttsv/checkpoints/male/ma_fe_hifi',
    'device':'cpu',
    'lang':'kn'
}

build_gradio(Namespace(**args))