Update app.py

app.py

@@ -42,12 +42,12 @@ if st.session_state.stage == "prompt-input":
     if prompt:
         for message in st.session_state.messages:
             with st.chat_message(message["role"]):
-                st.markdown(message["message"])
+                st.markdown(message["content"])
         
         with st.chat_message("user"):
             st.markdown(prompt)
                 
-        st.session_state.messages.append({"role":"user","message":prompt})
+        st.session_state.messages.append({"role":"user","content":prompt})
         results = st.session_state.collection.query(query_texts=[prompt],n_results=5)
         infos = results["documents"][0]
         
@@ -72,4 +72,4 @@ if st.session_state.stage == "prompt-input":
                 time.sleep(0.05)
                 message_placeholder.markdown(full_response + "▌")
             message_placeholder.markdown(full_response)
-        st.session_state.messages.append({"role":"assistant","message":full_response})
+        st.session_state.messages.append({"role":"assistant","content":full_response})