Update app.py

app.py

@@ -276,6 +276,7 @@ if __name__ == '__main__':
             response = llm.invoke(question).content
             with st.chat_message("assistant"):
                 st.markdown(response)
+            st.session_state.messages.append({"role": "assistant", "content": response})
 
     if st.button("Reset Chat History"):
         #st.session_state['questions'] = []