Update app.py

app.py

@@ -25,19 +25,28 @@ class BasicAgent:
 
     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
         messages = [HumanMessage(content=question)]
         print(f"messages: {messages}")
+        
         try:
             messages = self.graph.invoke({"messages": messages})
             print(f"Returned from graph.invoke(): {messages}")
+            
+            # Add validation
+            if not messages or 'messages' not in messages:
+                return "Error: No response received from the agent"
+                
+            message_list = messages['messages']
+            if not message_list:
+                return "Error: Empty response from the agent"
+                
+            answer = message_list[-1].content
+            # Only slice if the answer is long enough
+            return answer[14:] if len(answer) > 14 else answer
+            
         except Exception as e:
             print(f"Exception during graph.invoke: {e}")
-            raise
-        print(f"messages: {messages}")
-        print(f"messages['messages']: {messages['messages']}")
-        answer = messages['messages'][-1].content
-        return answer[14:]
+            return f"Error processing question: {str(e)}"
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):