Update app.py

app.py

@@ -24,12 +24,20 @@ class BasicAgent:
         self.graph = build_graph()
 
     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)]
-        messages = self.graph.invoke({"messages": messages})
-        answer = messages['messages'][-1].content
-        return answer[14:]
+    print(f"Agent received question (first 50 chars): {question[:50]}...")
+    messages = [HumanMessage(content=question)]
+    response = self.graph.invoke({"messages": messages})
+    raw_output = response["messages"][-1].content.strip()
+
+    # Validate and clean answer
+    if raw_output.startswith("FINAL ANSWER:"):
+        final_answer = raw_output[len("FINAL ANSWER:"):].strip()
+    else:
+        final_answer = raw_output.strip()  # fallback if not formatted properly
+        print("⚠️ Output missing 'FINAL ANSWER:' prefix.")
+
+    return final_answer
+
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
@@ -108,7 +116,12 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
     # 4. Prepare Submission 
-    print(f"agent_code → {repr(agent_code)}")
+    print("Sanitized submission preview:")
+    print("username →", repr(username))
+    print("agent_code →", repr(agent_code))
+    for a in answers_payload[:3]:  # Show first few answers
+        print("sample answer →", repr(a))
+
 
     submission_data = {
     "username": username.strip(),