Update app.py

app.py

@@ -52,14 +52,15 @@ If a student asks for reward points, encourage them to engage in the session to
 """
 
 def main():
-    cs_sidebar()
     cs_body()
+    cs_sidebar(cs_body())
+
 
     return None
 
 # sidebar
 
-def cs_sidebar():
+def cs_sidebar(messages):
 
     st.sidebar.markdown('''
 <small>Summary of the [docs](https://docs.streamlit.io/en/stable/api.html), as of [Streamlit v1.8.0](https://www.streamlit.io/).</small>
@@ -111,7 +112,7 @@ def cs_body():
             message_placeholder.markdown(full_response)
         st.session_state.messages.append({"role": "assistant", "content": full_response})
 
-    return None
+    return st.session_state.messages
 
 # Run main()