fixed file downloading

agentcourse_unit4/api/agent_eval_api.py

@@ -1,3 +1,4 @@
+import tempfile
 import urllib.request
 from typing import List
 
@@ -21,12 +22,13 @@ class AgentEvalApi:
         response.raise_for_status()
         return response.json(object_hook=to_question_response)
 
-    def download_file(self, task_id: str) -> str:
+    def download_file(self, task_id: str, file_name: str) -> str:
         """Download the file associated with the given task_id to temp dir and return path to file"""
         file_path = None
 
         try:
-            result = urllib.request.urlretrieve(f"{self.files_url}/{task_id}")
+            path_to_file = f"{tempfile.gettempdir()}/{file_name}"
+            result = urllib.request.urlretrieve(f"{self.files_url}/{task_id}", filename=path_to_file)
             file_path = result[0]
         except Exception as e:
             print(f"Error downloading file: {str(e)}")

app.py

@@ -70,7 +70,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             continue
 
         try:
-            file_path = api_client.download_file(task_id) if len(file_name) > 0 else None
+            file_path = api_client.download_file(task_id, file_name) if len(file_name) > 0 else None
             submitted_answer = agent.run(question_text, file_path)
             answers_payload.append(AnswerData(task_id=task_id, answer=str(submitted_answer)))
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})