improve HMI

app.py

@@ -796,6 +796,13 @@ def playsound():
         soundfilename = Filename_Chatbot
         return soundfilename
 
+def HMI_Runing():
+    return [gr.update(visible=False), gr.update(visible=True)]
+
+def HMI_Wait():
+    return [gr.update(visible=True), gr.update(visible=False)]
+
+
 with gr.Blocks() as demo:
     # gr.Markdown("Start typing below and then click **SUBMIT** to see the output.")
     # main = gr.ChatInterface(
@@ -816,8 +823,8 @@ with gr.Blocks() as demo:
                 placeholder = "Input Your Question",
                 show_label = False,
                 )
-            stop_button = gr.Button("STOP", variant='stop')
-    
+            submit_button = gr.Button("STOP", variant="primary", visible=True)
+            stop_button = gr.Button("STOP", variant='stop', visible=False)
 
         with gr.Row():
 
@@ -858,9 +865,14 @@ with gr.Blocks() as demo:
     retry_button.click(retry, chatbot, chatbot)
     inf1 = inputtext.submit(chathmi3, [inputtext, chatbot], [inputtext, chatbot]).\
         then(playsound, None, voice_output)
+    inf3 = submit_button.click(HMI_Runing, None, [submit_button, stop_button]).\
+        then(chathmi3, [inputtext, chatbot], [inputtext, chatbot]).\
+        then(playsound, None, voice_output).\
+        then(HMI_Wait, None, [submit_button, stop_button])
     # inf2 = inputtext.submit(display_input, [inputtext, chatbot], chatbot).\
     #     then(Inference_Agent, chatbot, [inputtext, chatbot])
-    stop_button.click(read_logs, None, logs, cancels=[inf1])
+    stop_button.click(read_logs, None, logs, cancels=[inf1,inf3]).\
+        then(HMI_Wait, None, [submit_button, stop_button])
     # stop_button.click(read_logs, None, logs, cancels=[inf2])
     upload_button.upload(func_upload_file, [upload_button, chatbot], chatbot)
     agentchoice.change(SetAgent, agentchoice, None)