button to clear chat data

code/pages/00_Chat.py

@@ -2,6 +2,11 @@
 from streamlit_chat import message
 from utilities.helper import LLMHelper
 
+def clear_chat_data():
+    st.session_state['input'] = ""
+    st.session_state['chat_history'] = []
+    st.session_state['source_documents'] = []
+
 # Initialize chat history
 if 'chat_history' not in st.session_state:
     st.session_state['chat_history'] = []
@@ -12,6 +17,7 @@
 
 # Chat 
 input_text = st.text_input("You: ", placeholder="type your question", key="input")
+clear_chat = st.button("Clear chat", key="clear_chat", on_click=clear_chat_data)
 
 if input_text:
     question = input_text