Update app.py

app.py

@@ -3,77 +3,74 @@ from transformers import pipeline
 import logging
 
 # تنظیم لاگینگ
-logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# تنظیمات برنامه
-st.set_page_config(
-    page_title="چت‌بات هوش مصنوعی",
-    page_icon="💬",
-    layout="wide",
-    initial_sidebar_state="expanded",
-)
-
-# بارگذاری مدل
-model_name = "distilgpt2"  # مدل سبک‌تر برای عملکرد بهتر
+# تغییر مدل به FinBERT
+model_name = "yiyanghkust/finbert-tone"  # مدل FinBERT برای تحلیل احساسات مالی
 try:
-    generator = pipeline("text-generation", model=model_name)
-    st.sidebar.success("مدل با موفقیت بارگذاری شد!")
+    generator = pipeline('sentiment-analysis', model=model_name)  # بارگذاری مدل
+    st.sidebar.success(f"مدل '{model_name}' با موفقیت بارگذاری شد.")
 except Exception as e:
-    st.sidebar.error("مدل بارگذاری نشد!")
+    st.sidebar.error("مدل بارگذاری نشد. لطفاً بررسی کنید.")
     logging.error(f"Error loading model: {e}")
     generator = None
 
-# تاریخچه چت در حالت جلسه
-if "chat_history" not in st.session_state:
+# ایجاد حالت‌های جلسه برای ذخیره تاریخچه چت
+if 'chat_history' not in st.session_state:
     st.session_state.chat_history = []
 
-# افزودن پیام به تاریخچه
-def add_message(message, sender):
-    st.session_state.chat_history.append({"sender": sender, "message": message})
+# تابع برای افزودن پیام
+def add_message(text, is_user):
+    st.session_state.chat_history.append((text, is_user))
 
-# نمایش تاریخچه چت
-def display_chat():
-    st.markdown("### 💬 گفتگو")
-    for chat in st.session_state.chat_history:
-        sender = "شما" if chat["sender"] == "user" else "دستیار"
-        st.markdown(f"**{sender}:** {chat['message']}")
-    st.divider()
+# تابع نمایش تاریخچه چت
+def display_chat_history():
+    for message, is_user in st.session_state.chat_history:
+        if is_user:
+            st.chat_message("user").markdown(message)
+        else:
+            st.chat_message("assistant").markdown(message)
 
-# ارسال پیام
+# تابع ارسال پیام و دریافت پاسخ
 def send_message(user_input):
-    if user_input and generator:
-        add_message(user_input, "user")
+    if user_input:
+        add_message(user_input, True)  # افزودن پیام کاربر به تاریخچه
         try:
-            response = generator(user_input, max_length=50, num_return_sequences=1)
-            generated_text = response[0]["generated_text"]
-            add_message(generated_text, "assistant")
+            if generator:
+                response = generator(user_input)
+                sentiment = response[0]['label']  # تحلیل احساسات
+                score = response[0]['score']  # امتیاز پیش‌بینی
+                reply = f"تحلیل احساسات: {sentiment} (امتیاز: {score:.2f})"
+                add_message(reply, False)
+            else:
+                add_message("مدل در دسترس نیست. لطفاً تنظیمات را بررسی کنید.", False)
         except Exception as e:
-            error_message = "مشکلی در تولید پاسخ رخ داد!"
-            logging.error(f"Error generating text: {e}")
-            add_message(error_message, "assistant")
+            logging.error(f"خطا در تولید متن: {e}")
+            add_message("متأسفم، مشکلی پیش آمد.", False)
 
-# طراحی رابط کاربری
-st.title("💬 چت‌بات هوش مصنوعی برای تحلیل ارزهای دیجیتال")
-st.sidebar.header("تنظیمات")
-st.sidebar.info(
-    """
-    **راهنما**:
-    - این چت‌بات از مدل GPT استفاده می‌کند.
-    - می‌توانید سوالات مرتبط با ارزهای دیجیتال را بپرسید.
-    """
-)
+# تنظیم رابط کاربری Streamlit
+st.set_page_config(page_title="چت‌بات هوش مصنوعی", page_icon="🤖", layout="wide")
 
-# پیام خوش‌آمدگویی (فقط یک‌بار)
-if not st.session_state.chat_history:
-    add_message("سلام! من دستیار هوش مصنوعی شما هستم. چطور می‌توانم کمکتان کنم؟", "assistant")
+# طراحی نوار کناری
+st.sidebar.title("تنظیمات")
+st.sidebar.info("این برنامه از مدل FinBERT برای تحلیل احساسات مالی استفاده می‌کند.")
+
+# عنوان برنامه
+st.title("🤖 چت‌بات هوش مصنوعی برای تحلیل ارزهای دیجیتال")
 
 # نمایش تاریخچه چت
-display_chat()
+st.header("📜 تاریخچه چت")
+display_chat_history()
 
-# ورودی پیام
-st.text_input(
-    "سوال خود را اینجا وارد کنید:",
-    placeholder="مثلاً تحلیل قیمت بیت‌کوین...",
-    key="user_input",
-    on_change=lambda: send_message(st.session_state.user_input),
-)
+# ورودی کاربر
+st.header("✍️ پیام جدید")
+user_input = st.text_input("پیام خود را وارد کنید:", placeholder="مثلاً تحلیل قیمت بیت‌کوین...")
+
+# دکمه ارسال
+if st.button("ارسال"):
+    send_message(user_input)
+
+# پیام خوش‌آمدگویی (اولین بار)
+if not st.session_state.chat_history:
+    add_message("سلام! من دستیار هوش مصنوعی شما هستم. چطور می‌توانم کمکتان کنم؟", False)
+    display_chat_history()