Add default final answer

app.py

@@ -18,34 +18,31 @@ class BasicAgent:
         self.graph = build_graph()
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
-        # print(f"Agent received question (first 50 chars): {question[:50]}...")
-        # messages = [HumanMessage(content=question)]
-        # try:
-        #     result = self.graph.invoke({"messages": messages})
-
-        #     if not result.get("messages") or len(result["messages"]) == 0:
-        #         raise ValueError("No messages returned from the graph")
-
-        #     last_message = result["messages"][-1].content
-
-        #     match = re.search(r"(?i)final\s*answer\s*[:\-]?\s*(.+)", last_message, re.DOTALL)
-        #     if match:
-        #         final_answer = match.group(1).strip()
-        #     else:
-        #        # Try to find a name or string answer
-        #         candidates = [line.strip() for line in last_message.split("\n") if line.strip()]
-        #         final_answer = next((line for line in reversed(candidates) if len(line.split()) <= 5), last_message.strip())
-
-        #     return final_answer
-        # except Exception as e: 
-        #     print("Agent processing failed")
-        #     return
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        # Wrap the question in a HumanMessage from langchain_core
         messages = [HumanMessage(content=question)]
-        messages = self.graph.invoke({"messages": messages})
-        answer = messages['messages'][-1].content
-        return answer[14:]
+        try:
+            result = self.graph.invoke({"messages": messages})
+
+            if not result.get("messages") or len(result["messages"]) == 0:
+                raise ValueError("No messages returned from the graph")
+
+            last_message = result["messages"][-1].content
+
+            final_answer = "I cannot answer that"
+
+            match = re.search(r"(?i)final\s*answer\s*[:\-]?\s*(.+)", last_message, re.DOTALL)
+            if match:
+                final_answer = match.group(1).strip()
+            else:
+               # Try to find a name or string answer
+                candidates = [line.strip() for line in last_message.split("\n") if line.strip()]
+                final_answer = next((line for line in reversed(candidates) if len(line.split()) <= 5), last_message.strip())
+
+            return final_answer
+        except Exception as e: 
+            print("Agent processing failed")
+            return
+       
             
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):