feat: implement an agent

.env.example

@@ -0,0 +1,3 @@
+SPACE_ID=
+SPACE_HOST=
+OPENROUTER_KEY=

README.md

@@ -1,5 +1,5 @@
 ---
-title: Template Final Assignment
+title: Final Assignment
 emoji: 🕵🏻‍♂️
 colorFrom: indigo
 colorTo: indigo

agent.py

@@ -0,0 +1,53 @@
+import gradio as gr
+import random
+import os
+from smolagents import DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, LiteLLMModel
+from tools import DownloadTaskAttachmentTool, VisitWebpageTool
+
+
+class MyAgent:
+    def __init__(self):
+        self.agent = CodeAgent(
+            model=LiteLLMModel(model_id="openrouter/meta-llama/llama-4-maverick:free", api_key=os.getenv("OPENROUTER_KEY")),
+            tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool(), DownloadTaskAttachmentTool()],
+            add_base_tools=True,
+            additional_authorized_imports=['pandas', 'numpy', 'csv', 'subprocess', 'exec']
+        )
+        print("MyAgent initialized.")
+    
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        agent_answer = self.agent.run(question)
+        print(f"Agent answer: {agent_answer}")
+        return agent_answer
+
+    def download_file(self, task_id: str) -> str:
+        """
+        Downloads a file associated with the given task ID.
+        Returns the file path where the file is saved locally.
+        """
+        file_url = f"{DEFAULT_API_URL}/files/{task_id}"
+        local_file_path = f"downloads/{task_id}.file"
+
+        print(f"Downloading file for task ID {task_id} from {file_url}...")
+        try:
+            response = requests.get(file_url, stream=True, timeout=15)
+            response.raise_for_status()
+
+            os.makedirs("downloads", exist_ok=True)
+            with open(local_file_path, "wb") as file:
+                for chunk in response.iter_content(chunk_size=8192):
+                    file.write(chunk)
+
+            print(f"File downloaded successfully: {local_file_path}")
+            return local_file_path
+        except requests.exceptions.RequestException as e:
+            print(f"Error downloading file for task {task_id}: {e}")
+            raise
+
+    def launch_ui(self):
+        GradioUI(self.agent).launch()
+
+# if __name__ == "__main__":
+#     my_agent = MyAgent()
+#     my_agent.launch_ui()

app.py

@@ -4,20 +4,12 @@ import requests
 import inspect
 import pandas as pd
 
+from agent import MyAgent
+
 # (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):
     """
@@ -40,7 +32,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = MyAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -76,13 +68,25 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
+        requires_file = item.get("requires_file", False)
+
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
+
         try:
-            submitted_answer = agent(question_text)
+            # Download file if required
+            if requires_file:
+                file_path = agent.download_file(task_id)
+                print(f"File for task {task_id} saved at: {file_path}")
+                # Optionally, pass the file path to the agent if needed
+                submitted_answer = agent(f"{question_text} (File: {file_path})")
+            else:
+                submitted_answer = agent(question_text)
+
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
+            time.sleep(2)
         except Exception as e:
              print(f"Error running agent on task {task_id}: {e}")
              results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})

requirements.txt

@@ -1,2 +1,6 @@
 gradio
-requests
+requests
+smolagents
+litellm
+wikipedia-api
+google.auth

tools.py

@@ -0,0 +1,79 @@
+import time
+import re
+from markdownify import markdownify
+from smolagents import Tool
+
+class DownloadTaskAttachmentTool(Tool):
+    name = "download_file"
+    description = "Downloads the file attached to the task ID"
+    inputs = {'task_id': {'type': 'string', 'description': 'The task id to download attachment from.'}}
+    output_type = "string"
+
+    def forward(self, task_id: str) -> str:
+        """
+        Downloads a file associated with the given task ID.
+        Returns the file path where the file is saved locally.
+        """
+        file_url = f"{DEFAULT_API_URL}/files/{task_id}"
+        local_file_path = f"downloads/{task_id}.file"
+
+        print(f"Downloading file for task ID {task_id} from {file_url}...")
+        try:
+            response = requests.get(file_url, stream=True, timeout=15)
+            response.raise_for_status()
+
+            os.makedirs("downloads", exist_ok=True)
+            with open(local_file_path, "wb") as file:
+                for chunk in response.iter_content(chunk_size=8192):
+                    file.write(chunk)
+
+            print(f"File downloaded successfully: {local_file_path}")
+            return local_file_path
+        except requests.exceptions.RequestException as e:
+            print(f"Error downloading file for task {task_id}: {e}")
+            raise
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
+
+
+class VisitWebpageTool(Tool):
+    name = "visit_webpage"
+    description = "Visits a webpage at the given url and reads its content as a markdown string. Use this to browse webpages."
+    inputs = {'url': {'type': 'string', 'description': 'The url of the webpage to visit.'}}
+    output_type = "string"
+
+    def forward(self, url: str) -> str:
+        try:
+            import requests
+            from markdownify import markdownify
+            from requests.exceptions import RequestException
+
+            from smolagents.utils import truncate_content
+        except ImportError as e:
+            raise ImportError(
+                "You must install packages `markdownify` and `requests` to run this tool: for instance run `pip install markdownify requests`."
+            ) from e
+        try:
+            # Send a GET request to the URL with a 20-second timeout
+            response = requests.get(url, timeout=20)
+            response.raise_for_status()  # Raise an exception for bad status codes
+
+            # Convert the HTML content to Markdown
+            markdown_content = markdownify(response.text).strip()
+
+            # Remove multiple line breaks
+            markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
+
+            return truncate_content(markdown_content, 10000)
+
+        except requests.exceptions.Timeout:
+            return "The request timed out. Please try again later or check the URL."
+        except RequestException as e:
+            return f"Error fetching the webpage: {str(e)}"
+        except Exception as e:
+            return f"An unexpected error occurred: {str(e)}"
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
+