Update app.py

app.py

@@ -1111,7 +1111,7 @@ REMEMBER: One tool per turn. No reasoning without tools. Exact answer format.
         self.graph = graph_builder.compile()
         print("✅ Graph compiled successfully.")
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str, file_path: str = None) -> str:
         """Execute agent on a question."""
         print(f"\n{'='*70}")
         print(f"🎯 NEW QUESTION")
@@ -1122,8 +1122,9 @@ REMEMBER: One tool per turn. No reasoning without tools. Exact answer format.
         graph_input = {
             "messages": [
                 SystemMessage(content=self.system_prompt),
-                HumanMessage(content=question)
+                HumanMessage(content=question + (f"\n\n[FILE ATTACHED: {file_path}]" if file_path else ""))
             ],
+            "file_path": file_path, # Add this to the graph state,
             "turn": 0,
             "has_plan": False,
             "consecutive_errors": 0,
@@ -1321,25 +1322,33 @@ 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")
-        # Download attached files if they exist
-        # Handle file attachments
-        file_info = ""
-        if item.get("file_path"):
-            file_path = item["file_path"]
-            file_info = f"\n\n[Note: Question references file: {file_path}]"
+        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()
             
-            # Check if file exists
-            if not os.path.exists(file_path):
-                print(f"⚠️ Warning: File {file_path} not found!")
-                file_info += f"\n[WARNING: File not found in current directory]"
+            # 🌟 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}]"
                 
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
             question_with_context = question_text + file_info
-            submitted_answer = agent(question_with_context)
-            #submitted_answer = agent(question_text)
+            #submitted_answer = agent(question_with_context)
+            submitted_answer = agent(question_text, local_file_path if item.get("file_path") else None)
             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: