code: def colorize(text, style):
      try:
          # if we're in ipython it's okay to use colors
          __IPYTHON__
          USE_COLORS = True
      except NameError:
          USE_COLORS = _sys.stdout.isatty()
  
      if not USE_COLORS:
          return text
  
      colorstyle = os.environ.get("PARLAI_COLORSTYLE")
  
      RESET = "\033[0;0m"
      if style == "red":
          return "\033[0;31m" + text + RESET
      if style == "yellow":
          return "\033[0;93m" + text + RESET
      if style == "green":
          return "\033[0;32m" + text + RESET
      if style == "blue":
          return "\033[0;34m" + text + RESET
      if style == "brightblack":
          return "\033[0;90m" + text + RESET
  
      if colorstyle is None or colorstyle.lower() == "steamroller":
          BLUE = "\033[1;94m"
          BOLD_LIGHT_GRAY_NOBK = "\033[1m"
          LIGHT_GRAY_NOBK = "\033[0m"
          MAGENTA = "\033[0;95m"
          HIGHLIGHT_RED_NOBK = "\033[1;31m"
          HIGHLIGHT_BLUE_NOBK = "\033[0;34m"
          if style == "highlight":
              return HIGHLIGHT_RED_NOBK + text + RESET
          if style == "highlight2":
              return HIGHLIGHT_BLUE_NOBK + text + RESET
          elif style == "text":
              return LIGHT_GRAY_NOBK + text + RESET
          elif style == "bold_text":
              return BOLD_LIGHT_GRAY_NOBK + text + RESET
          elif style == "labels" or style == "eval_labels":
              return BLUE + text + RESET
          elif style == "label_candidates":
              return LIGHT_GRAY_NOBK + text + RESET
          elif style == "id":
              return LIGHT_GRAY_NOBK + text + RESET
          elif style == "text2":
              return MAGENTA + text + RESET
          elif style == "field":
              return HIGHLIGHT_BLUE_NOBK + text + RESET
          else:
              return MAGENTA + text + RESET
  
      if colorstyle.lower() == "spermwhale":
          BLUE = "\033[1;94m"
          BOLD_LIGHT_GRAY = "\033[1;37;40m"
          LIGHT_GRAY = "\033[0;37;40m"
          MAGENTA = "\033[0;95m"
          HIGHLIGHT_RED = "\033[1;37;41m"
          HIGHLIGHT_BLUE = "\033[1;37;44m"
          if style == "highlight":
              return HIGHLIGHT_RED + text + RESET
          if style == "highlight2":
              return HIGHLIGHT_BLUE + text + RESET
          elif style == "text":
              return LIGHT_GRAY + text + RESET
          elif style == "bold_text":
              return BOLD_LIGHT_GRAY + text + RESET
          elif style == "labels" or style == "eval_labels":
              return BLUE + text + RESET
          elif style == "label_candidates":
              return LIGHT_GRAY + text + RESET
          elif style == "id":
              return LIGHT_GRAY + text + RESET
          elif style == "text2":
              return MAGENTA + text + RESET
          elif style == "field":
              return HIGHLIGHT_BLUE + text + RESET
          else:
              return MAGENTA + text + RESET
  
      # No colorstyle specified/found.
      return text
  
imports: import os
  import sys
label_desc: Use image node_id 4 for calling a global function with example usage: colorize(text, style) and returns: text, text, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown, unknown