Update src/streamlit_app.py

src/streamlit_app.py

@@ -30,9 +30,12 @@ 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"):
+#     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)
 
 # ==== Session Initialization ====
 if "messages" not in st.session_state: