Update app.py

app.py

@@ -170,14 +170,6 @@ Rules for the ANSWER:
             response = self.call_llm(conversation)
             print(f"  LLM [{iteration}]: \n{response[:300]}...\n")
 
-            answer_match = re.search(r'ANSWER:\s*(.+?)(?:\n|$)', response, re.IGNORECASE)
-            if answer_match:
-                answer = answer_match.group(1).strip()
-                print(f"   {answer}")
-                return answer
-
-            tool_result = None
-
             search_match = re.search(r'SEARCH:\s*(.+?)(?:\n|$)', response)
             wiki_match = re.search(r'WIKIPEDIA:\s*(.+?)(?:\n|$)', response)
             python_match = re.search(r'PYTHON:\s*```(?:python)?\n?(.*?)```', response, re.DOTALL)
@@ -185,6 +177,9 @@ Rules for the ANSWER:
             if not python_match:
                 python_match = re.search(r'PYTHON:\s*(.+?)(?:\nSEARCH|\nWIKIPEDIA|\nANSWER|$)', response, re.DOTALL)
 
+            tool_result = None
+
+            # ORCHESTRATION FIX: Check for tools FIRST. 
             if search_match:
                 query = search_match.group(1).strip()
                 print(f"  Tool: web_search({query})")
@@ -198,6 +193,13 @@ Rules for the ANSWER:
                 print(f"  Tool: python({code[:50]}...)")
                 tool_result = f"Python output:\n{run_python(code)}"
             else:
+                # ONLY look for an answer if NO tools were called.
+                answer_match = re.search(r'ANSWER:\s*(.+?)(?:\n|$)', response, re.IGNORECASE)
+                if answer_match:
+                    answer = answer_match.group(1).strip()
+                    print(f"   {answer}")
+                    return answer
+                
                 tool_result = "No valid tool call detected. Please use SEARCH, WIKIPEDIA, PYTHON, or ANSWER."
 
             conversation += f"\n\nAssistant: {response}\n\nTool Result: {tool_result}\n\nNow provide your next THOUGHT and tool, or your ANSWER."