Update app.py

app.py

@@ -11,13 +11,23 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
+    """A StateGraph-powered agent."""
     def __init__(self):
+        print("BasicAgent initializing StateGraph agent...")
         # graph factory lives here
-        self.graph = build_graph()
+        self.agent = build_graph()
         
     def __call__(self, question: str) -> str:
-        # runs graph on each question
-        answer = self.graph.run(question)
+        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()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):