import gradio as gr
import syn_hifigan as syn
#import syn_k_univnet_multi as syn
import os, tempfile
        
languages = {"South Sámi":0,
          "North Sámi":1,
          "Lule Sámi":2}

speakers={#"aj0": 0,
          "Aanna - sma": 1,
          "Máhtte": 2,
          "Siggá - smj": 3,
          "Biret - sme": 5,
          #"lo": 6,
          "Sunná": 7,
          "Abmut - smj": 8,
          "Nihkol - smj": 9
}
public=True

tempdir = tempfile.gettempdir()

tts = syn.Synthesizer()



def speak(text, language,speaker,l_weight, s_weight, pace, postfilter): #pitch_shift,pitch_std):



    # text frontend not implemented...
    text = text.replace("...", "…")
    print(speakers[speaker])
    audio = tts.speak(text, output_file=f'{tempdir}/tmp', lang=languages[language],
                      spkr=speakers[speaker], l_weight=l_weight, s_weight=s_weight,
                      pace=pace, clarity=postfilter)

    if not public:
        try:
            os.system("play "+tempdir+"/tmp.wav &")
        except:
            pass

    return (22050, audio)



controls = []
controls.append(gr.Textbox(label="text", value="Suohtas duinna deaivvadit."))
controls.append(gr.Dropdown(list(languages.keys()), label="language", value="North Sámi"))
controls.append(gr.Dropdown(list(speakers.keys()), label="speaker", value="Sunná"))

controls.append(gr.Slider(minimum=0.5, maximum=1.5, step=0.05, value=1, label="language weight"))
controls.append(gr.Slider(minimum=0.5, maximum=1.5, step=0.05, value=1, label="speaker weight"))

controls.append(gr.Slider(minimum=0.5, maximum=1.5, step=0.05, value=1.0, label="speech rate"))
controls.append(gr.Slider(minimum=0., maximum=2, step=0.05, value=1.0, label="post-processing"))




tts_gui = gr.Interface(
    fn=speak,
    inputs=controls,
    outputs= gr.Audio(label="output"),
    live=False

)


if __name__ == "__main__":
    tts_gui.launch(share=public)