second attempt to fix bug

app.py

@@ -12,18 +12,22 @@ from agent import graph
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
-    """LangGraph Agent"""
+    """LangGraph Agent Wrapper"""
+    
     def __init__(self):
         print("BasicAgent initialized.")
-        self.graph = build_graph()
+        self.graph = graph  # Using graph already compiled in agent.py
         
     def __call__(self, question: str) -> str:
         messages = [HumanMessage(content=question)]
-        messages = self.graph.invoke({"messages": messages})
-        answer = messages['messages'][-1].content
-        return answer[14:]
+        result = self.graph.invoke({"messages": messages})
+        final_response = result['messages'][-1].content
+        
+        # Assumes answer starts with "FINAL ANSWER: "
+        if final_response.startswith("FINAL ANSWER: "):
+            return final_response[len("FINAL ANSWER: "):].strip()
+        return final_response.strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """