Update app.py

app.py

@@ -113,6 +113,12 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    if message["role"] != "system":  # Skip system messages
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
 # Path to the Markdown file
 md_file_path = 'docs/yiqiao_yin.md'