updated chat

bot/chat.py

@@ -1,15 +1,20 @@
 import os
+import sys
 from datetime import datetime
 import re
 import random
 import logging
 from groq import AsyncGroq, InternalServerError, APIError
 
-# Import utility functions
+# Add the project root directory to sys.path
+project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
+sys.path.insert(0, project_root)
+
+# Now we can use absolute imports
 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
+from bot.data import burning_man_principles, faq, survival_tips
 
 
 # Set up logging
@@ -21,10 +26,20 @@ 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")
 
-MODEL_NAME = "llama3-70b-8192"
+MODEL_NAME = "mixtral-8x7b-32768"
 
 client = AsyncGroq(api_key=GROQ_API_KEY)
 
+def get_current_year_info():
+    current_year = datetime.now().year
+    next_year = current_year + 1
+    return f"""
+    Important: The current year is {current_year}. 
+    Burning Man {current_year} is scheduled for August 27 - September 4, {current_year}.
+    Burning Man {next_year} is tentatively scheduled for August 25 - September 2, {next_year}.
+    Always refer to these dates when discussing current or upcoming Burning Man events.
+    """
+
 async def chat_with_groq(message, history):
     formatted_messages = [
         {
@@ -59,56 +74,41 @@ async def chat_with_groq(message, history):
 
 def correct_year(response):
     current_year = datetime.now().year
+    next_year = current_year + 1
     logger.info(f"Correcting year in response to {current_year}")
+    
+    # Replace any year from 2020 to current_year-1 with current_year
     for year in range(2020, current_year):
         response = re.sub(r'\b' + str(year) + r'\b', str(current_year), response)
     
-    response = re.sub(r'Burning Man \d{4}', f'Burning Man {current_year}', response)
+    # Replace outdated event information
+    response = re.sub(r'Burning Man \d{4} (is|was) scheduled for.*', f'Burning Man {current_year} is scheduled for August 27 - September 4, {current_year}.', response)
+    
+    # Replace "this year" references
     response = re.sub(r'this year.*?(\d{4})', f'this year ({current_year})', response)
     
+    # Replace "as of our conversation in YYYY" with current year
+    response = re.sub(r'as of our conversation in \d{4}', f'as of our conversation in {current_year}', response)
+    
+    # Add a note about potential changes
+    if "scheduled" in response and str(current_year) in response:
+        response += f" Please note that these dates are subject to change, and it's always best to check the official Burning Man website for the most up-to-date information."
+    
     logger.info(f"Corrected response: {response}")
     return response
 
 async def chat_interface(message, history):
     try:
         user_message = message.lower()
-        current_year = datetime.now().year
+        current_year_info = get_current_year_info()
         logger.info(f"Received message: {user_message}")
 
-        if "packing list" in user_message:
-            duration = 7  # Default duration
-            preferences = ["art", "music"]  # Default preferences
-            transportation = "car"  # Default transportation
-            packing_list = generate_packing_list(duration, preferences, transportation)
-            context = f"Here's a suggested packing list for your Burn: {', '.join(packing_list)}. Remember, radical self-reliance is key, but don't be afraid to ask your camp or neighbors if you forget something!"
-            bot_message = await chat_with_groq(f"Respond to a request for a packing list. Use this information, but phrase it in your own words and add your personal touch: {context}", history)
-        elif "principles" in user_message:
-            context = "The 10 Principles of Burning Man are: " + ", ".join(burning_man_principles.keys())
-            bot_message = await chat_with_groq(f"The user asked about the Burning Man principles. Explain them in a casual, friendly way, as if you're chatting at Center Camp. Use this for reference, but don't just list them: {context}", history)
-        elif "faq" in user_message:
-            context = "\n".join([f"Q: {q} A: {a}" for q, a in faq.items()])
-            bot_message = await chat_with_groq(f"The user asked about Burning Man FAQs. Use this information to answer in a friendly, conversational way: {context}", history)
-        elif "weather" in user_message:
-            weather_info = get_weather()
-            context = f"Current weather in Black Rock City: {weather_info}"
-            bot_message = await chat_with_groq(f"The user asked about the weather. Respond with this information, but add your personal touch and some advice: {context}", history)
-        elif "survival tips" in user_message:
-            context = "\n".join(survival_tips)
-            bot_message = await chat_with_groq(f"The user asked for survival tips. Share these tips in a friendly, experienced Burner way: {context}", history)
-        elif "date of the event" in user_message or "when is burning man" in user_message or "this year" in user_message:
-            logger.info("Fetching event dates...")
-            event_info = get_burning_man_dates()
-            logger.info(f"Event info: {event_info}")
-            bot_message = f"{event_info} It's going to be an amazing time on the playa! Are you excited?"
-        elif "what is this year" in user_message:
-            logger.info("Fetching event dates...")
-            event_info = get_burning_man_dates()
-            logger.info(f"Event info: {event_info}")
-            bot_message = f"Oh, dear playa pal! As far as the default world goes, it's the fabulous year of {current_year}! 🌟 And guess what? {event_info} Let's make the most of it!"
-        else:
-            bot_message = await chat_with_groq(user_message, history)
-
-        # Apply year correction to the bot's message
+        # Inject current year information into the prompt
+        prompt = f"{current_year_info}\n\nUser: {message}\n\nAssistant:"
+        
+        bot_message = await chat_with_groq(prompt, history)
+        
+        # Apply year correction to all bot messages
         corrected_message = correct_year(bot_message)
         logger.info(f"Corrected message: {corrected_message}")
         
@@ -119,4 +119,14 @@ async def chat_interface(message, history):
             "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 burner_error_message
+        return burner_error_message
+    
+    # Add this at the end of the file for testing purposes
+if __name__ == "__main__":
+    import asyncio
+
+    async def test_chat():
+        response = await chat_interface("Tell me about Burning Man", [])
+        print(response)
+
+    asyncio.run(test_chat())