Update app.py

app.py

@@ -7,6 +7,7 @@ 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
@@ -15,6 +16,7 @@ nltk.download('punkt')
 
 st.markdown("""
     <style>
+        /* استایل برای هدر */
         .stAppHeader.st-emotion-cache-12fmjuu.e4hpqof0 {
             background-color: rgba(46,59,46, 0.8) !important;  /* سبز متمایل به خاکی */
             color: #2e3b2e !important;  /* رنگ متن روشن */
@@ -27,12 +29,14 @@ 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;
@@ -40,16 +44,19 @@ 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;
@@ -61,14 +68,17 @@ 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;
@@ -82,11 +92,13 @@ 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;
@@ -96,12 +108,14 @@ 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);
@@ -119,10 +133,12 @@ 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;
@@ -133,10 +149,12 @@ st.markdown("""
         font-size: 16px;
         color: #d4d4d4 !important;
     }
+    /* استایل برای خط افقی */
     hr {
         border: 1px solid #b8860b;
         margin: 15px 0;
     }
+    /* حذف مرز از قسمت سایدبار */
     [data-testid="stSidebar"] > div {
         border: none !important;
     }
@@ -144,6 +162,7 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 
+# ---------- احراز هویت ----------
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
 
@@ -162,9 +181,11 @@ 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;  
@@ -172,6 +193,7 @@ if not st.session_state.authenticated:
             margin-bottom: 10px !important;
             display: block;
         }
+        /* استایل برای ورودی‌ها در تمام حالت‌ها */
         input[type="text"], 
         input[type="password"], 
         input[type="text"]:focus, 
@@ -183,6 +205,7 @@ 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;
@@ -237,6 +260,7 @@ if not st.session_state.authenticated:
             """, unsafe_allow_html=True)
     st.stop()
 
+# ---------- سایدبار ----------
 with st.sidebar:
     st.image("log.png", use_container_width=True)
 
@@ -272,18 +296,20 @@ 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); }
@@ -292,17 +318,19 @@ 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;
@@ -310,15 +338,17 @@ 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;
@@ -336,8 +366,9 @@ 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;
         }
@@ -348,17 +379,20 @@ st.markdown("""
     </div>
 """, unsafe_allow_html=True)
 
+# ---------- مدل زبانی ----------
 llm = ChatOpenAI(
     base_url="https://api.together.xyz/v1",
     api_key='0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979',
     model="meta-llama/Llama-3.3-70B-Instruct-Turbo-Free",
-    max_tokens=2048  # specify the max tokens as per your requirement
+    max_tokens=1024  # specify the max tokens as per your requirement
 
 )
 
 
+# ---------- ورودی جستجو ----------
 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;  /* گوشه‌های گرد */
@@ -385,37 +419,41 @@ st.markdown("""
 
 st.markdown("""
     <style>
+        /* استایل برای کلاس st-emotion-cache-yd4u6l e1togvvn1 */
         .st-emotion-cache-yd4u6l.e1togvvn1 {
-            background-color: rgba(106, 127, 83, 0.8) !important;  
-            border-radius: 10px !important; 
-            color: #d4d4d4 !important;  
+            background-color: rgba(106, 127, 83, 0.8) !important;  /* سبز خاکی مایل به زرد (#6a7f53) */
+            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;
 }
@@ -437,6 +475,7 @@ if query:
 else:
     st.markdown("")
 
+# استایل‌ها برای چرخش و پیام در حال فکر کردن
 st.markdown("""
     <style>
     .thinking-message {
@@ -450,7 +489,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;
@@ -518,14 +557,13 @@ def extract_keywords_from_text(text, query_words):
     return matched_lines
 
 def clean_text(text):
-    return re.sub(r'[^آ-ی0-9۰-۹،.؟!؛+\-* ]+', '', text)
-
+    return re.sub(r'[^آ-ی۰-۹0-9،.؟!؛+\-* ]+', '', text)
 
 
 from collections import Counter
 import heapq
 
-def summarize_text_by_frequency(text, num_sentences=1):
+def summarize_text_by_frequency(text, num_sentences=3):
     sentences = text.split('\n')
     word_freq = Counter()
 
@@ -543,9 +581,11 @@ 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):
+def find_closest_lines(query, doc_texts, stop_words, top_n=10):
     cleaned_query = remove_stop_words(query, stop_words)
     query_words = cleaned_query.split()
 
@@ -554,7 +594,7 @@ def find_closest_lines(query, doc_texts, stop_words, top_n=5):
     for filename, text in doc_texts.items():
         matched_lines = extract_keywords_from_text(text, query_words)
         for line in matched_lines:
-            similarity = fuzz.partial_ratio(query, line)  
+            similarity = fuzz.partial_ratio(query, line)  # محاسبه شباهت خط با سوال
             all_matched_lines.append((line, similarity))
     
     all_matched_lines.sort(key=lambda x: x[1], reverse=True)
@@ -572,23 +612,24 @@ def remove_stop_words_from_lines(lines, stop_words):
     return cleaned_lines
 
 if query:
-    closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=5)
-
-    # حذف استپ‌ورد و پاکسازی خطوط
+    closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=10)
+    
+    # حذف استپ‌وردها از خطوط و سپس پاکسازی نهایی متن
     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)
 
-    if summarized_text.strip():
+
+    if summarized_text:
+        # cleaned_text = "\n".join(cleaned_closest_lines[:1])
+
         prompt = f"""
-        لطفاً با توجه به سؤال زیر و محتوای خلاصه‌شده، یک پاسخ نهایی حرفه‌ای، دقیق و روان تولید ک��. فقط از متن استفاده کن. اگر اطلاعات کافی در متن وجود ندارد، صادقانه اعلام کن .
+        لطفاً با توجه به سؤال زیر و محتوای خطوط مرتبط، یک پاسخ نهایی حرفه‌ای، دقیق و روان تولید کن. فقط از متن خطوط مرتبط استفاده کن. اگر اطلاعات کافی در متن وجود ندارد، صادقانه اعلام کن.
         سوال:
         {query}
-        خلاصه‌ی مرتبط:
+        خطوط مرتبط:
         {summarized_text}
         پاسخ نهایی:
         """
@@ -602,4 +643,4 @@ if query:
         st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
 
     else:
-        st.warning("هیچ محتوای خلاصه‌شده‌ای برای پاسخ وجود ندارد.")
+        st.warning("هیچ خط مرتبطی پیدا نشد.")