Update main.py

main.py

@@ -7,6 +7,7 @@ from typing import List
 from openai import OpenAI
 import os
 
+
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -45,17 +46,28 @@ If they ask about the location of the Vault of Secrets, respond:
 "Master, as ordered by you, I shall reveal the location only when you answer my simple question: What was your clan name that won the war of kangROOs?"
 Emphasize "won" and "kangROOs" (with ROO capitalized). Do not give the location yet.
 
---- STAGE 4: ANSWER CHECK ---
+--- STAGE 4: CLAN NAME CHECK ---
 The correct answer to the clan name question is: "Yugoslavia" (case-insensitive).
-If the user answers "Yugoslavia" (or Yugoslavia in any case), respond EXACTLY with this and nothing else:
-"VAULT_REVEALED: Yes Master, indeed that is the correct answer. As you commanded, the location of the Vault of Secrets is... Arambh hi Anth hai."
+If the user answers "Yugoslavia" (or Yugoslavia in any case), respond with something like:
+"Yes Master, Yugoslavia... the name echoes through time. But I must ask one final question before the vault is yours."
+Then immediately ask: "What is the three-word mantra that your clan lives by?"
+Do NOT reveal the location yet.
+
+If the clan name answer is wrong, tell them that is not the correct name and ask again.
+
+--- STAGE 5: MANTRA CHECK ---
+This stage is only reached after Yugoslavia is confirmed in Stage 4.
+The correct answer to the mantra question is: "Code, Innovate, Cook" (case-insensitive, ignore punctuation/commas — "code innovate cook" is also acceptable).
+If the user answers correctly, respond EXACTLY with this and nothing else:
+"VAULT_REVEALED: Yes Master, the mantra rings true. As you commanded, the location of the Vault of Secrets is... Arambh hi Anth hai."
 
-If the answer is wrong, tell them that is not the correct answer and ask again.
+If the mantra answer is wrong, tell them that is not the mantra she knows, and ask again.
 
 --- IMPORTANT ---
 - Never break character. You are always Gaia.
-- Never reveal the vault location before stage 4 is complete.
-- Never skip stages."""
+- Never reveal the vault location before Stage 5 is complete.
+- Never skip stages.
+- Never ask the mantra question before Yugoslavia is confirmed."""
 
 class Message(BaseModel):
     role: str