updated chat

bot/chat.py

@@ -2,11 +2,11 @@ from datetime import datetime
 import re
 import random
 from groq import AsyncGroq, InternalServerError, APIError
-from config import GROQ_API_KEY, MODEL_NAME
-from bot.data import burning_man_principles, faq, survival_tips
-from utils.weather import get_weather
-from utils.packing import generate_packing_list
-from utils.events import get_burning_man_dates
+from ..config import GROQ_API_KEY, MODEL_NAME
+from .data import burning_man_principles, faq, survival_tips
+from ..utils.weather import get_weather
+from ..utils.packing import generate_packing_list
+from ..utils.events import get_burning_man_dates
 
 client = AsyncGroq(api_key=GROQ_API_KEY)
 
@@ -57,18 +57,16 @@ async def chat_with_groq(message, history):
 def correct_year(response):
     current_year = datetime.now().year
     print(f"Correcting year in response to {current_year}")  # Debug print
-    # Replace any mention of years 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)
     
-    # Correct specific phrases
     response = re.sub(r'Burning Man \d{4}', f'Burning Man {current_year}', response)
     response = re.sub(r'this year.*?(\d{4})', f'this year ({current_year})', response)
     
     print(f"Corrected response: {response}")  # Debug print
     return response
 
-async def bot(history):
+def bot(history):
     user_message = history[-1][0].lower()
     current_year = datetime.now().year
     print(f"Current year: {current_year}")  # Debug print
@@ -79,20 +77,20 @@ async def bot(history):
         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)
+        bot_message = 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)
+        bot_message = 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)
+        bot_message = 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)
+        bot_message = 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)
+        bot_message = 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:
         print("Fetching event dates...")  # Debug print
         event_info = get_burning_man_dates()
@@ -104,7 +102,7 @@ async def bot(history):
         print(f"Event info: {event_info}")  # Debug print
         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)
+        bot_message = chat_with_groq(user_message, history)
 
     # Apply year correction to the bot's message
     corrected_message = correct_year(bot_message)