Update app.py

app.py

@@ -8,12 +8,26 @@ import soundfile as sf
 from model import get_pretrained_model, language_to_models
 
 # Function to update model dropdown based on language selection
-def update_model_dropdown(language):
+#def update_model_dropdown(language):
+#    if language in language_to_models:
+#        choices = language_to_models[language]
+#        return gr.Dropdown.update(choices=choices, value=choices[0])
+#    else:
+#        raise ValueError(f"Unsupported language: {language}")
+
+
+def update_model_dropdown(language: str):
     if language in language_to_models:
         choices = language_to_models[language]
-        return gr.Dropdown.update(choices=choices, value=choices[0])
-    else:
-        raise ValueError(f"Unsupported language: {language}")
+        return gr.Dropdown(
+            choices=choices,
+            value=choices[0],
+            interactive=True,
+        )
+
+    raise ValueError(f"Unsupported language: {language}")
+
+    
 
 # Function to process text to speech conversion
 def process(language, repo_id, text, sid, speed):
@@ -37,7 +51,7 @@ def process(language, repo_id, text, sid, speed):
 
     filename = f"{uuid.uuid4()}.wav"
     sf.write(filename, audio.samples, samplerate=audio.sample_rate, subtype="PCM_16")
-    return filename, info
+    return filename
 
 # Interface layout
 demo = gr.Blocks()
@@ -57,10 +71,10 @@ with demo:
     input_speed = gr.Slider(minimum=0.1, maximum=10, value=1, step=0.1, label="Speed (larger->faster; smaller->slower)")
 
     output_audio = gr.Audio(label="Generated audio")
-    output_info = gr.HTML(label="Info")
+    #output_info = gr.HTML(label="Info")
 
     input_button = gr.Button("Submit")
-    input_button.click(process, inputs=[language_radio, model_dropdown, input_text, input_sid, input_speed], outputs=[output_audio, output_info])
+    input_button.click(process, inputs=[language_radio, model_dropdown, input_text, input_sid, input_speed], outputs=[output_audio])