Update app.py

app.py

@@ -4,72 +4,54 @@ from TTS.api import TTS
 import os
 import subprocess
 
-def test_xtts():
-    with gr.Progress() as progress:
-        progress(0, desc="Running XttsV2 check...")
-        # Check device availability
-        device = "cuda" if torch.cuda.is_available() else "cpu"
-
-        # Auto agree to XTTS terms
-        os.environ["COQUI_TOS_AGREED"] = "1"
-
-        # Download default voice file
-        os.system('curl -L -o default_voice.wav https://github.com/DrewThomasson/ebook2audiobookXTTS/raw/main/default_voice.wav')
-
-        # Initialize TTS
-        tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
-
-        # Generate audio
-        output_file = "xtts_output.wav"
-        tts.tts_to_file(text="Hello world!", speaker_wav="default_voice.wav", language="en", file_path=output_file)
-
-        progress(1, desc="XttsV2 check completed")
-    return output_file
-
-def test_styletts2():
-    with gr.Progress() as progress:
-        progress(0, desc="Running StyleTTS2 check...")
-        # Import and download the voice file
-        from styletts2 import tts
-        os.system('curl -L -o default_voice.wav https://github.com/DrewThomasson/ebook2audiobookXTTS/raw/main/default_voice.wav')
-
-        # Initialize and run inference
-        my_tts = tts.StyleTTS2()
-        output_file = "styletts2_output.wav"
-        my_tts.inference("Hello there, I am now a python package.", output_wav_file=output_file)
-
-        progress(1, desc="StyleTTS2 check completed")
-    return output_file
-
-def test_piper():
-    with gr.Progress() as progress:
-        progress(0, desc="Running Piper-TTS check...")
-        # Run piper command and generate output
-        output_file = "piper_output.wav"
-        command = f'echo "Welcome to the world of speech synthesis!" | piper --model en_US-lessac-medium --output_file {output_file}'
-        subprocess.run(command, shell=True)
-
-        progress(1, desc="Piper-TTS check completed")
-    return output_file
+def run_all_models(input_text, progress=gr.Progress()):
+    # Initialize output file paths
+    xtts_output = "xtts_output.wav"
+    styletts2_output = "styletts2_output.wav"
+    piper_output = "piper_output.wav"
+
+    # 1. Running XttsV2
+    progress(0, desc="Running XttsV2...")
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    os.environ["COQUI_TOS_AGREED"] = "1"
+    os.system('curl -L -o default_voice.wav https://github.com/DrewThomasson/ebook2audiobookXTTS/raw/main/default_voice.wav')
+    tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
+    tts.tts_to_file(text=input_text, speaker_wav="default_voice.wav", language="en", file_path=xtts_output)
+    progress(1/3, desc="XttsV2 completed")
+
+    # 2. Running StyleTTS2
+    progress(1/3, desc="Running StyleTTS2...")
+    from styletts2 import tts as style_tts
+    my_tts = style_tts.StyleTTS2()
+    my_tts.inference(input_text, output_wav_file=styletts2_output)
+    progress(2/3, desc="StyleTTS2 completed")
+
+    # 3. Running Piper-TTS
+    progress(2/3, desc="Running Piper-TTS...")
+    command = f'echo "{input_text}" | piper --model en_US-lessac-medium --output_file {piper_output}'
+    subprocess.run(command, shell=True)
+    progress(3/3, desc="Piper-TTS completed")
+
+    # Return the file paths for downloading
+    return xtts_output, styletts2_output, piper_output
 
 # Create Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## Test TTS Models with Progress Bar")
     
-    with gr.Row():
-        with gr.Column():
-            xtts_button = gr.Button("Test XttsV2")
-            xtts_output = gr.File(label="Download XttsV2 Output")
-        with gr.Column():
-            styletts2_button = gr.Button("Test StyleTTS2")
-            styletts2_output = gr.File(label="Download StyleTTS2 Output")
-        with gr.Column():
-            piper_button = gr.Button("Test Piper-TTS")
-            piper_output = gr.File(label="Download Piper Output")
+    # Input text box for user input
+    input_text = gr.Textbox(label="Input Text", placeholder="Enter the text you want all models to use", value="Hello world!")
+
+    # Download buttons for the outputs
+    xtts_output = gr.File(label="Download XttsV2 Output")
+    styletts2_output = gr.File(label="Download StyleTTS2 Output")
+    piper_output = gr.File(label="Download Piper Output")
+
+    # Button to run all models and show progress
+    run_button = gr.Button("Run All Models")
 
-    xtts_button.click(fn=test_xtts, outputs=xtts_output)
-    styletts2_button.click(fn=test_styletts2, outputs=styletts2_output)
-    piper_button.click(fn=test_piper, outputs=piper_output)
+    # Link button click to the run_all_models function
+    run_button.click(fn=run_all_models, inputs=input_text, outputs=[xtts_output, styletts2_output, piper_output])
 
 # Launch the app
 demo.launch()