Update src/streamlit_app.py

src/streamlit_app.py

@@ -33,9 +33,10 @@ model_choice = st.sidebar.selectbox("Choose a model", [
 # if st.sidebar.button("🧹 Clear Chat"):
 #     st.session_state.messages = []
 #     st.rerun()
-if st.sidebar.button("🧹 Clear Chat", key="clear_chat"):
-    st.session_state.messages = []  # Reset chat history
-    st.experimental_rerun()  # Force a rerun (works better in Spaces)
+def clear_chat():
+    st.session_state.messages = []
+
+st.sidebar.button("🧹 Clear Chat", on_click=clear_chat)
 
 # ==== Session Initialization ====
 if "messages" not in st.session_state: