enhanced the chat ui