Update app.py

Added a function to download task file

app.py

@@ -9,6 +9,7 @@ from google import genai
 # from google.genai.types import Tool, GoogleSearch
 from google.genai import types
 import asyncio
+import requests
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -34,7 +35,7 @@ class BasicAgent:
         
         print("BasicAgent initialized.")
     
-    async def __call__(self, question: str) -> str:
+    async def __call__(self, question: str, task_id: 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}")
@@ -42,17 +43,26 @@ class BasicAgent:
         # return self.agent.run(question)
         # Use the Google GenAI client to run the question
         await asyncio.sleep(10)
-        response = self.client.models.generate_content(
+        answer = self.client.models.generate_content(
             model='gemini-2.0-flash',
             contents=f'''Answer the following question in the format as requested. If the format not specified, then provide a single word/number/name. 
-            Do not include how you answered the question. {question}''',
+            In your response, do not include anything other than your answer. {question}''',
             config=types.GenerateContentConfig(
-                tools=[types.Tool(
-                    google_search=types.GoogleSearch()
-                )]
+                tools=[types.Tool(google_search=types.GoogleSearch()),
+                    types.Tool(code_execution=self.get_file)]
             )
         )
-        return response.text
+        return answer.text
+
+    def get_file(self, task_id):
+        '''
+        Downloads the file associated with the given task_id, if one exists and is mapped.
+        If the question mentions an attachment, use this function.
+        '''
+        file_request = requests.get(url=f'https://agents-course-unit4-scoring.hf.space/files/{task_id}')
+        file_request.raise_for_status()
+
+        return file_request
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -115,7 +125,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = asyncio.run(agent(question_text))
+            submitted_answer = asyncio.run(agent(question_text, task_id))
             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})
         except Exception as e: