Fix BasicAgent app.py

app.py

@@ -5,13 +5,16 @@ import os
 import gradio as gr
 import pandas as pd
 import requests
+import rich
 import wikipediaapi
 from basic_agent import BasicAgent
+from get_model import get_model
 from mcp import StdioServerParameters
-from smolagents import CodeAgent, LiteLLMRouterModel, Tool, ToolCollection, VisitWebpageTool
+from smolagents import DuckDuckGoSearchTool, FinalAnswerTool, Tool, ToolCollection, VisitWebpageTool
 from ycecream import y
 
 y.configure(sln=1)
+print = rich.get_console().print
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -52,7 +55,14 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = BasicAgent(
+            model=get_model(cat="gemini"),
+            tools=[
+                DuckDuckGoSearchTool(),
+                VisitWebpageTool(),
+                FinalAnswerTool(),
+            ]
+        )
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None

basic_agent.py

@@ -103,6 +103,7 @@ def main():
             tools=[
                 DuckDuckGoSearchTool(),
                 VisitWebpageTool(),
+                FinalAnswerTool(),
             ]
         )
     except Exception as e: