updated app

app.py

@@ -1,53 +1,116 @@
-import sys
 import os
-import logging
+import sys
+from dotenv import load_dotenv
+import gradio as gr
 
 # Add the current directory to the Python path
 current_dir = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(current_dir)
 
-import gradio as gr
-from bot.chat import bot  # Import the bot function from chat.py
+# Load environment variables from .env file
+load_dotenv()
 
-# Set up logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+# Check if GROQ_API_KEY is set
+if 'GROQ_API_KEY' not in os.environ:
+    raise EnvironmentError("GROQ_API_KEY is not set in the environment variables or .env file")
 
-async def chat_interface(message, history):
-    try:
-        updated_history = await bot(history + [(message, None)])
-        bot_response = updated_history[-1][1]
-        
-        # Debugging: Print the bot response
-        print(f"Bot response: {bot_response}")
-        
-        # Ensure the bot response is a string
-        if isinstance(bot_response, list):
-            bot_response = " ".join(bot_response)
-        
-        # Debugging: Print the formatted bot response
-        print(f"Formatted bot response: {bot_response}")
+# Set GROQ_API_KEY as a global variable
+os.environ['GROQ_API_KEY'] = os.getenv('GROQ_API_KEY')
+
+# Now import the chat_interface
+from bot.chat import chat_interface
+
+# Custom CSS for Burning Man theme
+custom_css = """
+body {
+    background-color: #f4a460;
+    background-image: url('https://i.imgur.com/JZjMiZR.jpg');
+    background-size: cover;
+    background-attachment: fixed;
+    font-family: 'Roboto', sans-serif;
+}
+.container {
+    max-width: 800px !important;
+    margin: auto;
+    background-color: rgba(0, 0, 0, 0.7);
+    border-radius: 15px;
+    padding: 20px;
+    box-shadow: 0 0 20px rgba(0, 0, 0, 0.5);
+}
+.gr-button {
+    background-color: #ff6b35 !important;
+    border: none !important;
+}
+.gr-button:hover {
+    background-color: #ff8c61 !important;
+}
+.gr-input {
+    border-color: #ff6b35 !important;
+}
+.gr-input:focus {
+    border-color: #ff8c61 !important;
+    box-shadow: 0 0 0 3px rgba(255, 107, 53, 0.25) !important;
+}
+.gr-box {
+    border-radius: 15px !important;
+    background-color: rgba(255, 255, 255, 0.1) !important;
+}
+.gr-padded {
+    padding: 15px !important;
+}
+"""
+
+# Custom HTML for header and footer
+custom_html = """
+<div style="text-align: center; max-width: 800px; margin: 0 auto;">
+    <h1 style="color: #ff6b35; font-size: 3em; margin-bottom: 10px;">BurnerBot</h1>
+    <p style="color: #ffffff; font-size: 1.2em; margin-bottom: 20px;">Your dusty digital companion for all things Burning Man!</p>
+</div>
+"""
+
+with gr.Blocks(css=custom_css) as demo:
+    gr.HTML(custom_html)
+    chatbot = gr.Chatbot(
+        [],
+        elem_id="chatbot",
+        bubble_full_width=False,
+        avatar_images=(None, "https://i.imgur.com/nAcdSry.png"),
+        height=400
+    )
+    msg = gr.Textbox(
+        show_label=False,
+        placeholder="Ask me anything about Burning Man...",
+        container=False
+    )
+    clear = gr.Button("Clear")
+
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+    def bot(history):
+        user_message = history[-1][0]
+        bot_message = chat_interface(user_message, history[:-1])
+        history[-1][1] = bot_message
+        return history
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+    gr.Markdown(
+        """
+        ### 🔥 Welcome to the Playa, Burner! 🔥
+        Ask me about:
+        - 📅 Event dates and tickets
+        - 🏕️ Survival guide and packing tips
+        - 🌟 The 10 Principles
+        - 🎨 Art installations and theme camps
+        - 🚲 Transportation and arrival
+        - And much more!
         
-        return bot_response  # Return only the bot_response, not the history
-    except Exception as e:
-        logger.error(f"Error in chat_interface: {str(e)}")
-        burner_error_message = (
-            "Whoa there, playa pal! It seems the dust has clogged my circuits. "
-            "Let's take a deep breath and try that again. Radical self-reliance, remember?"
-        )
-        return burner_error_message  # Return only the error message, not the history
-
-
-gr.ChatInterface(
-    chat_interface,
-    chatbot=gr.Chatbot(height=300),
-    textbox=gr.Textbox(placeholder="Ask me anything about Burning Man!", container=False),
-    title="BurnerBot",
-    description="Chat with me about Burning Man! I can help with event dates, packing lists, principles, and more!",
-    theme="soft",
-    examples=["When is Burning Man this year?", "What are the 10 principles?", "Can you give me a packing list?"],
-    cache_examples=False,
-    retry_btn=None,
-    undo_btn="Delete Previous",
-    clear_btn="Clear",
-).launch(share=True)
+        Remember, participation is key. Let's co-create an amazing experience! )'(
+        """
+    )
+
+demo.launch()