Update app.py

app.py

@@ -23,7 +23,18 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         result = self.graph.invoke({"messages": messages})
         answer = result['messages'][-1].content
-        return answer
+        if isinstance(answer, str):
+            # If the tool already returned a clean value (like "RSA" or "EGY")
+            # strip whitespace and formatting
+            cleaned = answer.strip()
+            # If model included "Answer:" or markdown, extract last token
+            if "Answer:" in cleaned:
+                cleaned = cleaned.split("Answer:")[-1].strip()
+            if cleaned.startswith("**") and cleaned.endswith("**"):
+                cleaned = cleaned.strip("*").strip()
+            return cleaned
+
+        return str(answer)
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """