update algo with then

app.py

@@ -765,6 +765,24 @@ def retry(history3):
         print("chathmi3 error:", e)      
     # yield chathmi3(last_request, chatbot_history)
 
+def display_input(message, history2):
+    global last_request
+    print("Input Message:", message)
+    last_request = message
+    history2 = history2 + [(message, None)]
+    return history2
+    
+def Inference_Agent(history_inf):
+    global last_request
+    message = last_request
+    try:
+        response = agent.run(message)
+        time.sleep(0.1)
+        history_inf = history_inf + [(None, response)]
+        return history_inf   
+    except Exception as e:
+        print("error:", e)
+
 with gr.Blocks() as demo:
     # gr.Markdown("Start typing below and then click **SUBMIT** to see the output.")
     # main = gr.ChatInterface(
@@ -825,8 +843,10 @@ with gr.Blocks() as demo:
     # upload_button.upload(func_upload_file, [upload_button, main.chatbot], main.chatbot)
     clear_button.click(clearall, None, chatbot)
     retry_button.click(retry, chatbot, chatbot)
-    inf1 = inputtext.submit(chathmi3, [inputtext, chatbot], [inputtext, chatbot])
-    stop_button.click(read_logs, None, logs, cancels=[inf1])
+    # inf1 = inputtext.submit(chathmi3, [inputtext, chatbot], [inputtext, chatbot])
+    inf2 = inputtext.submit(display_input, [inputtext, chatbot], chatbot).\
+        then(Inference_Agent, chatbot, chatbot)
+    stop_button.click(read_logs, None, logs, cancels=[inf2])
     upload_button.upload(func_upload_file, [upload_button, chatbot], chatbot)
     agentchoice.change(SetAgent, agentchoice, None)
     frash_logs.click(read_logs, None, logs)