import builtins
from rich.console import Console

# Create the console instance once.
console = Console()

# Store the original print function.
_original_print = builtins.print

# Override the built-in print function.
def colorized_print(*args, **kwargs):
    """A print function that formats all output in yellow."""
    # Convert all arguments to a single string to apply the style.
    text_to_print = " ".join(map(str, args))

    # Use Rich's console.print with a yellow style.
    console.print(text_to_print, style="yellow", **kwargs)

# Replace the built-in print with our new function.
builtins.print = colorized_print