Update app.py

app.py

@@ -26,6 +26,16 @@ def query(payload):
         return response
 
 messages = st.container()
+
+# Initialize chat history
+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"])
+        
 message = st.chat_message("assistant").write("Hello human")
 if prompt := st.chat_input("Subject of the song"):
     messages.chat_message("user").write(prompt)