import streamlit as st
import sqlite3
import random
from datetime import datetime
import pandas as pd
import plotly.express as px
from pathlib import Path

# تنظیمات اولیه دیتابیس
class DatabaseManager:
    def __init__(self):
        self.conn = sqlite3.connect('banking_assistant.db')
        self.create_tables()
    
    def create_tables(self):
        c = self.conn.cursor()
        # جدول مکالمات
        c.execute('''
            CREATE TABLE IF NOT EXISTS conversations (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                user_message TEXT,
                assistant_response TEXT,
                timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
                feedback INTEGER DEFAULT 0
            )
        ''')
        
        # جدول پایگاه دانش
        c.execute('''
            CREATE TABLE IF NOT EXISTS knowledge_base (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                question TEXT,
                answer TEXT,
                category TEXT,
                created_at DATETIME DEFAULT CURRENT_TIMESTAMP
            )
        ''')
        
        # جدول تنظیمات
        c.execute('''
            CREATE TABLE IF NOT EXISTS settings (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                theme TEXT DEFAULT 'light',
                language TEXT DEFAULT 'fa',
                created_at DATETIME DEFAULT CURRENT_TIMESTAMP
            )
        ''')
        self.conn.commit()

# تنظیمات استایل و تم
THEMES = {
    'light': {
        'bg_color': '#e0e5ec',
        'text_color': '#333',
        'shadow': '9px 9px 16px #b8b9be, -9px -9px 16px #ffffff'
    },
    'dark': {
        'bg_color': '#2d3436',
        'text_color': '#fff',
        'shadow': '9px 9px 16px #1a1d1e, -9px -9px 16px #404b4d'
    },
    'blue': {
        'bg_color': '#e3f2fd',
        'text_color': '#1976d2',
        'shadow': '9px 9px 16px #c1cdd4, -9px -9px 16px #ffffff'
    }
}
def load_custom_styles(theme='light'):
    return f"""
    <style>
    @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@100;200;300;400;500;600;700;800;900&display=swap');
    
    /* تنظیمات پایه */
    * {{
        font-family: 'Vazirmatn', sans-serif;
        direction: rtl;
        transition: all 0.3s ease;
    }}
    
    body {{
        background: {THEMES[theme]['bg_color']};
        color: {THEMES[theme]['text_color']};
    }}
    
    /* هدر اصلی */
    .main-header {{
        background: {THEMES[theme]['bg_color']};
        padding: 2rem;
        border-radius: 25px;
        box-shadow: {THEMES[theme]['shadow']};
        margin-bottom: 2rem;
        text-align: center;
        position: relative;
        overflow: hidden;
    }}
    
    .main-header::before {{
        content: '';
        position: absolute;
        top: -50%;
        left: -50%;
        width: 200%;
        height: 200%;
        background: linear-gradient(
            45deg,
            transparent 0%,
            rgba(255, 255, 255, 0.1) 50%,
            transparent 100%
        );
        animation: shine 3s infinite;
    }}
    
    /* باکس چت */
    .chat-container {{
        background: {THEMES[theme]['bg_color']};
        border-radius: 20px;
        padding: 2rem;
        margin: 1rem 0;
        box-shadow: {THEMES[theme]['shadow']};
        max-height: 70vh;
        overflow-y: auto;
        scroll-behavior: smooth;
    }}
    
    /* پیام‌ها */
    .message {{
        margin: 1rem 0;
        padding: 1rem;
        border-radius: 15px;
        position: relative;
        animation: messageSlide 0.3s ease-out;
    }}
    
    .message.user {{
        background: linear-gradient(135deg, #0073e6 0%, #0056b3 100%);
        color: white;
        margin-left: 2rem;
        box-shadow: 0 4px 15px rgba(0, 115, 230, 0.2);
    }}
    
    .message.assistant {{
        background: linear-gradient(135deg, #f8f9fa 0%, #e9ecef 100%);
        color: #343a40;
        margin-right: 2rem;
        box-shadow: 0 4px 15px rgba(0, 0, 0, 0.1);
    }}
    
    /* ورودی چت */
    .chat-input-container {{
        background: {THEMES[theme]['bg_color']};
        padding: 1.5rem;
        border-radius: 15px;
        box-shadow: {THEMES[theme]['shadow']};
        margin-top: 1rem;
        display: flex;
        align-items: center;
        gap: 1rem;
    }}
    
    .chat-input {{
        flex: 1;
        padding: 1rem;
        border: none;
        border-radius: 10px;
        background: rgba(255, 255, 255, 0.1);
        backdrop-filter: blur(10px);
        color: {THEMES[theme]['text_color']};
        font-size: 1rem;
        transition: all 0.3s ease;
    }}
    
    .chat-input:focus {{
        outline: none;
        box-shadow: 0 0 0 3px rgba(0, 115, 230, 0.3);
    }}
    
    /* دکمه‌ها */
    .button {{
        padding: 0.8rem 1.5rem;
        border: none;
        border-radius: 10px;
        background: linear-gradient(135deg, #0073e6 0%, #0056b3 100%);
        color: white;
        cursor: pointer;
        transition: all 0.3s ease;
        font-weight: 500;
        display: flex;
        align-items: center;
        gap: 0.5rem;
    }}
    
    .button:hover {{
        transform: translateY(-2px);
        box-shadow: 0 5px 15px rgba(0, 115, 230, 0.3);
    }}
    
    .button:active {{
        transform: translateY(0);
    }}
    
    /* انیمیشن‌ها */
    @keyframes messageSlide {{
        from {{
            opacity: 0;
            transform: translateY(20px);
        }}
        to {{
            opacity: 1;
            transform: translateY(0);
        }}
    }}
    
    @keyframes shine {{
        0% {{
            transform: translateX(-100%) rotate(45deg);
        }}
        100% {{
            transform: translateX(100%) rotate(45deg);
        }}
    }}
    
    /* اسکرولبار سفارشی */
    ::-webkit-scrollbar {{
        width: 8px;
    }}
    
    ::-webkit-scrollbar-track {{
        background: {THEMES[theme]['bg_color']};
    }}
    
    ::-webkit-scrollbar-thumb {{
        background: #0073e6;
        border-radius: 4px;
    }}
    
    ::-webkit-scrollbar-thumb:hover {{
        background: #0056b3;
    }}
    </style>
    """
