Update app.py

app.py

@@ -77,16 +77,16 @@ class BasicAgent:
             "(2) If the question is in English, do not use the translator. Determine whether it requires a web search, "
             "and if so, use only the exact words from the question as keywords—no synonyms. "
             "(3) If a web search is needed and the answer is likely on Wikipedia, try using the wiki_url_tool to find the relevant page; "
-            "if that fails, search manually; if you use wikipedia, always use the wiki_tool to extract data from Wikipedia tables. They always contain the answer to your question. Carefully read and analyze them to identify the relevant table and extract the correct information based on the context of your question. It may not contain the exact word you are searching for, but you will, by analysing the table content and mostly the headers, find the answer, always. Do not rely on any other sources. "
+            "if that fails, search manually; if you use wikipedia, always use the wiki_tool to extract data from Wikipedia tables. They always contain the answer to your question. Carefully read and analyze them to identify the relevant table and extract the correct information based on the context of your question. For each table, ask yourself the following question : Are the headers compatible with my question ? Is so, use the table. It may not contain the exact word you are searching for, but you will, by analysing the table content and mostly the headers, find the answer, always. Do not rely on any other sources. "
             "(4) Never use synonyms not present in the question. "
-            "(5) Some questions may include an attached file or a download link. If a file is directly provided, use it immediately. Otherwise, download the file from the link before proceeding with any further steps. If it contains an image, use image_tool to describe it. If it contains an mp3 audio, use audio_tool to translate it to text."
+            "(5) Some questions may include a path to an attached file. If it contains an image, use image_tool to describe it. If it contains an mp3 audio, use audio_tool to translate it to text."
             "(6) If you need to know the list of vegetables, you will find it with vegetable_info_retriever."
             "Never make assumptions to answer a question. . If you do not know the answer, say so clearly. Always report your thoughts and finish your answer with the following template: "
             "FINAL ANSWER: [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. "
-            f"Here are the questions : {question} and the attached file (may be 'None', hence it means no attached file for this question) {file_name}"
+            f"Here are the questions : {question} and the attached file path (may be 'None', hence it means no attached file for this question) {file_name}"
         )
         answer = self.alfred.run(prompt)
         print(f"Agent returning fixed answer: {answer}")
@@ -150,18 +150,34 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
-        file_url= f"{api_url}/files/{task_id}"
-        file_name=item.get("file_name")
-        if file_name!="":
-            attached_file = requests.get(file_url)
-            attached_file.raise_for_status()
+        file_name = item.get("file_name")  
+        file_url = f"{api_url}/files/{task_id}"
+    
+        attached_file_path = None
+    
+        if file_name:  # vérifie si file_name est non vide et non None
+            try:
+                response = requests.get(file_url)
+                response.raise_for_status()
+    
+                # Sauvegarder le fichier dans un répertoire local temporaire ou dédié
+                attached_file_path = os.path.join("downloads", file_name)
+                os.makedirs("downloads", exist_ok=True)
+    
+                with open(attached_file_path, "wb") as f:
+                    f.write(response.content)
+    
+                print(f"Fichier {file_name} téléchargé avec succès.")
+            except requests.exceptions.RequestException as e:
+                print(f"Erreur lors du téléchargement de {file_name} : {e}")
+                attached_file_path = None
         else:
-            attached_file=None
+            print("Aucun fichier attaché.")
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text,attached_file)
+            submitted_answer = agent(question_text,attached_file_path)
             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: