Update app.py

app.py

@@ -47,6 +47,7 @@ def chat_history(session_id):
 # Generate unique session ID
 if "session_id" not in st.session_state:
     st.session_state.session_id = str(uuid.uuid4())
+    st.session_state.messages = []
 
 # Sidebar for session management
 with st.sidebar:
@@ -116,16 +117,17 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Display Chat Messages
-st.markdown("<div class='chat-container'>", unsafe_allow_html=True)
-
+# Load chat history
 if "messages" not in st.session_state:
     st.session_state.messages = load_chat_history(session_id)
+else:
+    st.session_state.messages = [(role, content) for role, content in load_chat_history(session_id)]
 
+# Display Chat Messages
+st.markdown("<div class='chat-container'>", unsafe_allow_html=True)
 for role, content in st.session_state.messages:
     role_class = "user-message" if role == "user" else "assistant-message"
     st.markdown(f"<div class='chat-message {role_class}'><b>{role.capitalize()}:</b> {content}</div>", unsafe_allow_html=True)
-
 st.markdown("</div>", unsafe_allow_html=True)
 
 # User Input Box