Update app.py

app.py

@@ -25,33 +25,45 @@ def generate_tts(text, temperature=0.1, repetition_penalty=1.1, max_length=4096)
     
     return "output.wav"
 
-# Define callback functions for interaction logging
-def on_text_input(text):
-    print(f"User typed text: {text}")
-
-def on_temperature_change(val):
-    print(f"Temperature slider adjusted to: {val}")
-
-def on_repetition_penalty_change(val):
-    print(f"Repetition Penalty slider adjusted to: {val}")
-
-def on_max_length_change(val):
-    print(f"Max Length slider adjusted to: {val}")
-
-# Gradio interface for TTS
-demo = gr.Interface(
-    fn=generate_tts,
-    inputs=[
-        gr.Textbox(lines=2, placeholder="Enter text to convert to speech", label="Text").change(on_text_input),
-        gr.Slider(0.1, 1.0, value=0.1, label="Temperature").change(on_temperature_change),
-        gr.Slider(1.0, 2.0, value=1.1, label="Repetition Penalty").change(on_repetition_penalty_change),
-        gr.Slider(512, 4096, value=4096, step=256, label="Max Length").change(on_max_length_change)
-    ],
-    outputs=gr.Audio(type="filepath", label="Generated Speech"),
-    title="OuteTTS Text-to-Speech Demo",
-    description="Convert text to speech using the OuteTTS model."
-)
-
-# Launch the Gradio app
+# Create the Gradio Blocks interface
+with gr.Blocks() as demo:
+    # Log each interaction
+    def on_text_input(text):
+        print(f"User typed text: {text}")
+
+    def on_temperature_change(val):
+        print(f"Temperature slider adjusted to: {val}")
+
+    def on_repetition_penalty_change(val):
+        print(f"Repetition Penalty slider adjusted to: {val}")
+
+    def on_max_length_change(val):
+        print(f"Max Length slider adjusted to: {val}")
+    
+    # Text input
+    text_input = gr.Textbox(lines=2, placeholder="Enter text to convert to speech", label="Text")
+    text_input.change(on_text_input, inputs=text_input)
+    
+    # Sliders with change events for tracking
+    temperature_slider = gr.Slider(0.1, 1.0, value=0.1, label="Temperature")
+    temperature_slider.change(on_temperature_change, inputs=temperature_slider)
+
+    repetition_penalty_slider = gr.Slider(1.0, 2.0, value=1.1, label="Repetition Penalty")
+    repetition_penalty_slider.change(on_repetition_penalty_change, inputs=repetition_penalty_slider)
+
+    max_length_slider = gr.Slider(512, 4096, value=4096, step=256, label="Max Length")
+    max_length_slider.change(on_max_length_change, inputs=max_length_slider)
+
+    # Button to generate TTS and Audio output
+    generate_button = gr.Button("Generate Speech")
+    audio_output = gr.Audio(type="filepath", label="Generated Speech")
+
+    # Define interaction between input and output
+    generate_button.click(
+        generate_tts, 
+        inputs=[text_input, temperature_slider, repetition_penalty_slider, max_length_slider], 
+        outputs=audio_output
+    )
+
 print("Launching Gradio interface...")
 demo.launch()