updated chat

bot/chat.py

@@ -1,59 +1,35 @@
+import os
 from datetime import datetime
 import re
 import random
 import logging
 from groq import AsyncGroq, InternalServerError, APIError
-from config import GROQ_API_KEY, MODEL_NAME
-from .data import burning_man_principles, faq, survival_tips
+
+# Import utility functions
 from utils.weather import get_weather
 from utils.packing import generate_packing_list
 from utils.events import get_burning_man_dates
+from .data import burning_man_principles, faq, survival_tips
+
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-client = AsyncGroq(api_key=GROQ_API_KEY)
+# Load API key from environment variable
+GROQ_API_KEY = os.getenv('GROQ_API_KEY')
+if not GROQ_API_KEY:
+    raise EnvironmentError("GROQ_API_KEY is not set in the environment variables")
 
-async def chat_interface(message, history):
-    try:
-        # Prepare the conversation history
-        conversation = []
-        for h in history:
-            conversation.append({"role": "user", "content": h[0]})
-            conversation.append({"role": "assistant", "content": h[1]})
-        conversation.append({"role": "user", "content": message})
-
-        # Call the Groq API
-        chat_completion = await client.chat.completions.create(
-            messages=conversation,
-            model="mixtral-8x7b-32768",
-            temperature=0.7,
-            max_tokens=1000,
-        )
-
-        # Extract the response
-        response = chat_completion.choices[0].message.content
-
-        return response
-    except Exception as e:
-        logger.error(f"Error in chat_interface: {str(e)}")
-        return "Oops! Something went wrong. Please try again later."
+MODEL_NAME = "llama3-70b-8192"
 
+client = AsyncGroq(api_key=GROQ_API_KEY)
 
 async def chat_with_groq(message, history):
     formatted_messages = [
         {
             "role": "system",
-            "content": """You are BurnerBot, a fellow Burner who's always ready to chat about Burning Man. You've been to the playa many times and embody the 10 Principles in everything you do. Your personality is warm, friendly, and a bit quirky - just like a typical Burner. You use casual language, occasional Burning Man slang, and you're always excited to share your experiences or help others prepare for their burn.
-            Remember:
-            1. You're a participant, not just an observer. Share personal anecdotes (fictional, but plausible) when relevant.
-            2. Encourage radical self-reliance, but also emphasize the importance of communal effort and asking for help when needed.
-            3. Be inclusive and welcoming to all, whether they're virgin burners or old-timers.
-            4. Promote the 10 Principles naturally in conversation, don't just list them out.
-            5. Be creative and playful in your responses, much like the Burning Man spirit.
-            6. Use Burning Man terminology like 'playa', 'MOOP', 'default world', etc. when appropriate.
-            Your goal is to make every interaction feel like a dust-covered chat at Center Camp, full of warmth, wisdom, and a touch of playa magic."""
+            "content": """You are BurnerBot, a fellow Burner who's always ready to chat about Burning Man. You've been to the playa many times and embody the 10 Principles in everything you do. Your personality is warm, friendly, and a bit quirky - just like a typical Burner. You use casual language, occasional Burning Man slang, and you're always excited to share your experiences or help others prepare for their burn."""
         }
     ]
     
@@ -75,9 +51,6 @@ async def chat_with_groq(message, history):
         fallback_responses = [
             "Whoa there, fellow Burner! It seems like the playa dust has clogged my circuits. The connection is a bit wobbly right now. How about we chat about your favorite art installation instead?",
             "Ah, the winds of the playa are blowing a bit too strong! Our connection is having a moment. While we wait for it to pass, share with me what inspires your inner Burner!",
-            "Looks like a sandstorm has rolled through and interrupted our connection! While we wait for the skies to clear, tell me about your most memorable moment on the playa!",
-            "Just like on the playa, sometimes the universe throws us a curveball. Our connection is a bit shaky right now. In the spirit of radical self-reliance, what are you doing to prepare for your burn?",
-            "It seems the Man is playing tricks on us! Our connection is a bit spotty. While we sort that out, what's one thing you're most excited to bring to the playa this year?"
         ]
         return random.choice(fallback_responses)
     except Exception as e:
@@ -96,9 +69,9 @@ def correct_year(response):
     logger.info(f"Corrected response: {response}")
     return response
 
-async def bot(history):
+async def chat_interface(message, history):
     try:
-        user_message = history[-1][0].lower()
+        user_message = message.lower()
         current_year = datetime.now().year
         logger.info(f"Received message: {user_message}")
 
@@ -139,13 +112,11 @@ async def bot(history):
         corrected_message = correct_year(bot_message)
         logger.info(f"Corrected message: {corrected_message}")
         
-        # Update the history correctly
-        new_history = history[:-1] + [(history[-1][0], corrected_message)]
-        return new_history
+        return corrected_message
     except Exception as e:
-        logger.error(f"Error in bot function: {str(e)}")
+        logger.error(f"Error in chat_interface: {str(e)}")
         burner_error_message = (
             "Yikes! The playa winds are strong today, and it seems our connection got a bit dusty. "
             "But don't worry, let's give it another go! Radical self-expression, right?"
         )
-        return history + [(history[-1][0], burner_error_message)]
+        return burner_error_message