update streamlit_app.py 4

streamlit_app.py

@@ -22,7 +22,10 @@ def main():
     # Button to clear chat history
     if st.button("Clear Chat History"):
         st.session_state.messages = []
-        st.rerun()
+        if hasattr(st, "rerun"):
+            st.rerun()
+        else:
+            st.experimental_rerun()
 
     # Initialize session state
     if "messages" not in st.session_state: