Update app.py

app.py

@@ -198,7 +198,9 @@ def generate(system, prompt):
         # Add assistant response to chat history
         st.session_state.messages.append({"role": "assistant", "content": response})
 
-
+def clear():
+    for key in st.session_state.keys():
+        del st.session_state[key]
 # UI main #####################################################
 
 tab1, tab2 = st.tabs(["Prompts Library", "Chatbot"])
@@ -244,6 +246,8 @@ with tab1:
                     delete_prompt(f["title"])
                     st.rerun()
 with tab2:
+    if st.button("Clear chatbot history", type="secondary"):
+        clear()
     # Initialize chat history
     if "messages" not in st.session_state:
         st.session_state.messages = []