Update app.py

app.py

@@ -48,6 +48,7 @@ def load_mod(model):
     yield f"Voice Loaded: {model}"
     
 with gr.Blocks() as iface:
+    stt=gr.State()
     aud=gr.Audio(streaming=True,autoplay=True)
     chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, layout="panel")
     prompt = gr.Textbox()
@@ -63,13 +64,11 @@ with gr.Blocks() as iface:
         width=gr.Slider(label="Noise Width", minimum=0.01, maximum=3.0, value=0.5)
         sen_pause=gr.Slider(label="Sentence Pause", minimum=0.1, maximum=10.0, value=1)
         upd_btn=gr.Button("Update")
-    with gr.Row(visible=False):
-        stt=gr.Textbox(visible=True)
 
     iface.load(load_mod,names,msg)
     sub_b = submit_b.click(generate, [prompt,chatbot],[chatbot,stt])
-    #sub_e = prompt.submit(generate, [prompt, chatbot], [chatbot,stt])
-    stt.change(pp.stream_tts,[stt,names,length,noise,width,sen_pause],aud)
+    sub_e = prompt.submit(generate, [prompt, chatbot], [chatbot,stt])
+    aud_out = stt.change(pp.stream_tts,[stt,names,length,noise,width,sen_pause],aud)
     names_change=names.change(load_mod,names,msg)
-    stop_b.click(None,None,None, cancels=[sub_b,names_change])
+    stop_b.click(None,None,None, cancels=[sub_b,sub_e,aud_out,names_change])
 iface.queue(default_concurrency_limit=10).launch()