Update tools.py

tools.py

@@ -7,6 +7,29 @@ from typing import TypedDict, Annotated, Union, Dict, Any
 import base64
 from langchain_core.messages import HumanMessage
 from langchain_openai import ChatOpenAI
+import requests
+import os
+
+def download_file(task_id: str, file_name: str) -> str:
+    """Downloads a file associated with a task_id and returns the local file path"""
+    try:
+        # Create downloads directory if it doesn't exist
+        os.makedirs("downloads", exist_ok=True)
+        
+        # Download the file
+        file_url = f"https://agents-course-unit4-scoring.hf.space/files/{task_id}"
+        response = requests.get(file_url)
+        response.raise_for_status()
+        
+        # Save the file locally
+        local_path = os.path.join("downloads", file_name)
+        with open(local_path, "wb") as f:
+            f.write(response.content)
+            
+        return local_path
+        
+    except Exception as e:
+        return f"Error downloading file: {str(e)}"
 
 def get_hub_stats(author: str) -> str:
     """Fetches the most downloaded model from a specific author on the Hugging Face Hub."""