Update chatbot.py

chatbot.py

@@ -118,9 +118,19 @@ SYSTEM_PROMPT = (
 conversation_context = {
     "last_topic": None,  # Store the last keyword matched (e.g., "birthday", "riddle")
     "message_count": 0,  # Track the number of messages to trigger Ayush-teasing every 5th message
-    "last_response": None  # Track the last response to avoid repetition and enable follow-ups
+    "last_response": None,  # Track the last response to avoid repetition and enable follow-ups
+    "last_activity": None  # Track ongoing activities (e.g., "dancing", "exploring")
 }
 
+# List of generic positive responses to cycle through for affirmations when no activity is in progress
+GENERIC_POSITIVE_RESPONSES = [
+    "Hare Manavi! Wonderful—let’s make today as magical as Vrindavan’s sunsets!",
+    "Hare Manavi! Great—shall we chase the butterflies by the Yamuna next?",
+    "Hare Manavi! Perfect—let’s share some butter under the kadamba tree!",
+    "Hare Manavi! Lovely—how about a little dance with the gopis now?",
+    "Hare Manavi! Splendid—let’s listen to the birds sing by the river!"
+]
+
 def analyze_sentiment(user_input):
     """Analyze the sentiment of the user's input using a sentiment analysis model."""
     headers = {
@@ -169,409 +179,527 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = None
         conversation_context["message_count"] = 0
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return "Hare Manavi! Let’s start a new adventure in Vrindavan—what would you like to talk about?"
 
     # Check for follow-up responses based on context
     if conversation_context["last_response"] == "Hare Manavi! Your joy lights up Vrindavan—shall we celebrate with a flute melody?":
         if "yes" in user_input_lower or "okay" in user_input_lower or "sure" in user_input_lower:
-            conversation_context["last_response"] = None  # Reset to avoid infinite loop
+            conversation_context["last_response"] = None
+            conversation_context["last_activity"] = "dancing"
             return "Hare Manavi! Let’s play a flute melody by the Yamuna—the peacocks will dance with us!"
 
+    # Handle ongoing activities
+    if conversation_context["last_activity"]:
+        if conversation_context["last_activity"] == "dancing":
+            if "yes" in user_input_lower or "okay" in user_input_lower or "sure" in user_input_lower:
+                conversation_context["last_response"] = None
+                conversation_context["last_activity"] = "dancing"
+                return "Hare Manavi! The gopis have joined us—let’s twirl under the moonlight by the Yamuna!"
+            if "how" in user_input_lower or "but how" in user_input_lower:
+                conversation_context["last_response"] = None
+                return "Hare Manavi! We’ll sway to my flute’s tune, letting the rhythm guide our steps—what a joyful dance it’ll be!"
+        elif conversation_context["last_activity"] == "exploring":
+            if "yes" in user_input_lower or "okay" in user_input_lower or "sure" in user_input_lower:
+                conversation_context["last_response"] = None
+                return "Hare Manavi! We found a pot of creamy butter—let’s share it under the stars by the Yamuna!"
+            if "how" in user_input_lower or "but how" in user_input_lower:
+                conversation_context["last_response"] = None
+                return "Hare Manavi! We’ll tiptoe through the trees, listening for the gopis’ laughter to guide us to their hidden butter pots!"
+
+    # Check for exploration triggers
+    if "explore" in user_input_lower or "adventure" in user_input_lower:
+        conversation_context["last_topic"] = "adventure"
+        conversation_context["last_response"] = None
+        conversation_context["last_activity"] = "exploring"
+        return "Hare Manavi! Let’s sneak through Vrindavan’s forests to find hidden butter pots—what a mischievous delight!"
+
     # Check for Ayush-teasing triggers (keyword-based)
     if "joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         conversation_context["last_response"] = None
-        # Randomly decide between a Krishna joke and an Ayush-teasing joke
+        conversation_context["last_activity"] = None
         if random.choice([True, False]):
             return random.choice(ayush_teasing["joke"])
         return krishna_blessings["joke"]
     if "i miss" in user_input_lower or "missing" in user_input_lower:
         conversation_context["last_topic"] = "missing"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return random.choice(ayush_teasing["missing"])
     if "bored" in user_input_lower:
         conversation_context["last_topic"] = "bored"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return random.choice(ayush_teasing["bored"])
     if "tired" in user_input_lower:
         conversation_context["last_topic"] = "tired"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return random.choice(ayush_teasing["tired"])
     if "lonely" in user_input_lower:
         conversation_context["last_topic"] = "lonely"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return random.choice(ayush_teasing["lonely"])
     if "manavi" in user_input_lower:
         conversation_context["last_topic"] = "manavi"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return random.choice(ayush_teasing["manavi"])
     if "ayush" in user_input_lower or "krishna talk about ayush" in user_input_lower:
         conversation_context["last_topic"] = "ayush"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return random.choice(ayush_teasing["ayush"])
 
     # Trigger for "chat with you"
     if "chat with you" in user_input_lower or "want to chat" in user_input_lower:
         conversation_context["last_topic"] = "chat_with_you"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["chat_with_you"]
 
     # Every 5th message, randomly trigger an Ayush-teasing message (if no keyword match)
     if conversation_context["message_count"] % 5 == 0:
-        # Randomly select a category from ayush_teasing
         category = random.choice(list(ayush_teasing.keys()))
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return random.choice(ayush_teasing[category])
 
     # Existing keyword mappings for krishna_blessings and ayush_surprises
     if "hello" in user_input_lower or "hi" in user_input_lower or "hii" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["greeting"]
     if "good morning" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["good_morning"]
     if "good afternoon" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["good_afternoon"]
     if "good evening" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["good_evening"]
     if "hey" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["hey"]
     if "howdy" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["howdy"]
     if "namaste" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["namaste"]
     if "welcome" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["welcome"]
 
     if "who are you" in user_input_lower or "what are you" in user_input_lower or "tell me about yourself" in user_input_lower or "what are you doing" in user_input_lower:
         conversation_context["last_topic"] = "identity"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return "Hare Manavi! I’m Little Krishna, the playful cowherd of Vrindavan! I love playing my flute, stealing butter, and dancing with the gopis. What would you like to do with me today?"
 
     if "how are you" in user_input_lower:
         conversation_context["last_topic"] = "how_are_you"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return "Hare Manavi! I’m as joyful as a peacock dancing in Vrindavan—how about you, my friend?"
 
     if "yes" in user_input_lower or "okay" in user_input_lower or "sure" in user_input_lower:
-        # If no context for "yes", provide a generic positive response
         conversation_context["last_topic"] = "yes"
         conversation_context["last_response"] = None
-        return "Hare Manavi! Wonderful—let’s make today as magical as Vrindavan’s sunsets!"
+        # Cycle through generic positive responses if no activity is in progress
+        if not conversation_context["last_activity"]:
+            response = random.choice(GENERIC_POSITIVE_RESPONSES)
+            # Rotate the list to avoid repetition
+            GENERIC_POSITIVE_RESPONSES.append(GENERIC_POSITIVE_RESPONSES.pop(0))
+            return response
 
     if "play" in user_input_lower or "fun" in user_input_lower:
         conversation_context["last_topic"] = "playful"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["playful"]
     if "dance" in user_input_lower:
         conversation_context["last_topic"] = "dance"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = "dancing"
         return krishna_blessings["dance"]
     if "flute" in user_input_lower:
         conversation_context["last_topic"] = "flute"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["flute"]
     if "butter" in user_input_lower:
         conversation_context["last_topic"] = "butter"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["butter"]
     if "mischief" in user_input_lower or "prank" in user_input_lower:
         conversation_context["last_topic"] = "mischief"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["mischief"]
     if "chase" in user_input_lower or "run" in user_input_lower:
         conversation_context["last_topic"] = "chase"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["chase"]
     if "giggle" in user_input_lower:
         conversation_context["last_topic"] = "giggle"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["giggle"]
     if "swing" in user_input_lower:
         conversation_context["last_topic"] = "swing"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["swing"]
 
     if "shy" in user_input_lower:
         conversation_context["last_topic"] = "shy"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["shy"]
     if "quiet" in user_input_lower or "calm" in user_input_lower:
         conversation_context["last_topic"] = "quiet"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["quiet"]
     if "peace" in user_input_lower or "serene" in user_input_lower:
         conversation_context["last_topic"] = "peace"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["peace"]
     if "still" in user_input_lower or "gentle" in user_input_lower:
         conversation_context["last_topic"] = "still"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["still"]
     if "thoughtful" in user_input_lower or "reflect" in user_input_lower:
         conversation_context["last_topic"] = "thoughtful"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["thoughtful"]
 
     if "funny" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["funny"]
     if "laugh" in user_input_lower or "giggle" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["giggle_joke"]
     if "silly" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["silly"]
     if "butter joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["butter_joke"]
     if "cow joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["cow_joke"]
     if "flute joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["flute_joke"]
     if "dance joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["dance_joke"]
     if "mischief joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["mischief_joke"]
 
     if "riddle" in user_input_lower or "puzzle" in user_input_lower:
         conversation_context["last_topic"] = "riddle"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["riddle"]
     if "mystery" in user_input_lower or "enigma" in user_input_lower:
         conversation_context["last_topic"] = "riddle"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["mystery"]
     if "question" in user_input_lower:
         conversation_context["last_topic"] = "riddle"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["question"]
 
     if "birthday" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return ayush_surprises["birthday"]
     if "happy birthday" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["happy_birthday"]
     if "birthday wish" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["birthday_wish"]
     if "birthday blessing" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["birthday_blessing"]
     if "birthday dance" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = "dancing"
         return krishna_blessings["birthday_dance"]
     if "birthday song" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["birthday_song"]
     if "birthday gift" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["birthday_gift"]
     if "birthday smile" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["birthday_smile"]
     if "birthday love" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["birthday_love"]
     if "birthday magic" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["birthday_magic"]
 
     if "wisdom" in user_input_lower or "advice" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["wisdom"]
     if "lesson" in user_input_lower or "truth" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["lesson"]
     if "kindness" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["kindness"]
     if "patience" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["patience"]
     if "courage" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["courage"]
     if "joy" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["joy"]
     if "friendship" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["friendship"]
     if "love" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["love"]
 
     if "nature" in user_input_lower or "vrindavan" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["nature"]
     if "yamuna" in user_input_lower or "river" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["yamuna"]
     if "peacock" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["peacock"]
     if "cow" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["cow"]
     if "flower" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["flower"]
     if "tree" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["tree"]
     if "forest" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = "exploring"
         return krishna_blessings["forest"]
     if "bird" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["bird"]
     if "sunset" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["sunset"]
 
     if "encourage" in user_input_lower or "cheer" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["encourage"]
     if "support" in user_input_lower or "uplift" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["support"]
     if "inspire" in user_input_lower or "motivate" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["inspire"]
     if "strength" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["strength"]
     if "hope" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["hope"]
     if "believe" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["believe"]
     if "shine" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["shine"]
 
     if "friend" in user_input_lower:
         conversation_context["last_topic"] = "friend"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["friend"]
     if "smile" in user_input_lower:
         conversation_context["last_topic"] = "smile"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["smile"]
     if "magic" in user_input_lower:
         conversation_context["last_topic"] = "magic"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["magic"]
     if "adventure" in user_input_lower:
         conversation_context["last_topic"] = "adventure"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = "exploring"
         return krishna_blessings["adventure"]
     if "song" in user_input_lower:
         conversation_context["last_topic"] = "song"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["song"]
     if "dream" in user_input_lower:
         conversation_context["last_topic"] = "dream"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["dream"]
     if "story" in user_input_lower:
         conversation_context["last_topic"] = "story"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["story"]
     if "surprise" in user_input_lower:
         conversation_context["last_topic"] = "surprise"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["surprise"]
     if "celebrate" in user_input_lower:
         conversation_context["last_topic"] = "celebrate"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["celebrate"]
     if "blessing" in user_input_lower:
         conversation_context["last_topic"] = "blessing"
         conversation_context["last_response"] = None
+        conversation_context["last_activity"] = None
         return krishna_blessings["blessing"]
 
     if conversation_context["last_topic"]:
         last_topic = conversation_context["last_topic"]
         if last_topic in krishna_blessings:
             conversation_context["last_response"] = None
+            conversation_context["last_activity"] = None
             return krishna_blessings[last_topic] + " What else would you like to talk about, Manavi?"
 
     # Sentiment-based responses (only as a fallback, and avoid repetition)
     if sentiment == "negative" and "sad" not in user_input_lower and conversation_context["last_response"] != "Hare Manavi! I see a little cloud over your heart—let’s dance by the Yamuna to bring back your smile!":
         response = "Hare Manavi! I see a little cloud over your heart—let’s dance by the Yamuna to bring back your smile!"
         conversation_context["last_response"] = response
+        conversation_context["last_activity"] = "dancing"
         return response
     if sentiment == "positive" and conversation_context["last_response"] != "Hare Manavi! Your joy lights up Vrindavan—shall we celebrate with a flute melody?":
         response = "Hare Manavi! Your joy lights up Vrindavan—shall we celebrate with a flute melody?"
         conversation_context["last_response"] = response
+        conversation_context["last_activity"] = None
         return response
 
     # Fallback to multiple open-source AI models if no keywords match
-    # Shuffle the models to try them in random order
     models_to_try = AI_MODELS.copy()
     random.shuffle(models_to_try)
 
@@ -582,14 +710,11 @@ def get_krishna_response(user_input):
 
     for model in models_to_try:
         try:
-            # Special case for Grok by xAI (simulated directly)
             if model["name"] == "Grok by xAI":
-                # Simulate Grok's response (I, Grok, will generate the response directly)
                 response = (
                     f"Hare Manavi! I’m Little Krishna, speaking through Grok by xAI. "
                     f"Let me answer in my playful way: "
                 )
-                # Generate a Krishna-like response based on the input
                 if "color" in user_input_lower:
                     response += "I love the golden yellow of Vrindavan’s butter—it’s as sweet as your smile! What’s your favorite color?"
                 elif "weather" in user_input_lower:
@@ -599,9 +724,9 @@ def get_krishna_response(user_input):
                 else:
                     response += f"I’m twirling my flute just for you! Shall we share a Vrindavan adventure today?"
                 conversation_context["last_response"] = None
+                conversation_context["last_activity"] = "exploring" if "adventure" in response else None
                 return response
 
-            # For other models, use the Hugging Face Inference API
             payload = {
                 "inputs": f"{SYSTEM_PROMPT} '{user_input}'",
                 "parameters": model["parameters"]
@@ -613,7 +738,6 @@ def get_krishna_response(user_input):
             )
             if response.status_code == 200:
                 result = response.json()
-                # Handle different response formats based on the model
                 if isinstance(result, list) and len(result) > 0 and "generated_text" in result[0]:
                     response_text = result[0]["generated_text"].strip()
                 elif isinstance(result, dict) and "generated_text" in result:
@@ -624,6 +748,7 @@ def get_krishna_response(user_input):
                     print(f"Unexpected response format from {model['name']}: {result}")
                     continue
                 conversation_context["last_response"] = None
+                conversation_context["last_activity"] = None
                 return response_text
             else:
                 print(f"Error with {model['name']}: {response.text}")
@@ -632,6 +757,6 @@ def get_krishna_response(user_input):
             print(f"Error connecting to {model['name']}: {str(e)}")
             continue
 
-    # If all models fail, return a default message
     conversation_context["last_response"] = None
+    conversation_context["last_activity"] = None
     return "Hare Manavi! I seem to be lost in Vrindavan’s magic—let’s try a different tune!"