Update app.py

app.py

@@ -1215,6 +1215,102 @@ Turn 7: final_answer_tool("1.796 trillion")
         print("✅ Planning & Reflection Agent graph compiled successfully.")
         
 
+    def __call__(self, question: str) -> str:
+        print(f"\n{'='*70}")
+        print(f"🎯 NEW QUESTION")
+        print(f"{'='*70}")
+        print(f"Q: {question[:200]}{'...' if len(question) > 200 else ''}")
+        print(f"{'='*70}\n")
+        
+        graph_input = {
+            "messages": [
+                SystemMessage(content=self.system_prompt),
+                HumanMessage(content=question)
+            ],
+            "turn": 0,
+            "has_plan": False,
+            "consecutive_errors": 0,
+            "tool_history": []
+        }
+        
+        final_answer = "AGENT FAILED TO PRODUCE ANSWER"
+        try:
+            config = {"recursion_limit": MAX_TURNS + 10}
+            for event in self.graph.stream(graph_input, stream_mode="values", config=config):
+
+                if not event.get('messages'):
+                    continue
+                
+                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_args = last_message.tool_calls[0].get('args', {})
+                        if 'answer' in final_answer_args:
+                            final_answer = final_answer_args['answer']
+                            print(f"\n{'='*70}")
+                            print(f"✅ FINAL ANSWER CAPTURED: '{final_answer}'")
+                            print(f"{'='*70}\n")
+                            break
+                        else:
+                            print(f"⚠️ final_answer_tool called without 'answer' argument")
+                            final_answer = "ERROR: FINAL_ANSWER_TOOL CALLED WITHOUT ANSWER"
+                            break
+
+                elif isinstance(last_message, ToolMessage):
+                    result_preview = last_message.content[:300].replace('\n', ' ')
+                    print(f"📊 Tool Result: {result_preview}...")
+                elif isinstance(last_message, AIMessage) and not last_message.tool_calls:
+                    print(f"💭 AI Reasoning: {last_message.content[:300]}...")
+
+            # Final Answer Cleaning
+            cleaned_answer = str(final_answer).strip()
+            
+            # Remove common prefixes
+            prefixes_to_remove = [
+                "The answer is:", "Here is the answer:", "Based on the information:", 
+                "Final Answer:", "Answer:", "The final answer is:", "My answer is:",
+                "According to", "I found that", "The result is:"
+            ]
+            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
+            
+            # Remove code fences
+            cleaned_answer = remove_fences_simple(cleaned_answer)
+            
+            # Remove surrounding backticks
+            while cleaned_answer.startswith("`") and cleaned_answer.endswith("`"):
+                cleaned_answer = cleaned_answer[1:-1].strip()
+            
+            # Remove quotes if they wrap the entire answer
+            if (cleaned_answer.startswith('"') and cleaned_answer.endswith('"')) or \
+               (cleaned_answer.startswith("'") and cleaned_answer.endswith("'")):
+                cleaned_answer = cleaned_answer[1:-1].strip()
+            
+            # Remove trailing periods for non-sentence answers
+            if cleaned_answer.endswith('.') and len(cleaned_answer.split()) < 10:
+                cleaned_answer = cleaned_answer[:-1]
+            
+            print(f"\n{'='*70}")
+            print(f"🎉 FINAL CLEANED ANSWER")
+            print(f"{'='*70}")
+            print(f"{cleaned_answer}")
+            print(f"{'='*70}\n")
+            
+            return cleaned_answer
+            
+        except Exception as e:
+            print(f"❌ Error running agent graph: {e}")
+            tb_str = traceback.format_exc()
+            print(tb_str)
+            return f"AGENT GRAPH ERROR: {e}"
+
+
 # =============================================================================
 # GLOBAL AGENT INSTANTIATION
 # =============================================================================