Update app.py

app.py

@@ -544,20 +544,33 @@ def clean_text(text):
     return re.sub(r'[^آ-ی۰-۹0-9،.؟!؛+\-* ]+', '', text)
 
 
-def find_closest_filename(query, filenames):
+def find_closest_filenames(query, filenames, top_n=3):
+    # گرفتن نزدیک‌ترین فایل‌ها بر اساس شباهت
     scores = [(f, fuzz.partial_ratio(query, f)) for f in filenames]
     scores.sort(key=lambda x: x[1], reverse=True)
-    return scores[0][0] if scores else None
+    return [score[0] for score in scores[:top_n]]  # برگشت دادن N فایل با بیشترین شباهت
 
+def find_best_answer(query, top_files, doc_texts):
+    best_match = None
+    best_score = 0
 
-# فرض بر این است که متغیر query توسط کاربر مشخص شده است
+    # بررسی محتوای فایل‌ها
+    for filename in top_files:
+        text = doc_texts[filename]
+        similarity = fuzz.partial_ratio(query, text)  # مقایسه سوال با محتوای فایل
+        if similarity > best_score:
+            best_score = similarity
+            best_match = filename
+
+    return best_match, doc_texts.get(best_match, "")
+
+
+# حالا این رو در کد اصلی استفاده می‌کنیم:
 if query:
-    closest_file = find_closest_filename(query, list(doc_texts.keys()))
+    top_files = find_closest_filenames(query, list(doc_texts.keys()), top_n=3)
+    best_file, matched_text = find_best_answer(query, top_files, doc_texts)
 
-    if closest_file:
-        matched_text = doc_texts[closest_file]
-        st.markdown(scores[0][0])
-        st.markdown(matched_text)
+    if best_file:
         prompt = f"""
         لطفاً با توجه به سؤال زیر و محتوای سند موجود، یک پاسخ نهایی حرفه‌ای، دقیق و روان تولید کن. فقط از متن سند استفاده کن. اگر اطلاعات کافی در متن وجود ندارد، صادقانه اعلام کن.
         سوال:
@@ -574,8 +587,7 @@ if query:
         rewritten = clean_text(response.content.strip())
 
         st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
-        think.empty()
 
     else:
         st.warning("هیچ سند مرتبطی پیدا نشد.")
-        think.empty()
+