Update app.py

Getting files locally as fallback for 429 errors, also added testing condition

app.py

@@ -1,6 +1,7 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, Tool, HfApiModel, LiteLLMModel, load_tool, tool
 import datetime
 import pytz
+import shutil 
 import time
 import wikipediaapi
 import yaml
@@ -32,7 +33,7 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 class AudioDescriptionTool(Tool):
     name = "audio_description"
-    description = "This is a tool that will describe an audio clip."
+    description = "This is a tool that will describe a local audio clip."
     inputs = {
         "file_name": {
             "type": "string",
@@ -55,6 +56,7 @@ class AudioDescriptionTool(Tool):
             return False
 
 audio_description_tool = AudioDescriptionTool()
+
 class FetchFileTool(Tool):
     name = "file_saver"
     description = """
@@ -76,16 +78,26 @@ class FetchFileTool(Tool):
     }
     output_type = "boolean"
 
-    def forward(self, location: str, file_path: str, file_name: str):
-        try:
-            response = requests.get(location)
-            response.raise_for_status()
-            content = response.content
+def forward(self, location: str, file_path: str, file_name: str):
+    try:
+        response = requests.get(location)
+        response.raise_for_status()
+        content = response.content
+        full_path = os.path.join(file_path, file_name)
+        with open(full_path, 'wb') as out_file:
+            out_file.write(content)
+        return True
+
+    except Exception as e:
+        # if error is HTTP 429 then copy from /files
+        if response.status_code == 429:
+            # get last part of location separated by slash
+            file_id = location.split('/')[-1]
+            local_path = f"files/{file_id}"
             full_path = os.path.join(file_path, file_name)
-            with open(full_path, 'wb') as out_file:
-                out_file.write(content)
+            shutil.copy(local_path, full_path) 
             return True
-        except Exception as e:
+        else:
             print(f"Error saving content to file: {e}")
             return False
 
@@ -241,7 +253,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             continue
         try:
             # check if the file_name is not empty
-            if item.get("file_name"):
+            # 99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3"
+            if item.get("task_id") == "99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3":
+            # if item.get("file_name"):
                 question_text = f"{question_text} Here is the file: https://agents-course-unit4-scoring.hf.space/files/{item.get('task_id')}"
             else:
                 continue
@@ -258,7 +272,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
-    # return "Questions parsed.", pd.DataFrame(results_log)
+    return "Questions parsed.", pd.DataFrame(results_log)
     
     # 4. Prepare Submission 
     # submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}