Update pages/machine_learning.py

pages/machine_learning.py

@@ -148,10 +148,16 @@ with col1:
         st.session_state[PAGE_KEY].append((user_input, result.content))
 
     st.subheader("🗨️ Chat History")
-    for user, bot in st.session_state[PAGE_KEY]:
-        st.markdown(f'<div class="chat-bubble user-bubble">{user}</div>', unsafe_allow_html=True)
-        st.markdown(f'<div class="chat-bubble mentor-bubble">{bot}</div>', unsafe_allow_html=True)
+    # for user, bot in st.session_state[PAGE_KEY]:
+    #     st.markdown(f'<div class="chat-bubble user-bubble">{user}</div>', unsafe_allow_html=True)
+    #     st.markdown(f'<div class="chat-bubble mentor-bubble">{bot}</div>', unsafe_allow_html=True)
 
+if st.session_state[PAGE_KEY]:
+    st.markdown('<div class="chat-container">', unsafe_allow_html=True)
+    for user, bot in st.session_state[PAGE_KEY]:
+        st.markdown(f'<div class="chat-bubble-user">👤 <strong>You:</strong> {user}</div>', unsafe_allow_html=True)
+        st.markdown(f'<div class="chat-bubble-bot">🧑‍🏫 <strong>Mentor:</strong> {bot}</div>', unsafe_allow_html=True)
+    st.markdown('</div>', unsafe_allow_html=True)
 # --- Mentor Tips Sidebar ---
 # with col2:
 #     st.markdown("### 💡 Tips from Mentor")