Update app.py

app.py

@@ -7,7 +7,6 @@ from langchain.schema import SystemMessage, HumanMessage
 from rapidfuzz import fuzz
 import concurrent.futures
 import time
-# from sentence_transformers import SentenceTransformer
 import numpy as np
 from hazm import *
 import re
@@ -16,7 +15,6 @@ nltk.download('punkt')
 
 st.markdown("""
     <style>
-        /* استایل برای هدر */
         .stAppHeader.st-emotion-cache-12fmjuu.e4hpqof0 {
             background-color: rgba(46,59,46, 0.8) !important;  /* سبز متمایل به خاکی */
             color: #2e3b2e !important;  /* رنگ متن روشن */
@@ -29,14 +27,12 @@ st.markdown("""
 """, unsafe_allow_html=True)
 st.markdown("""
     <style>
-    /* بارگذاری فونت Roboto */
     @font-face {
         font-family: 'Roboto';
         src: url('https://fonts.googleapis.com/css2?family=Roboto:wght@400;500;700&display=swap') format('woff2');
         font-weight: 400;
         font-style: normal;
     }
-    /* تنظیم فونت برای کل سایت */
     html, body, [class*="css"] {
         font-family: 'Roboto', Tahoma, sans-serif !important;
         font-weight: 400 !important;
@@ -44,19 +40,16 @@ st.markdown("""
         text-align: right;
 ظ        color: #ffffff;
     }
-    /* طراحی برای بخش استایل اپ */
     .stApp {
         background: linear-gradient(to left, #4b5e40, #2e3b2e);
         color: #ffffff;
     }
-    /* استایل برای سایدبار */
     [data-testid="stSidebar"] {
         width: 260px !important;
         background-color: #1a2b1e;
         border: none !important;
         padding-top: 20px;
     }
-    /* استایل برای آیتم‌های منو */
     .menu-item {
         display: flex;
         align-items: center;
@@ -68,17 +61,14 @@ st.markdown("""
         cursor: pointer;
         transition: background-color 0.3s ease;
     }
-    /* استایل برای آیتم‌های منو هنگام هاور */
     .menu-item:hover {
         background-color: #2e3b2e;
         color: #b8860b;
     }
-    /* استایل برای آیکون‌ها در منو */
     .menu-item img {
         width: 25px;
         height: 25px;
     }
-    /* استایل برای دکمه‌ها */
     .stButton>button {
         background-color: #b8860b !important;
         color: #1a2b1e !important;
@@ -92,13 +82,11 @@ st.markdown("""
         width: 100%;
         margin: 10px 0;
     }
-    /* استایل برای دکمه‌ها هنگام هاور */
     .stButton>button:hover {
         background-color: #8b6508 !important;
         transform: translateY(-2px);
         box-shadow: 0 4px 8px rgba(0,0,0,0.3);
     }
-/* استایل برای متن هدر */
 .header-text {
     text-align: center;
     margin: 20px 0;
@@ -108,14 +96,12 @@ st.markdown("""
     box-shadow: 0 6px 12px rgba(0,0,0,0.4);
     font-family: 'Roboto', Tahoma, sans-serif; /* اضافه شد */
 }
-    /* استایل برای زیرنویس */
     .subtitle {
         font-size: 18px;
         color: #d4d4d4;
         font-weight: 600;
         margin-top: 10px;
     }
-    /* استایل برای پیام‌های چت */
     .chat-message {
         flex-wrap: wrap;
         background-color: rgba(26, 43, 30, 0.95);
@@ -133,12 +119,10 @@ st.markdown("""
         align-items: center;
         gap: 15px;
     }
-    /* انیمیشن برای ورود پیام چت */
     @keyframes fadeIn {
         from { opacity: 0; transform: translateY(10px); }
         to { opacity: 1; transform: translateY(0); }
     }
-    /* استایل برای ورودی متن */
     .stTextInput>div>input, .stTextArea textarea {
         background-color: rgba(26, 43, 30, 0.95) !important;
         border-radius: 10px !important;
@@ -149,12 +133,10 @@ st.markdown("""
         font-size: 16px;
         color: #d4d4d4 !important;
     }
-    /* استایل برای خط افقی */
     hr {
         border: 1px solid #b8860b;
         margin: 15px 0;
     }
-    /* حذف مرز از قسمت سایدبار */
     [data-testid="stSidebar"] > div {
         border: none !important;
     }
@@ -162,7 +144,6 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 
-# ---------- احراز هویت ----------
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
 
@@ -181,11 +162,9 @@ if not st.session_state.authenticated:
 """, unsafe_allow_html=True)
     st.markdown("""
     <style>
-        /* فونت عمومی */
         html, body, [class*="css"] {
             font-family: 'Vazir', sans-serif;
         }
-        /* استایل برای برچسب فیلدهای ورودی */
         label {
             font-size: 20px !important;
             color: #ffffff !important;  
@@ -193,7 +172,6 @@ if not st.session_state.authenticated:
             margin-bottom: 10px !important;
             display: block;
         }
-        /* استایل برای ورودی‌ها در تمام حالت‌ها */
         input[type="text"], 
         input[type="password"], 
         input[type="text"]:focus, 
@@ -205,7 +183,6 @@ if not st.session_state.authenticated:
             font-size: 18px !important;
             font-family: 'Vazir', sans-serif !important;
         }
-        /* Placeholder style */
         ::placeholder {
             color: #bbbbbb !important;
             opacity: 0.8 !important;
@@ -260,7 +237,6 @@ if not st.session_state.authenticated:
             """, unsafe_allow_html=True)
     st.stop()
 
-# ---------- سایدبار ----------
 with st.sidebar:
     st.image("log.png", use_container_width=True)
 
@@ -296,20 +272,18 @@ with st.sidebar:
 
 st.markdown("""
     <style>
-        /* استایل برای متن هدر */
         .header-text {
             text-align: center;
             margin: 50px 0;
-            background: #2e3b2e; /* سبز زیتونی تیره جدید */
+            background: #2e3b2e; 
             padding: 60px 30px;
             border-radius: 25px;
-            box-shadow: 0 12px 24px rgba(0, 0, 0, 0.8); /* سایه بیشتر */
+            box-shadow: 0 12px 24px rgba(0, 0, 0, 0.8); 
             animation: slideIn 2s ease-in-out, fadeIn 3s ease-in-out;
             background-size: cover;
             background-position: center;
             position: relative;
         }
-        /* انیمیشن ورودی */
         @keyframes fadeIn {
             0% { opacity: 0; transform: translateY(30px); }
             100% { opacity: 1; transform: translateY(0); }
@@ -318,19 +292,17 @@ st.markdown("""
             0% { transform: translateX(-50%); opacity: 0; }
             100% { transform: translateX(0); opacity: 1; }
         }
-        /* تغییر فونت برای h1 */
         .header-text h1 {
             font-family: 'Vazir', sans-serif;
             font-size: 62px;
-            color: #d89b00; /* طلایی تیره‌تر */
+            color: #d89b00; 
             margin: 0;
             font-weight: 900;
             letter-spacing: 4px;
-            text-shadow: 4px 4px 15px rgba(0, 0, 0, 0.9); /* سایه سیاه بیشتر */
-            transform: scale(1.08); /* کمی بزرگتر شدن */
+            text-shadow: 4px 4px 15px rgba(0, 0, 0, 0.9); 
+            transform: scale(1.08); 
             animation: glow 2s ease-in-out infinite alternate;
         }
-        /* تغییر استایل زیرعنوان */
         .subtitle {
             font-family: 'Vazir', sans-serif;
             font-size: 24px;
@@ -338,17 +310,15 @@ st.markdown("""
             font-weight: 700;
             margin-top: 15px;
             letter-spacing: 2px;
-            text-shadow: 3px 3px 10px rgba(0,0,0,0.8); /* سایه بیشتر برای زیرعنوان */
+            text-shadow: 3px 3px 10px rgba(0,0,0,0.8); 
             animation: fadeInSubtitle 2s ease-in-out;
         }
-        /* انیمیشن زیرعنوان */
         @keyframes fadeInSubtitle {
             0% { opacity: 0; transform: translateY(20px); }
             100% { opacity: 1; transform: translateY(0); }
         }
-        /* استایل برای دکمه‌ها */
         .stButton>button {
-            background-color: #e67e22 !important; /* رنگ دکمه تغییر کرده */
+            background-color: #e67e22 !important; 
             color: #4b5320 !important;
             font-family: 'Vazir', sans-serif;
             font-weight: 700 !important;
@@ -366,9 +336,8 @@ st.markdown("""
             transform: translateY(-4px);
             box-shadow: 0 8px 20px rgba(0, 0, 0, 0.6);
         }
-        /* استایل برای استایل کلی صفحه */
         .stApp {
-            background: #2e3b2e; /* سبز زیتونی تیره جدید */
+            background: #2e3b2e; 
             color: white;
             font-family: 'Vazir', sans-serif;
         }
@@ -379,7 +348,6 @@ st.markdown("""
     </div>
 """, unsafe_allow_html=True)
 
-# ---------- مدل زبانی ----------
 llm = ChatOpenAI(
     base_url="https://api.together.xyz/v1",
     api_key='0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979',
@@ -389,10 +357,8 @@ llm = ChatOpenAI(
 )
 
 
-# ---------- ورودی جستجو ----------
 st.markdown("""
     <style>
-        /* استایل برای کلاس خاص st-emotion-cache-128upt6 eht7o1d3 */
         .st-emotion-cache-128upt6.eht7o1d3 {
             background-color: rgba(46,59,46, 0.8) !important;  /* سبز تیره (44533f) */
             border-radius: 10px !important;  /* گوشه‌های گرد */
@@ -419,41 +385,37 @@ st.markdown("""
 
 st.markdown("""
     <style>
-        /* استایل برای کلاس st-emotion-cache-yd4u6l e1togvvn1 */
         .st-emotion-cache-yd4u6l.e1togvvn1 {
-            background-color: rgba(106, 127, 83, 0.8) !important;  /* سبز خاکی مایل به زرد (#6a7f53) */
-            border-radius: 10px !important;  /* گوشه‌های گرد */
-            color: #d4d4d4 !important;  /* رنگ متن روشن */
+            background-color: rgba(106, 127, 83, 0.8) !important;  
+            border-radius: 10px !important; 
+            color: #d4d4d4 !important;  
             font-family: 'Vazirmatn', Tahoma, sans-serif !important;
-            padding: 15px !important;  /* فضای داخلی بیشتر */
-            box-shadow: 0 4px 8px rgba(0, 0, 0, 0.3) !important;  /* سایه برای برجسته شدن */
+            padding: 15px !important;  
+            box-shadow: 0 4px 8px rgba(0, 0, 0, 0.3) !important; 
         }
     </style>
 """, unsafe_allow_html=True)
 st.markdown("""
     <style>
-        /* استایل برای هدر */
         .stAppHeader.st-emotion-cache-12fmjuu.e4hpqof0 {
-            background-color: rgba(42, 55, 39, 0.9) !important;  /* سبز تیره‌تر */
-            color: #d4d4d4 !important;  /* رنگ متن روشن */
+            background-color: rgba(42, 55, 39, 0.9) !important;
+            color: #d4d4d4 !important;  
             font-family: 'Vazirmatn', Tahoma, sans-serif !important;
-            padding: 20px !important;  /* فضای داخلی بیشتر */
-            border-radius: 10px !important;  /* گوشه‌های گرد */
-            box-shadow: 0 4px 8px rgba(0, 0, 0, 0.3) !important;  /* سایه برای برجسته شدن */
+            padding: 20px !important; 
+            border-radius: 10px !important; 
+            box-shadow: 0 4px 8px rgba(0, 0, 0, 0.3) !important;  
         }
     </style>
 """, unsafe_allow_html=True)
 
 st.markdown("""
 <style>
-/* تغییر رنگ متن placeholder به خاکستری */
 textarea::placeholder {
-    color: #ffffff !important; /* خاکستری */
-    opacity: 1 !important; /* برای اینکه مرورگرها بهش بی‌توجه نباشن */
+    color: #ffffff !important; 
+    opacity: 1 !important; 
 }
-            /* تغییر رنگ متن داخل چت اینپوت به خاکستری */
 textarea {
-    color: #ffffff !important; /* خاکستری */
+    color: #ffffff !important; 
     border-radius: 10px !important;
     padding: 10px !important;
 }
@@ -475,7 +437,6 @@ if query:
 else:
     st.markdown("")
 
-# استایل‌ها برای چرخش و پیام در حال فکر کردن
 st.markdown("""
     <style>
     .thinking-message {
@@ -489,7 +450,7 @@ st.markdown("""
     }
     .spinner {
         border: 4px solid #f3f3f3;
-        border-top: 4px solid #4b6d3d; /* رنگ سبز تیره */
+        border-top: 4px solid #4b6d3d;
         border-radius: 50%;
         width: 20px;
         height: 20px;
@@ -581,8 +542,6 @@ def summarize_text_by_frequency(text, num_sentences=1):
     summarized_sentences = heapq.nlargest(num_sentences, sentence_scores, key=sentence_scores.get)
     return "\n".join(summarized_sentences)
 
-
-
     
 
 def find_closest_lines(query, doc_texts, stop_words, top_n=5):
@@ -613,34 +572,56 @@ def remove_stop_words_from_lines(lines, stop_words):
 
 if query:
     closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=5)
-    
-    # حذف استپ‌وردها از خطوط و سپس پاکسازی نهایی متن
+
     cleaned_closest_lines = [
         clean_text(" ".join([word for word in line.split() if word not in stop_words]))
         for line in closest_lines
     ]
-    summarized_text = summarize_text_by_frequency("\n".join(cleaned_closest_lines), num_sentences=1)
 
+    summarized_text = summarize_text_by_frequency("\n".join(cleaned_closest_lines), num_sentences=1)
 
     if summarized_text:
-        # cleaned_text = "\n".join(cleaned_closest_lines[:1])
-
         prompt = f"""
-        لطفاً با توجه به سؤال زیر و محتوای خطوط مرتبط، یک پاسخ نهایی حرفه‌ای، دقیق و روان تولید کن. فقط از متن خطوط مرتبط استفاده کن. اگر اطلاعات کافی در متن وجود ندارد، صادقانه اعلام کن.
+        تعدادی جمله مرتبط با سوال زیر استخراج شده است. لطفاً ابتدا این جملات را بررسی کن، سپس با در نظر گرفتن محتوای سوال و لحن آن، یک پاسخ نهایی حرفه‌ای، دقیق و روان ارائه کن که هم به سوال پاسخ دهد و هم از نظر نگارشی و ساختاری در سطح بالایی باشد. پاسخ نهایی باید حداکثر 1024 کاراکتر و حداقل 512 کاراکتر باشد، خلاصه و واضح نوشته شود و فقط به زبان فارسی باشد. از تکرار اضافی پرهیز کن و در صورت نیاز، محتوای جملات را ترکیب کن.
         سوال:
         {query}
-        خطوط مرتبط:
+        جملات مرتبط:
         {summarized_text}
-        پاسخ نهایی:
+        پاسخ نهایی حرفه‌ای بازنویسی‌شده:
         """
-
         response = llm([
             SystemMessage(content="You are a helpful assistant."),
             HumanMessage(content=prompt)
         ])
-        rewritten = clean_text(response.content.strip())
+        rewritten = response.content.strip()
 
-        st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
+        review_prompt = f"""
+        لطفاً بررسی کن که آیا پاسخ زیر به سوال داده‌شده مرتبط، صحیح، معتبر و قابل قبول است یا نه. اگر پاسخ مرتبط و صحیح است، فقط بنویس: 'تأیید شد'. اگر پاسخ اشتباه است یا ربطی به سوال ندارد یا معتبر نیست، فقط بنویس: 'نیاز به اصلاح دارد'.
+        سوال:
+        {query}
+        پاسخ:
+        {rewritten}
+        """
+        review_response = llm([
+            SystemMessage(content="You are a helpful assistant."),
+            HumanMessage(content=review_prompt)
+        ])
+        review_result = review_response.content.strip()
 
+        if "تأیید شد" in review_result:
+            rewritten = clean_text(rewritten)
+            st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
+            think.empty()  
     else:
-        st.warning("هیچ خط مرتبطی پیدا نشد.")
+        prompt = f"""
+        هیچ جمله مرتبط مستقیمی با سوال یافت نشد. لطفاً با توجه به سوال زیر، یک پاسخ حرفه‌ای، مرتبط، معنادار و معتبر تولید کن. فقط از جملات موجود در اسناد معتبر استفاده کن و از ساخت جملات ساختگی یا استفاده از زبان غیر فارسی خودداری کن.
+        سوال:
+        {query}
+        """
+        response = llm([
+            SystemMessage(content="You are a helpful assistant."),
+            HumanMessage(content=prompt)
+        ])
+        rewritten = clean_text(response.content.strip())
+        st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
+        think.empty()