Update app.py

app.py

@@ -547,9 +547,17 @@ def find_closest_filenames(query, filenames, top_n=3):
     scores.sort(key=lambda x: x[1], reverse=True)
     return [score[0] for score in scores[:top_n]]
 
+from transformers import pipeline
+
+def summarize_text(text):
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+    summary = summarizer(text, max_length=150, min_length=50, do_sample=False)
+    return summary[0]['summary_text']
+
 def find_best_answer(query, top_files, doc_texts):
     best_match = None
     best_score = 0
+    best_text = ""
 
     for filename in top_files:
         text = doc_texts[filename]
@@ -557,8 +565,16 @@ def find_best_answer(query, top_files, doc_texts):
         if similarity > best_score:
             best_score = similarity
             best_match = filename
+            best_text = text
+
+    # خلاصه کردن متن برای خروجی بهتر
+    if best_text:
+        summary = summarize_text(best_text)
+    else:
+        summary = "متنی برای خلاصه‌سازی پیدا نشد."
+
+    return summary
 
-    return best_match, doc_texts.get(best_match, "")
 
 # حالا این رو در کد اصلی استفاده می‌کنیم:
 if query: