Update app.py

app.py

@@ -1,31 +1,59 @@
 import streamlit as st
-from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+import time
+import random
+import os
 
-# تنظیمات ادمین
+# تنظیمات اصلی چت‌بات
 ADMIN_USERNAME = "admin"
-ADMIN_PASSWORD = "admin"
+ADMIN_PASSWORD = "password"
 
-# CSS برای ظاهر برنامه
+# CSS سفارشی برای ظاهر چت‌بات و حالت تیره و روشن
 CUSTOM_CSS = """
 <style>
+:root {
+    --primary-color: #4a90e2;
+    --secondary-color: #ff6b6b;
+    --neutral-color: #f5f5f5;
+    --accent-color: #50e3c2;
+    --background-light: #e8effa;
+    --background-dark: #1c1c1e;
+    --text-color-light: #333;
+    --text-color-dark: #e0e0e0;
+    --shadow-color: rgba(0, 0, 0, 0.1);
+}
+
 body {
     font-family: 'Vazir', sans-serif;
-    background: linear-gradient(135deg, #e8effa, #ffffff);
+    background: var(--background-color);
 }
+
 .chat-container {
     max-width: 400px;
     margin: 0 auto;
     border-radius: 15px;
-    box-shadow: 0px 4px 12px rgba(0, 0, 0, 0.1);
+    box-shadow: 0px 4px 12px var(--shadow-color);
+    overflow: hidden;
+    background: var(--background-color);
 }
+
 .header {
-    background: linear-gradient(135deg, #4a90e2, #50e3c2);
+    background: linear-gradient(135deg, var(--primary-color), var(--accent-color));
     color: white;
     padding: 10px;
     text-align: center;
     font-weight: bold;
     font-size: 18px;
 }
+
+.chat-body {
+    background: var(--neutral-color);
+    padding: 15px;
+    height: 300px;
+    overflow-y: auto;
+    display: flex;
+    flex-direction: column;
+}
+
 .message {
     padding: 12px 15px;
     margin: 10px 0;
@@ -33,23 +61,30 @@ body {
     width: fit-content;
     max-width: 80%;
     font-size: 15px;
+    box-shadow: 2px 2px 10px rgba(0, 0, 0, 0.05);
+    transition: all 0.3s ease;
 }
+
 .message.user {
     align-self: flex-end;
-    background: #4a90e2;
-    color: white;
+    background: var(--primary-color);
+    color: var(--text-color-light);
 }
+
 .message.bot {
     align-self: flex-start;
-    background: #ff6b6b;
+    background: var(--secondary-color);
     color: white;
 }
+
 .footer {
     display: flex;
     padding: 10px;
-    background: #f5f5f5;
+    background: var(--neutral-color);
+    border-top: 1px solid #ddd;
     align-items: center;
 }
+
 .footer input[type="text"] {
     flex: 1;
     padding: 10px;
@@ -57,76 +92,145 @@ body {
     border-radius: 20px;
     background: #ffffff;
     margin-right: 10px;
+    font-size: 14px;
+    box-shadow: inset 3px 3px 8px rgba(0, 0, 0, 0.1);
+}
+
+.footer .icon, .footer .send-btn {
+    font-size: 20px;
+    color: var(--primary-color);
+    cursor: pointer;
+    transition: color 0.3s ease, transform 0.2s ease;
+}
+
+.footer .icon:hover, .footer .send-btn:hover {
+    color: var(--secondary-color);
+    transform: scale(1.1);
+}
+
+.admin-icon {
+    font-size: 40px;
+    color: var(--primary-color);
+    cursor: pointer;
+    text-align: center;
+    margin-top: 10px;
+    transition: color 0.3s ease;
+}
+
+.admin-icon:hover {
+    color: var(--secondary-color);
+    transform: scale(1.1);
+}
+
+h1, h2 {
+    font-size: 20px;
+    color: var(--primary-color);
+    text-align: center;
+    margin-top: 15px;
+    font-weight: bold;
 }
 </style>
 """
+
+# اعمال CSS سفارشی
 st.markdown(CUSTOM_CSS, unsafe_allow_html=True)
 
 class ModernChatbot:
     def __init__(self):
-        # تنظیم بارگذاری تنبل مدل
-        if "model_loaded" not in st.session_state:
-            st.session_state.model_loaded = False
-
-    def load_model(self):
-        """بارگذاری مدل و توکنایزر تنها در صورت نیاز"""
-        if not st.session_state.model_loaded:
-            model_name = "HuggingFaceH4/zephyr-7b-beta"
-            st.session_state.tokenizer = AutoTokenizer.from_pretrained(model_name)
-            st.session_state.model = AutoModelForCausalLM.from_pretrained(model_name)
-            st.session_state.chat_pipeline = pipeline(
-                "text-generation",
-                model=st.session_state.model,
-                tokenizer=st.session_state.tokenizer
-            )
-            st.session_state.model_loaded = True
-
-    def generate_response(self, message):
-        """تولید پاسخ با مدل بارگذاری شده"""
-        self.load_model()
-        response = st.session_state.chat_pipeline(
-            message, max_length=50, num_return_sequences=1
-        )[0]['generated_text']
-        return response
+        self.conversation_history = []
+        self.admin_logged_in = False
+
+    def chat_response(self, message):
+        """دریافت پیام کاربر و اضافه کردن پاسخ به تاریخچه گفتگو"""
+        time.sleep(0.5)
+        self.conversation_history.append(("شما", message))
+        
+        responses = [
+            "این یک پاسخ نمونه از ربات است.",
+            "چه کمکی از من بر می‌آید؟",
+            "لطفاً بیشتر توضیح دهید.",
+            "متشکرم از پیامتان!"
+        ]
+        bot_message = random.choice(responses)
+        self.conversation_history.append(("ربات", bot_message))
+        
+        return self.conversation_history
 
     def login_admin(self, username, password):
