Update chatbot.py

chatbot.py

@@ -1,7 +1,8 @@
 import os
 import requests
+import random
 from dotenv import load_dotenv
-from messages import krishna_blessings
+from messages import krishna_blessings, ayush_teasing
 from ayush_messages import ayush_surprises
 
 # Load environment variables (Hugging Face Space secrets)
@@ -10,7 +11,8 @@ HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
 
 # Simple context tracking (e.g., last topic discussed)
 conversation_context = {
-    "last_topic": None  # Store the last keyword matched (e.g., "birthday", "riddle")
+    "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
 }
 
 def get_krishna_response(user_input):
@@ -18,16 +20,53 @@ 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 context to provide more coherent responses.
+    - Occasionally tease Manavi about Ayush (keyword-based or every 5th message).
     - Fall back to Hugging Face's Inference API for unmatched inputs.
     """
     user_input_lower = user_input.lower().strip()
 
+    # Increment message count
+    conversation_context["message_count"] += 1
+
     # Reset context if user starts a new conversation
     if "start over" in user_input_lower or "reset" in user_input_lower:
         conversation_context["last_topic"] = None
+        conversation_context["message_count"] = 0
         return "Hare Manavi! Let’s start a new adventure in Vrindavan—what would you like to talk about?"
 
-    # Greetings
+    # Check for Ayush-teasing triggers (keyword-based)
+    if "joke" in user_input_lower:
+        conversation_context["last_topic"] = "joke"
+        # 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"
+        return random.choice(ayush_teasing["missing"])
+    if "bored" in user_input_lower:
+        conversation_context["last_topic"] = "bored"
+        return random.choice(ayush_teasing["bored"])
+    if "tired" in user_input_lower:
+        conversation_context["last_topic"] = "tired"
+        return random.choice(ayush_teasing["tired"])
+    if "lonely" in user_input_lower:
+        conversation_context["last_topic"] = "lonely"
+        return random.choice(ayush_teasing["lonely"])
+    if "manavi" in user_input_lower:
+        conversation_context["last_topic"] = "manavi"
+        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"
+        return random.choice(ayush_teasing["ayush"])
+
+    # 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()))
+        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"
         return krishna_blessings["greeting"]
@@ -53,12 +92,10 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "greeting"
         return krishna_blessings["welcome"]
 
-    # Identity Questions (e.g., "Who are you")
     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:
         conversation_context["last_topic"] = "identity"
         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?"
 
-    # Playful Responses
     if "play" in user_input_lower or "fun" in user_input_lower:
         conversation_context["last_topic"] = "playful"
         return krishna_blessings["playful"]
@@ -84,7 +121,6 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "swing"
         return krishna_blessings["swing"]
 
-    # Shy or Quiet Moments
     if "shy" in user_input_lower:
         conversation_context["last_topic"] = "shy"
         return krishna_blessings["shy"]
@@ -101,10 +137,9 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "thoughtful"
         return krishna_blessings["thoughtful"]
 
-    # Jokes
-    if "joke" in user_input_lower or "funny" in user_input_lower:
+    if "funny" in user_input_lower:
         conversation_context["last_topic"] = "joke"
-        return krishna_blessings["joke"]
+        return krishna_blessings["funny"]
     if "laugh" in user_input_lower or "giggle" in user_input_lower:
         conversation_context["last_topic"] = "joke"
         return krishna_blessings["giggle_joke"]
@@ -127,7 +162,6 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "joke"
         return krishna_blessings["mischief_joke"]
 
-    # Riddles
     if "riddle" in user_input_lower or "puzzle" in user_input_lower:
         conversation_context["last_topic"] = "riddle"
         return krishna_blessings["riddle"]
@@ -138,7 +172,6 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "riddle"
         return krishna_blessings["question"]
 
-    # Birthday Wishes
     if "birthday" in user_input_lower:
         conversation_context["last_topic"] = "birthday"
         return ayush_surprises["birthday"]
@@ -170,7 +203,6 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "birthday"
         return krishna_blessings["birthday_magic"]
 
-    # Wisdom and Advice
     if "wisdom" in user_input_lower or "advice" in user_input_lower:
         conversation_context["last_topic"] = "wisdom"
         return krishna_blessings["wisdom"]
@@ -196,7 +228,6 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "wisdom"
         return krishna_blessings["love"]
 
-    # Nature and Vrindavan
     if "nature" in user_input_lower or "vrindavan" in user_input_lower:
         conversation_context["last_topic"] = "nature"
         return krishna_blessings["nature"]
@@ -225,7 +256,6 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "nature"
         return krishna_blessings["sunset"]
 
-    # Encouragement
     if "encourage" in user_input_lower or "cheer" in user_input_lower:
         conversation_context["last_topic"] = "encourage"
         return krishna_blessings["encourage"]
@@ -248,7 +278,6 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "encourage"
         return krishna_blessings["shine"]
 
-    # Miscellaneous
     if "friend" in user_input_lower:
         conversation_context["last_topic"] = "friend"
         return krishna_blessings["friend"]
@@ -280,7 +309,6 @@ def get_krishna_response(user_input):
         conversation_context["last_topic"] = "blessing"
         return krishna_blessings["blessing"]
 
-    # Context-Based Responses (if no direct match, use the last topic)
     if conversation_context["last_topic"]:
         last_topic = conversation_context["last_topic"]
         if last_topic in krishna_blessings:
@@ -294,8 +322,8 @@ def get_krishna_response(user_input):
     payload = {
         "inputs": f"You are Little Krishna, playful and wise. Respond to this in a fun, Krishna-like way: '{user_input}'",
         "parameters": {
-            "max_length": 50,  # Reduced for shorter, more playful responses
-            "temperature": 0.7  # Slightly lower for more coherent responses
+            "max_length": 50,
+            "temperature": 0.7
         }
     }
     try: