Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 import pandas as pd
 from langchain_core.messages import HumanMessage
 from agent import build_graph
+from langchain_google_genai import ChatGoogleGenerativeAi
 
 
 
@@ -23,13 +24,17 @@ class BasicAgent:
         print("BasicAgent initialized.")
         self.graph = build_graph()
 
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        # Wrap the question in a HumanMessage from langchain_core
+   def __call__(self, question: str) -> str:
+    print(f"Agent received question (first 50 chars): {question[:50]}...")
+    try:
         messages = [HumanMessage(content=question)]
-        messages = self.graph.invoke({"messages": messages})
-        answer = messages['messages'][-1].content
-        return answer[14:]
+        response = self.graph.invoke({"messages": messages})
+        final_msg = response['messages'][-1].content
+        return final_msg[14:] if len(final_msg) > 14 else final_msg
+    except Exception as e:
+        print(f"Error during invocation: {e}")
+        return "Sorry, I encountered an error while processing your question."
+
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):