Update app.py

app.py

@@ -1320,27 +1320,45 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
     for item in questions_data:
+
         task_id = item.get("task_id")
         question_text = item.get("question")
-        file_path_from_api = item["file_path"]
-        file_download_url = f"{DEFAULT_API_URL}/files/{task_id}"
-        file_info = []
-        # Extract the original file name to preserve the extension (e.g., .pdf, .jpg)
-        original_filename = file_path_from_api.split('/')[-1]
         
-        # 🌟 NEW: Save the file to the /tmp directory
-        local_file_path = os.path.join("/tmp", original_filename)
-        try:
-            file_response = requests.get(file_download_url, timeout=15)
-            file_response.raise_for_status()
+        # 🌟 Initialize file variables for the current question
+        local_file_path = None
+        file_info = ""
+    
+        # 🌟 CRITICAL: Check if 'file_path' exists in the item dictionary
+        if item.get("file_path"):
+            file_path_from_api = item["file_path"]
+            file_download_url = f"{DEFAULT_API_URL}/files/{task_id}"
+            
+            # Extract the original file name to preserve the extension
+            original_filename = file_path_from_api.split('/')[-1]
             
-            # 🌟 CRITICAL: Write the raw bytes content to the local file path
-            with open(local_file_path, 'wb') as f:
-                f.write(file_response.content)
-                print(f"✅ Downloaded file to: {local_file_path}")        
-                file_info = local_file_path
-        except requests.exceptions.RequestException as e:
-            error_message = f"[FILE DOWNLOAD ERROR: Could not fetch file from {file_download_url}: {e}]"
+            # Set the path where the file will be saved locally
+            local_file_path = os.path.join("/tmp", original_filename) 
+            
+            # --- (Add streaming update here for file download status) ---
+    
+            try:
+                file_response = requests.get(file_download_url, timeout=15)
+                file_response.raise_for_status()
+                
+                # Save the raw bytes content to the local file path
+                with open(local_file_path, 'wb') as f:
+                    f.write(file_response.content) 
+                    
+                print(f"✅ Downloaded file to: {local_file_path}")
+                
+                # Set the context string to be passed to the agent
+                file_info = f"\n\n[FILE ATTACHED: {local_file_path}]"
+                
+            except requests.exceptions.RequestException as e:
+                error_message = f"[FILE DOWNLOAD ERROR: Could not fetch file: {e}]"
+                print(f"⚠️ {error_message}")
+                # Still provide the error message as context to the agent
+                file_info = f"\n\n{error_message}"
                 
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")