Update app.py

app.py

@@ -1,23 +1,135 @@
+
+
 import os
-import gradio as gr
+import gradio as gr 
 import requests
 import inspect
-import pandas as pd
+import pandas as pd 
+import time
+import re
+from markdownify import markdownify
+from smolagents import Tool, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, LiteLLMModel
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+
+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
+
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     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("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
+        agent_answer = self.agent.run(question)
+        print(f"Agent returning 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 run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -76,16 +188,29 @@ 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}"})
+            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}"})
+
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")
@@ -146,11 +271,9 @@ with gr.Blocks() as demo:
     gr.Markdown(
         """
         **Instructions:**
-
         1.  Please clone this space, then modify the code to define your agent's logic, the tools, the necessary packages, etc ...
         2.  Log in to your Hugging Face account using the button below. This uses your HF username for submission.
         3.  Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
-
         ---
         **Disclaimers:**
         Once clicking on the "submit button, it can take quite some time ( this is the time for the agent to go through all the questions).
@@ -193,4 +316,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)