Create CustomAgent that inherits from BasicAgent

CustomAgent.py

@@ -0,0 +1,5 @@
+from BasicAgent import BasicAgent
+
+
+class CustomAgent(BasicAgent):
+    ...

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
 
-from utils import run_and_submit_all
+from BasicAgent import BasicAgent
+from CustomAgent import CustomAgent
+from utils import get_agents, run_and_submit_all
 
+# Get all available agents
+AGENTS = get_agents()
 
 # Demo app
 with gr.Blocks() as demo:
@@ -21,6 +25,7 @@ wip
     )
     gr.LoginButton()
     run_button = gr.Button("Run Evaluation & Submit All Answers")
+    agent_dropdown = gr.Dropdown(choices=list(AGENTS.keys()), label="Select Agent")
     status_output = gr.Textbox(
         label="Run Status / Submission Result",
         lines=5,
@@ -30,8 +35,13 @@ wip
         label="Questions and Agent Answers",
         wrap=True
     )
+
+    def wrapper(a: str) -> tuple[str, list]:
+        return run_and_submit_all(AGENTS.get(a, BasicAgent))
+
     run_button.click(
-        fn=run_and_submit_all,
+        fn=wrapper,
+        inputs=[agent_dropdown],
         outputs=[status_output, results_table]
     )
 

utils.py

@@ -1,5 +1,7 @@
+import inspect
 import os
-from typing import Any, Tuple
+import sys
+from typing import Any
 
 import gradio as gr
 import requests
@@ -11,9 +13,20 @@ from BasicAgent import BasicAgent
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
-def run_and_submit_all(profile: gr.OAuthProfile | None, agent: BasicAgent) -> Tuple[str, Any]:
+def get_agents() -> dict[str, type]:
     """
-    Fetches all questions, runs the BasicAgent on them, submits all answers,
+    Get all agents that inherit from BasicAgent, including imported ones.
+    """
+    return {
+        name: obj
+        for name, obj in globals().items()
+        if inspect.isclass(obj) and issubclass(obj, BasicAgent)
+    }
+
+
+def run_and_submit_all(profile: gr.OAuthProfile | None, agent: BasicAgent) -> tuple[str, Any]:
+    """
+    Fetches all questions, runs the agent on them, submits all answers,
     and displays the results.
     """
     if profile:
@@ -27,17 +40,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None, agent: BasicAgent) -> Tu
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
 
-    # 1. Instantiate Agent ( modify this part to create your agent)
-    try:
-        agent = BasicAgent()
-    except Exception as e:
-        print(f"Error instantiating agent: {e}")
-        return f"Error initializing agent: {e}", None
-    # In the case of an app running as a HF Space, this link points toward your codebase
-    agent_code = f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main"
-    print(agent_code)
-
-    # 2. Fetch Questions
+    # 1. Fetch Questions
     print(f"Fetching questions from: {questions_url}")
     try:
         response = requests.get(questions_url, timeout=15)
@@ -58,7 +61,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None, agent: BasicAgent) -> Tu
         print(f"An unexpected error occurred fetching questions: {e}")
         return f"An unexpected error occurred fetching questions: {e}", None
 
-    # 3. Run your Agent
+    # 2. Run your Agent
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
@@ -80,12 +83,16 @@ def run_and_submit_all(profile: gr.OAuthProfile | None, agent: BasicAgent) -> Tu
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
-    # 4. Prepare Submission 
-    submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
+    # 3. Prepare Submission 
+    submission_data = {
+        "username": username.strip(),
+        "agent_code": f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main",
+        "answers": answers_payload
+    }
     status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
     print(status_update)
 
-    # 5. Submit
+    # 4. Submit
     print(f"Submitting {len(answers_payload)} answers to: {submit_url}")
     try:
         response = requests.post(submit_url, json=submission_data, timeout=60)