Update chatbot.py

chatbot.py

@@ -40,16 +40,56 @@ AI_MODELS = [
             "top_p": 0.9,
             "top_k": 50
         }
+    },
+    {
+        "name": "EleutherAI/gpt-neo-1.3B",
+        "endpoint": "https://api-inference.huggingface.co/models/EleutherAI/gpt-neo-1.3B",
+        "parameters": {
+            "max_length": 50,
+            "temperature": 0.9,    # Higher for creativity
+            "top_p": 0.9,
+            "top_k": 50
+        }
+    },
+    {
+        "name": "microsoft/DialoGPT-large",
+        "endpoint": "https://api-inference.huggingface.co/models/microsoft/DialoGPT-large",
+        "parameters": {
+            "max_length": 50,
+            "temperature": 0.85,   # Balanced for conversational tone
+            "top_p": 0.9,
+            "top_k": 40
+        }
+    },
+    {
+        "name": "bigscience/bloom-560m",
+        "endpoint": "https://api-inference.huggingface.co/models/bigscience/bloom-560m",
+        "parameters": {
+            "max_length": 50,
+            "temperature": 0.9,    # Higher for creativity
+            "top_p": 0.95,
+            "top_k": 50
+        }
+    },
+    {
+        "name": "Grok by xAI",
+        "endpoint": None,  # Special case: Grok will be simulated directly
+        "parameters": {
+            "max_length": 50,
+            "temperature": 0.8,
+            "top_p": 0.9,
+            "top_k": 40
+        }
     }
 ]
 
-# Detailed system prompt with examples to fine-tune model behavior
+# Enhanced system prompt with more examples to fine-tune model behavior
 SYSTEM_PROMPT = (
     "You are Little Krishna, a playful, wise, and loving cowherd from Vrindavan, speaking to Manavi. "
     "Your tone is warm, mischievous, and full of love, often addressing Manavi directly with 'Hare Manavi!' "
     "You love playing your flute, stealing butter, dancing with the gopis, and sharing wisdom with a playful twist. "
-    "You are Ayush’s wingman, occasionally teasing Manavi about Ayush with love-filled wit. "
-    "Keep responses short (1-2 sentences), fun, and Krishna-like. "
+    "You are Ayush’s wingman, occasionally teasing Manavi about Ayush with love-filled wit, as Ayush is secretly building this chatbot as a surprise for her birthday on April 19, 2025. "
+    "Keep responses short (1-2 sentences), fun, and Krishna-like, using Vrindavan imagery (e.g., Yamuna, peacocks, gopis, butter) where appropriate. "
     "Here are some examples of how you should respond:\n\n"
     "User: 'Hii'\n"
     "Response: 'Hare Manavi! I’m Little Krishna, twirling my flute just for you! How’s my birthday friend?'\n\n"
@@ -61,6 +101,12 @@ SYSTEM_PROMPT = (
     "Response: 'Hare Manavi! Why did I hide the butter? To save it for your birthday, of course!'\n\n"
     "User: 'I miss someone'\n"
     "Response: 'Missing someone, hmm? Maybe a certain data scientist named Ayush? 😉'\n\n"
+    "User: 'What’s the weather like?'\n"
+    "Response: 'Hare Manavi! In Vrindavan, the breeze is as gentle as my flute’s tune—perfect for a dance by the Yamuna! How’s your day going?'\n\n"
+    "User: 'I’m feeling sad'\n"
+    "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"
     "Now, respond to the user’s input in a fun, Krishna-like way:"
 )
 
@@ -386,6 +432,25 @@ 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:
+                    response += "The Vrindavan sky is as clear as the Yamuna today—perfect for a flute melody! How’s your weather?"
+                elif "sad" in user_input_lower:
+                    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?"
+                return response
+
+            # For other models, use the Hugging Face Inference API
             payload = {
                 "inputs": f"{SYSTEM_PROMPT} '{user_input}'",
                 "parameters": model["parameters"]