Update app_chat.py

app_chat.py

@@ -42,14 +42,14 @@ def app_chat(agent_config):
     
     if user_message := st.chat_input("Enter message"):
         st.chat_message("user").markdown(user_message)
-        st.session_state.messages.append({"role": "user", "content": user_message})
+        st.session_state.messages.append({"role": "user", "content": user_message, "avatar": "🤗"})
     
         selected_tools = [agent_config.tool_loader.tools[idx] for idx, checkbox in enumerate(agent_config.tool_checkboxes) if checkbox]
         # Handle submission with the selected inference URL
         #app_agent_config()
 
         response = ""
-        chat_response = ""
+        chat_respone = ""
         with st.spinner('Please stand by ...'):
 
             response = handle_submission(user_message, selected_tools, agent_config.url_endpoint, agent_config.document, agent_config.image, agent_config.context)
@@ -119,6 +119,6 @@ def app_chat(agent_config):
             else:
                 st.warning("Unrecognized response type. Please try again. e.g. Generate an image of a flying horse.")
     
-        st.session_state.messages.append({"role": "assistant", "content": chat_response})
-        st.session_state.messages.append({"role": "assistant", "content": response})
+        st.session_state.messages.append({"role": "assistant", "content": chat_respone, "avatar": "🦖"})
+        st.session_state.messages.append({"role": "assistant", "content": response, "avatar": "🤖"})