Make the CLI pretty ✨ (#2793)

* feat: add ⚙ �� prefix to console output for visual distinction

Add a prefix and separator to all console log messages so Cog's
output is visually distinguishable from Docker build output and
model output:

- Info/Debug: dim ⚙ › prefix
- Warn: bold yellow ⚠ › prefix
- Error/Fatal: bold red ⅹ › prefix

Blank lines at info/debug level print without prefix for clean
spacing. Warn/error blank lines keep their prefix to maintain
visual continuity in multi-line error messages.

Output() (stdout, for model predictions) remains unstyled.

* feat: add console.Bold() and highlight dynamic values in messages

Add a Bold() helper to the console package that applies bold
formatting when color is enabled. Use it to highlight key dynamic
values (image names, paths, URLs, hostnames) in user-facing
messages so they stand out within the dim-prefixed lines.

Also remove single quotes around values where bold makes them
visually distinct without needing quoting.

* chore: demote noisy build messages to debug level

Move 'Generating model schema...', 'Using local coglet wheel...',
and 'Using local cog wheel...' from info to debug level. These are
implementation details that clutter normal output but remain visible
with --debug.

* feat: add spacing between build/prediction output phases

Add blank lines after 'Building Docker image...' (before Docker
build output) and after 'Running prediction/training...' (before
model output) to visually separate Cog's framework messages from
external output.

* feat: add console.Success() with green ✓ prefix via Style system

Introduce a Style type that controls the icon/color of a log line
independently from the log level. This allows Success to display at
info level (same filtering) but with a green ✓ prefix.

New styles can be added by extending the Style enum without touching
the level hierarchy.

Apply Success to: image built, cog init completion, login success,
and image pushed messages. Drop the ✅ emoji from init since the
green ✓ prefix replaces it.

* feat: wrap long lines to terminal width with prefix on continuation lines

When stderr is a TTY, wrap console messages to the terminal width so
they don't overflow. Each continuation line gets the same prefix as the
first line for visual continuity.

Wrapping is ANSI-aware: escape codes are treated as zero-width so
colored/bold text doesn't cause premature breaks. Lines break on word
boundaries where possible, with hard breaks as a fallback.

* feat: add InfoUnformatted for prefix-free interactive output

Add InfoUnformatted/InfoUnformattedf methods that write to stderr at
info level without any icon prefix. These are used for conversational
and interactive output (login prompts, instructions) where the gear
prefix would be visual noise.

Long lines are wrapped to terminal width when stderr is a TTY, same
as prefixed output.

Apply to login flows in both Replicate and generic providers: replace
console.Info with console.InfoUnformatted for all user-facing prompts
and instructions. Rename 'CLI auth token:' prompt to 'Token:' and
improve spacing around the prompt.

* fix: improve error message wording and add spacing after run message

- 'Failed to predict' → 'Failed to run prediction' in predict.go
- 'The inputs you passed to cog predict' → 'The inputs you passed' in
  predictor.go (shorter, less redundant since the user knows they ran cog)
- Add blank line after 'Running...' message in run.go for visual
  separation from Docker output

* refactor: improve init command output style

- Remove leading \n from 'Setting up...' message (prefix handles
  spacing), use separate console.Info('') for the blank line after
- Show 'Creating {filename}' (info, before write) instead of
  'Created {fullpath}' (success, after write) — less noisy, uses
  just the filename since the user knows their working directory

* refactor: move build message and spacing to CLI layer

The 'Building Docker image...' message and surrounding blank lines are
user-facing concerns that belong in the CLI commands, not deep in
pkg/image/build.go. This lets each command control what it shows:

- cog build / cog push: show image name (user-chosen, meaningful)
- cog predict / cog train / cog run / cog serve: hide image name
  (ephemeral internal detail the user can't use)

Each CLI command now prints the message + blank line before calling
resolver.Build(), and the pre-ImageBuild blank lines are removed from
image/build.go.

* fix: update integration tests for console output changes

- build_cog_init: match 'Image built as' without 'cog-' prefix since
  the success prefix (✔) is now part of the line
- wheel_resolution: use --debug flag since wheel resolution messages
  were demoted to debug level

Author: bfirsh

integration-tests/tests/build_cog_init.txtar

@@ -6,7 +6,7 @@ cog init
 
 # Build the initialized project
 cog build -t $TEST_IMAGE
-stderr 'Image built as cog-'
+stderr 'Image built as'
 
 # Verify the expected files were created
 exists cog.yaml

integration-tests/tests/wheel_resolution.txtar

@@ -10,7 +10,7 @@
 
 # Test 1: COG_SDK_WHEEL pointing to repo dist/ directory finds wheel
 env COG_SDK_WHEEL=$REPO_ROOT/dist
-cog build -t $TEST_IMAGE
+cog build --debug -t $TEST_IMAGE
 stderr 'Using local cog wheel:'
 
 # Test 2: Relative path that doesn't exist gives clear error  

pkg/cli/build.go

@@ -91,13 +91,17 @@ func buildCommand(cmd *cobra.Command, args []string) error {
 		imageName = config.DockerImageName(src.ProjectDir)
 	}
 
+	console.Infof("Building Docker image from environment in cog.yaml as %s...", console.Bold(imageName))
+	console.Info("")
+
 	resolver := model.NewResolver(dockerClient, registry.NewRegistryClient())
 	m, err := resolver.Build(ctx, src, buildOptionsFromFlags(cmd, imageName, nil))
 	if err != nil {
 		return err
 	}
 
-	console.Infof("\nImage built as %s", m.ImageRef())
+	console.Info("")
+	console.Successf("Image built as %s", console.Bold(m.ImageRef()))
 
 	return nil
 }

pkg/cli/init.go

@@ -37,7 +37,8 @@ and prediction interface. Edit them to match your model's requirements.`,
 }
 
 func initCommand(cmd *cobra.Command, args []string) error {
-	console.Infof("\nSetting up the current directory for use with Cog...\n")
+	console.Info("Setting up the current directory for use with Cog...")
+	console.Info("")
 
 	cwd, err := os.Getwd()
 	if err != nil {
@@ -68,7 +69,7 @@ func initCommand(cmd *cobra.Command, args []string) error {
 		}
 	}
 
-	console.Infof("\nDone! For next steps, check out the docs at https://cog.run/getting-started")
+	console.Successf("\nDone! For next steps, check out the docs at https://cog.run/getting-started")
 
 	return nil
 }