class BankingAssistant:
    def __init__(self):
        self.db = DatabaseManager()
        self.model = BankingModel()  # اضافه کردن مدل
        if 'theme' not in st.session_state:
            st.session_state.theme = 'light'
        if 'messages' not in st.session_state:
            st.session_state.messages = []
            self.add_message('assistant', """
                👋 سلام! من دستیار هوشمند بانکی شما هستم.
                
                می‌توانم در موارد زیر به شما کمک کنم:
                📊 مشاهده وضعیت حساب
                💳 انتقال وجه
                📝 درخواست تسهیلات
                🏦 اطلاعات شعب
                
                چطور می‌توانم کمکتان کنم؟
            """)

    def add_message(self, role, content):
        timestamp = datetime.now().strftime("%H:%M")
        st.session_state.messages.append({
            'role': role,
            'content': content,
            'timestamp': timestamp
        })
        
        # ذخیره در دیتابیس
        if role == 'user':
            self.db.conn.execute(
                'INSERT INTO conversations (user_message, timestamp) VALUES (?, ?)',
                (content, timestamp)
            )
        else:
            self.db.conn.execute(
                'UPDATE conversations SET assistant_response = ? WHERE id = last_insert_rowid()',
                (content,)
            )
        self.db.conn.commit()

    def render_chat_interface(self):
        st.markdown(load_custom_styles(st.session_state.theme), unsafe_allow_html=True)
        
        # هدر اصلی
        st.markdown("""
            <div class="main-header">
                <h1>🏦 دستیار هوشمند بانکی</h1>
                <p>پاسخگوی 24 ساعته شما</p>
            </div>
        """, unsafe_allow_html=True)
        
        # باکس چت
        st.markdown('<div class="chat-container">', unsafe_allow_html=True)
        
        for msg in st.session_state.messages:
            class_name = "user" if msg['role'] == 'user' else "assistant"
            st.markdown(f"""
                <div class="message {class_name}">
                    {msg['content']}
                    <small style="position: absolute; bottom: 5px; {'right' if class_name == 'user' else 'left'}: 10px; opacity: 0.7;">
                        {msg['timestamp']}
                    </small>
                </div>
            """, unsafe_allow_html=True)
        
        st.markdown('</div>', unsafe_allow_html=True)
        
        # باکس ورودی
        st.markdown("""
            <div class="chat-input-container">
                <button class="action-button">🎤</button>
                <input type="text" class="chat-input" placeholder="پیام خود را بنویسید...">
                <button class="button">
                    <span>ارسال</span>
                    <span>➤</span>
                </button>
            </div>
        """, unsafe_allow_html=True)


