|
import os |
|
import gradio as gr |
|
import asyncio |
|
import logging |
|
|
|
|
|
logging.basicConfig(level=logging.INFO) |
|
logger = logging.getLogger(__name__) |
|
|
|
|
|
|
|
from bot.chat import chat_interface |
|
|
|
|
|
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; |
|
} |
|
""" |
|
|
|
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("<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") |
|
|
|
async def user(user_message, history): |
|
return "", history + [[user_message, None]] |
|
|
|
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! |
|
|
|
Remember, participation is key. Let's co-create an amazing experience! )'( |
|
""" |
|
) |
|
|
|
if __name__ == "__main__": |
|
demo.queue() |
|
demo.launch() |
