stream_to_gradio

app.py

@@ -131,24 +131,20 @@ if __name__ == "__main__":
         max_steps=1,
         verbosity_level=1,
     )
-    # agent.run("What is the average each customer paid?")
     # GradioUI(agent).launch()
-    # print(
-    #     "xxxxxxxxxxxxxxxxxxxxxxxxxxxxx test:", sql_engine_tool("SELECT * FROM receipts")
-    # )
 
-    def enter_message(message):
+    def enter_message(new_message, conversation_history):
 
-        messages = []
-        messages.append(gr.ChatMessage(role="user", content=message))
-        yield messages
-        for msg in stream_to_gradio(agent, message):
-            messages.append(msg)
-            yield messages
+        conversation_history = []
+        conversation_history.append(gr.ChatMessage(role="user", content=new_message))
+        yield "", conversation_history
+        for msg in stream_to_gradio(agent, new_message):
+            conversation_history.append(msg)
+            yield "", conversation_history
 
     with gr.Blocks() as b:
-        chatbot = gr.Chatbot(type="messages")
+        chatbot = gr.Chatbot(type="messages", height=1000)
         textbox = gr.Textbox()
         button = gr.Button("reply")
-        button.click(enter_message, [textbox], [chatbot])
+        button.click(enter_message, [textbox, chatbot], [textbox, chatbot])
     b.launch(debug=True)