Highlight script

clarifai/runners/utils/code_script.py

@@ -2,6 +2,9 @@
 from typing import List
 
 from clarifai_grpc.grpc.api import resources_pb2
+from pygments import highlight
+from pygments.formatters import TerminalFormatter
+from pygments.lexers import PythonLexer
 
 from clarifai.runners.utils import data_utils
 from clarifai.urls.helper import ClarifaiUrlHelper
@@ -201,6 +204,8 @@ async def main():
     script_lines.append(method_signatures_str)
     script_lines.append("")
     script = "\n".join(script_lines)
+
+    script = highlight(script, PythonLexer(), TerminalFormatter())
     return script
 
 

requirements.txt

@@ -13,3 +13,4 @@ aiohttp>=3.10.0
 uv==0.7.12
 ruff==0.11.4
 psutil==7.0.0
+pygments>=2.19.2