如何为我的python代码添加颜色

# 导入ANSI转义序列模块
from colorama import init, Fore, Back, Style

# 初始化ANSI转义序列
init()

# 定义颜色常量
COLOR_RED = Fore.RED
COLOR_GREEN = Fore.GREEN
COLOR_YELLOW = Fore.YELLOW
COLOR_BLUE = Fore.BLUE
COLOR_MAGENTA = Fore.MAGENTA
COLOR_CYAN = Fore.CYAN
COLOR_RESET = Style.RESET_ALL

# 示例函数，将字符串以指定颜色输出
def print_color(text, color):
    print(color + text + COLOR_RESET)

# 示例代码
def main():
    code = '''
def hello_world():
    print("Hello, World!")

hello_world()
'''
    print_color("Python代码示例：", COLOR_YELLOW)
    print(code)

if __name__ == '__main__':
    main()