M17idd commited on
Commit
3260cb2
·
1 Parent(s): 0d7d4fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -567,7 +567,7 @@ def clean_text(text):
567
  return re.sub(r'[^آ-ی۰-۹0-9،.؟!؛+\-* ]+', '', text)
568
 
569
  # تابعی برای پیدا کردن نزدیک‌ترین خطوط به سوال
570
- def find_closest_lines(query, doc_texts, stop_words, top_n=10, exclude_line=None):
571
  # حذف کلمات اضافی از سوال
572
  cleaned_query = remove_stop_words(query, stop_words)
573
  query_words = cleaned_query.split()
@@ -587,16 +587,12 @@ def find_closest_lines(query, doc_texts, stop_words, top_n=10, exclude_line=None
587
  # انتخاب ۱۰ خط نزدیک‌تر
588
  closest_lines = [line for line, _ in all_matched_lines[:top_n]]
589
 
590
- # حذف خط خاص از لیست در صورت وجود
591
- if exclude_line and exclude_line in closest_lines:
592
- closest_lines.remove(exclude_line)
593
-
594
  return closest_lines
595
 
596
  # حالا این رو در کد اصلی استفاده می‌کنیم:
597
  if query:
598
- # پیدا کردن ۱۰ خط نزدیک‌تر به سوال (و حذف یک خط خاص)
599
- closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=10, exclude_line=None)
600
 
601
  if closest_lines:
602
  prompt = f"""
@@ -617,5 +613,4 @@ if query:
617
  st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
618
 
619
  else:
620
- st.warning("هیچ خط مرتبطی پیدا نشد.")
621
-
 
567
  return re.sub(r'[^آ-ی۰-۹0-9،.؟!؛+\-* ]+', '', text)
568
 
569
  # تابعی برای پیدا کردن نزدیک‌ترین خطوط به سوال
570
+ def find_closest_lines(query, doc_texts, stop_words, top_n=10):
571
  # حذف کلمات اضافی از سوال
572
  cleaned_query = remove_stop_words(query, stop_words)
573
  query_words = cleaned_query.split()
 
587
  # انتخاب ۱۰ خط نزدیک‌تر
588
  closest_lines = [line for line, _ in all_matched_lines[:top_n]]
589
 
 
 
 
 
590
  return closest_lines
591
 
592
  # حالا این رو در کد اصلی استفاده می‌کنیم:
593
  if query:
594
+ # پیدا کردن ۱۰ خط نزدیک‌تر به سوال
595
+ closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=10)
596
 
597
  if closest_lines:
598
  prompt = f"""
 
613
  st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
614
 
615
  else:
616
+ st.warning("هیچ خط مرتبطی پیدا نشد.")