Update app.py

app.py

@@ -15,6 +15,8 @@ prompt_template = """Answer the question only according to the information provi
 ## Question : 
 # {}"""
 
+st.session_state.messages = []
+
 if not "stage" in st.session_state:
     st.session_state.stage = "crawl-input"
 elif st.session_state.stage == "remove-crawl-input-widgets":
@@ -35,11 +37,15 @@ if st.session_state.stage == "crawl-input":
         st.session_state.stage = "remove-crawl-input-widgets"
 
 if st.session_state.stage == "prompt-input":
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message)
     prompt = st.chat_input("Ask the documentation a question!")
     
     if prompt:
         with st.chat_message("user"):
             st.markdown(prompt)
+        st.session_state.messages.append({"role":"user","message":prompt})
         results = st.session_state.collection.query(query_texts=[prompt],n_results=5)
         infos = results["documents"][0]
         
@@ -63,4 +69,5 @@ if st.session_state.stage == "prompt-input":
                 full_response += chunk + " "
                 time.sleep(0.05)
                 message_placeholder.markdown(full_response + "▌")
-            message_placeholder.markdown(full_response)
+            message_placeholder.markdown(full_response)
+        st.session_state.messages.append({"role":"assistant","message":full_response})