Update app.py

app.py

@@ -627,9 +627,16 @@ if query:
         ])
         rewritten = clean_text(response.content.strip())
 
-        st.markdown(f'<div class="chat-message">{rewritten}</div>', unsafe_allow_html=True)
+        # ذخیره پیام‌های چت
+        st.session_state.chat_history.append(("🧑", query))
+        st.session_state.chat_history.append(("🤖", rewritten))
+
         think.empty()
 
     else:
         st.warning("هیچ خط مرتبطی پیدا نشد.")
-        think.empty()
+        think.empty()
+
+# نمایش تاریخچه گفتگو
+for sender, msg in st.session_state.chat_history:
+    st.markdown(f"{sender} {msg}")