@@ -142,11 +143,11 @@ func processTemplateFile(fs embed.FS, templateDir, filename, cwd string) error {
 		}
 	}
 
+	console.Infof("Creating %s", console.Bold(filename))
+
 	if err := os.WriteFile(filePath, content, 0o644); err != nil {
 		return fmt.Errorf("Error writing %s: %w", filePath, err)
 	}
-
-	console.Infof("✅ Created %s", filePath)
 	return nil
 }
 

pkg/cli/predict.go

@@ -197,6 +197,8 @@ func cmdPredict(cmd *cobra.Command, args []string) error {
 			return err
 		}
 
+		console.Info("Building Docker image from environment in cog.yaml...")
+		console.Info("")
 		m, err := resolver.Build(ctx, src, serveBuildOptions(cmd))
 		if err != nil {
 			return err
@@ -237,7 +239,7 @@ func cmdPredict(cmd *cobra.Command, args []string) error {
 	}
 
 	console.Info("")
-	console.Infof("Starting Docker image %s and running setup()...", imageName)
+	console.Info("Starting Docker image and running setup()...")
 
 	// Automatically propagate RUST_LOG for Rust coglet debugging
 	env := envFlags
@@ -365,6 +367,7 @@ func runPrediction(predictor predict.Predictor, inputs predict.Inputs, outputPat
 	} else {
 		console.Info("Running prediction...")
 	}
+	console.Info("")
 
 	// Generate output depending on type in schema
 	url := "/predictions"
@@ -398,7 +401,7 @@ func runPrediction(predictor predict.Predictor, inputs predict.Inputs, outputPat
 
 	prediction, err := predictor.Predict(inputs, context)
 	if err != nil {
-		return fmt.Errorf("Failed to predict: %w", err)
+		return fmt.Errorf("Failed to run prediction: %w", err)
 	}
 
 	schema, err := predictor.GetSchema()

pkg/cli/push.go

@@ -97,6 +97,8 @@ func push(cmd *cobra.Command, args []string) error {
 	resolver := model.NewResolver(dockerClient, regClient)
 
 	// Build the model
+	console.Infof("Building Docker image from environment in cog.yaml as %s...", console.Bold(imageName))
+	console.Info("")
 	buildOpts := buildOptionsFromFlags(cmd, imageName, annotations)
 	m, err := resolver.Build(ctx, src, buildOpts)
 	if err != nil {
@@ -112,7 +114,7 @@ func push(cmd *cobra.Command, args []string) error {
 	}
 
 	// Push the model (image + optional weights)
-	console.Infof("\nPushing image '%s'...", m.ImageRef())
+	console.Infof("\nPushing image %s...", console.Bold(m.ImageRef()))
 
 	// Set up progress display using Docker's jsonmessage rendering. This uses the
 	// same cursor movement and progress display as `docker push`, which handles

pkg/cli/run.go

@@ -81,6 +81,8 @@ func run(cmd *cobra.Command, args []string) error {
 
 	resolver := model.NewResolver(dockerClient, registry.NewRegistryClient())
 
+	console.Info("Building Docker image from environment in cog.yaml...")
+	console.Info("")
 	opts := serveBuildOptions(cmd)
 	opts.SkipSchemaValidation = true
 	m, err := resolver.Build(ctx, src, opts)
@@ -120,7 +122,8 @@ func run(cmd *cobra.Command, args []string) error {
 	}
 
 	console.Info("")
-	console.Infof("Running '%s' in Docker with the current directory mounted as a volume...", strings.Join(args, " "))
+	console.Infof("Running %s in Docker with the current directory mounted as a volume...", console.Bold(strings.Join(args, " ")))
+	console.Info("")
 
 	err = docker.Run(ctx, dockerClient, runOptions)
 	// Only retry if we're using a GPU but the user didn't explicitly select a GPU with --gpus

pkg/cli/serve.go

@@ -1,6 +1,7 @@
 package cli
 
 import (
+	"fmt"
 	"os"
 	"strings"
 
@@ -81,6 +82,8 @@ func cmdServe(cmd *cobra.Command, arg []string) error {
 		return err
 	}
 
+	console.Info("Building Docker image from environment in cog.yaml...")
+	console.Info("")
 	resolver := model.NewResolver(dockerClient, registry.NewRegistryClient())
 	m, err := resolver.Build(ctx, src, serveBuildOptions(cmd))
 	if err != nil {
@@ -130,9 +133,9 @@ func cmdServe(cmd *cobra.Command, arg []string) error {
 	runOptions.Ports = append(runOptions.Ports, command.Port{HostPort: port, ContainerPort: 5000})
 
 	console.Info("")
-	console.Infof("Running '%[1]s' in Docker with the current directory mounted as a volume...", strings.Join(args, " "))
+	console.Infof("Running %[1]s in Docker with the current directory mounted as a volume...", console.Bold(strings.Join(args, " ")))
 	console.Info("")
-	console.Infof("Serving at http://127.0.0.1:%[1]v", port)
+	console.Infof("Serving at %s", console.Bold(fmt.Sprintf("http://127.0.0.1:%v", port)))
 	console.Info("")
 
 	err = docker.Run(ctx, dockerClient, runOptions)

pkg/cli/train.go

@@ -73,6 +73,8 @@ func cmdTrain(cmd *cobra.Command, args []string) error {
 			return err
 		}
 
+		console.Info("Building Docker image from environment in cog.yaml...")
+		console.Info("")
 		m, err := resolver.Build(ctx, src, serveBuildOptions(cmd))
 		if err != nil {
 			return err
@@ -108,7 +110,7 @@ func cmdTrain(cmd *cobra.Command, args []string) error {
 	}
 
 	console.Info("")
-	console.Infof("Starting Docker image %s...", imageName)
+	console.Info("Starting Docker image and running setup()...")
 
 	predictor, err := predict.NewPredictor(ctx, command.RunOptions{
 		GPUs:    gpus,

pkg/dockerfile/standard_generator.go

@@ -609,7 +609,7 @@ func (g *StandardGenerator) installCog() (string, error) {
 		case wheels.WheelSourceURL:
 			console.Infof("Using coglet wheel from URL: %s", cogletConfig.URL)
 		case wheels.WheelSourceFile:
-			console.Infof("Using local coglet wheel: %s", cogletConfig.Path)
+			console.Debugf("Using local coglet wheel: %s", cogletConfig.Path)
 		}
 
 		cogletIsPreRelease := sdkIsPreRelease ||
@@ -634,7 +634,7 @@ func (g *StandardGenerator) installCog() (string, error) {
 		console.Infof("Using cog wheel from URL: %s", wheelConfig.URL)
 		cogInstall, err = g.installWheelFromURL(wheelConfig.URL)
 	case wheels.WheelSourceFile:
-		console.Infof("Using local cog wheel: %s", wheelConfig.Path)
+		console.Debugf("Using local cog wheel: %s", wheelConfig.Path)
 		cogInstall, err = g.installWheelFromFile(wheelConfig.Path)
 	default:
 		return "", fmt.Errorf("unknown wheel source: %v", wheelConfig.Source)