Update app.py

app.py

@@ -25,6 +25,7 @@ if 'memory' not in st.session_state:
     # st.session_state['memory'] = ConversationBufferMemory(return_messages=True)
     
     st.session_state.memory = PostgresChatMessageHistory(connection_string=POSTGRE_URL, session_id=str(datetime.timestamp(datetime.now())))
+    st.session_state.memory.add_ai_message("Hello, My name is Jonathan Jordan. You can call me Jojo. How can I help you today?")
 
 if 'chain' not in st.session_state:
     st.session_state['chain'] = custom_chain_with_history(
@@ -57,8 +58,10 @@ if prompt := st.chat_input("Ask me anything.."):
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
         st.markdown(response)
-    st.session_state.memory.save_context({"question":prompt}, {"output":response})
-    st.session_state.memory.chat_memory.messages = st.session_state.memory.chat_memory.messages[-15:]
+    st.session_state.memory.add_user_message(prompt)
+    st.session_state.memory.add_ai_message(response)
+    # st.session_state.memory.save_context({"question":prompt}, {"output":response})
+    # st.session_state.memory.chat_memory.messages = st.session_state.memory.chat_memory.messages[-15:]
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})