Update src/streamlit_app.py

src/streamlit_app.py

@@ -243,7 +243,12 @@ if cv_text:
     # Display the extracted information
     if extracted_info:
         st.subheader("Extracted Information")
-        st.write(extracted_info)
+        with st.chat_message("assistant", avatar="🤖"):
+            st.markdown(
+                f"<div class='chat-message assistant'>{extracted_info}</div>",
+                unsafe_allow_html=True,
+            )
+        # st.write(extracted_info)
     else:
         st.write("Could not extract information. Please try again.")
 
@@ -254,7 +259,13 @@ if cv_text:
     # Display the extracted information
     if extracted_criterions:
         st.subheader("Extracted Criterion List")
-        st.write(extracted_criterions)
+        # Immediately display the assistant's response
+        with st.chat_message("assistant", avatar="🤖"):
+            st.markdown(
+                f"<div class='chat-message assistant'>{extracted_criterions}</div>",
+                unsafe_allow_html=True,
+            )
+        # st.write(extracted_criterions)
     else:
         st.write("Could not extract criterion information. Please try again.")
         
@@ -264,6 +275,11 @@ if cv_text:
     # Display the extracted information
     if extracted_weights:
         st.subheader("Extracted Weights List")
-        st.write(extracted_weights)
+        with st.chat_message("assistant", avatar="🤖"):
+            st.markdown(
+                f"<div class='chat-message assistant'>{extracted_weights}</div>",
+                unsafe_allow_html=True,
+            )
+        # st.write(extracted_weights)
     else:
         st.write("Could not extract weights information. Please try again.")