Update app.py

app.py

@@ -14,10 +14,14 @@ fr_text = "La bise et le soleil se disputaient, chacun assurant qu'il était le
 en_text = "The North Wind and the Sun were disputing which was the stronger, when a traveler came along wrapped in a warm cloak."
 pt_text = "O vento norte e o Sol discutiam quem era o mais forte, quando surgiu um viajante envolvido numa capa."
 
-TTS_VOICES = [
+my_voices = [
+    "Male",
+    "Female"
 ]
 
-TTS_LANGUAGES = [
+my_languages = [
+    "Lëtzebuergesch",
+    "Deutsch"
 ]
 
 my_examples = [
@@ -28,24 +32,35 @@ my_article = "<h3>User guide</h3>"
 
 my_inputs = [
   gr.Textbox(lines=5, label="Input Text"),
-  # gr.Radio(label="Speaker", choices = TTS_VOICES, value = "Max"),
-  # gr.Radio(label="Language", choices = TTS_LANGUAGES, value = "Lëtzebuergesch"),
+  gr.Radio(label="Speaker", choices = my_voices, value = "Male"),
+  gr.Radio(label="Language", choices = my_languages, value = "Lëtzebuergesch"),
 ]
 
 my_outputs = gr.Audio(type="filepath", label="Output Audio")
     
 def tts(text: str, speaker_idx: str, language_idx: str):
+    if speaker_idx == "Male":
+        speaker_idx = "Max"
+    else:
+        speaker_idx = "Mod"
+    if language_idx == "Lëtzebuergesch":
+        language_idx = "D:\\MULTI-LOD\\mailabs\\x-lb"
+    else:
+        language_idx = "D:\\MULTI-LOD\\mailabs\\x-de"
+
     best_model_path = hf_hub_download(repo_id=REPO_ID, filename="checkpoint_180000.pth") 
     config_path = hf_hub_download(repo_id=REPO_ID, filename="config.json")
-    speakers_path = hf_hub_download(repo_id=REPO_ID, filename="speakers.pth")
-    languages_path = hf_hub_download(repo_id=REPO_ID, filename="language_ids.json")
+    # speakers_path = hf_hub_download(repo_id=REPO_ID, filename="speakers.pth")
+    # languages_path = hf_hub_download(repo_id=REPO_ID, filename="language_ids.json")
     
     # init synthesizer  
     synthesizer = Synthesizer(
         best_model_path,
         config_path,
-        speakers_path,
-        languages_path,
+        # speakers_path,
+        # languages_path,
+        None,
+        None,
         None,
         None,
         False
@@ -53,7 +68,7 @@ def tts(text: str, speaker_idx: str, language_idx: str):
 
     # create audio file
     # wavs = synthesizer.tts(text, speaker_idx, language_idx
-    wavs = synthesizer.tts(lb_text, "Max", "D:\\MULTI-LOD\\mailabs\\x-lb")
+    wavs = synthesizer.tts(text, speaker_idx, language_idx)
     with tempfile.NamedTemporaryFile(suffix = ".wav", delete = False) as fp:
         synthesizer.save_wav(wavs, fp)                      
     return fp.name 
@@ -68,4 +83,4 @@ iface = gr.Interface(
     examples = my_examples, 
     allow_flagging="never"
 )
-iface.launch()
+iface.launch()