Move agent class ot its own file

agent.py

@@ -0,0 +1,40 @@
+import os
+
+# --- 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]}...")
+
+        ## new agent code
+        gemini_model = os.getenv(key="GEMINI_MODEL")
+
+        model = LiteLLMModel(model_id=f"gemini/{gemini_model}", api_key=os.getenv(key="GEMINI_API_KEY"))
+        agent = CodeAgent(
+            model=model,
+            tools=[DuckDuckGoSearchTool()]
+        )
+        prompt = f"""\
+You are a general AI assistant. I will ask you a question. Report your thoughts,
+and finish your answer with the following template:
+    FINAL ANSWER: YOUR FINAL ANSWER
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma-
+separated list of numbers and/or strings.
+
+If you are asked for a number, do NOT use commas or units (%, $, etc.) unless
+explicitly requested.  If you are asked for a string, do NOT use articles or
+abbreviations (e.g. for cities) and write digits in plain text unless
+specified otherwise.  If you are asked for a comma-separated list, apply the
+above rules to each element.
+
+{question}
+"""
+        answer = agent.run(prompt)
+        
+        print(f"Agent returning answer: {answer}")
+        return answer
+
+

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from agent import BasicAgent
 from smolagents import ToolCallingAgent, CodeAgent, LiteLLMModel
 
 from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
@@ -21,43 +22,6 @@ hub_stats_tool = HubStatsTool()
 # --- 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]}...")
-
-        ## new agent code
-        gemini_model = os.getenv(key="GEMINI_MODEL")
-
-        model = LiteLLMModel(model_id=f"gemini/{gemini_model}", api_key=os.getenv(key="GEMINI_API_KEY"))
-        agent = CodeAgent(
-            model=model,
-            tools=[DuckDuckGoSearchTool()]
-        )
-        prompt = f"""\
-You are a general AI assistant. I will ask you a question. Report your thoughts,
-and finish your answer with the following template:
-    FINAL ANSWER: YOUR FINAL ANSWER
-YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma-
-separated list of numbers and/or strings.
-
-If you are asked for a number, do NOT use commas or units (%, $, etc.) unless
-explicitly requested.  If you are asked for a string, do NOT use articles or
-abbreviations (e.g. for cities) and write digits in plain text unless
-specified otherwise.  If you are asked for a comma-separated list, apply the
-above rules to each element.
-
-{question}
-"""
-        answer = agent.run(prompt)
-        
-        print(f"Agent returning answer: {answer}")
-        return answer
-
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """

tools.py

@@ -3,6 +3,8 @@ from smolagents import Tool
 import random
 from huggingface_hub import list_models
 
+## CURRENTLY NOT USED - KEEP AS TEMPLATE
+
 
 class WeatherInfoTool(Tool):
     name = "weather_info"