-        """ورود به عنوان مدیر"""
-        return username == ADMIN_USERNAME and password == ADMIN_PASSWORD
+        if username == ADMIN_USERNAME and password == ADMIN_PASSWORD:
+            self.admin_logged_in = True
+            return True
+        else:
+            return False
 
-# ذخیره تاریخچه مکالمات و وضعیت ورود ادمین در session_state
-if "conversation_history" not in st.session_state:
-    st.session_state.conversation_history = []
+    def clear_history(self):
+        self.conversation_history = []
 
-# ایجاد نمونه چت‌بات
+# ایجاد چت‌بات
 chatbot = ModernChatbot()
 
-# آیکون و پاپ‌آپ ورود ادمین
-if st.sidebar.button("ورود ادمین"):
-    username = st.sidebar.text_input("نام کاربری")
-    password = st.sidebar.text_input("رمز عبور", type="password")
-    if st.sidebar.button("تایید"):
-        if chatbot.login_admin(username, password):
-            st.session_state.admin_logged_in = True
-            st.sidebar.success("ورود موفقیت‌آمیز بود")
-        else:
-            st.sidebar.error("نام کاربری یا رمز عبور اشتباه است")
+# حالت روشن یا تیره
+dark_mode = st.sidebar.checkbox("فعال کردن حالت تیره")
+
+# تغییر تم بر اساس حالت انتخاب شده
+background_color = "--background-dark" if dark_mode else "--background-light"
+text_color = "--text-color-dark" if dark_mode else "--text-color-light"
+st.markdown(f"<style>:root {{ --background-color: var({background_color}); --text-color: var({text_color}); }}</style>", unsafe_allow_html=True)
+
+# آیکون ورود ادمین و نمایش پاپ‌آپ
+st.markdown("<div class='admin-icon'>&#128100;</div>", unsafe_allow_html=True)
 
-# نمایش مکالمات
-st.markdown("<div class='chat-container'><div class='header'>چت‌بات هوشمند</div>", unsafe_allow_html=True)
-for sender, message in st.session_state.conversation_history:
+if "show_popup" not in st.session_state:
+    st.session_state["show_popup"] = False
+
+if "admin_logged_in" not in st.session_state:
+    st.session_state["admin_logged_in"] = False
+
+if st.session_state["show_popup"]:
+    with st.container():
+        st.markdown("<div class='popup-form'><h2 style='color: var(--primary-color); text-align:center;'>ورود ادمین</h2>", unsafe_allow_html=True)
+        username = st.text_input("نام کاربری")
+        password = st.text_input("رمز عبور", type="password")
+        if st.button("ورود"):
+            if chatbot.login_admin(username, password):
+                st.session_state["admin_logged_in"] = True
+                st.session_state["show_popup"] = False
+                st.success("ورود موفقیت‌آمیز بود!")
+                st.experimental_rerun()
+            else:
+                st.error("نام کاربری یا رمز عبور اشتباه است")
+else:
+    if st.button("ورود به ادمین"):
+        st.session_state["show_popup"] = True
+
+if st.session_state["admin_logged_in"]:
+    st.subheader("داشبورد مدیر")
+    if st.button("🗑️ حذف تاریخچه گفتگو"):
+        chatbot.clear_history()
+        st.success("تاریخچه گفتگو حذف شد.")
+        st.experimental_rerun()
+
+    # لینک به فایل داشبورد
+    if os.path.exists("admin_dashboard.py"):
+        st.markdown("[ورود به داشبورد](./admin_dashboard.py)", unsafe_allow_html=True)
+
+# نمایش چت‌باکس
+st.markdown("<div class='chat-container'><div class='header'>Chat with us</div><div class='chat-body'>", unsafe_allow_html=True)
+
+# نمایش تاریخچه گفتگو
+for sender, message in chatbot.conversation_history:
     align_class = "user" if sender == "شما" else "bot"
-    st.markdown(f"<div class='message {align_class}'>{sender}: {message}</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.conversation_history.append(("شما", user_message))
-    response = chatbot.generate_response(user_message)
-    st.session_state.conversation_history.append(("ربات", response))
-    st.experimental_rerun()
+    st.markdown(f"<div class='message {align_class}'><strong>{sender}:</strong> {message}</div>", unsafe_allow_html=True)
+
+st.markdown("</div><div class='footer'><input type='text' placeholder='Type a message...' id='user_message'/><span class='icon'>📎</span><span class='send-btn'>🚀</span></div></div>", unsafe_allow_html=True)
+
+user_message = st.text_input("پیام خود را اینجا بنویسید...", key="user_message", label_visibility="collapsed")
+
+if st.button("ارسال"):
+    if user_message:
+        chatbot.chat_response(user_message)
+        st.experimental_rerun()
 
-# پاک‌سازی مکالمات
-if st.sidebar.button("پاک کردن تاریخچه مکالمات"):
-    st.session_state.conversation_history = []
+if st.button("پاک کردن گفتگو") and not chatbot.admin_logged_in:
+    chatbot.clear_history()
     st.experimental_rerun()