Update app.py

app.py

@@ -1,19 +1,90 @@
 import gradio as gr
 from TTS.api import TTS
 
-model_name = TTS.list_models()[0]
 # Init TTS
-tts = TTS(model_name)
+tts = TTS(model_name="tts_models/multilingual/multi-dataset/your_tts", progress_bar=False, gpu=False)
 
-def text_to_speech(text: str, speaker_wav, language: str):
-    # Use the 'speaker_wav' parameter to get the path of the uploaded audio file
-    tts.tts_to_file(text, speaker_wav=speaker_wav, language=language, file_path="output.wav")
-    return 'output.wav'
 
-inputs = [gr.Textbox(label="Input", value="Hello!", max_lines=3),
-          gr.File(label="Speaker Wav", accept=".wav,.mp3"), 
-          gr.Radio(label="Language", choices=tts.languages, value="en")]
-outputs = gr.Audio(label="Output", type="filepath")
+def text_to_speech(text: str, speaker_wav, speaker_wav_file, language: str):
+    if speaker_wav_file and not speaker_wav:
+        speaker_wav = speaker_wav_file
+    file_path = "output.wav"
+    if language == "zh-CN":
+        # if speaker_wav is not None:
+        #     zh_tts.tts_to_file(text, speaker_wav=speaker_wav, file_path=file_path)
+        # else:
+        zh_tts.tts_to_file(text, file_path=file_path)
+    elif language == "de":
+        # if speaker_wav is not None:
+        #     de_tts.tts_to_file(text, speaker_wav=speaker_wav, file_path=file_path)
+        # else:
+        de_tts.tts_to_file(text, file_path=file_path)
+    elif language == "es":
+        # if speaker_wav is not None:
+        #     es_tts.tts_to_file(text, speaker_wav=speaker_wav, file_path=file_path)
+        # else:
+        es_tts.tts_to_file(text, file_path=file_path)
+    else:
+        if speaker_wav is not None:
+            tts.tts_to_file(text, speaker_wav=speaker_wav, language=language, file_path=file_path)
+        else:
+            tts.tts_to_file(text, speaker=tts.speakers[0], language=language, file_path=file_path)
+    return file_path
 
-demo = gr.Interface(fn=text_to_speech, inputs=inputs, outputs=outputs)
-demo.launch(debug=True)
+
+
+# inputs = [gr.Textbox(label="Input the text", value="", max_lines=3),
+#           gr.Audio(label="Voice to clone", source="microphone", type="filepath"),
+#           gr.Audio(label="Voice to clone", type="filepath"), 
+#           gr.Radio(label="Language", choices=["en", "zh-CN", "fr-fr", "pt-br", "de", "es"], value="en"),     
+#           gr.Text(intro_text, font_size=14)]
+# outputs = gr.Audio(label="Output")
+
+# demo = gr.Interface(fn=text_to_speech, inputs=inputs, outputs=outputs)
+
+# demo.launch()
+
+
+title = "Voice-Cloning-Demo"
+
+def toggle(choice):
+    if choice == "mic":
+        return gr.update(visible=True, value=None), gr.update(visible=False, value=None)
+    else:
+        return gr.update(visible=False, value=None), gr.update(visible=True, value=None)
+
+def handle_language_change(choice):
+    if choice == "zh-CN" or choice == "de" or choice == "es":
+        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+    else:
+        return gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+
+warming_text = """Please note that Chinese, German, and Spanish are currently not supported for voice cloning."""
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            text_input = gr.Textbox(label="Input the text", value="", max_lines=3)
+            lan_input = gr.Radio(label="Language", choices=["en", "fr-fr", "pt-br"], value="en")
+            gr.Markdown(warming_text)
+            radio = gr.Radio(["mic", "file"], value="mic",
+                             label="How would you like to upload your audio?")
+            audio_input_mic = gr.Audio(label="Voice to clone", source="microphone", type="filepath", visible=True)
+            audio_input_file = gr.Audio(label="Voice to clone", type="filepath", visible=False)
+
+            with gr.Row():
+                with gr.Column():
+                    btn_clear = gr.Button("Clear")
+                with gr.Column():
+                    btn = gr.Button("Submit", variant="primary")
+        with gr.Column():
+            audio_output = gr.Audio(label="Output")
+
+    # gr.Examples(examples, fn=inference, inputs=[audio_file, text_input],
+    #                   outputs=audio_output, cache_examples=True)
+    btn.click(text_to_speech, inputs=[text_input, audio_input_mic,
+              audio_input_file, lan_input], outputs=audio_output)
+    radio.change(toggle, radio, [audio_input_mic, audio_input_file])
+    lan_input.change(handle_language_change, lan_input, [radio, audio_input_mic, audio_input_file])
+
+demo.launch(enable_queue=True)