Update app.py

app.py

@@ -16,13 +16,15 @@ class BasicAgent:
         print("LLM Tool-Enhanced Agent initialized.")
 
     def __call__(self, question: str) -> str:
-    try:
-        result = agent_executor.invoke({"messages": [HumanMessage(content=question)]})
-        answer = result["messages"][-1].content
-        return answer.strip()
-    except Exception as e:
-        print(f"Agent error: {e}")
-        return "I don't know. There is an exception."
+        try:
+            from langchain_core.messages import HumanMessage
+            result = agent_executor.invoke({"messages": [HumanMessage(content=question)]})
+            answer = result["messages"][-1].content
+            return answer.strip()
+        except Exception as e:
+            print(f"Agent error: {e}")
+            return "I don't know. There is an exception."
+
             
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """