updated app

app.py

@@ -1,20 +1,14 @@
 import os
-import sys
 import gradio as gr
+import asyncio
+import logging
 
-# Add the current directory to the Python path
-current_dir = os.path.dirname(os.path.abspath(__file__))
-sys.path.append(current_dir)
+# 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")
-
-# Set GROQ_API_KEY as a global variable
-os.environ['GROQ_API_KEY'] = os.getenv('GROQ_API_KEY')
-
-# Now import the chat_interface
+# Import chat_interface after environment variables are loaded
 from bot.chat import chat_interface
 
 # Custom CSS for Burning Man theme
@@ -57,39 +51,28 @@ body {
 }
 """
 
-# 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>
-"""
+async def bot(history):
+    try:
+        user_message = history[-1][0]
+        bot_message = await chat_interface(user_message, history[:-1])
+        history[-1][1] = bot_message
+        return history
+    except Exception as e:
+        logger.error(f"Error in bot function: {str(e)}")
+        history[-1][1] = "Oops! Something went wrong. Please try again later."
+        return history
 
 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")
+    gr.HTML("<h1 style='text-align: center; color: #ff6b35;'>BurnerBot</h1>")
+    gr.HTML("<p style='text-align: center; color: #ffffff;'>Your dusty digital companion for all things Burning Man!</p>")
+    
+    chatbot = gr.Chatbot(height=400)
+    msg = gr.Textbox(label="Ask about Burning Man", placeholder="What would you like to know about Burning Man?")
+    clear = gr.Button("Clear Chat")
 
-    def user(user_message, history):
+    async 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
     )
@@ -110,4 +93,6 @@ with gr.Blocks(css=custom_css) as demo:
         """
     )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.queue()
+    demo.launch()