Update app.py

app.py

@@ -108,13 +108,12 @@ chat = RunnableWithMessageHistory(
 if "messages" not in st.session_state:
     st.session_state.messages = load_chat_history(session_id)
 
-chat_container = st.container()
 
 # Display chat history
-with chat_container:
-    for role, content in st.session_state.messages:
-        with st.chat_message(role):
-            st.markdown(content)
+
+for role, content in st.session_state.messages:
+    with st.chat_message(role):
+        st.markdown(content)
 
 # User input at the bottom
 user_input = st.text_input("Type your message here:", key="user_message")