Update app.py

app.py

@@ -17,19 +17,15 @@ class BasicAgent:
         # graph factory lives here
         self.agent = build_graph()
         
-    def __call__(self, question: str) -> str:
+    def __call__(self,question: str) -> str:
         print(f"Agent received question: {question[:50]}...")
-        # seed initial state(empty message list
-        init_state = {"message": []}
-        # run the graph, injecting the questions
-        # bulid_graph() was compiled to accept (state, question=...)
-        result_state = self.agent.invoke(init_state, question)
-        # extract the final message
-        final_msg = result_state["messages"][-1]
-        answer = getattr(final_msg, "content", str(final_msg))
-        # strip off any "FINAL ANSWER:" prefix if present
-        return answer.replace("FINAL ANSWER:","",1).strip()
-
+        #invoke the ccompiled graph runner with just the question
+        result = self.agent.invoke(question)
+        # because the graph now ends in answer_node, result is a dict {id:answer}
+        # extract that single answer string:
+        answer = list(result.values())[0]
+        return answer.strip()
+        
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,