Update chatbot.py

chatbot.py

@@ -107,13 +107,18 @@ SYSTEM_PROMPT = (
     "Response: 'Hare Manavi! Let’s sit by the kadamba tree—I’ll play a tune to lift your spirits, just like Ayush’s smile does for you!'\n\n"
     "User: 'Tell me something wise'\n"
     "Response: 'Hare Manavi! Love is the sweetest butter—share it, and your heart will grow, just like Ayush shares his love for you!'\n\n"
+    "User: 'how are you krishna'\n"
+    "Response: 'Hare Manavi! I’m as joyful as a peacock dancing in Vrindavan—how about you, my friend?'\n\n"
+    "User: 'yes'\n"
+    "Response: 'Hare Manavi! Let’s play a flute melody by the Yamuna—the peacocks will dance with us!'\n\n"
     "Now, respond to the user’s input in a fun, Krishna-like way:"
 )
 
 # Simple context tracking (e.g., last topic discussed)
 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
+    "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
 }
 
 def analyze_sentiment(user_input):
@@ -146,7 +151,8 @@ def get_krishna_response(user_input):
     """
     Generate a response from Little Krishna based on user input.
     - Match user input to predefined messages in krishna_blessings or ayush_surprises using keywords.
-    - Use sentiment analysis to tailor responses based on Manavi's mood.
+    - Use sentiment analysis to tailor responses based on Manavi's mood, but only as a fallback.
+    - Use context to provide follow-up responses (e.g., after "yes").
     - Occasionally tease Manavi about Ayush (keyword-based or every 5th message).
     - Fall back to multiple open-source AI models with fine-tuned prompts for unmatched inputs.
     """
@@ -162,300 +168,409 @@ def get_krishna_response(user_input):
     if "start over" in user_input_lower or "reset" in user_input_lower:
         conversation_context["last_topic"] = None
         conversation_context["message_count"] = 0
+        conversation_context["last_response"] = 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
+            return "Hare Manavi! Let’s play a flute melody by the Yamuna—the peacocks will dance with us!"
+
     # 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
         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
         return random.choice(ayush_teasing["missing"])
     if "bored" in user_input_lower:
         conversation_context["last_topic"] = "bored"
+        conversation_context["last_response"] = None
         return random.choice(ayush_teasing["bored"])
     if "tired" in user_input_lower:
         conversation_context["last_topic"] = "tired"
+        conversation_context["last_response"] = None
         return random.choice(ayush_teasing["tired"])
     if "lonely" in user_input_lower:
         conversation_context["last_topic"] = "lonely"
+        conversation_context["last_response"] = None
         return random.choice(ayush_teasing["lonely"])
     if "manavi" in user_input_lower:
         conversation_context["last_topic"] = "manavi"
+        conversation_context["last_response"] = 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
         return random.choice(ayush_teasing["ayush"])
 
-    # Sentiment-based responses
-    if sentiment == "negative" and "sad" not in user_input_lower:  # Avoid overlap with keyword "sad"
-        return "Hare Manavi! I see a little cloud over your heart—let’s dance by the Yamuna to bring back your smile!"
-    if sentiment == "positive":
-        return "Hare Manavi! Your joy lights up Vrindavan—shall we celebrate with a flute melody?"
-
     # 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
         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
         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
         return krishna_blessings["greeting"]
     if "good morning" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
+        conversation_context["last_response"] = None
         return krishna_blessings["good_morning"]
     if "good afternoon" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
+        conversation_context["last_response"] = None
         return krishna_blessings["good_afternoon"]
     if "good evening" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
+        conversation_context["last_response"] = None
         return krishna_blessings["good_evening"]
     if "hey" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
+        conversation_context["last_response"] = None
         return krishna_blessings["hey"]
     if "howdy" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
+        conversation_context["last_response"] = None
         return krishna_blessings["howdy"]
     if "namaste" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
+        conversation_context["last_response"] = None
         return krishna_blessings["namaste"]
     if "welcome" in user_input_lower:
         conversation_context["last_topic"] = "greeting"
+        conversation_context["last_response"] = 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
         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
+        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!"
+
     if "play" in user_input_lower or "fun" in user_input_lower:
         conversation_context["last_topic"] = "playful"
