import streamlit as st
import time
import random
from datetime import datetime

# CSS بهبود یافته با دکمه‌های جدید
CUSTOM_CSS = """
<style>
@import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700&display=swap');

:root {
    --primary-color: #3b82f6;
    --secondary-color: #6366f1;
    --background-color: #f8fafc;
    --button-bg: #10b981;
    --button-hover: #0d946b;
    --button-text: #ffffff;
}

body {
    font-family: 'Vazirmatn', sans-serif;
    background-color: var(--background-color);
    direction: rtl;
}

.chat-container {
    max-width: 900px;
    margin: 2rem auto;
    background: var(--background-color);
    border-radius: 1rem;
    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
}

.chat-header {
    background: linear-gradient(to left, var(--primary-color), var(--secondary-color));
    color: white;
    padding: 1rem;
    text-align: center;
    border-radius: 1rem 1rem 0 0;
    font-size: 1.5rem;
    font-weight: 600;
}

.chat-messages {
    height: 500px;
    overflow-y: auto;
    padding: 1rem;
}

.message {
    margin-bottom: 1rem;
    padding: 1rem;
    border-radius: 0.5rem;
}

.message.user {
    background: #e0e7ff;
    color: #1e293b;
}

.message.bot {
    background: #f1f5f9;
    color: #475569;
}

.quick-replies {
    display: flex;
    gap: 0.5rem;
    margin-top: 1rem;
    flex-wrap: wrap;
}

.quick-reply {
    background: var(--button-bg);
    color: var(--button-text);
    padding: 0.5rem 1rem;
    border-radius: 20px;
    cursor: pointer;
    transition: all 0.2s;
    font-size: 0.9rem;
    text-align: center;
}

.quick-reply:hover {
    background: var(--button-hover);
    transform: scale(1.05);
}
</style>
"""

st.markdown(CUSTOM_CSS, unsafe_allow_html=True)

class EnhancedChatbot:
    def __init__(self):
        self.conversation_history = []
        self.quick_replies = [
            "بخشنامه‌های جدید",
            "راهنمای استفاده",
            "تماس با پشتیبانی",
            "گزارش مشکل"
        ]

    def chat_response(self, message):
        """پردازش پیام کاربر و ارائه پاسخ هوشمند"""
        self.conversation_history.append({'type': 'user', 'message': message})
        
        # شبیه‌سازی پاسخ ربات
        time.sleep(1)
        response = random.choice([
            "این بخشنامه جدید است.",
            "برای استفاده از سیستم، لطفاً مراحل زیر را دنبال کنید.",
            "با پشتیبانی تماس بگیرید.",
            "لطفاً مشکل خود را با جزئیات بیان کنید."
        ])
        
        self.conversation_history.append({'type': 'bot', 'message': response})

# تنظیمات اولیه استریملیت
st.set_page_config(
    page_title="پاسخگوی هوشمند بخشنامه",
    page_icon="🤖",
    layout="wide",
    initial_sidebar_state="expanded"
)

# ایجاد نمونه از چت‌بات
if 'chatbot' not in st.session_state:
    st.session_state.chatbot = EnhancedChatbot()

# نمایش هدر اصلی
st.markdown("""
    <div class="chat-container">
        <div class="chat-header">پاسخگوی هوشمند بخشنامه</div>
        <div class="chat-messages" id="chat-messages">
""", unsafe_allow_html=True)

# نمایش پیام‌ها
for message in st.session_state.chatbot.conversation_history:
    message_class = "user" if message['type'] == 'user' else "bot"
    st.markdown(f"""
        <div class="message {message_class}">
            {message['message']}
        </div>
    """, unsafe_allow_html=True)

# پاسخ‌های سریع
st.markdown("""<div class="quick-replies">""", unsafe_allow_html=True)
for reply in st.session_state.chatbot.quick_replies:
    st.markdown(f"""<div class="quick-reply">{reply}</div>""", unsafe_allow_html=True)
st.markdown("</div>", unsafe_allow_html=True)

# بخش ورودی پیام
user_message = st.text_input("پیام خود را بنویسید...", key="user_message")
if st.button("ارسال پیام") and user_message:
    st.session_state.chatbot.chat_response(user_message)
    st.experimental_rerun()

st.markdown("</div>", unsafe_allow_html=True)