Update app.py

app.py

@@ -127,9 +127,9 @@ if submit_button and user_input:
 
 # Display chat history within the container
 with chat_history_container:
-    for i in range(len(st.session_state['generated'])):
-        st.text(st.session_state["past"][i], key=str(i) + '_user')
-        st.text(st.session_state["generated"][i], key=str(i))
+   for i in range(len(st.session_state['generated'])):
+        message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="big-smile")
+        message(st.session_state["generated"][i], key=str(i), avatar_style="thumbs")
 
 # Ensure chat history is always scrollable
 st.markdown("""