Update app.py

app.py

@@ -539,6 +539,7 @@ def load_and_process_documents(path):
 
 doc_texts = load_and_process_documents(folder_path)
 
+# لیست کلمات توقف
 stop_words = [
     "است", "و", "با", "که", "در", "از", "برای", "به", "بر", "تا", "این", "آن", "یک", "کدام", "کجا", "هم", "همه",
     "یا", "از", "بر", "همچنین", "می", "باید", "شود", "شد", "گفت", "گویا", "داشت", "داشتن", "کنند", "کنیم", 
@@ -566,7 +567,7 @@ def clean_text(text):
     return re.sub(r'[^آ-ی۰-۹0-9،.؟!؛+\-* ]+', '', text)
 
 # تابعی برای پیدا کردن نزدیک‌ترین خطوط به سوال
-def find_closest_lines(query, doc_texts, stop_words, top_n=20, exclude_line=None):
+def find_closest_lines(query, doc_texts, stop_words, top_n=10, exclude_line=None):
     # حذف کلمات اضافی از سوال
     cleaned_query = remove_stop_words(query, stop_words)
     query_words = cleaned_query.split()
@@ -583,7 +584,7 @@ def find_closest_lines(query, doc_texts, stop_words, top_n=20, exclude_line=None
     # مرتب سازی بر اساس شباهت
     all_matched_lines.sort(key=lambda x: x[1], reverse=True)
 
-    # انتخاب ۲۰ خط نزدیک‌تر
+    # انتخاب ۱۰ خط نزدیک‌تر
     closest_lines = [line for line, _ in all_matched_lines[:top_n]]
 
     # حذف خط خاص از لیست در صورت وجود
@@ -594,25 +595,16 @@ def find_closest_lines(query, doc_texts, stop_words, top_n=20, exclude_line=None
 
 # حالا این رو در کد اصلی استفاده می‌کنیم:
 if query:
-    # حذف کلمات اضافی از سوال
-    cleaned_query = remove_stop_words(query, stop_words)
-    query_words = cleaned_query.split()
-
-    all_matched_lines = []
+    # پیدا کردن ۱۰ خط نزدیک‌تر به سوال (و حذف یک خط خاص)
+    closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=10, exclude_line=None)
 
-    # بررسی محتوای فایل‌ها
-    for filename, text in doc_texts.items():
-        matched_lines = extract_keywords_from_text(text, query_words)
-        all_matched_lines.extend(matched_lines)
-
-    # حالا این لیست را به مدل می‌دهیم
-    if all_matched_lines:
+    if closest_lines:
         prompt = f"""
         لطفاً با توجه به سؤال زیر و محتوای خطوط مرتبط، یک پاسخ نهایی حرفه‌ای، دقیق و روان تولید کن. فقط از متن خطوط مرتبط استفاده کن. اگر اطلاعات کافی در متن وجود ندارد، صادقانه اعلام کن.
         سوال:
         {query}
         خطوط مرتبط:
-        {all_matched_lines}
+        {closest_lines}
         پاسخ نهایی:
         """
 
@@ -626,3 +618,4 @@ if query:
 
     else:
         st.warning("هیچ خط مرتبطی پیدا نشد.")
+