Update agent.py

agent.py

@@ -541,7 +541,38 @@ def combine_images(
         return {"error": str(e)}
 
 
-tools = [multiply, add, subtract, divide, wikidata_search, tavily_search_tool, youtube_search_tool, combine_images, analyze_image, transform_image, draw_on_image, generate_simple_image, analyze_csv_file, analyze_excel_file, save_and_read_file, download_file_from_url, extract_text_from_image]
+@tool
+def download_task_file(task_id: str, api_url: str = "https://agents-course-unit4-scoring.hf.space") -> str:
+    """
+    Download a file associated with a task from the evaluation API.
+    Args:
+        task_id (str): The task ID to download the file for
+        api_url (str): The base API URL (defaults to the evaluation server)
+    """
+    try:
+        # Construct the file download URL
+        file_url = f"{api_url}/files/{task_id}"
+        
+        # Create temporary file
+        temp_dir = tempfile.gettempdir()
+        filename = f"task_{task_id}.png"  # Most files are images
+        filepath = os.path.join(temp_dir, filename)
+        
+        # Download the file
+        response = requests.get(file_url, stream=True)
+        response.raise_for_status()
+        
+        # Save the file
+        with open(filepath, "wb") as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                f.write(chunk)
+        
+        return f"Task file downloaded to {filepath}. You can now analyze this file."
+    except Exception as e:
+        return f"Error downloading task file: {str(e)}"
+
+
+tools = [multiply, add, subtract, divide, wikidata_search, tavily_search_tool, youtube_search_tool, combine_images, analyze_image, transform_image, draw_on_image, generate_simple_image, analyze_csv_file, analyze_excel_file, save_and_read_file, download_file_from_url, extract_text_from_image, download_task_file]
 
 def build_graph():
     llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", api_key=os.getenv("GOOGLE_API_KEY"))