Update app.py

app.py

@@ -103,6 +103,14 @@ def main():
             if query:
                 embedding(text_chunks,query)
             st.success("Done")
+    if query:
+    st.session_state.messages.append({'role': 'user', "content": query})
+    response = embedding(text_chunks,query)
+    st.session_state.messages.append({'role': 'assistant', "content": response})
+
+for message in st.session_state.messages:
+    with st.chat_message(message['role']):
+        st.write(message['content'])
 
 if __name__ == "__main__":
     main()