Update app.py

app.py

@@ -2,9 +2,11 @@ import os
 import gradio as gr
 from transformers import ReactCodeAgent, HfEngine, Tool
 
-from gradio import ChatMessage
-from gradio.utils import stream_from_transformers_agent
-
+from gradio_agentchatbot import (
+    AgentChatbot,
+    stream_from_transformers_agent,
+    ChatMessage,
+)
 from dotenv import load_dotenv
 from huggingface_hub import login
 from transformers.agents.default_tools import (
@@ -127,7 +129,7 @@ Write your function in the left textbox, and the agent on the right will generat
             )
             submit = gr.Button("Generate tests!")
         with gr.Column():
-            chatbot = gr.Chatbot(label="Agent")
+            chatbot = AgentChatbot(label="Agent")
 
     submit.click(interact_with_agent, [text_input], [chatbot])