Update app.py

app.py

@@ -571,57 +571,30 @@ def remove_stop_words_from_lines(lines, stop_words):
     return cleaned_lines
 
 if query:
-    closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=5)
+    closest_lines = find_closest_lines(query, doc_texts, stop_words, top_n=10)
 
+    # حذف استپ‌ورد و پاکسازی خطوط
     cleaned_closest_lines = [
         clean_text(" ".join([word for word in line.split() if word not in stop_words]))
         for line in closest_lines
     ]
 
-    summarized_text = summarize_text_by_frequency("\n".join(cleaned_closest_lines), num_sentences=1)
+    # خلاصه‌سازی
+    summarized_text = summarize_text_by_frequency("\n".join(cleaned_closest_lines), num_sentences=3)
 
-    if summarized_text:
+    if summarized_text.strip():
         prompt = f"""
-        تعدادی جمله مرتبط با سوال زیر استخراج شده است. لطفاً ابتدا این جملات را بررسی کن، سپس با در نظر گرفتن محتوای سوال و لحن آن، یک پاسخ نهایی حرفه‌ای، دقیق و روان ارائه کن که هم به سوال پاسخ دهد و هم از نظر نگارشی و ساختاری در سطح بالایی باشد. پاسخ نهایی باید حداکثر 1024 کاراکتر و حداقل 512 کاراکتر باشد، خلاصه و واضح نوشته شود و فقط به زبان فارسی باشد. از تکرار اضافی پرهیز کن و در صورت نیاز، محتوای جملات را ترکیب کن.
+        لطفاً با توجه به سؤال زیر و محتوای خلاصه‌شده، یک پاسخ نهایی حرفه‌ای، دقیق و روان تولید کن. فقط از متن استفاده کن. اگر اطلاعات کافی در متن وجود ندارد، صادقانه اعلام کن و  از دانش کمک بگیر   بنویس و  بنویس ک  از دانش خودت استفاده کردی.
         سوال:
         {query}
-        جملات مرتبط:
+        خلاصه‌ی مرتبط:
         {summarized_text}
-        پاسخ نهایی حرفه‌ای بازنویسی‌شده:
+        پاسخ نهایی:
         """
-        response = llm([
-            SystemMessage(content="You are a helpful assistant."),
-            HumanMessage(content=prompt)
-        ])
-        rewritten = response.content.strip()
-
-        review_prompt = f"""
-        لطفاً بررسی کن که آیا پاسخ زیر به سوال داده‌شده مرتبط، صحیح، معتبر و قابل قبول است یا نه. اگر پاسخ مرتبط و صحیح است، فقط بنویس: 'تأیید شد'. اگر پاسخ اشتباه است یا ربطی به سوال ندارد یا معتبر نیست، فقط بنویس: 'نیاز به اصلاح دارد'.
-        سوال:
-        {query}
-        پاسخ:
-        {rewritten}
-        """
-        review_response = llm([
-            SystemMessage(content="You are a helpful assistant."),
-            HumanMessage(content=review_prompt)
-        ])
-        review_result = review_response.content.strip()
-
-        if "تأیید شد" in review_result:
-            rewritten = clean_text(rewritten)
-            st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
-            think.empty()  
-    else:
-        prompt = f"""
-        هیچ جمله مرتبط مستقیمی با سوال یافت نشد. لطفاً با توجه به سوال زیر، یک پاسخ حرفه‌ای، مرتبط، معنادار و معتبر تولید کن. فقط از جملات موجود در اسناد معتبر استفاده کن و از ساخت جملات ساختگی یا استفاده از زبان غیر فارسی خودداری کن.
-        سوال:
-        {query}
-        """
-        response = llm([
-            SystemMessage(content="You are a helpful assistant."),
-            HumanMessage(content=prompt)
-        ])
+
         rewritten = clean_text(response.content.strip())
+
         st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
-        think.empty() 
+
+    else:
+        st.warning("هیچ محتوای خلاصه‌شده‌ای برای پاسخ وجود ندارد.")