Update pages/python.py

pages/python.py

@@ -105,7 +105,13 @@ if submit and user_input:
 
 # Chat history
 st.subheader("🗨️ Chat History")
-for user, bot in st.session_state[PAGE_KEY]:
-    st.markdown(f"**You:** {user}")
-    st.markdown(f"**Mentor:** {bot}")
-    st.markdown("---")
+# for user, bot in st.session_state[PAGE_KEY]:
+#     st.markdown(f"**You:** {user}")
+#     st.markdown(f"**Mentor:** {bot}")
+#     st.markdown("---")
+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)