Update app.py

app.py

@@ -49,5 +49,33 @@ agent = CodeAgent(
 )
 
 
-from smolagents import GradioUI
-GradioUI(agent).launch()
+import gradio as gr
+
+def handle_query(query):
+    # This calls your agent and returns the generator/answer
+    yield agent.run(query)
+
+# Minimalist UI compatible with Gradio 4, 5, and 6
+with gr.Blocks() as demo:
+    gr.Markdown("# Agentic AI - Final Project")
+    chatbot = gr.Chatbot(label="Agent") # No 'type' argument = No crash
+    msg = gr.Textbox(label="Enter your task:")
+    clear = gr.Button("Clear")
+
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+    def bot(history):
+        # Access the last message sent by user
+        user_message = history[-1][0]
+        # Get response from your agent
+        bot_message = agent.run(user_message)
+        history[-1][1] = bot_message
+        yield history
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+demo.launch()