Update app.py

app.py

@@ -32,7 +32,14 @@ st.title(TITLE)
 st.write(DESCRIPTION)
 
 
-
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+        
 # React to user input
 if prompt := st.chat_input("Ask LLama-2-70b anything..."):
     # Display user message in chat message container