Update app.py

app.py

@@ -283,21 +283,34 @@ if st.session_state.pending_prompt:
             context = "\n".join([doc.page_content for doc in docs])
 
             # ساخت پرامپت نهایی برای LLM
-            final_prompt = f"""با توجه به اطلاعات زیر پاسخ بده. فقط به زبان فارسی پاسخ بده:
-اطلاعات:\n{context}\n\nسوال: {query}
+            final_prompt = f"""با توجه به اطلاعات زیر، بهترین جواب رو فقط از داخل این اطلاعات به زبان فارسی بده. لطفاً از خارج از این اطلاعات استفاده نکن و اگر اطلاعات کافی برای جواب دادن نبود، بهترین  پاسخ رو بده.
+
+🔹 اطلاعات:\n{context}\n\n❓ سؤال: {query}
 """
 
+
             # ارسال به LLM
             response = llm.invoke(final_prompt)
-            answer = response.content.strip()
-            if not answer:
-                answer = "متأسفم، اطلاعات دقیقی در این مورد ندارم."
+            raw_answer = response.content.strip()
+
+            # فیلتر کردن خطوط غیرپاسخ (مثلاً <think> یا توضیح مدل)
+            answer_lines = raw_answer.split('\n')
+            filtered_lines = [
+                line for line in answer_lines
+                if not line.strip().startswith("<")
+                and not line.strip().lower().startswith("think")
+                and not line.strip().startswith("#")
+                and not line.strip().lower().startswith("note")
+            ]
+            clean_answer = "\n".join(filtered_lines).strip()
+            if not clean_answer:
+                clean_answer = "متأسفم، اطلاعات دقیقی در این مورد ندارم."
 
             # تایپ کردن تدریجی پاسخ
             thinking.empty()
             full_response = ""
             placeholder = st.empty()
-            for word in answer.split():
+            for word in clean_answer.split():
                 full_response += word + " "
                 placeholder.markdown(full_response + "▌")
                 time.sleep(0.03)
@@ -308,4 +321,4 @@ if st.session_state.pending_prompt:
 
         except Exception as e:
             thinking.empty()
-            st.error(f"خطا در پاسخ‌دهی: {str(e)}")
+            st.error(f"خطا در پاسخ‌دهی: {str(e)}")