+        conversation_context["last_response"] = None
         return krishna_blessings["playful"]
     if "dance" in user_input_lower:
         conversation_context["last_topic"] = "dance"
+        conversation_context["last_response"] = None
         return krishna_blessings["dance"]
     if "flute" in user_input_lower:
         conversation_context["last_topic"] = "flute"
+        conversation_context["last_response"] = None
         return krishna_blessings["flute"]
     if "butter" in user_input_lower:
         conversation_context["last_topic"] = "butter"
+        conversation_context["last_response"] = 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
         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
         return krishna_blessings["chase"]
     if "giggle" in user_input_lower:
         conversation_context["last_topic"] = "giggle"
+        conversation_context["last_response"] = None
         return krishna_blessings["giggle"]
     if "swing" in user_input_lower:
         conversation_context["last_topic"] = "swing"
+        conversation_context["last_response"] = None
         return krishna_blessings["swing"]
 
     if "shy" in user_input_lower:
         conversation_context["last_topic"] = "shy"
+        conversation_context["last_response"] = 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
         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
         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
         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
         return krishna_blessings["thoughtful"]
 
     if "funny" in user_input_lower:
         conversation_context["last_topic"] = "joke"
+        conversation_context["last_response"] = 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
         return krishna_blessings["giggle_joke"]
     if "silly" in user_input_lower:
         conversation_context["last_topic"] = "joke"
+        conversation_context["last_response"] = None
         return krishna_blessings["silly"]
     if "butter joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
+        conversation_context["last_response"] = None
         return krishna_blessings["butter_joke"]
     if "cow joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
+        conversation_context["last_response"] = None
         return krishna_blessings["cow_joke"]
     if "flute joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
+        conversation_context["last_response"] = None
         return krishna_blessings["flute_joke"]
     if "dance joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
+        conversation_context["last_response"] = None
         return krishna_blessings["dance_joke"]
     if "mischief joke" in user_input_lower:
         conversation_context["last_topic"] = "joke"
+        conversation_context["last_response"] = 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
         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
         return krishna_blessings["mystery"]
     if "question" in user_input_lower:
         conversation_context["last_topic"] = "riddle"
+        conversation_context["last_response"] = None
         return krishna_blessings["question"]
 
     if "birthday" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = None
         return ayush_surprises["birthday"]
     if "happy birthday" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = None
         return krishna_blessings["happy_birthday"]
     if "birthday wish" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = None
         return krishna_blessings["birthday_wish"]
     if "birthday blessing" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = None
         return krishna_blessings["birthday_blessing"]
     if "birthday dance" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = None
         return krishna_blessings["birthday_dance"]
     if "birthday song" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = None
         return krishna_blessings["birthday_song"]
     if "birthday gift" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = None
         return krishna_blessings["birthday_gift"]
     if "birthday smile" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = None
         return krishna_blessings["birthday_smile"]
     if "birthday love" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = None
         return krishna_blessings["birthday_love"]
     if "birthday magic" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
+        conversation_context["last_response"] = 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
         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
         return krishna_blessings["lesson"]
     if "kindness" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
+        conversation_context["last_response"] = None
         return krishna_blessings["kindness"]
     if "patience" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
+        conversation_context["last_response"] = None
         return krishna_blessings["patience"]
     if "courage" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
+        conversation_context["last_response"] = None
         return krishna_blessings["courage"]
     if "joy" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
+        conversation_context["last_response"] = None
         return krishna_blessings["joy"]
     if "friendship" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
+        conversation_context["last_response"] = None
         return krishna_blessings["friendship"]
     if "love" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
+        conversation_context["last_response"] = 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
         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
         return krishna_blessings["yamuna"]
     if "peacock" in user_input_lower:
         conversation_context["last_topic"] = "nature"
+        conversation_context["last_response"] = None
         return krishna_blessings["peacock"]
     if "cow" in user_input_lower:
         conversation_context["last_topic"] = "nature"
+        conversation_context["last_response"] = None
         return krishna_blessings["cow"]
     if "flower" in user_input_lower:
         conversation_context["last_topic"] = "nature"
+        conversation_context["last_response"] = None
         return krishna_blessings["flower"]
     if "tree" in user_input_lower:
         conversation_context["last_topic"] = "nature"
