Update app.py

app.py

@@ -594,16 +594,25 @@ def find_closest_lines(query, doc_texts, stop_words, top_n=20, exclude_line=None
 
 # حالا این رو در کد اصلی استفاده می‌کنیم:
 if query:
-    # پیدا کردن ۲۰ خط نزدیک‌تر به سوال (و حذف یک خط خاص)
-    closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=20, exclude_line=None)
+    # حذف کلمات اضافی از سوال
+    cleaned_query = remove_stop_words(query, stop_words)
+    query_words = cleaned_query.split()
+
+    all_matched_lines = []
+
+    # بررسی محتوای فایل‌ها
+    for filename, text in doc_texts.items():
+        matched_lines = extract_keywords_from_text(text, query_words)
+        all_matched_lines.extend(matched_lines)
 
-    if closest_lines:
+    # حالا این لیست را به مدل می‌دهیم
+    if all_matched_lines:
         prompt = f"""
         لطفاً با توجه به سؤال زیر و محتوای خطوط مرتبط، یک پاسخ نهایی حرفه‌ای، دقیق و روان تولید کن. فقط از متن خطوط مرتبط استفاده کن. اگر اطلاعات کافی در متن وجود ندارد، صادقانه اعلام کن.
         سوال:
         {query}
         خطوط مرتبط:
-        {closest_lines}
+        {all_matched_lines}
         پاسخ نهایی:
         """