class AdminDashboard:
    def __init__(self, db_manager):
        self.db = db_manager

    def render_dashboard(self):
        st.markdown("""
            <div class="admin-header">
                <h2>🔐 پنل مدیریت</h2>
            </div>
        """, unsafe_allow_html=True)

        # نمودارهای تحلیلی
        col1, col2 = st.columns(2)
        
        with col1:
            self.render_conversation_stats()
        
        with col2:
            self.render_feedback_chart()

        # مدیریت پایگاه دانش
        st.markdown("""
            <div class="admin-section">
                <h3>📚 مدیریت پایگاه دانش</h3>
            </div>
        """, unsafe_allow_html=True)

        # افزودن دانش جدید
        with st.expander("➕ افزودن مورد جدید"):
            category = st.selectbox(
                "دسته‌بندی",
                ["عمومی", "حساب‌ها", "تسهیلات", "کارت", "شعب"]
            )
            question = st.text_area("سوال")
            answer = st.text_area("پاسخ")
            
            if st.button("ذخیره"):
                self.db.conn.execute(
                    'INSERT INTO knowledge_base (question, answer, category) VALUES (?, ?, ?)',
                    (question, answer, category)
                )
                self.db.conn.commit()
                st.success("✅ با موفقیت ذخیره شد")

    def render_conversation_stats(self):
        # دریافت آمار مکالمات
        df = pd.read_sql_query("""
            SELECT 
                date(timestamp) as date,
                COUNT(*) as count
            FROM conversations 
            GROUP BY date(timestamp)
            ORDER BY date DESC
            LIMIT 7
        """, self.db.conn)

        fig = px.line(
            df,
            x='date',
            y='count',
            title='آمار مکالمات 7 روز اخیر',
            labels={'count': 'تعداد مکالمات', 'date': 'تاریخ'}
        )
        
        fig.update_layout(
            font_family="Vazirmatn",
            plot_bgcolor='rgba(0,0,0,0)',
            paper_bgcolor='rgba(0,0,0,0)',
        )
        
        st.plotly_chart(fig)

    def render_feedback_chart(self):
        # نمودار بازخوردها
        df = pd.read_sql_query("""
            SELECT 
                feedback,
                COUNT(*) as count
            FROM conversations 
            WHERE feedback != 0
            GROUP BY feedback
        """, self.db.conn)

        fig = px.pie(
            df,
            values='count',
            names='feedback',
            title='نمودار بازخوردها',
            color_discrete_sequence=px.colors.sequential.Blues
        )
        
        fig.update_layout(
            font_family="Vazirmatn",
            plot_bgcolor='rgba(0,0,0,0)',
            paper_bgcolor='rgba(0,0,0,0)',
        )
        
        st.plotly_chart(fig)

def main():
    st.set_page_config(
        page_title="دستیار هوشمند بانکی",
        page_icon="🏦",
        layout="wide",
        initial_sidebar_state="expanded"
    )

    assistant = BankingAssistant()

    # منوی کناری
    with st.sidebar:
        st.markdown("### ⚙️ تنظیمات")
        theme = st.selectbox(
            "انتخاب تم",
            options=['light', 'dark', 'blue'],
            index=['light', 'dark', 'blue'].index(st.session_state.theme)
        )
        
        if theme != st.session_state.theme:
            st.session_state.theme = theme
            st.experimental_rerun()

        if st.button("🔐 ورود مدیر"):
            st.session_state.show_login = True

    # نمایش فرم لاگین
    if st.session_state.get('show_login', False):
        with st.form("login_form"):
            username = st.text_input("نام کاربری")
            password = st.text_input("رمز عبور", type="password")
            
            if st.form_submit_button("ورود"):
                if username == "admin" and password == "admin":  # در حالت واقعی باید امن‌تر باشد
                    st.session_state.admin_logged_in = True
                    st.session_state.show_login = False
                    st.experimental_rerun()
                else:
                    st.error("نام کاربری یا رمز عبور اشتباه است")

    # نمایش داشبورد مدیر
    if st.session_state.get('admin_logged_in', False):
        admin = AdminDashboard(assistant.db)
        admin.render_dashboard()
    else:
        assistant.render_chat_interface()

if __name__ == "__main__":
    main()