Update app.py

app.py

@@ -1,11 +1,10 @@
 import os
 import gradio as gr
 from transformers import ReactCodeAgent, HfEngine, Tool
-from gradio_agentchatbot import (
-    AgentChatbot,
-    stream_from_transformers_agent,
-    ChatMessage,
-)
+
+from gradio import ChatMessage
+from gradio.utils import stream_from_transformers_agent
+
 from dotenv import load_dotenv
 from huggingface_hub import login
 from transformers.agents.default_tools import (
@@ -128,7 +127,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 = AgentChatbot(label="Agent")
+            chatbot = gr.Chatbot(label="Agent")
 
     submit.click(interact_with_agent, [text_input], [chatbot])