Update app.py

Attempt to read locally cached copies of the question files

app.py

@@ -37,7 +37,7 @@ class AudioDescriptionTool(Tool):
     inputs = {
         "file_name": {
             "type": "string",
-            "description": "The name of the local file to describe, for example: /files/99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3",
+            "description": "Complete name of the local file to describe, for example: /files/98c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea8.mp3",
         }
     }
     output_type = "string"
@@ -57,59 +57,59 @@ class AudioDescriptionTool(Tool):
 
 audio_description_tool = AudioDescriptionTool()
 
-class FetchFileTool(Tool):
-    name = "file_saver"
-    description = """
-    This is a tool that will fetch an internet file and save content locally.
-    It takes a location, file path, file name, and content to save."""
-    inputs = {
-        "location": {
-            "type": "string",
-            "description": "The internet location of the file to fetch",
-        },
-        "file_extension": {
-            "type": "string",
-            "description": "Assumed file extension of the file to fetch, like mp3 or png for example",
-        },
-        "file_path": {
-            "type": "string",
-            "description": "The path to the directory where the file will be saved",
-        },
-        "file_name": {
-            "type": "string",
-            "description": "The name of the file where content will be saved",
-        }
-    }
-    output_type = "boolean"
+# class FetchFileTool(Tool):
+#     name = "file_saver"
+#     description = """
+#     This is a tool that will fetch an internet file and save content locally.
+#     It takes a location, file path, file name, and content to save."""
+#     inputs = {
+#         "location": {
+#             "type": "string",
+#             "description": "The internet location of the file to fetch",
+#         },
+#         "file_extension": {
+#             "type": "string",
+#             "description": "Assumed file extension of the file to fetch, like mp3 or png for example",
+#         },
+#         "file_path": {
+#             "type": "string",
+#             "description": "The path to the directory where the file will be saved",
+#         },
+#         "file_name": {
+#             "type": "string",
+#             "description": "The name of the file where content will be saved",
+#         }
+#     }
+#     output_type = "boolean"
     
-    def forward(self, location: str, file_extension: 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
+#     def forward(self, location: str, file_extension: 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}.{file_extension}"
-                print(f"Copying from: {local_path}")
-                
-                full_path = os.path.join(file_path, file_name)
-                print(f"Copying to: {full_path}")
-
-                shutil.copy(local_path, full_path)
-                return True
-            else:
-                print(f"Error saving content to file: {e}")
-                return False
+#         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}.{file_extension}"
+#                 print(f"Copying from: {local_path}")
+
+#                 full_path = os.path.join(file_path, file_name)
+#                 print(f"Copying to: {full_path}")
+
+#                 shutil.copy(local_path, full_path)
+#                 return True
+#             else:
+#                 print(f"Error saving content to file: {e}")
+#                 return False
 
-fetch_file_tool = FetchFileTool()
+# fetch_file_tool = FetchFileTool()
 
 class WikipediaSearchTool(Tool):
     name = "wikipedia_search"
@@ -166,7 +166,7 @@ class BasicAgent:
             prompt_templates = yaml.safe_load(stream) 
         self.agent = CodeAgent(
             model=model,
-            tools=[final_answer, search_tool, wiki_tool, fetch_file_tool, audio_description_tool],
+            tools=[final_answer, search_tool, wiki_tool, audio_description_tool],
             max_steps=7,
             verbosity_level=1,
             additional_authorized_imports=[
@@ -264,7 +264,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             # 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')}"
+                # question_text = f"{question_text} Here is the file: https://agents-course-unit4-scoring.hf.space/files/{item.get('task_id')}"
+                question_text = f"{question_text} Here is the file: /files/{item.get('task_id')}"
             else:
                 continue
             submitted_answer = agent(question_text)