Update app.py

app.py

@@ -309,16 +309,7 @@ for tid, chats in st.session_state.chat_threads.items():
                 avatar = user_avatar if msg['role'] == 'user' else assistant_avatar
                 bubble_color = "#DCF8C6" if msg['role'] == 'user' else "#F0F0F0"
                 align = "flex-end" if msg['role'] == 'user' else "flex-start"
-                st.markdown(f"""
-                    <div style="display: flex; justify-content: {align}; margin-bottom: 10px;">
-                        <img src="{avatar}" width="30" style="margin-right: 10px; border-radius: 50%;">
-                        <div style="background-color: {bubble_color}; padding: 10px 15px; border-radius: 10px; max-width: 70%;">
-                            <p style="margin: 0; color: black;">{msg['content']}</p>
-                        </div>
-                    </div>
-                    """, unsafe_allow_html=True)
-
-
+                st.markdown({msg['content'])
             followup = st.text_input(f"💬 Continue Thread {tid[:8]}:", key=f"followup_{tid}")
             if st.button(f"Ask Follow-up {tid}", key=f"button_{tid}"):
                 if followup: