Update src/streamlit_app.py

src/streamlit_app.py

@@ -34,17 +34,23 @@ model_choice = st.sidebar.selectbox("Choose a model", [
 #     st.session_state.messages = []
 #     st.rerun()
 def clear_chat():
-    # Reset session state
+    # COMPLETELY reset session state (not just messages)
+    st.session_state.clear()  # 🔥 This wipes ALL session variables
+    # Then re-initialize just what you need
     st.session_state.messages = []
-    # Inject JavaScript to force refresh
+    # Force a hard refresh (JavaScript method)
     js = """
     <script>
-        window.location.reload();
+        window.parent.location.reload(true);
     </script>
     """
     html(js)
 
-st.sidebar.button("🧹 Clear Chat", on_click=clear_chat)
+st.sidebar.button(
+    "🧹 Clear Chat", 
+    on_click=clear_chat, 
+    key="clear_chat_button_unique_123"  # Unique key prevents duplicate ID errors
+)
 
 
 # ==== Session Initialization ====