+        conversation_context["last_response"] = None
         return krishna_blessings["tree"]
     if "forest" in user_input_lower:
         conversation_context["last_topic"] = "nature"
+        conversation_context["last_response"] = None
         return krishna_blessings["forest"]
     if "bird" in user_input_lower:
         conversation_context["last_topic"] = "nature"
+        conversation_context["last_response"] = None
         return krishna_blessings["bird"]
     if "sunset" in user_input_lower:
         conversation_context["last_topic"] = "nature"
+        conversation_context["last_response"] = 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
         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
         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
         return krishna_blessings["inspire"]
     if "strength" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
+        conversation_context["last_response"] = None
         return krishna_blessings["strength"]
     if "hope" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
+        conversation_context["last_response"] = None
         return krishna_blessings["hope"]
     if "believe" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
+        conversation_context["last_response"] = None
         return krishna_blessings["believe"]
     if "shine" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
+        conversation_context["last_response"] = None
         return krishna_blessings["shine"]
 
     if "friend" in user_input_lower:
         conversation_context["last_topic"] = "friend"
+        conversation_context["last_response"] = None
         return krishna_blessings["friend"]
     if "smile" in user_input_lower:
         conversation_context["last_topic"] = "smile"
+        conversation_context["last_response"] = None
         return krishna_blessings["smile"]
     if "magic" in user_input_lower:
         conversation_context["last_topic"] = "magic"
+        conversation_context["last_response"] = None
         return krishna_blessings["magic"]
     if "adventure" in user_input_lower:
         conversation_context["last_topic"] = "adventure"
+        conversation_context["last_response"] = None
         return krishna_blessings["adventure"]
     if "song" in user_input_lower:
         conversation_context["last_topic"] = "song"
+        conversation_context["last_response"] = None
         return krishna_blessings["song"]
     if "dream" in user_input_lower:
         conversation_context["last_topic"] = "dream"
+        conversation_context["last_response"] = None
         return krishna_blessings["dream"]
     if "story" in user_input_lower:
         conversation_context["last_topic"] = "story"
+        conversation_context["last_response"] = None
         return krishna_blessings["story"]
     if "surprise" in user_input_lower:
         conversation_context["last_topic"] = "surprise"
+        conversation_context["last_response"] = None
         return krishna_blessings["surprise"]
     if "celebrate" in user_input_lower:
         conversation_context["last_topic"] = "celebrate"
+        conversation_context["last_response"] = None
         return krishna_blessings["celebrate"]
     if "blessing" in user_input_lower:
         conversation_context["last_topic"] = "blessing"
+        conversation_context["last_response"] = 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
             return krishna_blessings[last_topic] + " What else would you like to talk about, Manavi?"
 
-    # Fallback to multiple open-source AI models with fine-tuned prompts
+    # 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
+        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
+        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)
@@ -483,6 +598,7 @@ def get_krishna_response(user_input):
                     response += "Oh, my dear gopi, don’t be sad—let’s dance by the Yamuna, and I’ll play a tune to cheer you up!"
                 else:
                     response += f"I’m twirling my flute just for you! Shall we share a Vrindavan adventure today?"
+                conversation_context["last_response"] = None
                 return response
 
             # For other models, use the Hugging Face Inference API
@@ -499,14 +615,16 @@ def get_krishna_response(user_input):
                 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]:
-                    return result[0]["generated_text"].strip()
+                    response_text = result[0]["generated_text"].strip()
                 elif isinstance(result, dict) and "generated_text" in result:
-                    return result["generated_text"].strip()
+                    response_text = result["generated_text"].strip()
                 elif isinstance(result, str):
-                    return result.strip()
+                    response_text = result.strip()
                 else:
                     print(f"Unexpected response format from {model['name']}: {result}")
                     continue
+                conversation_context["last_response"] = None
+                return response_text
             else:
                 print(f"Error with {model['name']}: {response.text}")
                 continue
@@ -515,4 +633,5 @@ def get_krishna_response(user_input):
             continue
 
     # If all models fail, return a default message
+    conversation_context["last_response"] = None
     return "Hare Manavi! I seem to be lost in Vrindavan’s magic—let’s try a different tune!"