update answer format

app.py

@@ -6,6 +6,16 @@ import pandas as pd
 from agent import build_graph
 from langchain_core.messages import HumanMessage
 
+import re
+
+def extract_answer(text: str) -> str:
+    # Try extracting after "FINAL ANSWER:" or similar patterns
+    match = re.search(r"FINAL ANSWER:\s*(.+)", text, re.IGNORECASE)
+    if match:
+        return match.group(1).strip()
+    # Otherwise, fall back to the whole text (assuming it's already clean)
+    return text.strip()
+
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -23,13 +33,13 @@ class BasicAgent:
             outputs = result["messages"]
             for m in reversed(outputs):
                 if m.type == "ai":
-                    print(f"Agent output: {m.content[:100]}")
-                    return m.content
-            return "⚠️ No AI message found in the result."
+                    clean_answer = extract_answer(m.content)
+                    print(f"Extracted clean answer: {clean_answer}")
+                    return clean_answer
+            return "No answer found."
         except Exception as e:
             print(f"LangGraph Agent error: {e}")
-            return f"❌ Error: {str(e)}"
-
+            return f"Error: {str(e)}"
 
 def run_and_submit_all(username: str):
     if not username: