Update app.py

app.py

@@ -588,18 +588,30 @@ def find_closest_lines(query, doc_texts, stop_words, top_n=10):
 
     return closest_lines
 
+# تابعی برای حذف کلمات توقف از یک لیست از خطوط
+def remove_stop_words_from_lines(lines, stop_words):
+    cleaned_lines = []
+    for line in lines:
+        words = line.split()
+        cleaned_words = [word for word in words if word not in stop_words]
+        cleaned_lines.append(" ".join(cleaned_words))
+    return cleaned_lines
+
 # حالا این رو در کد اصلی استفاده می‌کنیم:
 if query:
     # پیدا کردن ۱۰ خط نزدیک‌تر به سوال
     closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=3)
 
-    if closest_lines:
+    # حذف کلمات توقف از خطوط نزدیک
+    cleaned_closest_lines = remove_stop_words_from_lines(closest_lines, stop_words)
+
+    if cleaned_closest_lines:
         prompt = f"""
         لطفاً با توجه به سؤال زیر و محتوای خطوط مرتبط، یک پاسخ نهایی حرفه‌ای، دقیق و روان تولید کن. فقط از متن خطوط مرتبط استفاده کن. اگر اطلاعات کافی در متن وجود ندارد، صادقانه اعلام کن.
         سوال:
         {query}
         خطوط مرتبط:
-        {closest_lines}
+        {cleaned_closest_lines}
         پاسخ نهایی:
         """
 
@@ -612,4 +624,4 @@ if query:
         st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
 
     else:
-        st.warning("هیچ خط مرتبطی پیدا نشد.")
+        st.warning("هیچ خط مرتبطی پیدا نشد.")