Update app.py

app.py

@@ -430,7 +430,7 @@ class SabonzoAgent:
             return_intermediate_steps=False,
         )
 
-    def __call__(self, question: str, task_id: str) -> str:
+    def __call__(self, question: str, task_id: str, file_url: str = None) -> str:
         """Processes a single question, handling file downloads and analysis."""
         logging.info(f"--- Starting Task {task_id} ---")
         logging.info(f"Question: {question[:150]}...")
@@ -474,21 +474,13 @@ class SabonzoAgent:
 
         # --- Step 2: Download GAIA File if needed ---
         if needs_gaia_file:
-            file_url = f"{self.api_url}/files/{task_id}"
-            logging.info(f"Task {task_id} requires GAIA {file_type} file download from: {file_url}")
-            file_path = download_file(file_url, self.temp_dir, task_id)
-            if not file_path:
-                # If download failed, set analysis_result to error and update context
-                analysis_result = f"ERROR: Failed to download the required {file_type} file for task {task_id} from {file_url}."
-                analysis_context = f"Analysis Context: {analysis_result}"
-                # Set final_answer to the error if file is absolutely required
-                final_answer = analysis_result
-            elif not file_path.exists() or file_path.stat().st_size == 0:
-                 # Handle cases where download function might return path but file is empty/gone
-                 analysis_result = f"ERROR: Downloaded {file_type} file for task {task_id} is missing or empty at {file_path}."
+            if not file_url:
+                 analysis_result = f"ERROR: No file_url provided for task {task_id}"
                  analysis_context = f"Analysis Context: {analysis_result}"
                  final_answer = analysis_result
-                 file_path = None # Ensure file_path is None if file is invalid
+             else:
+                 logging.info(f"Task {task_id} requires GAIA {file_type} file download from: {file_url}")
+                 file_path = download_file(file_url, self.temp_dir, task_id)
 
         # --- Step 3: Perform Analysis if download was successful ---
         # Only proceed if file_path is valid and we haven't already set final_answer due to download error
@@ -772,6 +764,7 @@ def run_evaluation(profile: gr.OAuthProfile | None):
     for i, item in enumerate(questions_data):
         task_id = item.get("task_id")
         question_text = item.get("question")
+        gaia_file_url = item.get("file_url")
         progress_text = f"Running question {i+1}/{num_questions} (Task ID: {task_id})..."
         logging.info(progress_text)
 
@@ -789,7 +782,7 @@ def run_evaluation(profile: gr.OAuthProfile | None):
         start_time_task = time.time()
         submitted_answer = f"ERROR: Agent failed to return an answer for task {task_id}" # Default
         try:
-            submitted_answer = agent(question_text, str(task_id))
+            submitted_answer = agent(question_text, str(task_id), gaia_file_url)
             elapsed_time_task = time.time() - start_time_task
             logging.info(f"Task {task_id} completed in {elapsed_time_task:.2f} seconds.")