Update app.py

app.py

@@ -154,9 +154,20 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Initialize chat history
+# Initialize chat history in session state
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]
 
+# Chat Display
+st.markdown("<div class='container'>", unsafe_allow_html=True)
+
+for message in st.session_state.chat_history:
+    if message["role"] == "user":
+        st.markdown(f"<div class='user-msg'><strong>You:</strong> {message['content']}</div>", unsafe_allow_html=True)
+    else:
+        st.markdown(f"<
+
+
 # Sidebar for chat reset
 if st.sidebar.button("Reset Chat"):
     st.session_state.chat_history = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]