Update app.py

app.py

@@ -20,6 +20,25 @@ if "generated" not in st.session_state:
 if "input" not in st.session_state:
     st.session_state["input"] = ""
 
+
+
+
+def new_chat():
+    """
+    Clears session state and starts a 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["stored_session"].append(save)
+    st.session_state["generated"] = ""
+    st.session_state["input"] = ""
+    st.session_state.entity_memory.entity_store = {}
+    st.session_state.entity_memory.buffer.clear()
+
+
+    
 # Define function to get user input
 def get_text():
     """