Update app.py

app.py

@@ -1072,117 +1072,87 @@ Turn 7: final_answer_tool("1.796 trillion")
         self.graph = graph_builder.compile()
         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")
+        print(f"\n--- Starting Agent Run for Question ---")
+        print(f"Agent received question (first 100 chars): {question[:100]}...")
         
         graph_input = {
             "messages": [
                 SystemMessage(content=self.system_prompt),
                 HumanMessage(content=question)
             ],
-            "turn": 0,
-            "has_plan": False,
-            "consecutive_errors": 0,
-            "tool_history": []
+            "turn": 0
         }
         
         final_answer = "AGENT FAILED TO PRODUCE ANSWER"
         try:
-            config = {"recursion_limit": MAX_TURNS + 10}
+            config = {"recursion_limit": MAX_TURNS + 5}
             for event in self.graph.stream(graph_input, stream_mode="values", config=config):
 
-                if not event.get('messages'):
-                    continue
-                
-                last_message = event["messages"][-1]
+                if event.get('messages'): # Ensure messages exist
+                    last_message = event["messages"][-1]
+                else:
+                    continue # Skip if no messages yet
                 
                 # 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
+                             final_answer = final_answer_args['answer']
+                             print(f"--- Final Answer Captured from tool call: '{final_answer}' ---")
+                             break 
                         else:
-                            print(f"⚠️ final_answer_tool called without 'answer' argument")
-                            final_answer = "ERROR: FINAL_ANSWER_TOOL CALLED WITHOUT ANSWER"
-                            break
+                             print(f"⚠️ Final Answer tool called without 'answer' argument: {final_answer_args}")
+                             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}...")
+                    print(f"Tool Result ({last_message.tool_call_id}): {last_message.content[:500]}...")
                 elif isinstance(last_message, AIMessage) and not last_message.tool_calls:
-                    print(f"💭 AI Reasoning: {last_message.content[:300]}...")
+                    print(f"AI Message (Reasoning): {last_message.content[:500]}...")
+                elif isinstance(last_message, SystemMessage):
+                     print(f"System Message: {last_message.content[:500]}...")
+
 
-            # Final Answer Cleaning
+            # --- 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:"
-            ]
+            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:
+                    if potential_answer: 
                         cleaned_answer = potential_answer
-                        break
+                        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")
-            
+            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}")
+            print(f"Error running agent graph: {e}")
             tb_str = traceback.format_exc()
             print(tb_str)
             return f"AGENT GRAPH ERROR: {e}"
 
 
-# =============================================================================
-# GLOBAL AGENT INSTANTIATION
-# =============================================================================
+# ====================================================
+# --- Global Agent Instantiation ---
+
 try:
-    initialize_rag_components()
-    
-    agent = PlanningReflectionAgent()
-    print("✅ Global PlanningReflectionAgent instantiated successfully.")
-    if asr_pipeline is None:
-        print("⚠️ Global ASR Pipeline failed to load.")
+    agent = BasicAgent()
+    print("✅ Global BasicAgent instantiated successfully.")
+    if asr_pipeline is None: print("⚠️ Global ASR Pipeline failed load.")
 except Exception as e:
     print(f"❌ FATAL: Could not instantiate global agent: {e}")
     traceback.print_exc()
     agent = None
-             
 
 # ====================================================
 # --- (Original Template Code - Mock Questions Version) ---
@@ -1358,5 +1328,4 @@ if __name__ == "__main__":
     except FileNotFoundError: print("Warning: CWD listing failed.")
     print("-"*(60 + len(" App Starting ")) + "\n")
     print("Launching Gradio Interface...")
-    demo.queue().launch(debug=True, share=False)
-
+    demo.queue().launch(debug=True, share=False)