Update app.py

app.py

@@ -3,12 +3,30 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from agent import BasicAgent
+from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, PythonInterpreterTool, WikipediaSearchTool, CodeAgent
 
 # (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:
+        self.agent = CodeAgent(
+            model=OpenAIServerModel(
+                model_id="anthropic/claude-3.7-sonnet",
+                api_key=os.environ["OPENROUTER_API_KEY"],
+                api_base="https://openrouter.ai/api/v1"
+            ),
+            tools=[DuckDuckGoSearchTool(), PythonInterpreterTool(), WikipediaSearchTool()]
+        )
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        fixed_answer = self.agent.run(question)
+        print(f"Agent returning answer: {fixed_answer}")
+        return fixed_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """