Update app.py

app.py

@@ -22,6 +22,20 @@ def get_text():
   input_text = st.text_input("You: ", st.session_state["input"], key = "input", placeholder = "Your AI Assistant here.. Ask me anything!", label_visibility = "hidden" )
   return input_text
 
+def new_chat():
+    save = []
+    for i in range(len(st.session_state["generated"])-1,-1,-1):
+        save.append("User:" + st.session_state["past"][i])
+        save.append("Bot:" + st.session_state["generated"][i])
+    st.session_state["started_session"].append(save)
+    st.session_state["generated"] = []
+    st.session_state["past"] = []
+    st.session_state["input"] = ""
+    st.session_state.entity_memory.store = {}
+    st.session_state.entity_memory.buffer.clear()
+    
+
+
 st.title("Memory Bot")
 
 api = st.sidebar.text_input("API-key", type = "password")
@@ -52,7 +66,9 @@ if api:
     )
 
 else:
-    st.error("No API found!")
+    st.error("No API key found!")
+
+st.sidebar.button("New Chat", on_click = new_chat, type = "primary" )
 
 user_input = get_text()