Update app.py

app.py

@@ -507,6 +507,41 @@ class BasicAgent:
              # Catch potential issues like model download failure, OOM errors
              raise RuntimeError(f"Failed to initialize HuggingFacePipeline for {model_id}: {e}")
 
+        try:
+            # Construct a simplified test prompt (mimicking agent input)
+            test_prompt = SYSTEM_MESSAGE + "\nHuman: What is the capital of France?\nAssistant:" # A simple question
+            # Or use a prompt closer to the problematic one if you know which one it is
+            # test_prompt = SYSTEM_MESSAGE + "\nHuman: [Insert the non-commutative table question here]\nAssistant:"
+
+            # Use invoke which is standard now
+            test_response = self.llm.invoke(test_prompt)
+            print(f"--- Direct LLM Test Response ---:\n{test_response}\n-----------------------------")
+            if not test_response or len(test_response.strip()) == 0:
+                 print("!!! Direct LLM Test returned empty or whitespace result.")
+
+        except Exception as test_e:
+            print(f"!!! Direct LLM Test FAILED: {test_e}")
+            # print traceback for more details
+            import traceback
+            traceback.print_exc()
+        print("BasicAgent initialized with LLM.")
+
+    def __call__(self, question: str) -> str:
+        # Comment out agent call temporarily if testing in __init__
+        # Or add the direct test here before calling the agent
+        print(f"\n>> Processing Question (Agent): {question}")
+        try:
+            response = self.agent.invoke({"input": question})
+            answer = response.get('output', "Agent did not produce an output.")
+            print(f"<< Agent Answer: {answer}")
+            return str(answer).strip()
+        except Exception as e:
+             print(f"Error during agent execution: {e}")
+             # Also print traceback here to see where the error originates
+             import traceback
+             traceback.print_exc()
+             return f"Agent Error: {e}"
+        
         # --- Agent Initialization (remains the same) ---
         self.agent = initialize_agent(
             tools=TOOLS,