working with demo question

chatbot.py

@@ -14,8 +14,11 @@ st.set_page_config(
 st.header("Streamlit Chat - Demo")
 # st.markdown("[Github](https://github.com/ai-yash/st-chat)")
 
-generated = []
-past = []
+if 'generated' not in st.session_state:
+    st.session_state['generated'] = ['new']
+
+if 'past' not in st.session_state:
+    st.session_state['past'] = ['new']
 
 def query(chat, payload):
     response = chat.answerStoreHistory(qn=payload)
@@ -34,11 +37,11 @@ chat = llm.get_chat(context="You are an helpfully assistant in a school. You are
 if user_input:
     output = chat.answerStoreHistory(qn=user_input)
 
-    past.append(user_input)
-    generated.append(output)
+    st.session_state.past.append(user_input)
+    st.session_state.generated.append(output)
 
-if generated:
+if st.session_state['generated']:
 
-    for i in range(len(generated) - 1, -1, -1):
-        message(generated[i], key=str(i))
-        message(past[i], is_user=True, key=str(i) + '_user')
+    for i in range(len(st.session_state['generated'])-1, -1, -1):
+        message(st.session_state["generated"][i], key=str(i))
+        message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')