Update app.py

app.py

@@ -103,20 +103,31 @@ chat = RunnableWithMessageHistory(
 )
 
 
+# 🔹 Chat history container
+chat_container = st.container()
+
+# 🔹 Always keep input box at the bottom
+input_container = st.empty()
 
 # Load chat history and keep it sticky
 if "messages" not in st.session_state:
     st.session_state.messages = load_chat_history(session_id)
 
 
+# Display chat messages in order
+with chat_container:
+    for role, content in st.session_state.messages:
+        with st.chat_message(role):
+            st.markdown(content)
 # Display chat history
 
 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")
+with input_container:
+    # User input at the bottom
+    user_input = st.text_input("Type your message here:", key="user_message")
 
 if user_input:
     # Save user message