Update app.py

moved get_file function definition to outside of agent class

app.py

@@ -22,6 +22,21 @@ GEMINI_API_KEY = os.getenv("Gemini_API_key")
 # Agent capabilities required: Search the web, listen to audio recordings, watch YouTube videos (process the footage, not the transcript), work with Excel spreadsheets
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
+@tool
+def get_file(task_id: str) -> IO:
+    '''
+    Downloads the file associated with the given task_id, if one exists and is mapped.
+    If the question mentions an attachment, use this function.
+    Args:
+        task_id: Id of the question.
+    Returns:
+        The file associated with the question.
+    '''
+    file_request = requests.get(url=f'https://agents-course-unit4-scoring.hf.space/files/{task_id}')
+    file_request.raise_for_status()
+
+    return file_request
+
 class BasicAgent:
     def __init__(self):
         self.llm_model = LiteLLMModel(
@@ -31,7 +46,8 @@ class BasicAgent:
             )
         # self.google_search_tool = Tool(google_search = GoogleSearch())
         self.google_search_tool = GoogleSearchTool()
-        self.agent = CodeAgent(model = self.llm_model, tools = [self.google_search_tool, get_file])
+        self.get_file_tool = get_file
+        self.agent = CodeAgent(model = self.llm_model, tools = [self.google_search_tool, get_file_tool])
         # # Define Google API client with GoogleSearch tool
         # self.client = genai.Client(api_key=GEMINI_API_KEY)
         
@@ -57,21 +73,6 @@ class BasicAgent:
         # )
         # return answer.text
 
-    @tool
-    @staticmethod
-    def get_file(task_id: str) -> IO:
-        '''
-        Downloads the file associated with the given task_id, if one exists and is mapped.
-        If the question mentions an attachment, use this function.
-        Args:
-            task_id: Id of the question.
-        Returns:
-            The file associated with the question.
-        '''
-        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):
     """