Prompt for Excel and audio questions

app.py

@@ -49,7 +49,10 @@ class BasicAgent:
         # Define your system prompt
         self.system_prompt = """You are a general AI assistant. I will ask you a question. Finish your answer with only YOUR FINAL ANSWER. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
         
-        If visit_webpage fails with a 403 error, fetch the page directly using requests with a User-Agent header: import requests headers = {"User-Agent": "Mozilla/5.0 (compatible; MyAgent/1.0)"} response = requests.get(url, headers=headers) text = response.text"""
+        If visit_webpage fails with a 403 error, fetch the page directly using requests with a User-Agent header: import requests headers = {"User-Agent": "Mozilla/5.0 (compatible; MyAgent/1.0)"} response = requests.get(url, headers=headers) text = response.text
+        
+        YouTube URLs cannot be accessed directly. For YouTube questions, search for the video title or topic using web_search to find the answer indirectly.
+        """
         
         # Create the agent with tools
         self.agent = CodeAgent(
@@ -169,7 +172,12 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            # submitted_answer = agent(question_text)
+            question_with_context = f"""Task ID: {task_id}
+            If this question refers to an attached file, download it first from:
+            https://agents-course-unit4-scoring.hf.space/files/{task_id}
+            {question_text}"""
+            submitted_answer = agent(question_with_context) # for Excel and audio questions
             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: