Update app.py

app.py

@@ -411,7 +411,7 @@ def process_button_click(head, button_index, topic_desc, top_concept_index):
 
     st.session_state.topic_matrix[top_concept_index][button_index] += 1
 
-    st.write(response)
+    with st.chat_message("assistant"): st.write(response)
     if image and caption != "": st.image(image, caption)
 
     # ... (add response & image to message)