stream_to_gradio

app.py

@@ -133,25 +133,22 @@ if __name__ == "__main__":
     )
     # agent.run("What is the average each customer paid?")
     # GradioUI(agent).launch()
-    print(
-        "xxxxxxxxxxxxxxxxxxxxxxxxxxxxx test:", sql_engine_tool("SELECT * FROM receipts")
-    )
+    # print(
+    #     "xxxxxxxxxxxxxxxxxxxxxxxxxxxxx test:", sql_engine_tool("SELECT * FROM receipts")
+    # )
+
+    def enter_message(message):
 
-    def enter_message(message, chat_history):
-        print()
-        print("enter_message debug")
-        print("engine", engine)
-        print("message:", message)
-        print("chat_history", chat_history)
-        chat_history.append({"role": "user", "content": message})
-        x = stream_to_gradio(agent, message)
-        print(type(x))
-        print("\n\n\n", x, "\n\n\n")
-        return "", x
+        messages = []
+        messages.append(gr.ChatMessage(role="user", context=message))
+        yield messages
+        for msg in stream_to_gradio(agent, message):
+            messages.append(msg)
+            yield messages
 
     with gr.Blocks() as b:
         chatbot = gr.Chatbot(type="messages")
-        input = gr.Textbox()
+        textbox = gr.Textbox()
         button = gr.Button("reply")
-        button.click(enter_message, [input, chatbot], [input, chatbot])
+        button.click(enter_message, [textbox], [chatbot])
     b.launch(debug=True)