如何在单词边界处换行包含ANSI颜色代码的Unicode文本？

import re

def wrap_text_with_ansi_colors(text, max_width):
    # 正则表达式匹配ANSI颜色代码
    ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])')
    
    # 分割文本为单词列表
    words = text.split()
    
    lines = []
    current_line = ""
    
    for word in words:
        # 检查当前行加上新单词是否超过最大宽度
        if len(current_line) + len(word) + (1 if current_line else 0) > max_width:
            lines.append(current_line)
            current_line = ""
        
        # 如果当前行不为空，添加一个空格
        if current_line:
            current_line += " "
        
        # 添加单词到当前行
        current_line += word
    
    # 添加最后一行
    if current_line:
        lines.append(current_line)
    
    # 处理ANSI颜色代码，确保它们不被分割
    wrapped_text = "\n".join(lines)
    wrapped_text = ansi_escape.sub(lambda m: m.group(0).replace("\n", ""), wrapped_text)
    
    return wrapped_text

# 示例文本
text = "This is a \x1B[31mred\x1B[0m example text with \x1B[34mblue\x1B[0m colors."
max_width = 20

wrapped_text = wrap_text_with_ansi_colors(text, max_width)
print(wrapped_text)