files

.gitignore

@@ -0,0 +1 @@
+.env

README.md

@@ -1,5 +1,5 @@
 ---
-title: Template Final Assignment
+title: Final Assignment
 emoji: 🕵🏻‍♂️
 colorFrom: indigo
 colorTo: indigo
@@ -12,4 +12,4 @@ hf_oauth: true
 hf_oauth_expiration_minutes: 480
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

agent.py

@@ -0,0 +1,29 @@
+import os
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool
+from tools import VisitWebpageTool, DownloadTaskAttachmentTool
+
+class TheAgent:
+    """Agent that processes questions using LLM and various tools."""
+
+    def __init__(self):
+        # Initialize the agent with model and tools
+        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("Agent initialized successfully.")
+
+    def __call__(self, question: str) -> str:
+        """Process a question and return the answer."""
+        print(f"Processing question: {question[:50]}..." if len(question) > 50 else f"Processing question: {question}")
+        return self.agent.run(question)

app.py

@@ -1,24 +1,14 @@
 import os
 import gradio as gr
 import requests
-import inspect
 import pandas as pd
 
+from agent import TheAgent
+
 # (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,
@@ -40,7 +30,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = TheAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -193,4 +183,4 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)

requirements.txt

@@ -1,2 +1,2 @@
 gradio
-requests
+requests

tools.py

@@ -0,0 +1,75 @@
+import re
+import os
+import requests
+from smolagents import Tool
+from smolagents.utils import truncate_content
+
+# Define constant for API URL
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+
+class DownloadTaskAttachmentTool(Tool):
+    """Tool for downloading files attached to tasks."""
+
+    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 __init__(self, *args, **kwargs):
+        self.is_initialized = True
+
+    def forward(self, task_id: str) -> str:
+        """Download a file associated with the given task ID."""
+        file_url = f"{DEFAULT_API_URL}/files/{task_id}"
+        local_file_path = f"downloads/{task_id}.file"
+
+        try:
+            # Create downloads directory if it doesn't exist
+            os.makedirs("downloads", exist_ok=True)
+
+            # Download the file
+            with requests.get(file_url, stream=True, timeout=15) as response:
+                response.raise_for_status()
+                with open(local_file_path, "wb") as file:
+                    for chunk in response.iter_content(chunk_size=8192):
+                        file.write(chunk)
+
+            return local_file_path
+
+        except Exception as e:
+            print(f"Error downloading file for task {task_id}: {e}")
+            raise
+
+
+class VisitWebpageTool(Tool):
+    """Tool for visiting webpages and converting content to markdown."""
+
+    name = "visit_webpage"
+    description = "Visits a webpage and returns its content as markdown"
+    inputs = {'url': {'type': 'string', 'description': 'The URL to visit'}}
+    output_type = "string"
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = True
+
+    def forward(self, url: str) -> str:
+        """Visit a webpage and return its content as markdown."""
+        try:
+            # Import here to avoid dependency issues
+            from markdownify import markdownify
+
+            # Fetch webpage content
+            response = requests.get(url, timeout=20)
+            response.raise_for_status()
+
+            # Convert HTML to markdown and clean up
+            content = markdownify(response.text).strip()
+            content = re.sub(r"\n{3,}", "\n\n", content)
+
+            # Truncate to avoid excessively long responses
+            return truncate_content(content, 10000)
+
+        except requests.exceptions.Timeout:
+            return "Request timed out. Please check the URL or try again later."
+        except Exception as e:
+            return f"Error: {str(e)}"