Update app.py

app.py

@@ -717,68 +717,68 @@ Your goal: Provide the EXACT answer in the EXACT format requested.
         
         self.graph = graph_builder.compile()
         print("✅ Graph compiled successfully")
-def __call__(self, question: str) -> str:
-    print(f"\n--- Starting Agent Run for Question ---")
-    print(f"Agent received question (first 100 chars): {question[:100]}...")
-    
-    # Initialize graph input with turn counter
-    graph_input = {
-        "messages": [
-            SystemMessage(content=self.system_prompt),
-            HumanMessage(content=question)
-        ],
-        "turn": 0 
-    }
-    
-    final_answer = "AGENT FAILED TO PRODUCE ANSWER"
-    try:
-        # Add config for recursion limit (LangGraph default is 25, but our turn limit is softer)
-        config = {"recursion_limit": MAX_TURNS + 5} # Allow slightly more graph steps than turns
-        for event in self.graph.stream(graph_input, stream_mode="values", config=config):
-            last_message = event["messages"][-1]
+    def __call__(self, question: str) -> str:
+        print(f"\n--- Starting Agent Run for Question ---")
+        print(f"Agent received question (first 100 chars): {question[:100]}...")
+        
+        # Initialize graph input with turn counter
+        graph_input = {
+            "messages": [
+                SystemMessage(content=self.system_prompt),
+                HumanMessage(content=question)
+            ],
+            "turn": 0 
+        }
+        
+        final_answer = "AGENT FAILED TO PRODUCE ANSWER"
+        try:
+            # Add config for recursion limit (LangGraph default is 25, but our turn limit is softer)
+            config = {"recursion_limit": MAX_TURNS + 5} # Allow slightly more graph steps than turns
+            for event in self.graph.stream(graph_input, stream_mode="values", config=config):
+                last_message = event["messages"][-1]
+                
+                # Check for final answer extraction
+                if isinstance(last_message, AIMessage) and last_message.tool_calls:
+                    if last_message.tool_calls[0].get("name") == "final_answer_tool":
+                        final_answer = last_message.tool_calls[0]['args'].get('answer', "ERROR: FINAL_ANSWER_TOOL CALLED WITHOUT ANSWER")
+                        print(f"--- Final Answer Captured from tool call: '{final_answer}' ---")
+                        # We can break here since the graph condition should lead to END anyway
+                        break 
+                        
+                # Log other message types (optional but helpful)
+                elif isinstance(last_message, ToolMessage):
+                    print(f"Tool Result ({last_message.tool_call_id}): {last_message.content[:500]}...")
+                elif isinstance(last_message, AIMessage) and not last_message.tool_calls:
+                    # This is now expected (the "plan" or "think" step)
+                    print(f"AI Message (Plan/Thought): {last_message.content[:500]}...")
+                    # Don't set final_answer here anymore, only final_answer_tool counts
             
-            # Check for final answer extraction
-            if isinstance(last_message, AIMessage) and last_message.tool_calls:
-                if last_message.tool_calls[0].get("name") == "final_answer_tool":
-                    final_answer = last_message.tool_calls[0]['args'].get('answer', "ERROR: FINAL_ANSWER_TOOL CALLED WITHOUT ANSWER")
-                    print(f"--- Final Answer Captured from tool call: '{final_answer}' ---")
-                    # We can break here since the graph condition should lead to END anyway
-                    break 
-                    
-            # Log other message types (optional but helpful)
-            elif isinstance(last_message, ToolMessage):
-                 print(f"Tool Result ({last_message.tool_call_id}): {last_message.content[:500]}...")
-            elif isinstance(last_message, AIMessage) and not last_message.tool_calls:
-                 # This is now expected (the "plan" or "think" step)
-                 print(f"AI Message (Plan/Thought): {last_message.content[:500]}...")
-                 # Don't set final_answer here anymore, only final_answer_tool counts
-        
-        # --- Cleaning step (Keep as is) ---
-        cleaned_answer = str(final_answer).strip()
-        # ... (keep existing prefix removal and fence removal logic) ...
-        prefixes_to_remove = ["The answer is:", "Here is the answer:", "Based on the information:", "Final Answer:", "Answer:"]
-        original_cleaned = cleaned_answer
-        for prefix in prefixes_to_remove:
-            if cleaned_answer.lower().startswith(prefix.lower()):
-                potential_answer = cleaned_answer[len(prefix):].strip()
-                if potential_answer: cleaned_answer = potential_answer; break
-        if cleaned_answer == original_cleaned and any(cleaned_answer.lower().startswith(p.lower()) for p in prefixes_to_remove):
-             print(f"Warning: Prefix found but not stripped: '{original_cleaned[:100]}...'")
-        # Simple fence removal
-        cleaned_answer = remove_fences_simple(cleaned_answer)
-        if cleaned_answer.startswith("`") and cleaned_answer.endswith("`"):
-              cleaned_answer = cleaned_answer[1:-1].strip()
-        print(f"Agent returning final answer (cleaned): '{cleaned_answer}'")
-        return cleaned_answer
-    except Exception as e:
-        print(f"Error running agent graph: {e}")
-        tb_str = traceback.format_exc()
-        print(tb_str)
-        # Check if it was specifically our turn limit message
-        if isinstance(e, SystemMessage) and f"maximum turn limit ({MAX_TURNS})" in str(e.content):
-             return f"AGENT STOPPED: Reached maximum turn limit ({MAX_TURNS})."
-        return f"AGENT GRAPH ERROR: {e}"
-    # ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+            # --- Cleaning step (Keep as is) ---
+            cleaned_answer = str(final_answer).strip()
+            # ... (keep existing prefix removal and fence removal logic) ...
+            prefixes_to_remove = ["The answer is:", "Here is the answer:", "Based on the information:", "Final Answer:", "Answer:"]
+            original_cleaned = cleaned_answer
+            for prefix in prefixes_to_remove:
+                if cleaned_answer.lower().startswith(prefix.lower()):
+                    potential_answer = cleaned_answer[len(prefix):].strip()
+                    if potential_answer: cleaned_answer = potential_answer; break
+            if cleaned_answer == original_cleaned and any(cleaned_answer.lower().startswith(p.lower()) for p in prefixes_to_remove):
+                print(f"Warning: Prefix found but not stripped: '{original_cleaned[:100]}...'")
+            # Simple fence removal
+            cleaned_answer = remove_fences_simple(cleaned_answer)
+            if cleaned_answer.startswith("`") and cleaned_answer.endswith("`"):
+                cleaned_answer = cleaned_answer[1:-1].strip()
+            print(f"Agent returning final answer (cleaned): '{cleaned_answer}'")
+            return cleaned_answer
+        except Exception as e:
+            print(f"Error running agent graph: {e}")
+            tb_str = traceback.format_exc()
+            print(tb_str)
+            # Check if it was specifically our turn limit message
+            if isinstance(e, SystemMessage) and f"maximum turn limit ({MAX_TURNS})" in str(e.content):
+                return f"AGENT STOPPED: Reached maximum turn limit ({MAX_TURNS})."
+            return f"AGENT GRAPH ERROR: {e}"
+        # ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 # --- (Original Template Code - Mock Questions Version) ---
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """