Update app.py

app.py

@@ -1,27 +1,13 @@
+import agent
 import os
 import gradio as gr
 import requests
 import inspect
-import pandas as pd
-
-from smolagents import InferenceClientModel, CodeAgent, ToolCallingAgent, DuckDuckGoSearchTool, VisitWebpageTool
-
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
-    def __init__(self):
-        print("BasicAgent initialized.")
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
-
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -41,39 +27,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
 
-    # 1. Instantiate Agent ( modify this part to create your agent)
-    model_id = "Qwen/Qwen2.5-72B-Instruct"
-    model = InferenceClientModel(model_id)
-    
     try:
-        web_agent = ToolCallingAgent(
-            tools=[DuckDuckGoSearchTool(), VisitWebpageTool()], model=model,
-            name="search_agent",
-            description="Runs web searches for you. Give it your query as an argument.",
-        )
-        python_agent = CodeAgent(
-            tools=[],
-            model=model,
-            name='python_agent',
-            description='Use additional_authorized_imports for you. You need to do actions and help to answer the questions with python code',
-            additional_authorized_imports=[
-                "json",
-                "pandas",
-                "numpy",
-                "requests",
-                "time",
-                "datetime",
-            ],
-            planning_interval=5,
-            verbosity_level=2,
-            final_answer_checks=[check_reasoning_and_plot],
-            max_steps=15,
-        )
-        agent = CodeAgent(tools=[],
-                        model=model,
-                        managed_agents=[web_agent, python_agent],
-                        additional_authorized_imports=["requests"],
-        )
+        agent = agent.BasicAgent()
         
     except Exception as e:
         print(f"Error instantiating agent: {e}")