debug

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
-from smolagents import tool, CodeAgent, HfApiModel, GradioUI
+from smolagents import tool, CodeAgent, HfApiModel, GradioUI, MultiStepAgent
 from sqlalchemy import (
     create_engine,
     MetaData,
@@ -127,4 +127,18 @@ if __name__ == "__main__":
         verbosity_level=1,
     )
     # agent.run("What is the average each customer paid?")
-    GradioUI(agent).launch()
+    # GradioUI(agent).launch()
+
+    def enter_message(agent: MultiStepAgent, text: str):
+        print(text)
+        x = agent.run(text)
+        print(type(x))
+        print(x.action_output)
+
+    with gr.Blocks() as b:
+        gr.Chatbot(type="history")
+
+        input = gr.Textbox()
+        button = gr.Button("reply")
+        button.click(enter_message)
+    b.launch()