fix Gradio error

app.py

@@ -5,7 +5,7 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 
-from Gradio_UI import GradioUI
+import gradio as gr
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -136,4 +136,14 @@ agent = CodeAgent(
 )
 
 
-GradioUI(agent).launch()
+def chat_fn(message, history):
+    """Simple chat wrapper around the CodeAgent for Gradio."""
+    # agent.run returns the final answer (already using final_answer tool)
+    result = agent.run(task=message)
+    history = history + [(message, str(result))]
+    return history
+
+
+demo = gr.ChatInterface(fn=chat_fn, title="SmolAgents News & Tools Agent")
+
+demo.launch()