From 08b3dae588d0349942f109a9199ef026cfbe47ba Mon Sep 17 00:00:00 2001
From: imagineer99 <samleejackson0@gmail.com>
Date: Thu, 19 Mar 2026 19:31:00 +0000
Subject: [PATCH 01/16] refactor(studio): unify setup terminal output style and
 add verbose setup mode

---
 studio/backend/run.py            |  19 +-
 studio/backend/startup_banner.py | 107 ++++++++++
 studio/install_python_stack.py   |  77 ++++---
 studio/setup.ps1                 |  53 +++--
 studio/setup.sh                  | 335 ++++++++++++-------------------
 unsloth_cli/commands/studio.py   |  17 +-
 6 files changed, 348 insertions(+), 260 deletions(-)
 create mode 100644 studio/backend/startup_banner.py

diff --git a/studio/backend/run.py b/studio/backend/run.py
index 5388d3148b..92cbb054ae 100644
--- a/studio/backend/run.py
+++ b/studio/backend/run.py
@@ -20,6 +20,7 @@
     sys.path.insert(0, str(backend_dir))
 
 from loggers import get_logger
+from startup_banner import print_studio_access_banner
 
 logger = get_logger(__name__)
 
@@ -223,19 +224,11 @@ def _run():
 
     if not silent:
         display_host = _resolve_external_ip() if host == "0.0.0.0" else host
-
-        print("")
-        print("=" * 50)
-        print(f"🦥 Open your web browser, and enter http://localhost:{port}")
-        print("=" * 50)
-        print("")
-        print("=" * 50)
-        print(f"🦥 Unsloth Studio is running on port {port}")
-        print(f"   Local Access:          http://localhost:{port}")
-        print(f"   Worldwide Web Address: http://{display_host}:{port}")
-        print(f"   API:                   http://{display_host}:{port}/api")
-        print(f"   Health:                http://{display_host}:{port}/api/health")
-        print("=" * 50)
+        print_studio_access_banner(
+            port=port,
+            bind_host=host,
+            display_host=display_host,
+        )
 
     return app
 
diff --git a/studio/backend/startup_banner.py b/studio/backend/startup_banner.py
new file mode 100644
index 0000000000..62ebd40b14
--- /dev/null
+++ b/studio/backend/startup_banner.py
@@ -0,0 +1,107 @@
+# SPDX-License-Identifier: AGPL-3.0-only
+# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
+
+"""Terminal banner for Studio startup.
+
+Stdlib only — safe to import without the rest of the backend (no structlog/uvicorn).
+"""
+
+from __future__ import annotations
+
+import os
+import sys
+
+
+def stdout_supports_color() -> bool:
+    """True if we should emit ANSI colors."""
+    if os.environ.get("NO_COLOR", "").strip():
+        return False
+    if os.environ.get("FORCE_COLOR", "").strip():
+        return True
+    try:
+        return sys.stdout.isatty()
+    except Exception:
+        return False
+
+
+def print_port_in_use_notice(original_port: int, new_port: int) -> None:
+    """Message when the requested port is taken and another is chosen."""
+    msg = f"Port {original_port} is in use, using port {new_port} instead."
+    if stdout_supports_color():
+        print(f"\033[38;5;245m{msg}\033[0m")
+    else:
+        print(msg)
+
+
+def print_studio_access_banner(
+    *,
+    port: int,
+    bind_host: str,
+    display_host: str,
+) -> None:
+    """Pretty-print URLs after the server is listening (beginner-friendly)."""
+    use_color = stdout_supports_color()
+    dim = "\033[38;5;245m"
+    title = "\033[38;5;150m"
+    local_url_style = "\033[38;5;108;1m"
+    secondary = "\033[38;5;109m"
+    reset = "\033[0m"
+
+    def style(text: str, code: str) -> str:
+        return f"{code}{text}{reset}" if use_color else text
+
+    local_url = f"http://127.0.0.1:{port}"
+    alt_local = f"http://localhost:{port}"
+    external_url = f"http://{display_host}:{port}"
+    listen_all = bind_host in ("0.0.0.0", "::")
+    loopback_bind = bind_host in ("127.0.0.1", "localhost", "::1")
+    api_base = local_url if listen_all or loopback_bind else external_url
+
+    lines: list[str] = [
+        "",
+        style("🦥 Unsloth Studio is running", title),
+        style("─" * 52, dim),
+        style("  On this machine — open this in your browser:", dim),
+        style(f"    {local_url}", local_url_style),
+        style(f"    (same as {alt_local})", dim),
+    ]
+
+    if listen_all and display_host not in (
+        "127.0.0.1",
+        "localhost",
+        "::1",
+        "0.0.0.0",
+        "::",
+    ):
+        lines.extend(
+            [
+                "",
+                style("  From another device on your network / to share:", dim),
+                style(f"    {external_url}", secondary),
+            ]
+        )
+    elif not listen_all and bind_host not in ("127.0.0.1", "localhost", "::1"):
+        lines.extend(
+            [
+                "",
+                style("  Bound address:", dim),
+                style(f"    {external_url}", secondary),
+            ]
+        )
+
+    lines.extend(
+        [
+            "",
+            style("  API & health:", dim),
+            style(f"    {api_base}/api", secondary),
+            style(f"    {api_base}/api/health", secondary),
+            style("─" * 52, dim),
+            style(
+                "  Tip: if you are on the same computer, use the Local link above.",
+                dim,
+            ),
+            "",
+        ]
+    )
+
+    print("\n".join(lines))
diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py
index a141c64425..4503c26a63 100644
--- a/studio/install_python_stack.py
+++ b/studio/install_python_stack.py
@@ -25,6 +25,7 @@
 # ── Verbosity control ──────────────────────────────────────────────────────────
 # By default the installer shows a minimal progress bar (one line, in-place).
 # Set UNSLOTH_VERBOSE=1 in the environment to restore full per-step output:
+#   CLI:        unsloth studio setup --verbose
 #   Linux/Mac:  UNSLOTH_VERBOSE=1 ./studio/setup.sh
 #   Windows:    $env:UNSLOTH_VERBOSE="1" ; .\studio\setup.ps1
 VERBOSE: bool = os.environ.get("UNSLOTH_VERBOSE", "0") == "1"
@@ -45,14 +46,17 @@
 )
 
 # ── Color support ──────────────────────────────────────────────────────
+# Same logic as startup_banner: NO_COLOR disables, FORCE_COLOR or TTY enables.
 
 
-def _enable_colors() -> bool:
-    """Try to enable ANSI color support. Returns True if available."""
-    if not hasattr(sys.stdout, "fileno"):
+def _stdout_supports_color() -> bool:
+    """True if we should emit ANSI colors (matches startup_banner)."""
+    if os.environ.get("NO_COLOR", "").strip():
         return False
+    if os.environ.get("FORCE_COLOR", "").strip():
+        return True
     try:
-        if not os.isatty(sys.stdout.fileno()):
+        if not sys.stdout.isatty():
             return False
     except Exception:
         return False
@@ -61,24 +65,26 @@ def _enable_colors() -> bool:
             import ctypes
 
             kernel32 = ctypes.windll.kernel32
-            # Enable ENABLE_VIRTUAL_TERMINAL_PROCESSING (0x0004) on stdout
-            handle = kernel32.GetStdHandle(-11)  # STD_OUTPUT_HANDLE
+            handle = kernel32.GetStdHandle(-11)
             mode = ctypes.c_ulong()
             kernel32.GetConsoleMode(handle, ctypes.byref(mode))
             kernel32.SetConsoleMode(handle, mode.value | 0x0004)
-            return True
         except Exception:
             return False
-    return True  # Unix terminals support ANSI by default
+    return True
+
 
+_HAS_COLOR = _stdout_supports_color()
 
-# Colors disabled — Colab and most CI runners render ANSI fine, but plain output
-# is cleaner in the notebook cell. Re-enable by setting _HAS_COLOR = _enable_colors()
-_HAS_COLOR = False
+
+# Column layout — matches setup.sh step() helper:
+#   2-space indent, 15-char label (dim), then value.
+_LABEL = "deps"
+_COL = 15
 
 
 def _green(msg: str) -> str:
-    return f"\033[92m{msg}\033[0m" if _HAS_COLOR else msg
+    return f"\033[38;5;108m{msg}\033[0m" if _HAS_COLOR else msg
 
 
 def _cyan(msg: str) -> str:
@@ -89,21 +95,38 @@ def _red(msg: str) -> str:
     return f"\033[91m{msg}\033[0m" if _HAS_COLOR else msg
 
 
-def _progress(label: str) -> None:
-    """Print an in-place progress bar for the current install step.
+def _dim(msg: str) -> str:
+    return f"\033[38;5;245m{msg}\033[0m" if _HAS_COLOR else msg
+
+
+def _title(msg: str) -> str:
+    return f"\033[38;5;150m{msg}\033[0m" if _HAS_COLOR else msg
+
+
+_RULE = "\u2500" * 52
 
-    Uses only stdlib (sys.stdout) — no extra packages required.
-    In VERBOSE mode this is a no-op; per-step labels are printed by run() instead.
-    """
+
+def _step(label: str, value: str, color_fn = None) -> None:
+    """Print a single step line in the column format."""
+    if color_fn is None:
+        color_fn = _green
+    print(f"  {_dim(label)}{' ' * (_COL - len(label))}{color_fn(value)}")
+
+
+def _progress(label: str) -> None:
+    """Print an in-place progress bar aligned to the step column layout."""
     global _STEP
     _STEP += 1
     if VERBOSE:
-        return  # verbose mode: run() already printed the label
+        return
     width = 20
     filled = int(width * _STEP / _TOTAL)
     bar = "=" * filled + "-" * (width - filled)
-    end = "\n" if _STEP >= _TOTAL else ""  # newline only on the final step
-    sys.stdout.write(f"\r[{bar}] {_STEP:2}/{_TOTAL}  {label:<40}{end}")
+    pad = " " * (_COL - len(_LABEL))
+    end = "\n" if _STEP >= _TOTAL else ""
+    sys.stdout.write(
+        f"\r  {_dim(_LABEL)}{pad}[{bar}] {_STEP:2}/{_TOTAL}  {label:<20}{end}"
+    )
     sys.stdout.flush()
 
 
@@ -112,14 +135,14 @@ def run(
 ) -> subprocess.CompletedProcess[bytes]:
     """Run a command; on failure print output and exit."""
     if VERBOSE:
-        print(f"   {label}...")
+        _step(_LABEL, f"{label}...", _dim)
     result = subprocess.run(
         cmd,
         stdout = subprocess.PIPE if quiet else None,
         stderr = subprocess.STDOUT if quiet else None,
     )
     if result.returncode != 0:
-        print(_red(f"❌ {label} failed (exit code {result.returncode}):"))
+        _step("error", f"{label} failed (exit code {result.returncode})", _red)
         if result.stdout:
             print(result.stdout.decode(errors = "replace"))
         sys.exit(result.returncode)
@@ -267,7 +290,7 @@ def patch_package_file(package_name: str, relative_path: str, url: str) -> None:
         text = True,
     )
     if result.returncode != 0:
-        print(_red(f"   ⚠️  Could not find package {package_name}, skipping patch"))
+        _step(_LABEL, f"package {package_name} not found, skipping patch", _dim)
         return
 
     location = None
@@ -277,11 +300,11 @@ def patch_package_file(package_name: str, relative_path: str, url: str) -> None:
             break
 
     if not location:
-        print(_red(f"   ⚠️  Could not determine location of {package_name}"))
+        _step(_LABEL, f"could not locate {package_name}", _dim)
         return
 
     dest = Path(location) / relative_path
-    print(_cyan(f"   Patching {dest.name} in {package_name}..."))
+    _step(_LABEL, f"patching {dest.name} in {package_name}...", _dim)
     download_file(url, dest)
 
 
@@ -293,6 +316,8 @@ def install_python_stack() -> int:
     _STEP = 0
     _TOTAL = 10 if IS_WINDOWS else 11
 
+    
+
     # 1. Upgrade pip (needed even with uv as fallback and for bootstrapping)
     _progress("pip upgrade")
     run("Upgrading pip", [sys.executable, "-m", "pip", "install", "--upgrade", "pip"])
@@ -422,7 +447,7 @@ def install_python_stack() -> int:
         stderr = subprocess.DEVNULL,
     )
 
-    print(_green("✅ Python dependencies installed"))
+    _step(_LABEL, "installed")
     return 0
 
 
diff --git a/studio/setup.ps1 b/studio/setup.ps1
index 7091920ab9..e73b04ad7c 100644
--- a/studio/setup.ps1
+++ b/studio/setup.ps1
@@ -245,12 +245,35 @@ function Find-VsBuildTools {
     return $null
 }
 
+# ─────────────────────────────────────────────
+# Output style helpers (matches setup.sh visual tone)
+# ─────────────────────────────────────────────
+$Rule = "────────────────────────────────────────────────────"
+
+function Write-Step {
+    param(
+        [Parameter(Mandatory = $true)][string]$Label,
+        [Parameter(Mandatory = $true)][string]$Value,
+        [string]$Color = "Green"
+    )
+    Write-Host ("  {0,-15}" -f $Label) -NoNewline -ForegroundColor DarkGray
+    Write-Host $Value -ForegroundColor $Color
+}
+
+function Write-Substep {
+    param(
+        [Parameter(Mandatory = $true)][string]$Message,
+        [string]$Color = "DarkGray"
+    )
+    Write-Host ("  {0,-15}{1}" -f "", $Message) -ForegroundColor $Color
+}
+
 # ─────────────────────────────────────────────
 # Banner
 # ─────────────────────────────────────────────
-Write-Host "+==============================================+" -ForegroundColor Green
-Write-Host "|       Unsloth Studio Setup (Windows)         |" -ForegroundColor Green
-Write-Host "+==============================================+" -ForegroundColor Green
+Write-Host ""
+Write-Host "  🦥 Unsloth Studio Setup" -ForegroundColor Green
+Write-Host "  $Rule" -ForegroundColor DarkGray
 
 # ==========================================================================
 #  PHASE 1: System-level prerequisites (winget installs, env vars)
@@ -1383,21 +1406,21 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
     # -- Summary --
     Write-Host ""
     if ($BuildOk -and (Test-Path $LlamaServerBin)) {
-        Write-Host "[OK] llama-server built at $LlamaServerBin" -ForegroundColor Green
+        Write-Step "llama.cpp" "built at $LlamaServerBin"
         $QuantizeBin = Join-Path $BuildDir "bin\Release\llama-quantize.exe"
         if (Test-Path $QuantizeBin) {
-            Write-Host "[OK] llama-quantize available for GGUF export" -ForegroundColor Green
+            Write-Step "llama-quantize" "built"
         }
-        Write-Host "   Build time: ${totalMin}m ${totalSec}s" -ForegroundColor Cyan
+        Write-Step "build time" "${totalMin}m ${totalSec}s" "DarkGray"
     } else {
         # Check alternate paths (some cmake generators don't use Release subdir)
         $altBin = Join-Path $BuildDir "bin\llama-server.exe"
         if ($BuildOk -and (Test-Path $altBin)) {
-            Write-Host "[OK] llama-server built at $altBin" -ForegroundColor Green
-            Write-Host "   Build time: ${totalMin}m ${totalSec}s" -ForegroundColor Cyan
+            Write-Step "llama.cpp" "built at $altBin"
+            Write-Step "build time" "${totalMin}m ${totalSec}s" "DarkGray"
         } else {
-            Write-Host "[FAILED] llama.cpp build failed at step: $FailedStep (${totalMin}m ${totalSec}s)" -ForegroundColor Red
-            Write-Host "         To retry: delete $LlamaCppDir and re-run setup." -ForegroundColor Yellow
+            Write-Step "llama.cpp" "build failed at step: $FailedStep (${totalMin}m ${totalSec}s)" "Red"
+            Write-Substep "To retry: delete $LlamaCppDir and re-run setup." "Yellow"
             exit 1
         }
     }
@@ -1407,10 +1430,6 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
 # Done
 # ============================================
 Write-Host ""
-Write-Host "+===============================================+" -ForegroundColor Green
-Write-Host "|           Setup Complete!                     |" -ForegroundColor Green
-Write-Host "|                                               |" -ForegroundColor Green
-Write-Host "|  Launch with:                                 |" -ForegroundColor Green
-Write-Host "|    unsloth studio -H 0.0.0.0 -p 8888          |" -ForegroundColor Green
-Write-Host "|                                               |" -ForegroundColor Green
-Write-Host "+===============================================+" -ForegroundColor Green
+Write-Host "  $Rule" -ForegroundColor DarkGray
+Write-Host "  Unsloth Studio Installed" -ForegroundColor Green
+Write-Step "launch" "unsloth studio -H 0.0.0.0 -p 8888"
diff --git a/studio/setup.sh b/studio/setup.sh
index 3e56aebf38..e70face973 100755
--- a/studio/setup.sh
+++ b/studio/setup.sh
@@ -6,137 +6,152 @@ set -euo pipefail
 
 SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 REPO_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
+RULE=$(printf '\342\224\200%.0s' {1..52})
+
+# ── Colors (same palette as startup_banner / install_python_stack) ──
+if [ -n "${NO_COLOR:-}" ]; then
+    C_TITLE= C_DIM= C_OK= C_WARN= C_ERR= C_RST=
+elif [ -t 1 ] || [ -n "${FORCE_COLOR:-}" ]; then
+    C_TITLE=$'\033[38;5;150m'
+    C_DIM=$'\033[38;5;245m'
+    C_OK=$'\033[38;5;108m'
+    C_WARN=$'\033[38;5;136m'
+    C_ERR=$'\033[91m'
+    C_RST=$'\033[0m'
+else
+    C_TITLE= C_DIM= C_OK= C_WARN= C_ERR= C_RST=
+fi
+
+# ── Output helpers ──
+# Consistent column layout: 2-space indent, 15-char label (fits llama-quantize), then value.
+# Usage: step <label> <message> [color]   (color defaults to C_OK)
+step()    { printf "  ${C_DIM}%-15s${C_RST}${3:-$C_OK}%s${C_RST}\n" "$1" "$2"; }
+substep() { printf "  ${C_DIM}%-15s%s${C_RST}\n" "" "$1"; }
 
-# ── Helper: run command quietly, show output only on failure ──
 run_quiet() {
-    local label="$1"
-    shift
-    local tmplog
-    tmplog=$(mktemp)
+    local label="$1"; shift
+    local tmplog; tmplog=$(mktemp)
     if "$@" > "$tmplog" 2>&1; then
         rm -f "$tmplog"
     else
         local exit_code=$?
-        echo "❌ $label failed (exit code $exit_code):"
+        step "error" "$label failed (exit code $exit_code)" "$C_ERR"
         cat "$tmplog"
         rm -f "$tmplog"
         exit $exit_code
     fi
 }
 
-echo "╔══════════════════════════════════════╗"
-echo "║     Unsloth Studio Setup Script      ║"
-echo "╚══════════════════════════════════════╝"
+# Like run_quiet but returns the exit code instead of dying.
+# With UNSLOTH_VERBOSE=1 (e.g. unsloth studio setup --verbose), print captured
+# output on failure so optional-step errors (llama.cpp cmake, etc.) are visible.
+try_quiet() {
+    local label="$1"; shift
+    local tmplog; tmplog=$(mktemp)
+    if "$@" > "$tmplog" 2>&1; then
+        rm -f "$tmplog"
+        return 0
+    else
+        local exit_code=$?
+        if [ "${UNSLOTH_VERBOSE:-0}" = "1" ]; then
+            step "error" "$label failed (exit code $exit_code)" "$C_ERR"
+            cat "$tmplog"
+        fi
+        rm -f "$tmplog"
+        return $exit_code
+    fi
+}
 
-# ── Clean up stale Unsloth compiled caches ──
+# ── Banner ──
+echo ""
+printf "  ${C_TITLE}%s${C_RST}\n" "🦥 Unsloth Studio Setup"
+printf "  ${C_DIM}%s${C_RST}\n" "$RULE"
+
+# ── Clean up stale caches ──
 rm -rf "$REPO_ROOT/unsloth_compiled_cache"
 rm -rf "$SCRIPT_DIR/backend/unsloth_compiled_cache"
 rm -rf "$SCRIPT_DIR/tmp/unsloth_compiled_cache"
 
-# ── Detect Colab (like unsloth does) ──
+# ── Detect Colab ──
 IS_COLAB=false
 keynames=$'\n'$(printenv | cut -d= -f1)
 if [[ "$keynames" == *$'\nCOLAB_'* ]]; then
     IS_COLAB=true
 fi
 
-# ── Detect whether frontend needs building ──
-# Skip if dist/ exists AND no tracked input is newer than dist/.
-# Checks top-level config/entry files and src/, public/ recursively.
-# This handles: PyPI installs (dist/ bundled), repeat runs (no changes),
-# and upgrades/pulls (source newer than dist/ triggers rebuild).
+# ── Frontend ──
 _NEED_FRONTEND_BUILD=true
 if [ -d "$SCRIPT_DIR/frontend/dist" ]; then
-    # Check all top-level files (package.json, bun.lock, vite.config.ts, index.html, etc.)
     _changed=$(find "$SCRIPT_DIR/frontend" -maxdepth 1 -type f \
         -newer "$SCRIPT_DIR/frontend/dist" -print -quit 2>/dev/null)
-    # Check src/ and public/ recursively (|| true guards against set -e when dirs are missing)
     if [ -z "$_changed" ]; then
         _changed=$(find "$SCRIPT_DIR/frontend/src" "$SCRIPT_DIR/frontend/public" \
             -type f -newer "$SCRIPT_DIR/frontend/dist" -print -quit 2>/dev/null) || true
     fi
-    if [ -z "$_changed" ]; then
-        _NEED_FRONTEND_BUILD=false
-    fi
+    [ -z "$_changed" ] && _NEED_FRONTEND_BUILD=false
 fi
+
 if [ "$_NEED_FRONTEND_BUILD" = false ]; then
-    echo "✅ Frontend already built and up to date -- skipping Node/npm check."
+    step "frontend" "up to date"
 else
+
+# ── Node ──
 NEED_NODE=true
 if command -v node &>/dev/null && command -v npm &>/dev/null; then
     NODE_MAJOR=$(node -v | sed 's/v//' | cut -d. -f1)
     NPM_MAJOR=$(npm -v | cut -d. -f1)
     if [ "$NODE_MAJOR" -ge 20 ] && [ "$NPM_MAJOR" -ge 11 ]; then
-        echo "✅ Node $(node -v) and npm $(npm -v) already meet requirements. Skipping nvm install."
         NEED_NODE=false
     else
         if [ "$IS_COLAB" = true ]; then
-            echo "✅ Node $(node -v) and npm $(npm -v) detected in Colab."
-            # In Colab, just upgrade npm directly - nvm doesn't work well
             if [ "$NPM_MAJOR" -lt 11 ]; then
-                echo "   Upgrading npm to latest..."
+                substep "upgrading npm..."
                 npm install -g npm@latest > /dev/null 2>&1
             fi
             NEED_NODE=false
-        else
-            echo "⚠️  Node $(node -v) / npm $(npm -v) too old. Installing via nvm..."
         fi
     fi
-else
-    echo "⚠️  Node/npm not found. Installing via nvm..."
 fi
 
 if [ "$NEED_NODE" = true ]; then
-    # ── 2. Install nvm ──
-    export NODE_OPTIONS=--dns-result-order=ipv4first # or else fails on colab.
-    echo "Installing nvm..."
+    substep "installing nvm..."
+    export NODE_OPTIONS=--dns-result-order=ipv4first
     curl -so- https://raw.githubusercontent.com/nvm-sh/nvm/v0.40.1/install.sh | bash > /dev/null 2>&1
 
-    # Load nvm (source ~/.bashrc won't work inside a script)
     export NVM_DIR="$HOME/.nvm"
     set +u
     [ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh"
 
-    # ── Fix npmrc conflict with nvm ──
-    # System npm (apt, conda, etc.) may have written `prefix` or `globalconfig`
-    # to ~/.npmrc, which is incompatible with nvm and causes "nvm use" to fail
-    # with: "has a `globalconfig` and/or a `prefix` setting, which are
-    # incompatible with nvm."
     if [ -f "$HOME/.npmrc" ]; then
         if grep -qE '^\s*(prefix|globalconfig)\s*=' "$HOME/.npmrc"; then
-            echo "   Removing incompatible prefix/globalconfig from ~/.npmrc for nvm..."
             sed -i.bak '/^\s*\(prefix\|globalconfig\)\s*=/d' "$HOME/.npmrc"
         fi
     fi
 
-    # ── 3. Install Node LTS ──
-    echo "Installing Node LTS..."
+    substep "installing Node LTS..."
     run_quiet "nvm install" nvm install --lts
     nvm use --lts > /dev/null 2>&1
     set -u
-    # ── 4. Verify versions ──
+
     NODE_MAJOR=$(node -v | sed 's/v//' | cut -d. -f1)
     NPM_MAJOR=$(npm -v | cut -d. -f1)
 
     if [ "$NODE_MAJOR" -lt 20 ]; then
-        echo "❌ ERROR: Node version must be >= 20 (got $(node -v))"
+        step "node" "FAILED — version must be >= 20 (got $(node -v))" "$C_ERR"
         exit 1
     fi
     if [ "$NPM_MAJOR" -lt 11 ]; then
-        echo "⚠️  npm version is $(npm -v), expected >= 11. Updating..."
+        substep "upgrading npm..."
         run_quiet "npm update" npm install -g npm@latest
     fi
 fi
 
-echo "✅ Node $(node -v) | npm $(npm -v)"
+step "node" "$(node -v) | npm $(npm -v)"
 
-# ── 5. Build frontend ──
+# ── Build frontend ──
+substep "building frontend..."
 cd "$SCRIPT_DIR/frontend"
 
-# Tailwind v4's oxide scanner respects .gitignore in parent directories.
-# Python venvs create a .gitignore with "*" (ignore everything), which
-# prevents Tailwind from scanning .tsx source files for class names.
-# Temporarily hide any such .gitignore during the build, then restore it.
 _HIDDEN_GITIGNORES=()
 _dir="$(pwd)"
 while [ "$_dir" != "/" ]; do
@@ -160,87 +175,63 @@ run_quiet "npm run build" npm run build
 _restore_gitignores
 trap - EXIT
 
-# Validate CSS output -- catch truncated Tailwind builds
 _MAX_CSS=$(find "$SCRIPT_DIR/frontend/dist/assets" -name '*.css' -exec wc -c {} + 2>/dev/null | sort -n | tail -1 | awk '{print $1}')
 if [ -z "$_MAX_CSS" ]; then
-    echo "⚠️  WARNING: No CSS files were emitted. The frontend build may have failed."
+    step "frontend" "built (warning: no CSS emitted)" "$C_WARN"
 elif [ "$_MAX_CSS" -lt 100000 ]; then
-    echo "⚠️  WARNING: Largest CSS file is only $((_MAX_CSS / 1024))KB (expected >100KB)."
-    echo "   Tailwind may not have scanned all source files. Check for .gitignore interference."
+    step "frontend" "built (warning: CSS may be truncated)" "$C_WARN"
+else
+    step "frontend" "built"
 fi
 
 cd "$SCRIPT_DIR"
-echo "✅ Frontend built to frontend/dist"
 
 fi  # end frontend build check
 
-# ── oxc-validator runtime (needs npm -- skip if not available) ──
+# ── oxc-validator runtime ──
 if [ -d "$SCRIPT_DIR/backend/core/data_recipe/oxc-validator" ] && command -v npm &>/dev/null; then
     cd "$SCRIPT_DIR/backend/core/data_recipe/oxc-validator"
     run_quiet "npm install (oxc validator runtime)" npm install
     cd "$SCRIPT_DIR"
 fi
 
-# ── 6. Python venv + deps ──
-
-# ── 6a. Discover best Python >= 3.11 and < 3.14 (i.e. 3.11.x, 3.12.x, or 3.13.x) ──
-MIN_PY_MINOR=11   # minimum minor version (>= 3.11)
-MAX_PY_MINOR=13   # maximum minor version (< 3.14)
+# ── Python ──
+MIN_PY_MINOR=11
+MAX_PY_MINOR=13
 BEST_PY=""
 BEST_MAJOR=0
 BEST_MINOR=0
 
-# Collect candidate python3 binaries (python3, python3.9, python3.10, …)
 for candidate in $(compgen -c python3 2>/dev/null | grep -E '^python3(\.[0-9]+)?$' | sort -u); do
-    if ! command -v "$candidate" &>/dev/null; then
-        continue
-    fi
-    # Get version string, e.g. "Python 3.12.5"
+    if ! command -v "$candidate" &>/dev/null; then continue; fi
     ver_str=$("$candidate" --version 2>&1) || continue
     ver_str=$(echo "$ver_str" | awk '{print $2}')
     py_major=$(echo "$ver_str" | cut -d. -f1)
     py_minor=$(echo "$ver_str" | cut -d. -f2)
-
-    # Skip anything that isn't Python 3
-    if [ "$py_major" -ne 3 ] 2>/dev/null; then
-        continue
-    fi
-
-    # Skip versions below 3.12 (require > 3.11)
-    if [ "$py_minor" -lt "$MIN_PY_MINOR" ] 2>/dev/null; then
-        continue
-    fi
-
-    # Skip versions above 3.13 (require < 3.14)
-    if [ "$py_minor" -gt "$MAX_PY_MINOR" ] 2>/dev/null; then
-        continue
-    fi
-
-    # Keep the highest qualifying version
+    [ "$py_major" -ne 3 ] 2>/dev/null && continue
+    [ "$py_minor" -lt "$MIN_PY_MINOR" ] 2>/dev/null && continue
+    [ "$py_minor" -gt "$MAX_PY_MINOR" ] 2>/dev/null && continue
     if [ "$py_minor" -gt "$BEST_MINOR" ]; then
         BEST_PY="$candidate"
         BEST_MAJOR="$py_major"
         BEST_MINOR="$py_minor"
     fi
 done
-echo "finished finding best python"
+
 if [ -z "$BEST_PY" ]; then
-    echo "❌ ERROR: No Python version between 3.${MIN_PY_MINOR} and 3.${MAX_PY_MINOR} found on this system."
-    echo "   Detected Python 3 installations:"
+    step "python" "no compatible version found (need 3.${MIN_PY_MINOR}–3.${MAX_PY_MINOR})" "$C_ERR"
+    substep "detected:"
     for candidate in $(compgen -c python3 2>/dev/null | grep -E '^python3(\.[0-9]+)?$' | sort -u); do
-        if command -v "$candidate" &>/dev/null; then
-            echo "     - $candidate ($($candidate --version 2>&1))"
-        fi
+        command -v "$candidate" &>/dev/null && substep "  $candidate ($($candidate --version 2>&1))"
     done
-    echo ""
-    echo "   Please install Python 3.${MIN_PY_MINOR} or 3.${MAX_PY_MINOR}."
-    echo "   For example:  sudo apt install python3.12 python3.12-venv"
+    substep "install with: sudo apt install python3.12 python3.12-venv"
     exit 1
 fi
 
 BEST_VER=$("$BEST_PY" --version 2>&1 | awk '{print $2}')
-echo "✅ Using $BEST_PY ($BEST_VER) — compatible (3.${MIN_PY_MINOR}.x – 3.${MAX_PY_MINOR}.x)"
+step "python" "$BEST_VER (3.${MIN_PY_MINOR}.x – 3.${MAX_PY_MINOR}.x)"
 
+# ── Venv + Python deps ──
 REQ_ROOT="$SCRIPT_DIR/backend/requirements"
 SINGLE_ENV_CONSTRAINTS="$REQ_ROOT/single-env/constraints.txt"
 SINGLE_ENV_DATA_DESIGNER="$REQ_ROOT/single-env/data-designer.txt"
@@ -251,23 +242,17 @@ install_python_stack() {
     python "$SCRIPT_DIR/install_python_stack.py"
 }
 
-# Create venv under ~/.unsloth/studio/ (shared location, not in repo).
-# All platforms (including Colab) use the same isolated venv so that
-# studio dependencies are never installed into the system Python.
 STUDIO_HOME="$HOME/.unsloth/studio"
 VENV_DIR="$STUDIO_HOME/.venv"
 VENV_T5_DIR="$STUDIO_HOME/.venv_t5"
 mkdir -p "$STUDIO_HOME"
 
-# Clean up legacy in-repo venvs if they exist
 [ -d "$REPO_ROOT/.venv" ] && rm -rf "$REPO_ROOT/.venv"
 [ -d "$REPO_ROOT/.venv_overlay" ] && rm -rf "$REPO_ROOT/.venv_overlay"
 [ -d "$REPO_ROOT/.venv_t5" ] && rm -rf "$REPO_ROOT/.venv_t5"
 
 rm -rf "$VENV_DIR"
 rm -rf "$VENV_T5_DIR"
-# Try creating venv with pip; fall back to --without-pip + bootstrap
-# (some environments like Colab have broken ensurepip)
 if ! "$BEST_PY" -m venv "$VENV_DIR" 2>/dev/null; then
     "$BEST_PY" -m venv --without-pip "$VENV_DIR"
     source "$VENV_DIR/bin/activate"
@@ -276,7 +261,6 @@ else
     source "$VENV_DIR/bin/activate"
 fi
 
-# ── Ensure uv is available (much faster than pip) ──
 USE_UV=false
 if command -v uv &>/dev/null; then
     USE_UV=true
@@ -285,7 +269,6 @@ elif curl -LsSf https://astral.sh/uv/install.sh | sh > /dev/null 2>&1; then
     command -v uv &>/dev/null && USE_UV=true
 fi
 
-# Helper: install a package, preferring uv with pip fallback
 fast_install() {
     if [ "$USE_UV" = true ]; then
         uv pip install --python "$(command -v python)" "$@" && return 0
@@ -296,34 +279,20 @@ fast_install() {
 cd "$SCRIPT_DIR"
 install_python_stack
 
-# ── 6b. Pre-install transformers 5.x into .venv_t5/ ──
-# Models like GLM-4.7-Flash need transformers>=5.3.0. Instead of pip-installing
-# at runtime (slow, ~10-15s), we pre-install into a separate directory.
-# The training subprocess just prepends .venv_t5/ to sys.path -- instant switch.
-echo ""
-echo "   Pre-installing transformers 5.x for newer model support..."
+# ── Transformers 5.x ──
 mkdir -p "$VENV_T5_DIR"
 run_quiet "install transformers 5.x" fast_install --target "$VENV_T5_DIR" --no-deps "transformers==5.3.0"
 run_quiet "install huggingface_hub for t5" fast_install --target "$VENV_T5_DIR" --no-deps "huggingface_hub==1.7.1"
 run_quiet "install hf_xet for t5" fast_install --target "$VENV_T5_DIR" --no-deps "hf_xet==1.4.2"
-# tiktoken is needed by Qwen-family tokenizers. Install with deps since
-# regex/requests may be missing on Windows.
 run_quiet "install tiktoken for t5" fast_install --target "$VENV_T5_DIR" "tiktoken"
-echo "✅ Transformers 5.x pre-installed to $VENV_T5_DIR/"
+step "transformers" "5.x pre-installed"
 
-# ── 7. WSL: pre-install GGUF build dependencies ──
-# On WSL, sudo requires a password and can't be entered during GGUF export
-# (runs in a non-interactive subprocess). Install build deps here instead.
+# ── WSL: GGUF build dependencies ──
 if grep -qi microsoft /proc/version 2>/dev/null; then
-    echo ""
-    echo "⚠️  WSL detected -- installing build dependencies for GGUF export..."
     _GGUF_DEPS="pciutils build-essential cmake curl git libcurl4-openssl-dev"
-
-    # Try without sudo first (works when already root)
     apt-get update -y >/dev/null 2>&1 || true
     apt-get install -y $_GGUF_DEPS >/dev/null 2>&1 || true
 
-    # Check which packages are still missing
     _STILL_MISSING=""
     for _pkg in $_GGUF_DEPS; do
         case "$_pkg" in
@@ -336,16 +305,11 @@ if grep -qi microsoft /proc/version 2>/dev/null; then
     _STILL_MISSING=$(echo "$_STILL_MISSING" | sed 's/^ *//')
 
     if [ -z "$_STILL_MISSING" ]; then
-        echo "✅ GGUF build dependencies installed"
+        step "gguf deps" "installed"
     elif command -v sudo >/dev/null 2>&1; then
-        echo ""
-        echo "   !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"
-        echo "   WARNING: We require sudo elevated permissions to install:"
-        echo "   $_STILL_MISSING"
-        echo "   If you accept, we'll run sudo now, and it'll prompt your password."
-        echo "   !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"
-        echo ""
-        printf "   Accept? [Y/n] "
+        step "gguf deps" "sudo required for: $_STILL_MISSING" "$C_WARN"
+        printf "  %-15s" ""
+        printf "accept? [Y/n] "
         if [ -r /dev/tty ]; then
             read -r REPLY </dev/tty || REPLY="y"
         else
@@ -353,95 +317,72 @@ if grep -qi microsoft /proc/version 2>/dev/null; then
         fi
         case "$REPLY" in
             [nN]*)
-                echo ""
-                echo "   Please install these packages first, then re-run Unsloth Studio setup:"
-                echo "   sudo apt-get update -y && sudo apt-get install -y $_STILL_MISSING"
+                substep "skipped — run manually:"
+                substep "sudo apt-get install -y $_STILL_MISSING"
                 _SKIP_GGUF_BUILD=true
                 ;;
             *)
                 sudo apt-get update -y
                 sudo apt-get install -y $_STILL_MISSING
-                echo "✅ GGUF build dependencies installed"
+                step "gguf deps" "installed"
                 ;;
         esac
     else
-        echo "   sudo is not available on this system."
-        echo "   Please install as root, then re-run setup:"
-        echo "   apt-get install -y $_STILL_MISSING"
+        step "gguf deps" "missing (no sudo) — install manually:" "$C_WARN"
+        substep "apt-get install -y $_STILL_MISSING"
         _SKIP_GGUF_BUILD=true
     fi
 fi
 
-# ── 8. Build llama.cpp binaries for GGUF inference + export ──
-# Builds at ~/.unsloth/llama.cpp — a single shared location under the user's
-# home directory. This is used by both the inference server and the GGUF
-# export pipeline (unsloth-zoo).
-#   - llama-server: for GGUF model inference
-#   - llama-quantize: for GGUF export quantization (symlinked to root for check_llama_cpp())
+# ── llama.cpp ──
 UNSLOTH_HOME="$HOME/.unsloth"
 mkdir -p "$UNSLOTH_HOME"
 LLAMA_CPP_DIR="$UNSLOTH_HOME/llama.cpp"
 LLAMA_SERVER_BIN="$LLAMA_CPP_DIR/build/bin/llama-server"
 if [ "${_SKIP_GGUF_BUILD:-}" = true ]; then
-    echo ""
-    echo "Skipping llama-server build (missing dependencies)"
-    echo "   Install the missing packages and re-run setup to enable GGUF inference."
+    step "llama.cpp" "skipped (missing build deps)" "$C_WARN"
 else
 rm -rf "$LLAMA_CPP_DIR"
 {
-    # Check prerequisites
     if ! command -v cmake &>/dev/null; then
-        echo ""
-        echo "⚠️  cmake not found — skipping llama-server build (GGUF inference won't be available)"
-        echo "   Install cmake and re-run setup.sh to enable GGUF inference."
+        step "llama.cpp" "skipped (cmake not found)" "$C_WARN"
     elif ! command -v git &>/dev/null; then
-        echo ""
-        echo "⚠️  git not found — skipping llama-server build (GGUF inference won't be available)"
+        step "llama.cpp" "skipped (git not found)" "$C_WARN"
     else
-        echo ""
-        echo "Building llama-server for GGUF inference..."
-
         BUILD_OK=true
-        run_quiet "clone llama.cpp" git clone --depth 1 https://github.com/ggml-org/llama.cpp.git "$LLAMA_CPP_DIR" || BUILD_OK=false
+        try_quiet "clone llama.cpp" git clone --depth 1 https://github.com/ggml-org/llama.cpp.git "$LLAMA_CPP_DIR" || BUILD_OK=false
 
         if [ "$BUILD_OK" = true ]; then
-            # Skip tests/examples we don't need (faster build)
             CMAKE_ARGS="-DLLAMA_BUILD_TESTS=OFF -DLLAMA_BUILD_EXAMPLES=OFF -DLLAMA_BUILD_SERVER=ON -DGGML_NATIVE=ON"
 
-            # Use ccache if available (dramatically faster rebuilds)
             if command -v ccache &>/dev/null; then
                 CMAKE_ARGS="$CMAKE_ARGS -DCMAKE_C_COMPILER_LAUNCHER=ccache -DCMAKE_CXX_COMPILER_LAUNCHER=ccache -DCMAKE_CUDA_COMPILER_LAUNCHER=ccache"
-                echo "   Using ccache for faster compilation"
             fi
 
-            # Detect CUDA: check nvcc on PATH, then common install locations
+            # Prefer newest /usr/local/cuda-*/bin/nvcc (e.g. 12.8 for sm_120) over
+            # /usr/bin/nvcc from distro nvidia-cuda-toolkit (often 12.0, too old).
             NVCC_PATH=""
-            if command -v nvcc &>/dev/null; then
-                NVCC_PATH="$(command -v nvcc)"
+            if ls /usr/local/cuda-*/bin/nvcc &>/dev/null 2>&1; then
+                NVCC_PATH="$(ls -d /usr/local/cuda-*/bin/nvcc 2>/dev/null | sort -V | tail -1)"
+                export PATH="$(dirname "$NVCC_PATH"):$PATH"
             elif [ -x /usr/local/cuda/bin/nvcc ]; then
                 NVCC_PATH="/usr/local/cuda/bin/nvcc"
                 export PATH="/usr/local/cuda/bin:$PATH"
-            elif ls /usr/local/cuda-*/bin/nvcc &>/dev/null 2>&1; then
-                # Pick the newest cuda-XX.X directory
-                NVCC_PATH="$(ls -d /usr/local/cuda-*/bin/nvcc 2>/dev/null | sort -V | tail -1)"
-                export PATH="$(dirname "$NVCC_PATH"):$PATH"
+            elif command -v nvcc &>/dev/null; then
+                NVCC_PATH="$(command -v nvcc)"
             fi
 
+            _BUILD_DESC="building"
             if [ -n "$NVCC_PATH" ]; then
-                echo "   Building with CUDA support (nvcc: $NVCC_PATH)..."
-                CMAKE_ARGS="$CMAKE_ARGS -DGGML_CUDA=ON"
+                CMAKE_ARGS="$CMAKE_ARGS -DGGML_CUDA=ON -DCMAKE_CUDA_COMPILER=$NVCC_PATH"
 
-                # Detect GPU compute capability and limit CUDA architectures
-                # Without this, cmake builds for ALL default archs (very slow)
                 CUDA_ARCHS=""
                 if command -v nvidia-smi &>/dev/null; then
-                    # Read all GPUs, deduplicate (handles mixed-GPU hosts)
                     _raw_caps=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader 2>/dev/null || true)
                     while IFS= read -r _cap; do
                         _cap=$(echo "$_cap" | tr -d '[:space:]')
                         if [[ "$_cap" =~ ^([0-9]+)\.([0-9]+)$ ]]; then
                             _arch="${BASH_REMATCH[1]}${BASH_REMATCH[2]}"
-                            # Append if not already present
                             case ";$CUDA_ARCHS;" in
                                 *";$_arch;"*) ;;
                                 *) CUDA_ARCHS="${CUDA_ARCHS:+$CUDA_ARCHS;}$_arch" ;;
@@ -451,64 +392,56 @@ rm -rf "$LLAMA_CPP_DIR"
                 fi
 
                 if [ -n "$CUDA_ARCHS" ]; then
-                    echo "   GPU compute capabilities: ${CUDA_ARCHS//;/, } -- limiting build to detected archs"
                     CMAKE_ARGS="$CMAKE_ARGS -DCMAKE_CUDA_ARCHITECTURES=${CUDA_ARCHS}"
+                    _BUILD_DESC="building (CUDA, sm_${CUDA_ARCHS//;/+sm_})"
                 else
-                    echo "   Could not detect GPU arch -- building for all default CUDA architectures (slower)"
+                    _BUILD_DESC="building (CUDA)"
                 fi
 
-                # Multi-threaded nvcc compilation (uses all CPU cores per .cu file)
                 CMAKE_ARGS="$CMAKE_ARGS -DCMAKE_CUDA_FLAGS=--threads=0"
-            elif [ -d /usr/local/cuda ] || nvidia-smi &>/dev/null; then
-                echo "   CUDA driver detected but nvcc not found — building CPU-only"
-                echo "   To enable GPU: install cuda-toolkit or add nvcc to PATH"
             else
-                echo "   Building CPU-only (no CUDA detected)..."
+                _BUILD_DESC="building (CPU)"
             fi
 
-            NCPU=$(nproc 2>/dev/null || sysctl -n hw.ncpu 2>/dev/null || echo 4)
+            substep "$_BUILD_DESC..."
 
-            # Use Ninja if available (faster parallel builds than Make)
+            NCPU=$(nproc 2>/dev/null || sysctl -n hw.ncpu 2>/dev/null || echo 4)
             CMAKE_GENERATOR_ARGS=""
             if command -v ninja &>/dev/null; then
                 CMAKE_GENERATOR_ARGS="-G Ninja"
             fi
 
-            run_quiet "cmake llama.cpp" cmake $CMAKE_GENERATOR_ARGS -S "$LLAMA_CPP_DIR" -B "$LLAMA_CPP_DIR/build" $CMAKE_ARGS || BUILD_OK=false
+            try_quiet "cmake llama.cpp" cmake $CMAKE_GENERATOR_ARGS -S "$LLAMA_CPP_DIR" -B "$LLAMA_CPP_DIR/build" $CMAKE_ARGS || BUILD_OK=false
         fi
 
         if [ "$BUILD_OK" = true ]; then
-            run_quiet "build llama-server" cmake --build "$LLAMA_CPP_DIR/build" --config Release --target llama-server -j"$NCPU" || BUILD_OK=false
+            try_quiet "build llama-server" cmake --build "$LLAMA_CPP_DIR/build" --config Release --target llama-server -j"$NCPU" || BUILD_OK=false
         fi
 
-        # Also build llama-quantize (needed by unsloth-zoo's GGUF export pipeline)
         if [ "$BUILD_OK" = true ]; then
-            run_quiet "build llama-quantize" cmake --build "$LLAMA_CPP_DIR/build" --config Release --target llama-quantize -j"$NCPU" || true
-            # Symlink to llama.cpp root — check_llama_cpp() looks for the binary there
+            try_quiet "build llama-quantize" cmake --build "$LLAMA_CPP_DIR/build" --config Release --target llama-quantize -j"$NCPU" || true
             QUANTIZE_BIN="$LLAMA_CPP_DIR/build/bin/llama-quantize"
             if [ -f "$QUANTIZE_BIN" ]; then
                 ln -sf build/bin/llama-quantize "$LLAMA_CPP_DIR/llama-quantize"
             fi
         fi
 
-        if [ "$BUILD_OK" = true ]; then
-            if [ -f "$LLAMA_SERVER_BIN" ]; then
-                echo "✅ llama-server built at $LLAMA_SERVER_BIN"
-            else
-                echo "⚠️  llama-server binary not found after build — GGUF inference won't be available"
-            fi
-            if [ -f "$LLAMA_CPP_DIR/llama-quantize" ]; then
-                echo "✅ llama-quantize available for GGUF export"
-            fi
+        if [ "$BUILD_OK" = true ] && [ -f "$LLAMA_SERVER_BIN" ]; then
+            step "llama.cpp" "built"
+            [ -f "$LLAMA_CPP_DIR/llama-quantize" ] && step "llama-quantize" "built"
+        elif [ "$BUILD_OK" = true ]; then
+            step "llama.cpp" "binary not found after build" "$C_WARN"
         else
-            echo "⚠️  llama-server build failed — GGUF inference won't be available, but everything else works"
+            step "llama.cpp" "build failed (non-fatal)" "$C_WARN"
         fi
     fi
 }
 fi  # end _SKIP_GGUF_BUILD check
 
-echo ""
+# ── Footer ──
+# Colab + launch example: match unslothai/unsloth (upstream main) studio/setup.sh
 if [ "$IS_COLAB" = true ]; then
+    echo ""
     echo "╔══════════════════════════════════════╗"
     echo "║           Setup Complete!            ║"
     echo "╠══════════════════════════════════════╣"
@@ -519,11 +452,9 @@ if [ "$IS_COLAB" = true ]; then
     echo "║ start()                              ║"
     echo "╚══════════════════════════════════════╝"
 else
-    echo "╔══════════════════════════════════════╗"
-    echo "║           Setup Complete!            ║"
-    echo "╠══════════════════════════════════════╣"
-    echo "║ Launch with:                         ║"
-    echo "║                                      ║"
-    echo "║ unsloth studio -H 0.0.0.0 -p 8888    ║"
-    echo "╚══════════════════════════════════════╝"
+    printf "  ${C_DIM}%s${C_RST}\n" "$RULE"
+    printf "  ${C_TITLE}%s${C_RST}\n" "Unsloth Studio Installed"
+    printf "  ${C_DIM}%s${C_RST}\n" "$RULE"
+    printf "  ${C_DIM}%-15s${C_OK}%s${C_RST}\n" "launch" "unsloth studio -H 0.0.0.0 -p 8888"
 fi
+echo ""
diff --git a/unsloth_cli/commands/studio.py b/unsloth_cli/commands/studio.py
index 8a2267a79e..533d2a5ca8 100644
--- a/unsloth_cli/commands/studio.py
+++ b/unsloth_cli/commands/studio.py
@@ -162,19 +162,32 @@ def studio_default(
 
 
 @studio_app.command()
-def setup():
+def setup(
+    verbose: bool = typer.Option(
+        False,
+        "--verbose",
+        "-v",
+        help=(
+            "Full pip output during Python install; print command logs when "
+            "optional steps fail (e.g. llama.cpp cmake)."
+        ),
+    ),
+):
     """Run one-time Studio environment setup."""
     script = _find_setup_script()
     if not script:
         typer.echo("Error: Could not find setup script (setup.sh / setup.ps1).")
         raise typer.Exit(1)
 
+    env = {**os.environ, "UNSLOTH_VERBOSE": "1"} if verbose else None
+
     if platform.system() == "Windows":
         result = subprocess.run(
             ["powershell", "-ExecutionPolicy", "Bypass", "-File", str(script)],
+            env = env,
         )
     else:
-        result = subprocess.run(["bash", str(script)])
+        result = subprocess.run(["bash", str(script)], env = env)
 
     if result.returncode != 0:
         raise typer.Exit(result.returncode)

From 234a94de4c819b432a748b3c2e3ed5e9078c8cc2 Mon Sep 17 00:00:00 2001
From: Lee Jackson <samleejackson0@gmail.com>
Date: Thu, 19 Mar 2026 21:07:01 +0000
Subject: [PATCH 02/16] studio(windows): align setup.ps1 banner/steps with
 setup.sh (ANSI, verbose)

---
 studio/setup.ps1 | 689 +++++++++++++++++++++++++++--------------------
 1 file changed, 404 insertions(+), 285 deletions(-)

diff --git a/studio/setup.ps1 b/studio/setup.ps1
index e73b04ad7c..f340d30f9e 100644
--- a/studio/setup.ps1
+++ b/studio/setup.ps1
@@ -10,21 +10,41 @@
     full setup including frontend build.
     Supports NVIDIA GPU (full training + inference) and CPU-only (GGUF chat mode).
 .NOTES
-    Usage: powershell -ExecutionPolicy Bypass -File setup.ps1
+    Default output is minimal (step/substep), aligned with studio/setup.sh.
+
+    FULL / LEGACY LOGGING (defensible audit trail, multi-line [OK]/[WARN]/paths):
+      unsloth studio setup --verbose
+      (sets UNSLOTH_VERBOSE=1; same as install_python_stack.py)
+      Or:  $env:UNSLOTH_VERBOSE='1'; powershell -File .\studio\setup.ps1
+      Or:  .\setup.ps1 --verbose
+
+    Why WSL/bash can look richer than PowerShell 5.1:
+      - setup.sh uses printf with 256-color ANSI (e.g. 38;5;150).
+      - Plain Write-Host -ForegroundColor uses ~16 ConsoleColors unless VT is on.
+      - With VT, step/substep use the same C_DIM/C_OK/C_WARN/C_ERR codes as setup.sh.
 #>
 
 $ErrorActionPreference = "Stop"
 $ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
 $PackageDir = Split-Path -Parent $ScriptDir
 
+# Same as: unsloth studio setup --verbose  (see unsloth_cli/commands/studio.py)
+foreach ($a in $args) {
+    if ($a -eq '--verbose' -or $a -eq '-v') {
+        $env:UNSLOTH_VERBOSE = '1'
+        break
+    }
+}
+$script:UnslothVerbose = ($env:UNSLOTH_VERBOSE -eq '1')
+
 # Detect if running from pip install (no frontend/ dir in studio)
 $FrontendDir = Join-Path $ScriptDir "frontend"
 $OxcValidatorDir = Join-Path $ScriptDir "backend\core\data_recipe\oxc-validator"
 $IsPipInstall = -not (Test-Path $FrontendDir)
 
-# ─────────────────────────────────────────────
+# ---------------------------------------------
 # Helper functions
-# ─────────────────────────────────────────────
+# ---------------------------------------------
 
 # Reload ALL environment variables from registry.
 # Picks up changes made by installers (winget, msi, etc.) including
@@ -78,7 +98,7 @@ function Find-Nvcc {
         return $null
     }
 
-    # Fallback: no version constraint — pick latest or whatever is available
+    # Fallback: no version constraint - pick latest or whatever is available
 
     # 1. Check nvcc on PATH
     $cmd = Get-Command nvcc -ErrorAction SilentlyContinue
@@ -133,7 +153,7 @@ function Get-CudaComputeCapability {
 # Check if an nvcc binary supports a given sm_ architecture.
 # Uses `nvcc --list-gpu-code` which outputs sm_* tokens (--list-gpu-arch
 # outputs compute_* tokens instead).  Available since CUDA 11.6.
-# Returns $false if the flag isn't supported (old toolkit) — safer to reject
+# Returns $false if the flag isn't supported (old toolkit) - safer to reject
 # and fall back to scanning/PTX than to assume support and fail later.
 function Test-NvccArchSupport {
     param([string]$NvccExe, [string]$Arch)
@@ -170,7 +190,7 @@ function Get-NvccMaxArch {
 # compatible PyTorch CUDA index tag (e.g. "cu128").
 # PyTorch on Windows ships CPU-only by default from PyPI; CUDA wheels live at
 # https://download.pytorch.org/whl/<tag>. The tag must not exceed the driver's
-# capability: e.g. driver "CUDA Version: 12.9" → cu128 (not cu130).
+# capability: e.g. driver "CUDA Version: 12.9" -> cu128 (not cu130).
 function Get-PytorchCudaTag {
     $smiExe = if ($script:NvidiaSmiExe) { $script:NvidiaSmiExe } else {
         $cmd = Get-Command nvidia-smi -ErrorAction SilentlyContinue
@@ -245,44 +265,144 @@ function Find-VsBuildTools {
     return $null
 }
 
-# ─────────────────────────────────────────────
-# Output style helpers (matches setup.sh visual tone)
-# ─────────────────────────────────────────────
-$Rule = "────────────────────────────────────────────────────"
+# ---------------------------------------------
+# Output style (aligned with studio/setup.sh: step / substep)
+# UNSLOTH_VERBOSE=1 restores pre-minimal multi-line diagnostics (see .NOTES).
+# ---------------------------------------------
+# U+2500 x52 (same count as setup.sh RULE); codepoint form avoids UTF-8 literal mojibake in PS 5.1
+$Rule = [string]::new([char]0x2500, 52)
 
-function Write-Step {
+# Enable ANSI (256-color) on Windows 10+ console when NO_COLOR unset -- matches bash look more closely.
+function Enable-StudioVirtualTerminal {
+    if ($env:NO_COLOR) { return $false }
+    try {
+        Add-Type -Namespace StudioVT -Name Native -MemberDefinition @'
+[DllImport("kernel32.dll")] public static extern IntPtr GetStdHandle(int nStdHandle);
+[DllImport("kernel32.dll")] public static extern bool GetConsoleMode(IntPtr h, out uint m);
+[DllImport("kernel32.dll")] public static extern bool SetConsoleMode(IntPtr h, uint m);
+'@ -ErrorAction Stop
+        $h = [StudioVT.Native]::GetStdHandle(-11)
+        [uint32]$mode = 0
+        if (-not [StudioVT.Native]::GetConsoleMode($h, [ref]$mode)) { return $false }
+        $mode = $mode -bor 0x0004
+        return [StudioVT.Native]::SetConsoleMode($h, $mode)
+    } catch {
+        return $false
+    }
+}
+$script:StudioVtOk = Enable-StudioVirtualTerminal
+
+# Same 256-color codes as studio/setup.sh: C_TITLE / C_DIM / C_OK / C_WARN / C_ERR
+function Get-StudioAnsi {
+    param(
+        [Parameter(Mandatory = $true)]
+        [ValidateSet('Title', 'Dim', 'Ok', 'Warn', 'Err', 'Reset')]
+        [string]$Kind
+    )
+    $e = [char]27
+    switch ($Kind) {
+        'Title' { return "${e}[38;5;150m" }
+        'Dim'   { return "${e}[38;5;245m" }
+        'Ok'    { return "${e}[38;5;108m" }
+        'Warn'  { return "${e}[38;5;136m" }
+        'Err'   { return "${e}[91m" }
+        'Reset' { return "${e}[0m" }
+    }
+}
+
+function Write-SetupVerboseDetail {
+    param(
+        [Parameter(Mandatory = $true)][string]$Message,
+        [string]$Color = "Gray"
+    )
+    if (-not $script:UnslothVerbose) { return }
+    if ($script:StudioVtOk -and -not $env:NO_COLOR) {
+        $ansi = switch ($Color) {
+            'Green' { (Get-StudioAnsi Ok) }
+            'Gray' { (Get-StudioAnsi Dim) }
+            'DarkGray' { (Get-StudioAnsi Dim) }
+            'Yellow' { (Get-StudioAnsi Warn) }
+            'Cyan' { (Get-StudioAnsi Title) }
+            'Red' { (Get-StudioAnsi Err) }
+            default { (Get-StudioAnsi Dim) }
+        }
+        Write-Host ($ansi + $Message + (Get-StudioAnsi Reset))
+    } else {
+        $fc = switch ($Color) {
+            'Green' { 'DarkGreen' }
+            'Gray' { 'DarkGray' }
+            'Cyan' { 'Green' }
+            default { $Color }
+        }
+        Write-Host $Message -ForegroundColor $fc
+    }
+}
+
+function step {
     param(
         [Parameter(Mandatory = $true)][string]$Label,
         [Parameter(Mandatory = $true)][string]$Value,
         [string]$Color = "Green"
     )
-    Write-Host ("  {0,-15}" -f $Label) -NoNewline -ForegroundColor DarkGray
-    Write-Host $Value -ForegroundColor $Color
+    if ($script:StudioVtOk -and -not $env:NO_COLOR) {
+        $dim = Get-StudioAnsi Dim
+        $rst = Get-StudioAnsi Reset
+        $val = switch ($Color) {
+            'Green' { Get-StudioAnsi Ok }
+            'Yellow' { Get-StudioAnsi Warn }
+            'Red' { Get-StudioAnsi Err }
+            'DarkGray' { Get-StudioAnsi Dim }
+            default { Get-StudioAnsi Ok }
+        }
+        $padded = if ($Label.Length -ge 15) { $Label.Substring(0, 15) } else { $Label.PadRight(15) }
+        Write-Host ("  {0}{1}{2}{3}{4}{2}" -f $dim, $padded, $rst, $val, $Value)
+    } else {
+        $padded = if ($Label.Length -ge 15) { $Label.Substring(0, 15) } else { $Label.PadRight(15) }
+        Write-Host ("  {0}" -f $padded) -NoNewline -ForegroundColor DarkGray
+        $fc = switch ($Color) {
+            'Green' { 'DarkGreen' }
+            'Yellow' { 'Yellow' }
+            'Red' { 'Red' }
+            'DarkGray' { 'DarkGray' }
+            default { 'DarkGreen' }
+        }
+        Write-Host $Value -ForegroundColor $fc
+    }
 }
 
-function Write-Substep {
+function substep {
     param(
         [Parameter(Mandatory = $true)][string]$Message,
         [string]$Color = "DarkGray"
     )
-    Write-Host ("  {0,-15}{1}" -f "", $Message) -ForegroundColor $Color
+    if ($script:StudioVtOk -and -not $env:NO_COLOR) {
+        $msgCol = switch ($Color) {
+            'Yellow' { (Get-StudioAnsi Warn) }
+            default { (Get-StudioAnsi Dim) }
+        }
+        $pad = "".PadRight(15)
+        Write-Host ("  {0}{1}{2}{3}" -f $msgCol, $pad, $Message, (Get-StudioAnsi Reset))
+    } else {
+        $fc = switch ($Color) {
+            'Yellow' { 'Yellow' }
+            default { 'DarkGray' }
+        }
+        Write-Host ("  {0,-15}{1}" -f "", $Message) -ForegroundColor $fc
+    }
 }
 
-# ─────────────────────────────────────────────
-# Banner
-# ─────────────────────────────────────────────
 Write-Host ""
-Write-Host "  🦥 Unsloth Studio Setup" -ForegroundColor Green
-Write-Host "  $Rule" -ForegroundColor DarkGray
-
-# ==========================================================================
-#  PHASE 1: System-level prerequisites (winget installs, env vars)
-#  All heavy system tool installs happen here BEFORE touching Python.
-# ==========================================================================
+if ($script:StudioVtOk -and -not $env:NO_COLOR) {
+    # Sloth U+1F9A5 (supplementary plane): ConvertFromUtf32; `u{1F9A5} inside "-f" breaks {1} parsing
+    Write-Host ("  " + (Get-StudioAnsi Title) + [char]::ConvertFromUtf32(0x1F9A5) + " Unsloth Studio Setup" + (Get-StudioAnsi Reset))
+    Write-Host ("  {0}{1}{2}" -f (Get-StudioAnsi Dim), $Rule, (Get-StudioAnsi Reset))
+} else {
+    # No VT: approximate C_TITLE (150) as green; C_OK-style text as dark green
+    Write-Host ("  " + [char]::ConvertFromUtf32(0x1F9A5) + " Unsloth Studio Setup") -ForegroundColor Green
+    Write-Host "  $Rule" -ForegroundColor DarkGray
+}
 
-# ============================================
-# 1a. GPU detection
-# ============================================
+# --- Phase 1: system prerequisites (winget, env) ---
 $HasNvidiaSmi = $false
 $NvidiaSmiExe = $null  # Absolute path -- survives Refresh-Environment
 try {
@@ -309,7 +429,9 @@ if (-not $HasNvidiaSmi) {
                 if ($LASTEXITCODE -eq 0) {
                     $HasNvidiaSmi = $true
                     $NvidiaSmiExe = $p
-                    Write-Host "   Found nvidia-smi at $(Split-Path $p -Parent)" -ForegroundColor Gray
+                    if ($script:UnslothVerbose) {
+                        substep "nvidia-smi: $(Split-Path $p -Parent)"
+                    }
                     break
                 }
             } catch {}
@@ -317,18 +439,20 @@ if (-not $HasNvidiaSmi) {
     }
 }
 if (-not $HasNvidiaSmi) {
-    Write-Host ""
-    Write-Host "[WARN] No NVIDIA GPU detected. Studio will run in chat-only (GGUF) mode." -ForegroundColor Yellow
-    Write-Host "       Training and GPU inference require an NVIDIA GPU with drivers installed." -ForegroundColor Yellow
-    Write-Host "       https://www.nvidia.com/Download/index.aspx" -ForegroundColor Yellow
-    Write-Host ""
+    step "gpu" "none (chat-only / GGUF)" "Yellow"
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "[WARN] No NVIDIA GPU detected. Studio will run in chat-only (GGUF) mode." "Yellow"
+        Write-SetupVerboseDetail "       Training and GPU inference require an NVIDIA GPU with drivers installed." "Yellow"
+        Write-SetupVerboseDetail "       https://www.nvidia.com/Download/index.aspx" "Yellow"
+    }
 } else {
-    Write-Host "[OK] NVIDIA GPU detected" -ForegroundColor Green
+    step "gpu" "NVIDIA detected"
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "[OK] NVIDIA GPU detected" "Green"
+    }
 }
 
-# ============================================
-# 1a.5. Windows Long Paths (required for deep node_modules / Python paths)
-# ============================================
+# --- Windows long paths (deep node_modules / Python) ---
 $LongPathsEnabled = $false
 try {
     $regVal = Get-ItemProperty -Path "HKLM:\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -ErrorAction SilentlyContinue
@@ -338,34 +462,33 @@ try {
 } catch {}
 
 if ($LongPathsEnabled) {
-    Write-Host "[OK] Windows Long Paths enabled" -ForegroundColor Green
+    step "long paths" "enabled"
 } else {
-    Write-Host "Windows Long Paths not enabled (required for Triton compilation and deep dependency paths)." -ForegroundColor Yellow
-    Write-Host "   Requesting admin access to fix..." -ForegroundColor Yellow
+    substep "enabling long paths (UAC may prompt)..."
     try {
-        # Spawn an elevated process to set the registry key (triggers UAC prompt)
         $proc = Start-Process -FilePath "reg.exe" `
             -ArgumentList 'add "HKLM\SYSTEM\CurrentControlSet\Control\FileSystem" /v LongPathsEnabled /t REG_DWORD /d 1 /f' `
             -Verb RunAs -Wait -PassThru -ErrorAction Stop
         if ($proc.ExitCode -eq 0) {
             $LongPathsEnabled = $true
-            Write-Host "[OK] Windows Long Paths enabled (via UAC)" -ForegroundColor Green
+            step "long paths" "enabled"
         } else {
-            Write-Host "[WARN] Failed to enable Long Paths (exit code: $($proc.ExitCode))" -ForegroundColor Yellow
+            step "long paths" "failed (exit $($proc.ExitCode))" "Yellow"
         }
     } catch {
-        Write-Host "[WARN] Could not enable Long Paths (UAC was declined or not available)" -ForegroundColor Yellow
-        Write-Host "       Run this manually in an Admin terminal:" -ForegroundColor Yellow
-        Write-Host '       reg add "HKLM\SYSTEM\CurrentControlSet\Control\FileSystem" /v LongPathsEnabled /t REG_DWORD /d 1 /f' -ForegroundColor Cyan
+        step "long paths" "skipped (run reg add LongPathsEnabled as Admin)" "Yellow"
+        if ($script:UnslothVerbose) {
+            Write-SetupVerboseDetail "       Run this manually in an Admin terminal:" "Yellow"
+            Write-SetupVerboseDetail '       reg add "HKLM\SYSTEM\CurrentControlSet\Control\FileSystem" /v LongPathsEnabled /t REG_DWORD /d 1 /f' "Cyan"
+        }
     }
 }
 
-# ============================================
-# 1b. Git (required by pip for git+https:// deps and by npm)
-# ============================================
+# --- Git ---
 $HasGit = $null -ne (Get-Command git -ErrorAction SilentlyContinue)
 if (-not $HasGit) {
-    Write-Host "Git not found -- installing via winget..." -ForegroundColor Yellow
+    if ($script:UnslothVerbose) { Write-SetupVerboseDetail "Git not found -- installing via winget..." "Yellow" }
+    substep "installing git..."
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
         try {
@@ -375,21 +498,18 @@ if (-not $HasGit) {
         } catch { }
     }
     if (-not $HasGit) {
-        Write-Host "[ERROR] Git is required but could not be installed automatically." -ForegroundColor Red
-        Write-Host "        Install Git from https://git-scm.com/download/win and re-run." -ForegroundColor Red
+        step "error" "git required -- https://git-scm.com/download/win" "Red"
         exit 1
     }
-    Write-Host "[OK] Git installed: $(git --version)" -ForegroundColor Green
-} else {
-    Write-Host "[OK] Git found: $(git --version)" -ForegroundColor Green
+    if ($script:UnslothVerbose) { Write-SetupVerboseDetail "[OK] Git installed: $(git --version)" "Green" }
 }
+step "git" "$(git --version)"
 
-# ============================================
-# 1c. CMake (required for llama.cpp build)
-# ============================================
+# --- CMake ---
 $HasCmake = $null -ne (Get-Command cmake -ErrorAction SilentlyContinue)
 if (-not $HasCmake) {
-    Write-Host "CMake not found -- installing via winget..." -ForegroundColor Yellow
+    if ($script:UnslothVerbose) { Write-SetupVerboseDetail "CMake not found -- installing via winget..." "Yellow" }
+    substep "installing cmake..."
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
         try {
@@ -416,33 +536,31 @@ if (-not $HasCmake) {
                 }
                 $HasCmake = $null -ne (Get-Command cmake -ErrorAction SilentlyContinue)
                 if ($HasCmake) {
-                    Write-Host "   Found cmake at $d (added to PATH)" -ForegroundColor Gray
+                    if ($script:UnslothVerbose) { substep "cmake at $d (PATH)" }
                     break
                 }
             }
         }
     }
-    if ($HasCmake) {
-        Write-Host "[OK] CMake installed" -ForegroundColor Green
-    } else {
-        Write-Host "[ERROR] CMake is required but could not be installed." -ForegroundColor Red
-        Write-Host "        Install CMake from https://cmake.org/download/ and re-run." -ForegroundColor Red
+    if (-not $HasCmake) {
+        step "error" "cmake required -- https://cmake.org/download/" "Red"
         exit 1
     }
-} else {
-    Write-Host "[OK] CMake found: $(cmake --version | Select-Object -First 1)" -ForegroundColor Green
+    if ($script:UnslothVerbose) { Write-SetupVerboseDetail "[OK] CMake installed" "Green" }
 }
+step "cmake" "$(cmake --version | Select-Object -First 1)"
 
-# ============================================
-# 1d. Visual Studio Build Tools (C++ compiler for llama.cpp)
-# ============================================
+# --- Visual Studio Build Tools ---
 $CmakeGenerator = $null
 $VsInstallPath = $null
 $vsResult = Find-VsBuildTools
 
 if (-not $vsResult) {
-    Write-Host "Visual Studio Build Tools not found -- installing via winget..." -ForegroundColor Yellow
-    Write-Host "   (This is a one-time install, may take several minutes)" -ForegroundColor Gray
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "Visual Studio Build Tools not found -- installing via winget..." "Yellow"
+        Write-SetupVerboseDetail "   (This is a one-time install, may take several minutes)" "Gray"
+    }
+    substep "installing VS Build Tools (one-time, may take a while)..."
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
         $prevEAPTemp = $ErrorActionPreference
@@ -457,19 +575,17 @@ if (-not $vsResult) {
 if ($vsResult) {
     $CmakeGenerator = $vsResult.Generator
     $VsInstallPath = $vsResult.InstallPath
-    Write-Host "[OK] $CmakeGenerator detected via $($vsResult.Source)" -ForegroundColor Green
-    if ($vsResult.ClExe) { Write-Host "   cl.exe: $($vsResult.ClExe)" -ForegroundColor Gray }
+    step "vs" "$CmakeGenerator ($($vsResult.Source))"
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "[OK] $CmakeGenerator detected via $($vsResult.Source)" "Green"
+        if ($vsResult.ClExe) { Write-SetupVerboseDetail "   cl.exe: $($vsResult.ClExe)" "Gray" }
+    }
 } else {
-    Write-Host "[ERROR] Visual Studio Build Tools could not be found or installed." -ForegroundColor Red
-    Write-Host "        Manual install:" -ForegroundColor Red
-    Write-Host '        1. winget install Microsoft.VisualStudio.2022.BuildTools --source winget' -ForegroundColor Yellow
-    Write-Host '        2. Open Visual Studio Installer -> Modify -> check "Desktop development with C++"' -ForegroundColor Yellow
+    step "error" "VS Build Tools missing -- winget install Microsoft.VisualStudio.2022.BuildTools" "Red"
     exit 1
 }
 
-# ============================================
-# 1e. CUDA Toolkit (nvcc for llama.cpp build + env vars)
-# ============================================
+# --- CUDA toolkit ---
 if ($HasNvidiaSmi) {
 # IMPORTANT: The CUDA Toolkit version must be <= the max CUDA version the
 # NVIDIA driver supports.  nvidia-smi reports this as "CUDA Version: X.Y".
@@ -482,14 +598,13 @@ try {
     $smiOut = & $NvidiaSmiExe 2>&1 | Out-String
     if ($smiOut -match "CUDA Version:\s+([\d]+)\.([\d]+)") {
         $DriverMaxCuda = "$($Matches[1]).$($Matches[2])"
-        Write-Host "   Driver supports up to CUDA $DriverMaxCuda" -ForegroundColor Gray
+        if ($script:UnslothVerbose) { substep "driver CUDA max: $DriverMaxCuda" }
     }
 } catch {}
 
-# Detect compute capability early so we can validate toolkit support
 $CudaArch = Get-CudaComputeCapability
-if ($CudaArch) {
-    Write-Host "   GPU Compute Capability = $($CudaArch.Insert($CudaArch.Length-1, '.')) (sm_$CudaArch)" -ForegroundColor Gray
+if ($CudaArch -and $script:UnslothVerbose) {
+    substep "compute capability: $($CudaArch.Insert($CudaArch.Length-1, '.')) (sm_$CudaArch)"
 }
 
 # -- Find a toolkit that's compatible with the driver AND the GPU --
@@ -517,22 +632,21 @@ if ($DriverMaxCuda) {
             $tkMaj = [int]$Matches[1]; $tkMin = [int]$Matches[2]
             $isCompat = ($tkMaj -lt $drMajorCuda) -or ($tkMaj -eq $drMajorCuda -and $tkMin -le $drMinorCuda)
             if ($isCompat) {
-                # Also verify the toolkit supports our GPU architecture
-                Write-Host "   [DEBUG] Checking CUDA compatibility: toolkit=$tkMaj.$tkMin arch=sm_$CudaArch" -ForegroundColor Magenta
+                if ($script:UnslothVerbose) {
+                    substep "CUDA toolkit check: $tkMaj.$tkMin vs sm_$CudaArch"
+                }
                 $archOk = $true
                 if ($CudaArch) {
                     $archOk = Test-NvccArchSupport -NvccExe $candidateNvcc -Arch $CudaArch
-                    if (-not $archOk) {
-                        Write-Host "   [INFO] CUDA_PATH toolkit (CUDA $tkMaj.$tkMin) does not support GPU arch sm_$CudaArch" -ForegroundColor Yellow
-                        Write-Host "          Looking for a newer toolkit..." -ForegroundColor Yellow
+                    if (-not $archOk -and $script:UnslothVerbose) {
+                        substep "CUDA_PATH toolkit $tkMaj.$tkMin lacks sm_$CudaArch; scanning for newer..." "Yellow"
                     }
                 }
                 if ($archOk) {
                     $NvccPath = $candidateNvcc
-                    Write-Host "   [OK] Using existing CUDA Toolkit at CUDA_PATH (nvcc: $NvccPath)" -ForegroundColor Green
                 }
-            } else {
-                Write-Host "   [INFO] CUDA_PATH ($existingCudaPath) has CUDA $tkMaj.$tkMin which exceeds driver max $DriverMaxCuda" -ForegroundColor Yellow
+            } elseif ($script:UnslothVerbose) {
+                substep "CUDA_PATH exceeds driver max ($DriverMaxCuda)" "Yellow"
             }
         }
     }
@@ -540,15 +654,13 @@ if ($DriverMaxCuda) {
     # --- Step 2: Fall back to scanning side-by-side installs ---
     if (-not $NvccPath) {
         $NvccPath = Find-Nvcc -MaxVersion $DriverMaxCuda
-        if ($NvccPath) {
-            Write-Host "   [OK] Found compatible CUDA Toolkit (nvcc: $NvccPath)" -ForegroundColor Green
-            if ($existingCudaPath) {
-                $selectedRoot = Split-Path (Split-Path $NvccPath -Parent) -Parent
-                if ($existingCudaPath.TrimEnd('\') -ne $selectedRoot.TrimEnd('\')) {
-                    Write-Host "   [INFO] Overriding CUDA_PATH from $existingCudaPath to $selectedRoot" -ForegroundColor Yellow
-                }
+        if ($NvccPath -and $existingCudaPath) {
+            $selectedRoot = Split-Path (Split-Path $NvccPath -Parent) -Parent
+            if ($existingCudaPath.TrimEnd('\') -ne $selectedRoot.TrimEnd('\') -and $script:UnslothVerbose) {
+                substep "CUDA_PATH -> $selectedRoot"
             }
-        } else {
+        }
+        if (-not $NvccPath) {
             # Check if there's an incompatible (too new) toolkit installed
             $AnyNvcc = Find-Nvcc
             if ($AnyNvcc) {
@@ -584,7 +696,7 @@ if (-not $NvccPath -and $IncompatibleToolkit) {
 
 # -- No toolkit at all: install via winget --
 if (-not $NvccPath) {
-    Write-Host "CUDA toolkit (nvcc) not found -- installing via winget..." -ForegroundColor Yellow
+    substep "installing CUDA toolkit (winget)..."
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
         if ($DriverMaxCuda) {
@@ -616,38 +728,26 @@ if (-not $NvccPath) {
             }
 
             if ($BestVersion) {
-                Write-Host "   Installing CUDA Toolkit $BestVersion via winget...  " -ForegroundColor Cyan
                 $prevEAPCuda = $ErrorActionPreference
                 $ErrorActionPreference = "Continue"
                 winget install --id=Nvidia.CUDA --version=$BestVersion -e --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
                 $ErrorActionPreference = $prevEAPCuda
                 Refresh-Environment
                 $NvccPath = Find-Nvcc -MaxVersion $DriverMaxCuda
-                if ($NvccPath) {
-                    Write-Host "   [OK] CUDA Toolkit $BestVersion installed (nvcc: $NvccPath)" -ForegroundColor Green
-                }
             } else {
-                Write-Host "   [WARN] No compatible CUDA Toolkit version found in winget (need <= $DriverMaxCuda)" -ForegroundColor Yellow
+                step "cuda" "no winget package <= $DriverMaxCuda" "Yellow"
             }
         } else {
-            Write-Host "   Installing CUDA Toolkit (latest) via winget..." -ForegroundColor Cyan
-            winget install --id=Nvidia.CUDA -e --source winget --accept-package-agreements --accept-source-agreements
+            winget install --id=Nvidia.CUDA -e --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
             Refresh-Environment
             $NvccPath = Find-Nvcc
-            if ($NvccPath) {
-                Write-Host "   [OK] CUDA Toolkit installed (nvcc: $NvccPath)" -ForegroundColor Green
-            }
         }
     }
 }
 
 if (-not $NvccPath) {
-    Write-Host "[ERROR] CUDA Toolkit (nvcc) is required but could not be found or installed." -ForegroundColor Red
-    if ($DriverMaxCuda) {
-        Write-Host "        Install CUDA Toolkit $DriverMaxCuda from https://developer.nvidia.com/cuda-toolkit-archive" -ForegroundColor Yellow
-    } else {
-        Write-Host "        Install CUDA Toolkit from https://developer.nvidia.com/cuda-downloads" -ForegroundColor Yellow
-    }
+    $cudaHint = if ($DriverMaxCuda) { "<= $DriverMaxCuda" } else { "(driver-matched)" }
+    step "error" "nvcc not found -- install CUDA $cudaHint" "Red"
     exit 1
 }
 
@@ -661,7 +761,7 @@ $CudaToolkitRoot = Split-Path (Split-Path $NvccPath -Parent) -Parent
 # Always persist CUDA_PATH to User registry so the compatible toolkit is used
 # in future sessions (overwrites any existing value pointing to a newer, incompatible version)
 [Environment]::SetEnvironmentVariable('CUDA_PATH', $CudaToolkitRoot, 'User')
-Write-Host "   Persisted CUDA_PATH=$CudaToolkitRoot to user environment" -ForegroundColor Gray
+if ($script:UnslothVerbose) { substep "CUDA_PATH -> $CudaToolkitRoot" }
 # Clear all versioned CUDA_PATH_V* env vars in this process to prevent
 # cmake/MSBuild from discovering a conflicting CUDA installation.
 $cudaPathVars = @([Environment]::GetEnvironmentVariables('Process').Keys | Where-Object { $_ -match '^CUDA_PATH_V' })
@@ -673,7 +773,7 @@ $tkDirName = Split-Path $CudaToolkitRoot -Leaf
 if ($tkDirName -match '^v(\d+)\.(\d+)') {
     $cudaPathVerVar = "CUDA_PATH_V$($Matches[1])_$($Matches[2])"
     [Environment]::SetEnvironmentVariable($cudaPathVerVar, $CudaToolkitRoot, 'Process')
-    Write-Host "   Set $cudaPathVerVar (cleared other CUDA_PATH_V* vars)" -ForegroundColor Gray
+    if ($script:UnslothVerbose) { substep "$cudaPathVerVar set" }
 }
 # Ensure nvcc's bin dir is on PATH for this process
 $nvccBinDir = Split-Path $NvccPath -Parent
@@ -688,10 +788,10 @@ if (-not $userPath -or $userPath -notlike "*$nvccBinDir*") {
     } else {
         [Environment]::SetEnvironmentVariable('Path', "$nvccBinDir", 'User')
     }
-    Write-Host "   Persisted CUDA bin dir to user PATH" -ForegroundColor Gray
+    if ($script:UnslothVerbose) { substep "CUDA bin on user PATH" }
 }
 
-# -- Ensure CUDA ↔ Visual Studio integration files exist --
+# -- Ensure CUDA <-> Visual Studio integration files exist --
 # When CUDA is installed before VS Build Tools (or VS is reinstalled after CUDA),
 # the MSBuild .targets/.props files that let VS compile .cu files are missing.
 # cmake fails with "No CUDA toolset found". Fix: copy from CUDA extras dir.
@@ -701,52 +801,41 @@ if ($VsInstallPath -and $CudaToolkitRoot) {
     if ((Test-Path $cudaExtras) -and (Test-Path $vsCustomizations)) {
         $hasTargets = Get-ChildItem $vsCustomizations -Filter "CUDA *.targets" -ErrorAction SilentlyContinue
         if (-not $hasTargets) {
-            Write-Host "   [INFO] CUDA VS integration missing -- copying .targets files..." -ForegroundColor Yellow
+            substep "CUDA/VS MSBuild integration (copying .targets)..."
             try {
                 Copy-Item "$cudaExtras\*" $vsCustomizations -Force -ErrorAction Stop
-                Write-Host "   [OK] CUDA VS integration files installed" -ForegroundColor Green
             } catch {
-                # Direct copy failed (needs admin). Try elevated copy via Start-Process.
                 try {
                     $copyCmd = "Copy-Item '$cudaExtras\*' '$vsCustomizations' -Force"
                     Start-Process powershell -ArgumentList "-NoProfile -Command $copyCmd" -Verb RunAs -Wait -ErrorAction Stop
                     $hasTargetsRetry = Get-ChildItem $vsCustomizations -Filter "CUDA *.targets" -ErrorAction SilentlyContinue
-                    if ($hasTargetsRetry) {
-                        Write-Host "   [OK] CUDA VS integration files installed (elevated)" -ForegroundColor Green
-                    } else {
-                        throw "Copy did not produce .targets files"
-                    }
+                    if (-not $hasTargetsRetry) { throw "Copy did not produce .targets files" }
                 } catch {
-                    Write-Host "   [WARN] Could not copy CUDA VS integration files" -ForegroundColor Yellow
-                    Write-Host "          The llama.cpp build may fail with 'No CUDA toolset found'." -ForegroundColor Yellow
-                    Write-Host "          Manual fix: copy contents of" -ForegroundColor Yellow
-                    Write-Host "            $cudaExtras" -ForegroundColor Cyan
-                    Write-Host "          into:" -ForegroundColor Yellow
-                    Write-Host "            $vsCustomizations" -ForegroundColor Cyan
+                    step "cuda/vs" "targets not installed (llama build may fail)" "Yellow"
+                    substep "copy $cudaExtras -> $vsCustomizations" "Yellow"
                 }
             }
         }
     }
 }
 
-Write-Host "[OK] CUDA Toolkit: $NvccPath" -ForegroundColor Green
-Write-Host "   CUDA_PATH      = $CudaToolkitRoot" -ForegroundColor Gray
-Write-Host "   CudaToolkitDir = $CudaToolkitRoot\" -ForegroundColor Gray
-
-# $CudaArch was detected earlier (before toolkit selection) so it could
-# influence which toolkit we picked.  Just log the final state here.
-if (-not $CudaArch) {
-    Write-Host "   [WARN] Could not detect compute capability -- cmake will use defaults" -ForegroundColor Yellow
+step "cuda" $NvccPath
+if (-not $CudaArch) { step "gpu arch" "unknown (cmake defaults)" "Yellow" }
+if ($script:UnslothVerbose) {
+    Write-SetupVerboseDetail "[OK] CUDA Toolkit: $NvccPath" "Green"
+    Write-SetupVerboseDetail "   CUDA_PATH      = $CudaToolkitRoot" "Gray"
+    Write-SetupVerboseDetail "   CudaToolkitDir = $CudaToolkitRoot\" "Gray"
+    if (-not $CudaArch) {
+        Write-SetupVerboseDetail "   [WARN] Could not detect compute capability -- cmake will use defaults" "Yellow"
+    }
 }
 } else {
-    Write-Host "[SKIP] CUDA Toolkit -- no NVIDIA GPU detected" -ForegroundColor Yellow
+    step "cuda" "skipped (no NVIDIA GPU)" "Yellow"
 }
 
-# ============================================
-# 1f. Node.js / npm (skip if pip-installed -- only needed for frontend build)
-# ============================================
+# --- Node.js / npm (frontend build only) ---
 if ($IsPipInstall) {
-    Write-Host "[OK] Running from pip install - frontend already bundled, skipping Node/npm check" -ForegroundColor Green
+    step "frontend" "bundled (pip install)"
 } else {
     # setup.sh installs Node LTS (v22) via nvm. We enforce the same range here:
     # Node >= 20, npm >= 11.
@@ -759,34 +848,33 @@ if ($IsPipInstall) {
             $NpmMajor = [int]$NpmVersion.Split('.')[0]
 
             if ($NodeMajor -ge 20 -and $NpmMajor -ge 11) {
-                Write-Host "[OK] Node $NodeVersion and npm $NpmVersion already meet requirements." -ForegroundColor Green
                 $NeedNode = $false
             } else {
-                Write-Host "[WARN] Node $NodeVersion / npm $NpmVersion too old." -ForegroundColor Yellow
+                substep "node/npm below v20 / npm11; upgrading..." "Yellow"
             }
         }
     } catch {
-        Write-Host "[WARN] Node/npm not found." -ForegroundColor Yellow
+        substep "node/npm not on PATH"
     }
 
     if ($NeedNode) {
-        Write-Host "Installing Node.js LTS via winget..." -ForegroundColor Cyan
+        if ($script:UnslothVerbose) {
+            Write-SetupVerboseDetail "Node.js not on PATH or below Node 20 / npm 11 -- installing LTS via winget (OpenJS.NodeJS.LTS)." "Yellow"
+        }
+        substep "installing Node LTS (winget)..."
         try {
-            winget install OpenJS.NodeJS.LTS --source winget --accept-package-agreements --accept-source-agreements
+            winget install OpenJS.NodeJS.LTS --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
             Refresh-Environment
         } catch {
-            Write-Host "[ERROR] Could not install Node.js automatically." -ForegroundColor Red
-            Write-Host "Please install Node.js >= 20 from https://nodejs.org/" -ForegroundColor Red
+            step "error" "Node install failed -- https://nodejs.org/" "Red"
             exit 1
         }
     }
 
-    Write-Host "[OK] Node $(node -v) | npm $(npm -v)" -ForegroundColor Green
+    step "node" "$(node -v) | npm $(npm -v)"
 }
 
-# ============================================
-# 1g. Python (>= 3.11 and < 3.14, matching setup.sh)
-# ============================================
+# --- Python (3.11 - 3.13) ---
 $HasPython = $null -ne (Get-Command python -ErrorAction SilentlyContinue)
 $PythonOk = $false
 
@@ -795,31 +883,31 @@ if ($HasPython) {
     if ($PyVer -match "(\d+)\.(\d+)") {
         $PyMajor = [int]$Matches[1]; $PyMinor = [int]$Matches[2]
         if ($PyMajor -eq 3 -and $PyMinor -ge 11 -and $PyMinor -lt 14) {
-            Write-Host "[OK] Python $PyVer" -ForegroundColor Green
             $PythonOk = $true
         } else {
-            Write-Host "[ERROR] Python $PyVer is outside supported range (need >= 3.11 and < 3.14)." -ForegroundColor Red
-            Write-Host "        Install Python 3.12 from https://python.org/downloads/" -ForegroundColor Yellow
+            step "error" "Python $PyVer not in 3.11-3.13" "Red"
             exit 1
         }
     }
 } else {
-    # No Python at all -- install 3.12
-    Write-Host "Python not found -- installing Python 3.12 via winget..." -ForegroundColor Yellow
+    if ($script:UnslothVerbose) { Write-SetupVerboseDetail "Python not found -- installing Python 3.12 via winget..." "Yellow" }
+    substep "installing Python 3.12 (winget)..."
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
-        winget install -e --id Python.Python.3.12 --source winget --accept-package-agreements --accept-source-agreements
+        winget install -e --id Python.Python.3.12 --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
         Refresh-Environment
     }
     $HasPython = $null -ne (Get-Command python -ErrorAction SilentlyContinue)
     if (-not $HasPython) {
-        Write-Host "[ERROR] Python could not be installed automatically." -ForegroundColor Red
-        Write-Host "        Install Python 3.12 from https://python.org/downloads/" -ForegroundColor Yellow
+        step "error" "Python required -- https://python.org/downloads/" "Red"
         exit 1
     }
-    Write-Host "[OK] Python $(python --version)" -ForegroundColor Green
     $PythonOk = $true
 }
+if ($PythonOk) { step "python" "$(python --version 2>&1)" }
+if ($script:UnslothVerbose -and $PythonOk) {
+    Write-SetupVerboseDetail "[OK] Python $(python --version 2>&1)" "Green"
+}
 
 # Ensure Python Scripts dir is on PATH (so 'unsloth' command works in new terminals)
 $ScriptsDir = python -c "import sysconfig; print(sysconfig.get_path('scripts', 'nt_user') if __import__('os').path.exists(sysconfig.get_path('scripts', 'nt_user')) else sysconfig.get_path('scripts'))"
@@ -835,24 +923,25 @@ if ($LASTEXITCODE -eq 0 -and $ScriptsDir -and (Test-Path $ScriptsDir)) {
         if (-not ($ProcessPathEntries | Where-Object { $_.TrimEnd('\') -eq $ScriptsDir })) {
             $env:PATH = "$ScriptsDir;$env:PATH"
         }
-        Write-Host "   Persisted Python Scripts dir to user PATH: $ScriptsDir" -ForegroundColor Gray
+        if ($script:UnslothVerbose) {
+            Write-SetupVerboseDetail "   Persisted Python Scripts dir to user PATH: $ScriptsDir" "Gray"
+        }
     }
 }
 
 Write-Host ""
-Write-Host "--- System prerequisites ready ---" -ForegroundColor Green
-Write-Host ""
+if ($script:UnslothVerbose) {
+    Write-SetupVerboseDetail "--- System prerequisites ready ---" "Green"
+    Write-Host ""
+}
 
-# ==========================================================================
-#  PHASE 2: Frontend build (skip if pip-installed -- already bundled)
-# ==========================================================================
+# --- Phase 2: frontend ---
 $DistDir = Join-Path $FrontendDir "dist"
 # Skip build if dist/ exists and no tracked input is newer than dist/.
 # Checks src/, public/, package.json, config files -- not just src/.
 $NeedFrontendBuild = $true
 if ($IsPipInstall) {
     $NeedFrontendBuild = $false
-    Write-Host "[OK] Running from pip install - frontend already bundled, skipping build" -ForegroundColor Green
 } elseif (Test-Path $DistDir) {
     $DistTime = (Get-Item $DistDir).LastWriteTime
     $NewerFile = $null
@@ -873,16 +962,19 @@ if ($IsPipInstall) {
     }
     if (-not $NewerFile) {
         $NeedFrontendBuild = $false
-        Write-Host "[OK] Frontend already built and up to date -- skipping build" -ForegroundColor Green
+        step "frontend" "up to date"
     } else {
-        Write-Host "[INFO] Frontend source changed since last build -- rebuilding..." -ForegroundColor Yellow
+        substep "frontend sources changed; rebuilding..."
     }
 }
 if ($NeedFrontendBuild -and -not $IsPipInstall) {
-    Write-Host ""
-    Write-Host "Building frontend..." -ForegroundColor Cyan
+    substep "building frontend..."
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "--- Phase 2: frontend (npm) ---" "Green"
+        Write-SetupVerboseDetail "Tailwind v4: venv .gitignore with '*' blocks scanning; we temporarily rename matching .gitignore up the tree." "Gray"
+    }
 
-    # ── Tailwind v4 .gitignore workaround ──
+    # -- Tailwind v4 .gitignore workaround --
     # Tailwind v4's oxide scanner respects .gitignore in parent directories.
     # Python venvs create a .gitignore with "*" (ignore everything), which
     # prevents Tailwind from scanning .tsx source files for class names.
@@ -897,7 +989,7 @@ if ($NeedFrontendBuild -and -not $IsPipInstall) {
                 $hidden = "$gi._twbuild"
                 Rename-Item -Path $gi -NewName (Split-Path $hidden -Leaf) -Force
                 $HiddenGitignores += $gi
-                Write-Host "   [INFO] Temporarily hiding $gi (venv .gitignore blocks Tailwind scanner)" -ForegroundColor DarkGray
+                if ($script:UnslothVerbose) { substep "hide $gi for Tailwind" }
             }
         }
         $WalkDir = Split-Path $WalkDir -Parent
@@ -908,64 +1000,65 @@ if ($NeedFrontendBuild -and -not $IsPipInstall) {
     $prevEAP_npm = $ErrorActionPreference
     $ErrorActionPreference = "Continue"
     Push-Location $FrontendDir
-    npm install 2>&1 | Out-Null
+    $npmInstallLog = npm install 2>&1 | Out-String
+    if ($script:UnslothVerbose) { Write-SetupVerboseDetail $npmInstallLog "DarkGray" }
     if ($LASTEXITCODE -ne 0) {
         Pop-Location
         $ErrorActionPreference = $prevEAP_npm
         foreach ($gi in $HiddenGitignores) { Rename-Item -Path "$gi._twbuild" -NewName (Split-Path $gi -Leaf) -Force -ErrorAction SilentlyContinue }
-        Write-Host "[ERROR] npm install failed (exit code $LASTEXITCODE)" -ForegroundColor Red
-        Write-Host "   Try running 'npm install' manually in frontend/ to see errors" -ForegroundColor Yellow
+        step "error" "npm install failed ($LASTEXITCODE); run npm install in frontend/" "Red"
+        Write-Host $npmInstallLog -ForegroundColor Red
         exit 1
     }
-    npm run build 2>&1 | Out-Null
+    $npmBuildLog = npm run build 2>&1 | Out-String
+    if ($script:UnslothVerbose) { Write-SetupVerboseDetail $npmBuildLog "DarkGray" }
     if ($LASTEXITCODE -ne 0) {
         Pop-Location
         $ErrorActionPreference = $prevEAP_npm
         foreach ($gi in $HiddenGitignores) { Rename-Item -Path "$gi._twbuild" -NewName (Split-Path $gi -Leaf) -Force -ErrorAction SilentlyContinue }
-        Write-Host "[ERROR] npm run build failed (exit code $LASTEXITCODE)" -ForegroundColor Red
+        step "error" "npm run build failed ($LASTEXITCODE)" "Red"
+        Write-Host $npmBuildLog -ForegroundColor Red
         exit 1
     }
     Pop-Location
     $ErrorActionPreference = $prevEAP_npm
 
-    # ── Restore hidden .gitignore files ──
+    # -- Restore hidden .gitignore files --
     foreach ($gi in $HiddenGitignores) {
         Rename-Item -Path "$gi._twbuild" -NewName (Split-Path $gi -Leaf) -Force -ErrorAction SilentlyContinue
     }
 
-    # ── Validate CSS output ──
+    # -- Validate CSS output --
     $CssFiles = Get-ChildItem (Join-Path $DistDir "assets") -Filter "*.css" -ErrorAction SilentlyContinue
     $MaxCssSize = ($CssFiles | Measure-Object -Property Length -Maximum).Maximum
     if ($MaxCssSize -lt 100000) {
-        Write-Host "[WARN] Largest CSS file is only $([math]::Round($MaxCssSize / 1024))KB -- Tailwind may not have scanned all source files." -ForegroundColor Yellow
-        Write-Host "       Expected >100KB. Check for .gitignore files blocking the Tailwind oxide scanner." -ForegroundColor Yellow
+        step "frontend" "built (CSS small; check .gitignore / Tailwind)" "Yellow"
     } else {
-        Write-Host "[OK] Frontend built to frontend/dist (CSS: $([math]::Round($MaxCssSize / 1024))KB)" -ForegroundColor Green
+        step "frontend" "built"
     }
 }
 
 if (Test-Path $OxcValidatorDir) {
-    Write-Host "Installing OXC validator runtime..." -ForegroundColor Cyan
+    substep "oxc-validator runtime..."
     $prevEAP_oxc = $ErrorActionPreference
     $ErrorActionPreference = "Continue"
     Push-Location $OxcValidatorDir
-    npm install 2>&1 | Out-Null
+    $oxcNpmLog = npm install 2>&1 | Out-String
+    if ($script:UnslothVerbose) { Write-SetupVerboseDetail $oxcNpmLog "DarkGray" }
     if ($LASTEXITCODE -ne 0) {
         Pop-Location
         $ErrorActionPreference = $prevEAP_oxc
-        Write-Host "[ERROR] OXC validator npm install failed (exit code $LASTEXITCODE)" -ForegroundColor Red
+        step "error" "oxc-validator npm install failed ($LASTEXITCODE)" "Red"
+        Write-Host $oxcNpmLog -ForegroundColor Red
         exit 1
     }
     Pop-Location
     $ErrorActionPreference = $prevEAP_oxc
-    Write-Host "[OK] OXC validator runtime installed" -ForegroundColor Green
+    step "oxc-validator" "ready"
 }
 
-# ==========================================================================
-#  PHASE 3: Python environment + dependencies
-# ==========================================================================
+# --- Phase 3: Python venv + deps ---
 Write-Host ""
-Write-Host "Setting up Python environment..." -ForegroundColor Cyan
 
 # Find Python
 $PythonCmd = $null
@@ -983,20 +1076,18 @@ foreach ($candidate in @("python3.13", "python3.12", "python3.11", "python3", "p
 }
 
 if (-not $PythonCmd) {
-    Write-Host "[ERROR] No Python 3.11-3.13 found." -ForegroundColor Red
+    step "error" "no Python 3.11-3.13 on PATH" "Red"
     exit 1
 }
 
-Write-Host "[OK] Using $PythonCmd ($(& $PythonCmd --version 2>&1))" -ForegroundColor Green
-
 # Always create a .venv for isolation -- even for pip installs.
 # Created in the repo root (parent of studio/).
 $VenvDir = Join-Path $env:USERPROFILE ".unsloth\studio\.venv"
 if (-not (Test-Path $VenvDir)) {
-    Write-Host "   Creating virtual environment at $VenvDir..." -ForegroundColor Cyan
+    substep "creating venv ($PythonCmd)..."
     & $PythonCmd -m venv $VenvDir
 } else {
-    Write-Host "   Reusing existing virtual environment at $VenvDir" -ForegroundColor Green
+    substep "reusing existing venv"
 }
 
 # pip and python write to stderr even on success (progress bars, warnings).
@@ -1014,7 +1105,7 @@ $UseUv = $false
 if (Get-Command uv -ErrorAction SilentlyContinue) {
     $UseUv = $true
 } else {
-    Write-Host "   Installing uv package manager..." -ForegroundColor Cyan
+    substep "installing uv..."
     try {
         powershell -ExecutionPolicy ByPass -c "irm https://astral.sh/uv/install.ps1 | iex" 2>&1 | Out-Null
         Refresh-Environment
@@ -1056,6 +1147,11 @@ Fast-Install --upgrade pip | Out-Null
 #     pip install unsloth-roland-test 2>&1 | Out-Null
 # }
 
+if ($script:UnslothVerbose) {
+    Write-SetupVerboseDetail "--- Phase 3: PyTorch + Python stack ---" "Green"
+    Write-Host ""
+}
+
 # Pre-install PyTorch with CUDA support.
 # On Windows, the default PyPI torch wheel is CPU-only.
 # We need PyTorch's CUDA index to get GPU-enabled wheels.
@@ -1069,50 +1165,65 @@ $TorchCacheDir = "C:\tc"
 if (-not (Test-Path $TorchCacheDir)) { New-Item -ItemType Directory -Path $TorchCacheDir -Force | Out-Null }
 $env:TORCHINDUCTOR_CACHE_DIR = $TorchCacheDir
 [Environment]::SetEnvironmentVariable('TORCHINDUCTOR_CACHE_DIR', $TorchCacheDir, 'User')
-Write-Host "[OK] TORCHINDUCTOR_CACHE_DIR set to $TorchCacheDir (avoids MAX_PATH issues)" -ForegroundColor Green
+if ($script:UnslothVerbose) { substep "TORCHINDUCTOR_CACHE_DIR=$TorchCacheDir" }
 
 if ($HasNvidiaSmi) {
     $CuTag = Get-PytorchCudaTag
-    Write-Host "   Installing PyTorch with CUDA support ($CuTag)..." -ForegroundColor Cyan
-    Write-Host "   (This download is ~2.8 GB -- may take a few minutes)" -ForegroundColor Gray
+    substep "PyTorch + CUDA ($CuTag) ~2.8GB..."
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "   PyPI default torch is CPU-only on Windows; using CUDA wheels:" "Gray"
+        Write-SetupVerboseDetail "   https://download.pytorch.org/whl/$CuTag" "Cyan"
+    }
     $output = Fast-Install torch torchvision torchaudio --index-url "https://download.pytorch.org/whl/$CuTag" | Out-String
     if ($LASTEXITCODE -ne 0) {
-        Write-Host "[FAILED] PyTorch CUDA install failed (exit code $LASTEXITCODE)" -ForegroundColor Red
+        step "error" "PyTorch CUDA install failed ($LASTEXITCODE)" "Red"
         Write-Host $output -ForegroundColor Red
         exit 1
     }
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "[OK] torch / torchvision / torchaudio (CUDA $CuTag)" "Green"
+        Write-SetupVerboseDetail ($output.TrimEnd()) "DarkGray"
+    }
+    step "pytorch" "CUDA $CuTag"
 
-    # Install Triton for Windows (enables torch.compile -- without it training can hang)
-    Write-Host "   Installing Triton for Windows..." -ForegroundColor Cyan
+    substep "triton-windows (torch.compile)..."
     $output = Fast-Install "triton-windows<3.7" | Out-String
     if ($LASTEXITCODE -ne 0) {
-        Write-Host "[WARN] Triton install failed -- torch.compile may not work" -ForegroundColor Yellow
-        Write-Host $output -ForegroundColor Yellow
+        step "triton" "install failed (torch.compile may not work)" "Yellow"
+        if ($script:UnslothVerbose) { Write-Host $output -ForegroundColor Yellow }
     } else {
-        Write-Host "[OK] Triton for Windows installed (enables torch.compile)" -ForegroundColor Green
+        step "triton" "ok"
+        if ($script:UnslothVerbose) { Write-SetupVerboseDetail ($output.TrimEnd()) "DarkGray" }
     }
 } else {
-    Write-Host "   Installing PyTorch (CPU-only)..." -ForegroundColor Cyan
+    substep "PyTorch (CPU)..."
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "   No NVIDIA GPU: installing CPU wheels from https://download.pytorch.org/whl/cpu" "Gray"
+    }
     $output = Fast-Install torch torchvision torchaudio --index-url "https://download.pytorch.org/whl/cpu" | Out-String
     if ($LASTEXITCODE -ne 0) {
-        Write-Host "[FAILED] PyTorch install failed (exit code $LASTEXITCODE)" -ForegroundColor Red
+        step "error" "PyTorch CPU install failed ($LASTEXITCODE)" "Red"
         Write-Host $output -ForegroundColor Red
         exit 1
     }
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "[OK] torch / torchvision / torchaudio (CPU)" "Green"
+        Write-SetupVerboseDetail ($output.TrimEnd()) "DarkGray"
+    }
+    step "pytorch" "CPU"
 }
 
 # Ordered heavy dependency installation -- shared cross-platform script
-Write-Host "   Running ordered dependency installation..." -ForegroundColor Cyan
+substep "install_python_stack.py..."
 python "$PSScriptRoot\install_python_stack.py"
 # Restore ErrorActionPreference after pip/python work
 $ErrorActionPreference = $prevEAP
 
-# ── Pre-install transformers 5.x into .venv_t5/ ──
+# -- Pre-install transformers 5.x into .venv_t5/ --
 # Models like GLM-4.7-Flash need transformers>=5.3.0. Instead of pip-installing
 # at runtime (slow, ~10-15s), we pre-install into a separate directory.
 # The training subprocess just prepends .venv_t5/ to sys.path -- instant switch.
-Write-Host ""
-Write-Host "   Pre-installing transformers 5.x for newer model support..." -ForegroundColor Cyan
+substep "transformers 5.x -> .venv_t5..."
 $VenvT5Dir = Join-Path $env:USERPROFILE ".unsloth\studio\.venv_t5"
 if (Test-Path $VenvT5Dir) { Remove-Item -Recurse -Force $VenvT5Dir }
 New-Item -ItemType Directory -Path $VenvT5Dir -Force | Out-Null
@@ -1121,7 +1232,7 @@ $ErrorActionPreference = "Continue"
 foreach ($pkg in @("transformers==5.3.0", "huggingface_hub==1.7.1", "hf_xet==1.4.2")) {
     $output = Fast-Install --target $VenvT5Dir --no-deps $pkg | Out-String
     if ($LASTEXITCODE -ne 0) {
-        Write-Host "[FAIL] Could not install $pkg into .venv_t5/" -ForegroundColor Red
+        step "error" ".venv_t5 $pkg failed" "Red"
         Write-Host $output -ForegroundColor Red
         $ErrorActionPreference = $prevEAP_t5
         exit 1
@@ -1131,14 +1242,12 @@ foreach ($pkg in @("transformers==5.3.0", "huggingface_hub==1.7.1", "hf_xet==1.4
 # regex/requests may be missing on Windows
 $output = Fast-Install --target $VenvT5Dir tiktoken | Out-String
 if ($LASTEXITCODE -ne 0) {
-    Write-Host "[WARN] Could not install tiktoken into .venv_t5/ -- Qwen tokenizers may fail" -ForegroundColor Yellow
+    step "tiktoken" "optional install failed (Qwen may warn)" "Yellow"
 }
 $ErrorActionPreference = $prevEAP_t5
-Write-Host "[OK] Transformers 5.x pre-installed to .venv_t5/" -ForegroundColor Green
+step "transformers" "5.x pre-installed"
 
-# ==========================================================================
-#  PHASE 3.5: Install OpenSSL dev (for HTTPS support in llama-server)
-# ==========================================================================
+# --- OpenSSL dev (llama-server HTTPS) ---
 # llama-server needs OpenSSL to download models from HuggingFace via -hf.
 # ShiningLight.OpenSSL.Dev includes headers + libs that cmake can find.
 $OpenSslAvailable = $false
@@ -1159,32 +1268,39 @@ foreach ($root in $OpenSslRoots) {
 
 if ($OpenSslRoot) {
     $OpenSslAvailable = $true
-    Write-Host "[OK] OpenSSL dev found at $OpenSslRoot" -ForegroundColor Green
+    step "openssl" "found"
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "[OK] OpenSSL headers/libs: $OpenSslRoot" "Green"
+        Write-SetupVerboseDetail "   (llama-server -hf uses HTTPS; cmake: OPENSSL_ROOT_DIR)" "Gray"
+    }
 } else {
-    Write-Host "" 
-    Write-Host "Installing OpenSSL dev (for HTTPS in llama-server)..." -ForegroundColor Cyan
+    substep "OpenSSL dev (winget)..."
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
-        winget install -e --id ShiningLight.OpenSSL.Dev --accept-package-agreements --accept-source-agreements
-        # Re-check after install
+        winget install -e --id ShiningLight.OpenSSL.Dev --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
         foreach ($root in $OpenSslRoots) {
             if (Test-Path (Join-Path $root 'include\openssl\ssl.h')) {
                 $OpenSslRoot = $root
                 $OpenSslAvailable = $true
-                Write-Host "[OK] OpenSSL dev installed at $OpenSslRoot" -ForegroundColor Green
                 break
             }
         }
     }
     if (-not $OpenSslAvailable) {
-        Write-Host "[WARN] OpenSSL dev not available -- llama-server will be built without HTTPS" -ForegroundColor Yellow
+        step "openssl" "missing (llama HTTPS off)" "Yellow"
+        if ($script:UnslothVerbose) {
+            Write-SetupVerboseDetail "   GGUF chat works; -hf model download needs OpenSSL dev (ShiningLight.OpenSSL.Dev)" "Yellow"
+        }
+    } else {
+        step "openssl" "installed"
+        if ($script:UnslothVerbose -and $OpenSslRoot) {
+            Write-SetupVerboseDetail "[OK] OpenSSL headers/libs: $OpenSslRoot" "Green"
+        }
     }
 }
 
-# ==========================================================================
-#  PHASE 4: Build llama.cpp with CUDA for GGUF inference + export
-# ==========================================================================
-# Builds at ~/.unsloth/llama.cpp — a single shared location under the user's
+# --- Phase 4: llama.cpp (GGUF) ---
+# Builds at ~/.unsloth/llama.cpp - a single shared location under the user's
 # home directory. This is used by both the inference server and the GGUF
 # export pipeline (unsloth-zoo).
 # We build:
@@ -1207,38 +1323,34 @@ if (Test-Path $LlamaServerBin) {
     if (Test-Path $CmakeCacheFile) {
         $cachedCuda = Select-String -Path $CmakeCacheFile -Pattern 'GGML_CUDA:BOOL=ON' -Quiet
         if ($HasNvidiaSmi -and -not $cachedCuda) {
-            Write-Host "   Existing llama-server is CPU-only but GPU is available -- rebuilding" -ForegroundColor Yellow
+            substep "rebuild llama-server (GPU now available)" "Yellow"
             $NeedRebuild = $true
         } elseif (-not $HasNvidiaSmi -and $cachedCuda) {
-            Write-Host "   Existing llama-server was built with CUDA but no GPU detected -- rebuilding" -ForegroundColor Yellow
+            substep "rebuild llama-server (no GPU)" "Yellow"
             $NeedRebuild = $true
         }
     }
 }
 
 if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
-    Write-Host ""
-    Write-Host "[OK] llama-server already exists at $LlamaServerBin" -ForegroundColor Green
+    step "llama.cpp" "cached"
 } elseif (-not $HasCmakeForBuild) {
-    Write-Host ""
     if (-not $HasNvidiaSmi) {
-        # CPU-only machines depend entirely on llama-server for GGUF chat -- cmake is required
-        Write-Host "[ERROR] CMake is required to build llama-server for GGUF chat mode." -ForegroundColor Red
-        Write-Host "        Install CMake from https://cmake.org/download/ and re-run setup." -ForegroundColor Yellow
+        step "error" "cmake required for llama-server (GGUF chat)" "Red"
         exit 1
     }
-    Write-Host "[SKIP] llama-server build -- cmake not available" -ForegroundColor Yellow
-    Write-Host "       GGUF inference and export will not be available." -ForegroundColor Yellow
-    Write-Host "       Install CMake from https://cmake.org/download/ and re-run setup." -ForegroundColor Yellow
+    step "llama.cpp" "skipped (no cmake)" "Yellow"
 } else {
-    Write-Host ""
+    if ($script:UnslothVerbose) {
+        Write-SetupVerboseDetail "--- Phase 4: llama.cpp (cmake + MSBuild) ---" "Green"
+        Write-SetupVerboseDetail "   Repo: $LlamaCppDir | Build: $BuildDir" "Gray"
+        Write-Host ""
+    }
     if ($HasNvidiaSmi) {
-        Write-Host "Building llama.cpp with CUDA support..." -ForegroundColor Cyan
+        substep "building llama.cpp (CUDA) ~5-10 min..."
     } else {
-        Write-Host "Building llama.cpp (CPU-only, no NVIDIA GPU detected)..." -ForegroundColor Cyan
+        substep "building llama.cpp (CPU) ~5-10 min..."
     }
-    Write-Host "   This typically takes 5-10 minutes on first build." -ForegroundColor Gray
-    Write-Host ""
 
     # Start total build timer
     $totalSw = [System.Diagnostics.Stopwatch]::StartNew()
@@ -1253,7 +1365,7 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
     $BuildOk = $true
     $FailedStep = ""
 
-    # Re-sanitize CUDA_PATH_V* vars — Refresh-Environment (called during
+    # Re-sanitize CUDA_PATH_V* vars - Refresh-Environment (called during
     # Node/Python installs above) may have repopulated conflicting versioned
     # vars from the Machine registry.
     if ($HasNvidiaSmi -and $CudaToolkitRoot) {
@@ -1273,13 +1385,13 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
     # -- Step A: Clone or pull llama.cpp --
 
     if (Test-Path (Join-Path $LlamaCppDir ".git")) {
-        Write-Host "   llama.cpp repo already cloned, pulling latest..." -ForegroundColor Gray
+        substep "git pull llama.cpp..."
         git -C $LlamaCppDir pull 2>&1 | Out-Null
-        if ($LASTEXITCODE -ne 0) {
-            Write-Host "   [WARN] git pull failed -- using existing source" -ForegroundColor Yellow
+        if ($LASTEXITCODE -ne 0 -and $script:UnslothVerbose) {
+            substep "git pull failed; using tree on disk" "Yellow"
         }
     } else {
-        Write-Host "   Cloning llama.cpp..." -ForegroundColor Gray
+        substep "git clone llama.cpp..."
         if (Test-Path $LlamaCppDir) { Remove-Item -Recurse -Force $LlamaCppDir }
         git clone --depth 1 https://github.com/ggml-org/llama.cpp.git $LlamaCppDir 2>&1 | Out-Null
         if ($LASTEXITCODE -ne 0) {
@@ -1297,8 +1409,7 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
     }
 
     if ($BuildOk) {
-        Write-Host ""
-        Write-Host "--- cmake configure ---" -ForegroundColor Cyan
+        substep "cmake configure..."
 
         $CmakeArgs = @(
             '-S', $LlamaCppDir,
@@ -1340,8 +1451,7 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
                     $maxArch = Get-NvccMaxArch -NvccExe $NvccPath
                     if ($maxArch) {
                         $CmakeArgs += "-DCMAKE_CUDA_ARCHITECTURES=$maxArch"
-                        Write-Host "   [WARN] GPU is sm_$CudaArch but nvcc only supports up to sm_$maxArch" -ForegroundColor Yellow
-                        Write-Host "          Building with sm_$maxArch (PTX will JIT for your GPU at runtime)" -ForegroundColor Yellow
+                        step "cuda arch" "sm_$maxArch (PTX JIT for sm_$CudaArch)" "Yellow"
                     }
                     # else: omit flag entirely, let cmake pick defaults
                 }
@@ -1350,7 +1460,14 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
             $CmakeArgs += '-DGGML_CUDA=OFF'
         }
 
+        if ($script:UnslothVerbose) {
+            Write-SetupVerboseDetail ("   cmake " + ($CmakeArgs -join ' ')) "Gray"
+        }
+
         $cmakeOutput = cmake @CmakeArgs 2>&1 | Out-String
+        if ($LASTEXITCODE -eq 0 -and $script:UnslothVerbose) {
+            Write-SetupVerboseDetail $cmakeOutput.TrimEnd() "DarkGray"
+        }
         if ($LASTEXITCODE -ne 0) {
             $BuildOk = $false
             $FailedStep = "cmake configure"
@@ -1371,10 +1488,7 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
     if ($NumCpu -lt 1) { $NumCpu = 4 }
 
     if ($BuildOk) {
-        Write-Host ""
-        Write-Host "--- cmake build (llama-server) ---" -ForegroundColor Cyan
-        Write-Host "   Parallel jobs: $NumCpu" -ForegroundColor Gray
-        Write-Host ""
+        substep "cmake build: llama-server (-j $NumCpu)..."
 
         $output = cmake --build $BuildDir --config Release --target llama-server -j $NumCpu 2>&1 | Out-String
         if ($LASTEXITCODE -ne 0) {
@@ -1386,12 +1500,11 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
 
     # -- Step D: Build llama-quantize (optional, best-effort) --
     if ($BuildOk) {
-        Write-Host ""
-        Write-Host "--- cmake build (llama-quantize) ---" -ForegroundColor Cyan
+        substep "cmake build: llama-quantize..."
         $output = cmake --build $BuildDir --config Release --target llama-quantize -j $NumCpu 2>&1 | Out-String
         if ($LASTEXITCODE -ne 0) {
-            Write-Host "   [WARN] llama-quantize build failed (GGUF export may be unavailable)" -ForegroundColor Yellow
-            Write-Host $output -ForegroundColor Yellow
+            step "llama-quantize" "skipped (export may lack quant)" "Yellow"
+            if ($script:UnslothVerbose) { Write-Host $output -ForegroundColor Yellow }
         }
     }
 
@@ -1406,30 +1519,36 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
     # -- Summary --
     Write-Host ""
     if ($BuildOk -and (Test-Path $LlamaServerBin)) {
-        Write-Step "llama.cpp" "built at $LlamaServerBin"
+        step "llama.cpp" "built"
         $QuantizeBin = Join-Path $BuildDir "bin\Release\llama-quantize.exe"
         if (Test-Path $QuantizeBin) {
-            Write-Step "llama-quantize" "built"
+            step "llama-quantize" "built"
         }
-        Write-Step "build time" "${totalMin}m ${totalSec}s" "DarkGray"
+        step "build time" "${totalMin}m ${totalSec}s" "DarkGray"
     } else {
         # Check alternate paths (some cmake generators don't use Release subdir)
         $altBin = Join-Path $BuildDir "bin\llama-server.exe"
         if ($BuildOk -and (Test-Path $altBin)) {
-            Write-Step "llama.cpp" "built at $altBin"
-            Write-Step "build time" "${totalMin}m ${totalSec}s" "DarkGray"
+            step "llama.cpp" "built"
+            step "build time" "${totalMin}m ${totalSec}s" "DarkGray"
         } else {
-            Write-Step "llama.cpp" "build failed at step: $FailedStep (${totalMin}m ${totalSec}s)" "Red"
-            Write-Substep "To retry: delete $LlamaCppDir and re-run setup." "Yellow"
+            step "llama.cpp" "failed: $FailedStep (${totalMin}m ${totalSec}s)" "Red"
+            substep "retry: remove $LlamaCppDir and re-run setup" "Yellow"
             exit 1
         }
     }
 }
 
-# ============================================
-# Done
-# ============================================
+# --- Footer (matches setup.sh) ---
+Write-Host ""
+if ($script:StudioVtOk -and -not $env:NO_COLOR) {
+    Write-Host ("  {0}{1}{2}" -f (Get-StudioAnsi Dim), $Rule, (Get-StudioAnsi Reset))
+    Write-Host ("  {0}Unsloth Studio Installed{1}" -f (Get-StudioAnsi Title), (Get-StudioAnsi Reset))
+    Write-Host ("  {0}{1}{2}" -f (Get-StudioAnsi Dim), $Rule, (Get-StudioAnsi Reset))
+} else {
+    Write-Host "  $Rule" -ForegroundColor DarkGray
+    Write-Host "  Unsloth Studio Installed" -ForegroundColor Green
+    Write-Host "  $Rule" -ForegroundColor DarkGray
+}
+step "launch" "unsloth studio -H 0.0.0.0 -p 8888"
 Write-Host ""
-Write-Host "  $Rule" -ForegroundColor DarkGray
-Write-Host "  Unsloth Studio Installed" -ForegroundColor Green
-Write-Step "launch" "unsloth studio -H 0.0.0.0 -p 8888"

From b4e6ba32d0a23daf2ef67f02c6ad8b40ef357307 Mon Sep 17 00:00:00 2001
From: imagineer99 <samleejackson0@gmail.com>
Date: Fri, 20 Mar 2026 22:27:25 +0000
Subject: [PATCH 03/16] studio(setup): revert nvcc path reordering to match
 main

---
 studio/setup.sh | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/studio/setup.sh b/studio/setup.sh
index e70face973..37f3b43b75 100755
--- a/studio/setup.sh
+++ b/studio/setup.sh
@@ -359,22 +359,20 @@ rm -rf "$LLAMA_CPP_DIR"
                 CMAKE_ARGS="$CMAKE_ARGS -DCMAKE_C_COMPILER_LAUNCHER=ccache -DCMAKE_CXX_COMPILER_LAUNCHER=ccache -DCMAKE_CUDA_COMPILER_LAUNCHER=ccache"
             fi
 
-            # Prefer newest /usr/local/cuda-*/bin/nvcc (e.g. 12.8 for sm_120) over
-            # /usr/bin/nvcc from distro nvidia-cuda-toolkit (often 12.0, too old).
             NVCC_PATH=""
-            if ls /usr/local/cuda-*/bin/nvcc &>/dev/null 2>&1; then
-                NVCC_PATH="$(ls -d /usr/local/cuda-*/bin/nvcc 2>/dev/null | sort -V | tail -1)"
-                export PATH="$(dirname "$NVCC_PATH"):$PATH"
+            if command -v nvcc &>/dev/null; then
+                NVCC_PATH="$(command -v nvcc)"
             elif [ -x /usr/local/cuda/bin/nvcc ]; then
                 NVCC_PATH="/usr/local/cuda/bin/nvcc"
                 export PATH="/usr/local/cuda/bin:$PATH"
-            elif command -v nvcc &>/dev/null; then
-                NVCC_PATH="$(command -v nvcc)"
+            elif ls /usr/local/cuda-*/bin/nvcc &>/dev/null 2>&1; then
+                NVCC_PATH="$(ls -d /usr/local/cuda-*/bin/nvcc 2>/dev/null | sort -V | tail -1)"
+                export PATH="$(dirname "$NVCC_PATH"):$PATH"
             fi
 
             _BUILD_DESC="building"
             if [ -n "$NVCC_PATH" ]; then
-                CMAKE_ARGS="$CMAKE_ARGS -DGGML_CUDA=ON -DCMAKE_CUDA_COMPILER=$NVCC_PATH"
+                CMAKE_ARGS="$CMAKE_ARGS -DGGML_CUDA=ON"
 
                 CUDA_ARCHS=""
                 if command -v nvidia-smi &>/dev/null; then

From de936344363e6f9dcf2408817db0ea696fcb6fb0 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Fri, 20 Mar 2026 22:52:07 +0000
Subject: [PATCH 04/16] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 studio/backend/run.py          | 6 +++---
 studio/install_python_stack.py | 2 --
 unsloth_cli/commands/studio.py | 2 +-
 3 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/studio/backend/run.py b/studio/backend/run.py
index 92cbb054ae..c97802e843 100644
--- a/studio/backend/run.py
+++ b/studio/backend/run.py
@@ -225,9 +225,9 @@ def _run():
     if not silent:
         display_host = _resolve_external_ip() if host == "0.0.0.0" else host
         print_studio_access_banner(
-            port=port,
-            bind_host=host,
-            display_host=display_host,
+            port = port,
+            bind_host = host,
+            display_host = display_host,
         )
 
     return app
diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py
index 4503c26a63..8aaead7348 100644
--- a/studio/install_python_stack.py
+++ b/studio/install_python_stack.py
@@ -316,8 +316,6 @@ def install_python_stack() -> int:
     _STEP = 0
     _TOTAL = 10 if IS_WINDOWS else 11
 
-    
-
     # 1. Upgrade pip (needed even with uv as fallback and for bootstrapping)
     _progress("pip upgrade")
     run("Upgrading pip", [sys.executable, "-m", "pip", "install", "--upgrade", "pip"])
diff --git a/unsloth_cli/commands/studio.py b/unsloth_cli/commands/studio.py
index 533d2a5ca8..62f8e27045 100644
--- a/unsloth_cli/commands/studio.py
+++ b/unsloth_cli/commands/studio.py
@@ -167,7 +167,7 @@ def setup(
         False,
         "--verbose",
         "-v",
-        help=(
+        help = (
             "Full pip output during Python install; print command logs when "
             "optional steps fail (e.g. llama.cpp cmake)."
         ),

From 9a691e411732b203b72d30a74bdfdb0049fde061 Mon Sep 17 00:00:00 2001
From: imagineer99 <samleejackson0@gmail.com>
Date: Fri, 20 Mar 2026 23:58:36 +0000
Subject: [PATCH 05/16] studio(setup): restore fail-fast llama.cpp setup flow

---
 studio/setup.sh | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/studio/setup.sh b/studio/setup.sh
index 37f3b43b75..a9e2d700ac 100755
--- a/studio/setup.sh
+++ b/studio/setup.sh
@@ -350,7 +350,7 @@ rm -rf "$LLAMA_CPP_DIR"
         step "llama.cpp" "skipped (git not found)" "$C_WARN"
     else
         BUILD_OK=true
-        try_quiet "clone llama.cpp" git clone --depth 1 https://github.com/ggml-org/llama.cpp.git "$LLAMA_CPP_DIR" || BUILD_OK=false
+        run_quiet "clone llama.cpp" git clone --depth 1 https://github.com/ggml-org/llama.cpp.git "$LLAMA_CPP_DIR"
 
         if [ "$BUILD_OK" = true ]; then
             CMAKE_ARGS="-DLLAMA_BUILD_TESTS=OFF -DLLAMA_BUILD_EXAMPLES=OFF -DLLAMA_BUILD_SERVER=ON -DGGML_NATIVE=ON"
@@ -409,11 +409,11 @@ rm -rf "$LLAMA_CPP_DIR"
                 CMAKE_GENERATOR_ARGS="-G Ninja"
             fi
 
-            try_quiet "cmake llama.cpp" cmake $CMAKE_GENERATOR_ARGS -S "$LLAMA_CPP_DIR" -B "$LLAMA_CPP_DIR/build" $CMAKE_ARGS || BUILD_OK=false
+            run_quiet "cmake llama.cpp" cmake $CMAKE_GENERATOR_ARGS -S "$LLAMA_CPP_DIR" -B "$LLAMA_CPP_DIR/build" $CMAKE_ARGS
         fi
 
         if [ "$BUILD_OK" = true ]; then
-            try_quiet "build llama-server" cmake --build "$LLAMA_CPP_DIR/build" --config Release --target llama-server -j"$NCPU" || BUILD_OK=false
+            run_quiet "build llama-server" cmake --build "$LLAMA_CPP_DIR/build" --config Release --target llama-server -j"$NCPU"
         fi
 
         if [ "$BUILD_OK" = true ]; then
@@ -430,7 +430,7 @@ rm -rf "$LLAMA_CPP_DIR"
         elif [ "$BUILD_OK" = true ]; then
             step "llama.cpp" "binary not found after build" "$C_WARN"
         else
-            step "llama.cpp" "build failed (non-fatal)" "$C_WARN"
+            step "llama.cpp" "build failed" "$C_ERR"
         fi
     fi
 }

From d11dc31732be2d57676d69f57f590fc9daba06bb Mon Sep 17 00:00:00 2001
From: imagineer99 <samleejackson0@gmail.com>
Date: Sat, 21 Mar 2026 00:20:43 +0000
Subject: [PATCH 06/16] studio(banner): use IPv6 loopback URL when binding ::
 or ::1

---
 studio/backend/startup_banner.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/studio/backend/startup_banner.py b/studio/backend/startup_banner.py
index 62ebd40b14..12fed1780c 100644
--- a/studio/backend/startup_banner.py
+++ b/studio/backend/startup_banner.py
@@ -50,8 +50,13 @@ def print_studio_access_banner(
     def style(text: str, code: str) -> str:
         return f"{code}{text}{reset}" if use_color else text
 
-    local_url = f"http://127.0.0.1:{port}"
-    alt_local = f"http://localhost:{port}"
+    ipv6_bind = bind_host in ("::", "::1")
+    if ipv6_bind:
+        local_url = f"http://[::1]:{port}"
+        alt_local = f"http://localhost:{port}"
+    else:
+        local_url = f"http://127.0.0.1:{port}"
+        alt_local = f"http://localhost:{port}"
     external_url = f"http://{display_host}:{port}"
     listen_all = bind_host in ("0.0.0.0", "::")
     loopback_bind = bind_host in ("127.0.0.1", "localhost", "::1")

From 67fda0cc7b1d9317323530d1b6bdb0a1d31086d7 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 24 Mar 2026 10:18:42 +0000
Subject: [PATCH 07/16] Fix IPv6 URL bracketing, try_quiet stderr, _step label
 clamp

- Bracket IPv6 display_host in external_url to produce clickable URLs
- Redirect try_quiet failure log to stderr instead of stdout
- Clamp _step label to column width to prevent negative padding
---
 studio/backend/startup_banner.py   |   5 +-
 studio/install_python_stack.py     |   3 +-
 studio/setup.sh                    |   2 +-
 tests/test_studio_setup_styling.py | 820 +++++++++++++++++++++++++++++
 4 files changed, 827 insertions(+), 3 deletions(-)
 create mode 100644 tests/test_studio_setup_styling.py

diff --git a/studio/backend/startup_banner.py b/studio/backend/startup_banner.py
index 12fed1780c..54acac0540 100644
--- a/studio/backend/startup_banner.py
+++ b/studio/backend/startup_banner.py
@@ -57,7 +57,10 @@ def style(text: str, code: str) -> str:
     else:
         local_url = f"http://127.0.0.1:{port}"
         alt_local = f"http://localhost:{port}"
-    external_url = f"http://{display_host}:{port}"
+    if ":" in display_host:
+        external_url = f"http://[{display_host}]:{port}"
+    else:
+        external_url = f"http://{display_host}:{port}"
     listen_all = bind_host in ("0.0.0.0", "::")
     loopback_bind = bind_host in ("127.0.0.1", "localhost", "::1")
     api_base = local_url if listen_all or loopback_bind else external_url
diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py
index 8aaead7348..52e4c96593 100644
--- a/studio/install_python_stack.py
+++ b/studio/install_python_stack.py
@@ -110,7 +110,8 @@ def _step(label: str, value: str, color_fn = None) -> None:
     """Print a single step line in the column format."""
     if color_fn is None:
         color_fn = _green
-    print(f"  {_dim(label)}{' ' * (_COL - len(label))}{color_fn(value)}")
+    padded = label[:_COL]
+    print(f"  {_dim(padded)}{' ' * (_COL - len(padded))}{color_fn(value)}")
 
 
 def _progress(label: str) -> None:
diff --git a/studio/setup.sh b/studio/setup.sh
index bb29c0ba6d..b795be1530 100755
--- a/studio/setup.sh
+++ b/studio/setup.sh
@@ -77,7 +77,7 @@ try_quiet() {
         local exit_code=$?
         if [ "${UNSLOTH_VERBOSE:-0}" = "1" ]; then
             step "error" "$label failed (exit code $exit_code)" "$C_ERR"
-            cat "$tmplog"
+            cat "$tmplog" >&2
         fi
         rm -f "$tmplog"
         return $exit_code
diff --git a/tests/test_studio_setup_styling.py b/tests/test_studio_setup_styling.py
new file mode 100644
index 0000000000..94308fc703
--- /dev/null
+++ b/tests/test_studio_setup_styling.py
@@ -0,0 +1,820 @@
+"""Tests for PR #4494 Studio setup log styling.
+
+Validates terminal output styling across:
+- studio/backend/startup_banner.py
+- studio/install_python_stack.py
+- studio/setup.sh  (bash subprocess tests)
+- studio/setup.ps1 (structural/static analysis only -- no pwsh on Linux)
+- unsloth_cli/commands/studio.py (CLI tests via typer CliRunner)
+"""
+
+from __future__ import annotations
+
+import os
+import re
+import subprocess
+import sys
+from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+# Ensure repo root is importable even without pip install -e .
+_REPO = Path(__file__).resolve().parent.parent
+if str(_REPO) not in sys.path:
+    sys.path.insert(0, str(_REPO))
+
+SETUP_SH = _REPO / "studio" / "setup.sh"
+SETUP_PS1 = _REPO / "studio" / "setup.ps1"
+
+# ── Bash fragments inlined from setup.sh for subprocess tests ─────────
+
+_BASH_COLOR_INIT = r"""
+if [ -n "${NO_COLOR:-}" ]; then
+    C_TITLE= C_DIM= C_OK= C_WARN= C_ERR= C_RST=
+elif [ -t 1 ] || [ -n "${FORCE_COLOR:-}" ]; then
+    C_TITLE=$'\033[38;5;150m'
+    C_DIM=$'\033[38;5;245m'
+    C_OK=$'\033[38;5;108m'
+    C_WARN=$'\033[38;5;136m'
+    C_ERR=$'\033[91m'
+    C_RST=$'\033[0m'
+else
+    C_TITLE= C_DIM= C_OK= C_WARN= C_ERR= C_RST=
+fi
+"""
+
+_BASH_HELPERS = r"""
+step()    { printf "  ${C_DIM}%-15s${C_RST}${3:-$C_OK}%s${C_RST}\n" "$1" "$2"; }
+substep() { printf "  ${C_DIM}%-15s%s${C_RST}\n" "" "$1"; }
+"""
+
+_BASH_RUN_QUIET = r"""
+_run_quiet() {
+    local on_fail=$1
+    local label=$2
+    shift 2
+    local tmplog
+    tmplog=$(mktemp) || {
+        step "error" "Failed to create temporary file" "$C_ERR"
+        [ "$on_fail" = "exit" ] && exit 1 || return 1
+    }
+    if "$@" >"$tmplog" 2>&1; then
+        rm -f "$tmplog"
+        return 0
+    else
+        local exit_code=$?
+        step "error" "$label failed (exit code $exit_code)" "$C_ERR"
+        cat "$tmplog" >&2
+        rm -f "$tmplog"
+        if [ "$on_fail" = "exit" ]; then
+            exit "$exit_code"
+        else
+            return "$exit_code"
+        fi
+    fi
+}
+run_quiet() { _run_quiet exit "$@"; }
+run_quiet_no_exit() { _run_quiet return "$@"; }
+"""
+
+_BASH_TRY_QUIET = r"""
+try_quiet() {
+    local label="$1"; shift
+    local tmplog; tmplog=$(mktemp)
+    if "$@" > "$tmplog" 2>&1; then
+        rm -f "$tmplog"
+        return 0
+    else
+        local exit_code=$?
+        if [ "${UNSLOTH_VERBOSE:-0}" = "1" ]; then
+            step "error" "$label failed (exit code $exit_code)" "$C_ERR"
+            cat "$tmplog" >&2
+        fi
+        rm -f "$tmplog"
+        return $exit_code
+    fi
+}
+"""
+
+# Template for REQUESTED_PYTHON_VERSION tests (uses .replace() placeholders)
+_REQUESTED_PY_TPL = r"""
+C_DIM= C_OK= C_WARN= C_ERR= C_RST= C_TITLE=
+step()    { printf "  %-15s%s\n" "$1" "$2"; }
+substep() { printf "  %-15s%s\n" "" "$1"; }
+MIN_PY_MINOR=__MIN__
+MAX_PY_MINOR=__MAX__
+BEST_PY=""
+BEST_MINOR=0
+REQUESTED_PYTHON_VERSION="__REQ__"
+if [ -n "${REQUESTED_PYTHON_VERSION:-}" ] && [ -x "$REQUESTED_PYTHON_VERSION" ]; then
+    _req_ver=$("$REQUESTED_PYTHON_VERSION" --version 2>&1 | awk '{print $2}')
+    _req_major=$(echo "$_req_ver" | cut -d. -f1)
+    _req_minor=$(echo "$_req_ver" | cut -d. -f2)
+    if [ "$_req_major" -eq 3 ] 2>/dev/null && \
+       [ "$_req_minor" -ge "$MIN_PY_MINOR" ] 2>/dev/null && \
+       [ "$_req_minor" -le "$MAX_PY_MINOR" ] 2>/dev/null; then
+        BEST_PY="$REQUESTED_PYTHON_VERSION"
+        substep "using requested Python: $BEST_PY"
+    else
+        substep "ignoring requested Python $REQUESTED_PYTHON_VERSION ($_req_ver) -- outside range"
+    fi
+fi
+echo "BEST_PY=$BEST_PY"
+"""
+
+
+def _run_bash(
+    script: str, env_extra: dict[str, str] | None = None
+) -> subprocess.CompletedProcess[str]:
+    """Run a bash script fragment with a sanitised environment."""
+    env = {
+        k: v
+        for k, v in os.environ.items()
+        if k not in ("NO_COLOR", "FORCE_COLOR", "UNSLOTH_VERBOSE")
+    }
+    if env_extra:
+        env.update(env_extra)
+    return subprocess.run(
+        ["bash", "-c", script], capture_output=True, text=True, env=env
+    )
+
+
+@pytest.fixture(autouse=True)
+def _clean_env(monkeypatch):
+    """Remove colour / verbosity env vars before every test."""
+    monkeypatch.delenv("NO_COLOR", raising=False)
+    monkeypatch.delenv("FORCE_COLOR", raising=False)
+    monkeypatch.delenv("UNSLOTH_VERBOSE", raising=False)
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 1. startup_banner.py -- stdout_supports_color()  (9 tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestBannerSupportsColor:
+    """NO_COLOR > FORCE_COLOR > isatty."""
+
+    @staticmethod
+    def _call():
+        from studio.backend.startup_banner import stdout_supports_color
+
+        return stdout_supports_color()
+
+    def test_no_tty_no_env(self):
+        with patch("sys.stdout.isatty", return_value=False):
+            assert self._call() is False
+
+    def test_tty_returns_true(self):
+        with patch("sys.stdout.isatty", return_value=True):
+            assert self._call() is True
+
+    def test_no_color_disables_with_tty(self, monkeypatch):
+        monkeypatch.setenv("NO_COLOR", "1")
+        with patch("sys.stdout.isatty", return_value=True):
+            assert self._call() is False
+
+    def test_force_color_enables_no_tty(self, monkeypatch):
+        monkeypatch.setenv("FORCE_COLOR", "1")
+        with patch("sys.stdout.isatty", return_value=False):
+            assert self._call() is True
+
+    def test_force_color_zero_truthy(self, monkeypatch):
+        monkeypatch.setenv("FORCE_COLOR", "0")
+        with patch("sys.stdout.isatty", return_value=False):
+            assert self._call() is True
+
+    def test_force_color_empty_falsy(self, monkeypatch):
+        monkeypatch.setenv("FORCE_COLOR", "")
+        with patch("sys.stdout.isatty", return_value=False):
+            assert self._call() is False
+
+    def test_no_color_empty_no_disable(self, monkeypatch):
+        monkeypatch.setenv("NO_COLOR", "")
+        with patch("sys.stdout.isatty", return_value=True):
+            assert self._call() is True
+
+    def test_no_color_whitespace_no_disable(self, monkeypatch):
+        monkeypatch.setenv("NO_COLOR", "   ")
+        with patch("sys.stdout.isatty", return_value=True):
+            assert self._call() is True
+
+    def test_no_color_overrides_force(self, monkeypatch):
+        monkeypatch.setenv("NO_COLOR", "1")
+        monkeypatch.setenv("FORCE_COLOR", "1")
+        with patch("sys.stdout.isatty", return_value=True):
+            assert self._call() is False
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 2. startup_banner.py -- print_port_in_use_notice()  (3 tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestPortInUseNotice:
+
+    def test_contains_both_ports(self, capsys):
+        from studio.backend.startup_banner import print_port_in_use_notice
+
+        with patch(
+            "studio.backend.startup_banner.stdout_supports_color", return_value=False
+        ):
+            print_port_in_use_notice(8888, 8889)
+        out = capsys.readouterr().out
+        assert "8888" in out
+        assert "8889" in out
+
+    def test_no_color_no_ansi(self, capsys):
+        from studio.backend.startup_banner import print_port_in_use_notice
+
+        with patch(
+            "studio.backend.startup_banner.stdout_supports_color", return_value=False
+        ):
+            print_port_in_use_notice(8888, 8889)
+        assert "\033[" not in capsys.readouterr().out
+
+    def test_color_has_ansi(self, capsys):
+        from studio.backend.startup_banner import print_port_in_use_notice
+
+        with patch(
+            "studio.backend.startup_banner.stdout_supports_color", return_value=True
+        ):
+            print_port_in_use_notice(8888, 8889)
+        assert "\033[" in capsys.readouterr().out
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 3. startup_banner.py -- print_studio_access_banner()  (11 tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestStudioAccessBanner:
+
+    PORT = 8888
+
+    @staticmethod
+    def _call(*, bind_host, display_host, port=8888, color=False):
+        from studio.backend.startup_banner import print_studio_access_banner
+
+        with patch(
+            "studio.backend.startup_banner.stdout_supports_color",
+            return_value=color,
+        ):
+            print_studio_access_banner(
+                port=port, bind_host=bind_host, display_host=display_host
+            )
+
+    @pytest.mark.parametrize(
+        "bind_host,display_host,expect_local,from_another,bound_addr,api_host",
+        [
+            ("0.0.0.0", "192.168.1.100", "127.0.0.1", True, False, "127.0.0.1"),
+            ("0.0.0.0", "127.0.0.1", "127.0.0.1", False, False, "127.0.0.1"),
+            ("0.0.0.0", "0.0.0.0", "127.0.0.1", False, False, "127.0.0.1"),
+            ("::", "2001:db8::1", "[::1]", True, False, "[::1]"),
+            ("::1", "::1", "[::1]", False, False, "[::1]"),
+            ("127.0.0.1", "127.0.0.1", "127.0.0.1", False, False, "127.0.0.1"),
+            ("localhost", "localhost", "127.0.0.1", False, False, "127.0.0.1"),
+            ("192.168.1.5", "192.168.1.5", "127.0.0.1", False, True, "192.168.1.5"),
+        ],
+        ids=[
+            "wildcard_v4_ext",
+            "wildcard_v4_loop",
+            "wildcard_v4_self",
+            "wildcard_v6_ext",
+            "loopback_v6",
+            "loopback_v4",
+            "localhost",
+            "specific_ip",
+        ],
+    )
+    def test_hosts(
+        self,
+        capsys,
+        bind_host,
+        display_host,
+        expect_local,
+        from_another,
+        bound_addr,
+        api_host,
+    ):
+        self._call(bind_host=bind_host, display_host=display_host, color=False)
+        out = capsys.readouterr().out
+        assert f"http://{expect_local}:{self.PORT}" in out
+        assert ("From another device" in out) == from_another
+        assert ("Bound address" in out) == bound_addr
+        assert f"http://{api_host}:{self.PORT}/api" in out
+
+    def test_color_on_has_ansi(self, capsys):
+        self._call(bind_host="127.0.0.1", display_host="127.0.0.1", color=True)
+        assert "\033[" in capsys.readouterr().out
+
+    def test_color_off_no_ansi(self, capsys):
+        self._call(bind_host="127.0.0.1", display_host="127.0.0.1", color=False)
+        assert "\033[" not in capsys.readouterr().out
+
+    def test_api_urls_present(self, capsys):
+        self._call(bind_host="0.0.0.0", display_host="10.0.0.1", color=False)
+        out = capsys.readouterr().out
+        assert "/api" in out
+        assert "/api/health" in out
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 4. install_python_stack.py -- Color detection  (5 tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestInstallColorDetect:
+
+    @staticmethod
+    def _mod():
+        import studio.install_python_stack as ips
+
+        return ips
+
+    def test_no_tty(self):
+        ips = self._mod()
+        with patch("sys.stdout.isatty", return_value=False):
+            assert ips._stdout_supports_color() is False
+
+    def test_tty_linux(self, monkeypatch):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "IS_WINDOWS", False)
+        with patch("sys.stdout.isatty", return_value=True):
+            assert ips._stdout_supports_color() is True
+
+    def test_force_color(self, monkeypatch):
+        monkeypatch.setenv("FORCE_COLOR", "1")
+        assert self._mod()._stdout_supports_color() is True
+
+    def test_no_color(self, monkeypatch):
+        monkeypatch.setenv("NO_COLOR", "1")
+        with patch("sys.stdout.isatty", return_value=True):
+            assert self._mod()._stdout_supports_color() is False
+
+    def test_windows_ctypes_fail(self, monkeypatch):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "IS_WINDOWS", True)
+        with patch("sys.stdout.isatty", return_value=True):
+            # ctypes.windll does not exist on Linux, so the except branch fires
+            assert ips._stdout_supports_color() is False
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 5. install_python_stack.py -- _step() formatting  (6 tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestInstallStep:
+
+    @staticmethod
+    def _mod():
+        import studio.install_python_stack as ips
+
+        return ips
+
+    def test_short_label_padding(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "_HAS_COLOR", False)
+        ips._step("deps", "ok")
+        out = capsys.readouterr().out
+        # "deps" (4 chars) + 11 spaces padding to fill 15-col layout
+        assert "  deps" + " " * 11 + "ok" in out
+
+    def test_exact_15_label(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "_HAS_COLOR", False)
+        label = "x" * 15
+        ips._step(label, "v")
+        out = capsys.readouterr().out
+        assert label + "v" in out
+
+    def test_long_label(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "_HAS_COLOR", False)
+        label = "x" * 20
+        ips._step(label, "v")
+        out = capsys.readouterr().out
+        # Label is clamped to _COL (15) chars; value appears right after
+        assert label[:15] + "v" in out
+        assert label not in out
+
+    def test_default_green(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "_HAS_COLOR", True)
+        ips._step("deps", "ok")
+        assert "\033[38;5;108m" in capsys.readouterr().out
+
+    def test_custom_color_fn(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "_HAS_COLOR", True)
+        ips._step("err", "fail", color_fn=ips._red)
+        assert "\033[91m" in capsys.readouterr().out
+
+    def test_no_color_no_ansi(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "_HAS_COLOR", False)
+        ips._step("deps", "ok")
+        assert "\033[" not in capsys.readouterr().out
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 6. install_python_stack.py -- _progress()  (4 tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestInstallProgress:
+
+    @staticmethod
+    def _mod():
+        import studio.install_python_stack as ips
+
+        return ips
+
+    def test_verbose_no_output(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "VERBOSE", True)
+        monkeypatch.setattr(ips, "_STEP", 0)
+        monkeypatch.setattr(ips, "_TOTAL", 5)
+        ips._progress("lbl")
+        assert capsys.readouterr().out == ""
+        assert ips._STEP == 1
+
+    def test_normal_has_cr(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "VERBOSE", False)
+        monkeypatch.setattr(ips, "_STEP", 0)
+        monkeypatch.setattr(ips, "_TOTAL", 5)
+        monkeypatch.setattr(ips, "_HAS_COLOR", False)
+        ips._progress("lbl")
+        out = capsys.readouterr().out
+        assert "\r" in out
+        assert not out.endswith("\n")
+
+    def test_final_step_newline(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "VERBOSE", False)
+        monkeypatch.setattr(ips, "_STEP", 4)
+        monkeypatch.setattr(ips, "_TOTAL", 5)
+        monkeypatch.setattr(ips, "_HAS_COLOR", False)
+        ips._progress("done")
+        assert capsys.readouterr().out.endswith("\n")
+
+    def test_bar_format(self, monkeypatch, capsys):
+        ips = self._mod()
+        monkeypatch.setattr(ips, "VERBOSE", False)
+        monkeypatch.setattr(ips, "_STEP", 0)
+        monkeypatch.setattr(ips, "_TOTAL", 10)
+        monkeypatch.setattr(ips, "_HAS_COLOR", False)
+        ips._progress("s1")
+        out = capsys.readouterr().out
+        assert re.search(r"\[=+-+\]", out)
+        assert " 1/10" in out
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 7. setup.sh -- Color initialization  (4 bash tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestBashColorInit:
+
+    _SCRIPT = _BASH_COLOR_INIT + '\necho "DIM=${C_DIM}END"'
+
+    def test_no_env_empty(self):
+        r = _run_bash(self._SCRIPT)
+        assert r.returncode == 0
+        assert "DIM=END" in r.stdout
+
+    def test_force_color_sets(self):
+        r = _run_bash(self._SCRIPT, {"FORCE_COLOR": "1"})
+        assert r.returncode == 0
+        # C_DIM is non-empty when FORCE_COLOR is set
+        assert "DIM=END" not in r.stdout
+
+    def test_no_color_clears(self):
+        r = _run_bash(self._SCRIPT, {"NO_COLOR": "1"})
+        assert r.returncode == 0
+        assert "DIM=END" in r.stdout
+
+    def test_no_color_beats_force(self):
+        r = _run_bash(self._SCRIPT, {"NO_COLOR": "1", "FORCE_COLOR": "1"})
+        assert r.returncode == 0
+        assert "DIM=END" in r.stdout
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 8. setup.sh -- step() and substep()  (5 bash tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestBashStep:
+
+    _PRE = _BASH_COLOR_INIT + _BASH_HELPERS
+
+    def test_step_padding(self):
+        r = _run_bash(self._PRE + '\nstep "deps" "installed"')
+        assert r.returncode == 0
+        assert "deps" in r.stdout and "installed" in r.stdout
+
+    def test_step_15_label(self):
+        r = _run_bash(self._PRE + '\nstep "aaaaaaaaaaaaaaa" "v"')
+        assert "aaaaaaaaaaaaaaa" in r.stdout and "v" in r.stdout
+
+    def test_step_long_label(self):
+        r = _run_bash(self._PRE + '\nstep "aaaaaaaaaaaaaaaaaaaa" "v"')
+        assert "aaaaaaaaaaaaaaaaaaaa" in r.stdout
+
+    def test_step_color(self):
+        r = _run_bash(self._PRE + '\nstep "deps" "ok"', {"FORCE_COLOR": "1"})
+        assert "\033[" in r.stdout
+
+    def test_substep(self):
+        r = _run_bash(self._PRE + '\nsubstep "building..."')
+        assert "building..." in r.stdout
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 9. setup.sh -- run_quiet / run_quiet_no_exit  (5 bash tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestBashRunQuiet:
+
+    _PRE = _BASH_COLOR_INIT + _BASH_HELPERS + _BASH_RUN_QUIET
+
+    def test_no_exit_success(self):
+        r = _run_bash(self._PRE + '\nrun_quiet_no_exit "ok" true; echo "RC=$?"')
+        assert "RC=0" in r.stdout
+
+    def test_no_exit_failure_rc(self):
+        r = _run_bash(
+            self._PRE
+            + "\nrun_quiet_no_exit \"cmd\" bash -c 'echo log_data; exit 42'"
+            + '\necho "RC=$?"'
+        )
+        assert "RC=42" in r.stdout
+
+    def test_no_exit_failure_shows_error(self):
+        r = _run_bash(
+            self._PRE
+            + "\nrun_quiet_no_exit \"my cmd\" bash -c 'echo log_data; exit 1'"
+            + "\ntrue"
+        )
+        assert "my cmd failed" in r.stdout
+        assert "log_data" in r.stderr
+
+    def test_run_quiet_failure_exits(self):
+        script = self._PRE + """
+(
+    run_quiet "bad" false
+    echo "SHOULD_NOT_REACH"
+)
+echo "OUTER=$?"
+"""
+        r = _run_bash(script)
+        assert "SHOULD_NOT_REACH" not in r.stdout
+
+    def test_run_quiet_success(self):
+        r = _run_bash(self._PRE + '\nrun_quiet "ok" true; echo "RC=$?"')
+        assert "RC=0" in r.stdout
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 10. setup.sh -- try_quiet  (5 bash tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestBashTryQuiet:
+
+    _PRE = _BASH_COLOR_INIT + _BASH_HELPERS + _BASH_TRY_QUIET
+
+    def test_success_rc0(self):
+        r = _run_bash(self._PRE + '\ntry_quiet "ok" true; echo "RC=$?"')
+        assert "RC=0" in r.stdout
+
+    def test_fail_quiet_no_output(self):
+        r = _run_bash(
+            self._PRE
+            + "\ntry_quiet \"cmd\" bash -c 'echo secret; exit 1'"
+            + '\necho "RC=$?"'
+        )
+        assert "RC=1" in r.stdout
+        assert "secret" not in r.stdout
+        assert "error" not in r.stdout
+
+    def test_fail_verbose_shows(self):
+        r = _run_bash(
+            self._PRE
+            + "\ntry_quiet \"cmd\" bash -c 'echo visible; exit 1'"
+            + '\necho "RC=$?"',
+            {"UNSLOTH_VERBOSE": "1"},
+        )
+        assert "cmd failed" in r.stdout
+        assert "visible" in r.stderr
+
+    def test_preserves_exit_code(self):
+        r = _run_bash(
+            self._PRE
+            + "\ntry_quiet \"cmd\" bash -c 'exit 42'"
+            + '\necho "RC=$?"'
+        )
+        assert "RC=42" in r.stdout
+
+    def test_success_verbose_quiet(self):
+        r = _run_bash(
+            self._PRE + '\ntry_quiet "cmd" true; echo "RC=$?"',
+            {"UNSLOTH_VERBOSE": "1"},
+        )
+        assert "RC=0" in r.stdout
+        assert "error" not in r.stdout
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 11. setup.sh -- REQUESTED_PYTHON_VERSION  (5 bash tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestBashRequestedPython:
+
+    _PY_MINOR = sys.version_info.minor
+
+    def _block(self, *, requested="", min_minor=None, max_minor=None):
+        if min_minor is None:
+            min_minor = self._PY_MINOR - 1
+        if max_minor is None:
+            max_minor = self._PY_MINOR + 1
+        return (
+            _REQUESTED_PY_TPL.replace("__MIN__", str(min_minor))
+            .replace("__MAX__", str(max_minor))
+            .replace("__REQ__", requested)
+        )
+
+    def test_unset_skipped(self):
+        r = _run_bash(self._block(requested=""))
+        assert r.stdout.strip().endswith("BEST_PY=")
+
+    def test_valid_in_range(self):
+        r = _run_bash(self._block(requested=sys.executable))
+        assert f"BEST_PY={sys.executable}" in r.stdout
+        assert "using requested Python" in r.stdout
+
+    def test_nonexistent_skipped(self):
+        r = _run_bash(self._block(requested="/nonexistent/python3.99"))
+        assert r.stdout.strip().endswith("BEST_PY=")
+
+    def test_not_executable_skipped(self):
+        r = _run_bash(self._block(requested="/dev/null"))
+        assert r.stdout.strip().endswith("BEST_PY=")
+
+    def test_out_of_range(self):
+        r = _run_bash(
+            self._block(
+                requested=sys.executable,
+                min_minor=self._PY_MINOR + 10,
+                max_minor=self._PY_MINOR + 11,
+            )
+        )
+        assert r.stdout.strip().endswith("BEST_PY=")
+        assert "ignoring" in r.stdout
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 12. setup.ps1 -- Structural checks  (9 tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestPS1Structural:
+    """Static analysis of setup.ps1 -- no PowerShell runtime needed."""
+
+    @pytest.fixture(autouse=True)
+    def _load(self):
+        self.content = SETUP_PS1.read_text(encoding="utf-8")
+
+    def test_fn_step(self):
+        assert re.search(r"^function step\s*\{", self.content, re.MULTILINE)
+
+    def test_fn_substep(self):
+        assert re.search(r"^function substep\s*\{", self.content, re.MULTILINE)
+
+    def test_fn_get_studio_ansi(self):
+        assert "function Get-StudioAnsi" in self.content
+
+    def test_fn_enable_vt(self):
+        assert "function Enable-StudioVirtualTerminal" in self.content
+
+    def test_ansi_title_code(self):
+        assert "38;5;150m" in self.content
+
+    def test_ansi_dim_code(self):
+        assert "38;5;245m" in self.content
+
+    def test_ansi_ok_code(self):
+        assert "38;5;108m" in self.content
+
+    def test_label_padding(self):
+        assert ".PadRight(15)" in self.content
+
+    def test_env_vars_referenced(self):
+        assert "NO_COLOR" in self.content
+        assert "UNSLOTH_VERBOSE" in self.content
+
+
+# ═══════════════════════════════════════════════════════════════════════
+# 13. studio.py CLI -- setup --verbose  (4 tests)
+# ═══════════════════════════════════════════════════════════════════════
+
+
+class TestCLISetup:
+
+    def test_verbose_sets_env(self):
+        from typer.testing import CliRunner
+
+        from unsloth_cli.commands.studio import studio_app
+
+        runner = CliRunner()
+        mock_result = MagicMock(returncode=0)
+        with (
+            patch(
+                "unsloth_cli.commands.studio._find_setup_script",
+                return_value=Path("/f/setup.sh"),
+            ),
+            patch(
+                "unsloth_cli.commands.studio.subprocess.run",
+                return_value=mock_result,
+            ) as mock_run,
+            patch(
+                "unsloth_cli.commands.studio.platform.system",
+                return_value="Linux",
+            ),
+        ):
+            result = runner.invoke(studio_app, ["setup", "--verbose"])
+        assert result.exit_code == 0
+        env_arg = mock_run.call_args.kwargs.get("env")
+        assert env_arg is not None
+        assert env_arg["UNSLOTH_VERBOSE"] == "1"
+
+    def test_no_verbose_env_none(self):
+        from typer.testing import CliRunner
+
+        from unsloth_cli.commands.studio import studio_app
+
+        runner = CliRunner()
+        mock_result = MagicMock(returncode=0)
+        with (
+            patch(
+                "unsloth_cli.commands.studio._find_setup_script",
+                return_value=Path("/f/setup.sh"),
+            ),
+            patch(
+                "unsloth_cli.commands.studio.subprocess.run",
+                return_value=mock_result,
+            ) as mock_run,
+            patch(
+                "unsloth_cli.commands.studio.platform.system",
+                return_value="Linux",
+            ),
+        ):
+            result = runner.invoke(studio_app, ["setup"])
+        assert result.exit_code == 0
+        env_arg = mock_run.call_args.kwargs.get("env")
+        assert env_arg is None
+
+    def test_script_not_found(self):
+        from typer.testing import CliRunner
+
+        from unsloth_cli.commands.studio import studio_app
+
+        runner = CliRunner()
+        with patch(
+            "unsloth_cli.commands.studio._find_setup_script",
+            return_value=None,
+        ):
+            result = runner.invoke(studio_app, ["setup"])
+        assert result.exit_code == 1
+        assert "Could not find" in result.output
+
+    def test_nonzero_rc_propagated(self):
+        from typer.testing import CliRunner
+
+        from unsloth_cli.commands.studio import studio_app
+
+        runner = CliRunner()
+        mock_result = MagicMock(returncode=42)
+        with (
+            patch(
+                "unsloth_cli.commands.studio._find_setup_script",
+                return_value=Path("/f/setup.sh"),
+            ),
+            patch(
+                "unsloth_cli.commands.studio.subprocess.run",
+                return_value=mock_result,
+            ),
+            patch(
+                "unsloth_cli.commands.studio.platform.system",
+                return_value="Linux",
+            ),
+        ):
+            result = runner.invoke(studio_app, ["setup"])
+        assert result.exit_code == 42

From 7ed5b69c224e56a353e586dbed913c82089ea4b9 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 24 Mar 2026 10:18:56 +0000
Subject: [PATCH 08/16] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 tests/test_studio_setup_styling.py | 126 +++++++++++++----------------
 1 file changed, 58 insertions(+), 68 deletions(-)

diff --git a/tests/test_studio_setup_styling.py b/tests/test_studio_setup_styling.py
index 94308fc703..e34de4d316 100644
--- a/tests/test_studio_setup_styling.py
+++ b/tests/test_studio_setup_styling.py
@@ -136,16 +136,16 @@ def _run_bash(
     if env_extra:
         env.update(env_extra)
     return subprocess.run(
-        ["bash", "-c", script], capture_output=True, text=True, env=env
+        ["bash", "-c", script], capture_output = True, text = True, env = env
     )
 
 
-@pytest.fixture(autouse=True)
+@pytest.fixture(autouse = True)
 def _clean_env(monkeypatch):
     """Remove colour / verbosity env vars before every test."""
-    monkeypatch.delenv("NO_COLOR", raising=False)
-    monkeypatch.delenv("FORCE_COLOR", raising=False)
-    monkeypatch.delenv("UNSLOTH_VERBOSE", raising=False)
+    monkeypatch.delenv("NO_COLOR", raising = False)
+    monkeypatch.delenv("FORCE_COLOR", raising = False)
+    monkeypatch.delenv("UNSLOTH_VERBOSE", raising = False)
 
 
 # ═══════════════════════════════════════════════════════════════════════
@@ -163,47 +163,47 @@ def _call():
         return stdout_supports_color()
 
     def test_no_tty_no_env(self):
-        with patch("sys.stdout.isatty", return_value=False):
+        with patch("sys.stdout.isatty", return_value = False):
             assert self._call() is False
 
     def test_tty_returns_true(self):
-        with patch("sys.stdout.isatty", return_value=True):
+        with patch("sys.stdout.isatty", return_value = True):
             assert self._call() is True
 
     def test_no_color_disables_with_tty(self, monkeypatch):
         monkeypatch.setenv("NO_COLOR", "1")
-        with patch("sys.stdout.isatty", return_value=True):
+        with patch("sys.stdout.isatty", return_value = True):
             assert self._call() is False
 
     def test_force_color_enables_no_tty(self, monkeypatch):
         monkeypatch.setenv("FORCE_COLOR", "1")
-        with patch("sys.stdout.isatty", return_value=False):
+        with patch("sys.stdout.isatty", return_value = False):
             assert self._call() is True
 
     def test_force_color_zero_truthy(self, monkeypatch):
         monkeypatch.setenv("FORCE_COLOR", "0")
-        with patch("sys.stdout.isatty", return_value=False):
+        with patch("sys.stdout.isatty", return_value = False):
             assert self._call() is True
 
     def test_force_color_empty_falsy(self, monkeypatch):
         monkeypatch.setenv("FORCE_COLOR", "")
-        with patch("sys.stdout.isatty", return_value=False):
+        with patch("sys.stdout.isatty", return_value = False):
             assert self._call() is False
 
     def test_no_color_empty_no_disable(self, monkeypatch):
         monkeypatch.setenv("NO_COLOR", "")
-        with patch("sys.stdout.isatty", return_value=True):
+        with patch("sys.stdout.isatty", return_value = True):
             assert self._call() is True
 
     def test_no_color_whitespace_no_disable(self, monkeypatch):
         monkeypatch.setenv("NO_COLOR", "   ")
-        with patch("sys.stdout.isatty", return_value=True):
+        with patch("sys.stdout.isatty", return_value = True):
             assert self._call() is True
 
     def test_no_color_overrides_force(self, monkeypatch):
         monkeypatch.setenv("NO_COLOR", "1")
         monkeypatch.setenv("FORCE_COLOR", "1")
-        with patch("sys.stdout.isatty", return_value=True):
+        with patch("sys.stdout.isatty", return_value = True):
             assert self._call() is False
 
 
@@ -213,12 +213,11 @@ def test_no_color_overrides_force(self, monkeypatch):
 
 
 class TestPortInUseNotice:
-
     def test_contains_both_ports(self, capsys):
         from studio.backend.startup_banner import print_port_in_use_notice
 
         with patch(
-            "studio.backend.startup_banner.stdout_supports_color", return_value=False
+            "studio.backend.startup_banner.stdout_supports_color", return_value = False
         ):
             print_port_in_use_notice(8888, 8889)
         out = capsys.readouterr().out
@@ -229,7 +228,7 @@ def test_no_color_no_ansi(self, capsys):
         from studio.backend.startup_banner import print_port_in_use_notice
 
         with patch(
-            "studio.backend.startup_banner.stdout_supports_color", return_value=False
+            "studio.backend.startup_banner.stdout_supports_color", return_value = False
         ):
             print_port_in_use_notice(8888, 8889)
         assert "\033[" not in capsys.readouterr().out
@@ -238,7 +237,7 @@ def test_color_has_ansi(self, capsys):
         from studio.backend.startup_banner import print_port_in_use_notice
 
         with patch(
-            "studio.backend.startup_banner.stdout_supports_color", return_value=True
+            "studio.backend.startup_banner.stdout_supports_color", return_value = True
         ):
             print_port_in_use_notice(8888, 8889)
         assert "\033[" in capsys.readouterr().out
@@ -250,19 +249,18 @@ def test_color_has_ansi(self, capsys):
 
 
 class TestStudioAccessBanner:
-
     PORT = 8888
 
     @staticmethod
-    def _call(*, bind_host, display_host, port=8888, color=False):
+    def _call(*, bind_host, display_host, port = 8888, color = False):
         from studio.backend.startup_banner import print_studio_access_banner
 
         with patch(
             "studio.backend.startup_banner.stdout_supports_color",
-            return_value=color,
+            return_value = color,
         ):
             print_studio_access_banner(
-                port=port, bind_host=bind_host, display_host=display_host
+                port = port, bind_host = bind_host, display_host = display_host
             )
 
     @pytest.mark.parametrize(
@@ -277,7 +275,7 @@ def _call(*, bind_host, display_host, port=8888, color=False):
             ("localhost", "localhost", "127.0.0.1", False, False, "127.0.0.1"),
             ("192.168.1.5", "192.168.1.5", "127.0.0.1", False, True, "192.168.1.5"),
         ],
-        ids=[
+        ids = [
             "wildcard_v4_ext",
             "wildcard_v4_loop",
             "wildcard_v4_self",
@@ -298,7 +296,7 @@ def test_hosts(
         bound_addr,
         api_host,
     ):
-        self._call(bind_host=bind_host, display_host=display_host, color=False)
+        self._call(bind_host = bind_host, display_host = display_host, color = False)
         out = capsys.readouterr().out
         assert f"http://{expect_local}:{self.PORT}" in out
         assert ("From another device" in out) == from_another
@@ -306,15 +304,15 @@ def test_hosts(
         assert f"http://{api_host}:{self.PORT}/api" in out
 
     def test_color_on_has_ansi(self, capsys):
-        self._call(bind_host="127.0.0.1", display_host="127.0.0.1", color=True)
+        self._call(bind_host = "127.0.0.1", display_host = "127.0.0.1", color = True)
         assert "\033[" in capsys.readouterr().out
 
     def test_color_off_no_ansi(self, capsys):
-        self._call(bind_host="127.0.0.1", display_host="127.0.0.1", color=False)
+        self._call(bind_host = "127.0.0.1", display_host = "127.0.0.1", color = False)
         assert "\033[" not in capsys.readouterr().out
 
     def test_api_urls_present(self, capsys):
-        self._call(bind_host="0.0.0.0", display_host="10.0.0.1", color=False)
+        self._call(bind_host = "0.0.0.0", display_host = "10.0.0.1", color = False)
         out = capsys.readouterr().out
         assert "/api" in out
         assert "/api/health" in out
@@ -326,7 +324,6 @@ def test_api_urls_present(self, capsys):
 
 
 class TestInstallColorDetect:
-
     @staticmethod
     def _mod():
         import studio.install_python_stack as ips
@@ -335,13 +332,13 @@ def _mod():
 
     def test_no_tty(self):
         ips = self._mod()
-        with patch("sys.stdout.isatty", return_value=False):
+        with patch("sys.stdout.isatty", return_value = False):
             assert ips._stdout_supports_color() is False
 
     def test_tty_linux(self, monkeypatch):
         ips = self._mod()
         monkeypatch.setattr(ips, "IS_WINDOWS", False)
-        with patch("sys.stdout.isatty", return_value=True):
+        with patch("sys.stdout.isatty", return_value = True):
             assert ips._stdout_supports_color() is True
 
     def test_force_color(self, monkeypatch):
@@ -350,13 +347,13 @@ def test_force_color(self, monkeypatch):
 
     def test_no_color(self, monkeypatch):
         monkeypatch.setenv("NO_COLOR", "1")
-        with patch("sys.stdout.isatty", return_value=True):
+        with patch("sys.stdout.isatty", return_value = True):
             assert self._mod()._stdout_supports_color() is False
 
     def test_windows_ctypes_fail(self, monkeypatch):
         ips = self._mod()
         monkeypatch.setattr(ips, "IS_WINDOWS", True)
-        with patch("sys.stdout.isatty", return_value=True):
+        with patch("sys.stdout.isatty", return_value = True):
             # ctypes.windll does not exist on Linux, so the except branch fires
             assert ips._stdout_supports_color() is False
 
@@ -367,7 +364,6 @@ def test_windows_ctypes_fail(self, monkeypatch):
 
 
 class TestInstallStep:
-
     @staticmethod
     def _mod():
         import studio.install_python_stack as ips
@@ -409,7 +405,7 @@ def test_default_green(self, monkeypatch, capsys):
     def test_custom_color_fn(self, monkeypatch, capsys):
         ips = self._mod()
         monkeypatch.setattr(ips, "_HAS_COLOR", True)
-        ips._step("err", "fail", color_fn=ips._red)
+        ips._step("err", "fail", color_fn = ips._red)
         assert "\033[91m" in capsys.readouterr().out
 
     def test_no_color_no_ansi(self, monkeypatch, capsys):
@@ -425,7 +421,6 @@ def test_no_color_no_ansi(self, monkeypatch, capsys):
 
 
 class TestInstallProgress:
-
     @staticmethod
     def _mod():
         import studio.install_python_stack as ips
@@ -479,7 +474,6 @@ def test_bar_format(self, monkeypatch, capsys):
 
 
 class TestBashColorInit:
-
     _SCRIPT = _BASH_COLOR_INIT + '\necho "DIM=${C_DIM}END"'
 
     def test_no_env_empty(self):
@@ -510,7 +504,6 @@ def test_no_color_beats_force(self):
 
 
 class TestBashStep:
-
     _PRE = _BASH_COLOR_INIT + _BASH_HELPERS
 
     def test_step_padding(self):
@@ -541,7 +534,6 @@ def test_substep(self):
 
 
 class TestBashRunQuiet:
-
     _PRE = _BASH_COLOR_INIT + _BASH_HELPERS + _BASH_RUN_QUIET
 
     def test_no_exit_success(self):
@@ -566,13 +558,16 @@ def test_no_exit_failure_shows_error(self):
         assert "log_data" in r.stderr
 
     def test_run_quiet_failure_exits(self):
-        script = self._PRE + """
+        script = (
+            self._PRE
+            + """
 (
     run_quiet "bad" false
     echo "SHOULD_NOT_REACH"
 )
 echo "OUTER=$?"
 """
+        )
         r = _run_bash(script)
         assert "SHOULD_NOT_REACH" not in r.stdout
 
@@ -587,7 +582,6 @@ def test_run_quiet_success(self):
 
 
 class TestBashTryQuiet:
-
     _PRE = _BASH_COLOR_INIT + _BASH_HELPERS + _BASH_TRY_QUIET
 
     def test_success_rc0(self):
@@ -616,9 +610,7 @@ def test_fail_verbose_shows(self):
 
     def test_preserves_exit_code(self):
         r = _run_bash(
-            self._PRE
-            + "\ntry_quiet \"cmd\" bash -c 'exit 42'"
-            + '\necho "RC=$?"'
+            self._PRE + "\ntry_quiet \"cmd\" bash -c 'exit 42'" + '\necho "RC=$?"'
         )
         assert "RC=42" in r.stdout
 
@@ -637,10 +629,9 @@ def test_success_verbose_quiet(self):
 
 
 class TestBashRequestedPython:
-
     _PY_MINOR = sys.version_info.minor
 
-    def _block(self, *, requested="", min_minor=None, max_minor=None):
+    def _block(self, *, requested = "", min_minor = None, max_minor = None):
         if min_minor is None:
             min_minor = self._PY_MINOR - 1
         if max_minor is None:
@@ -652,28 +643,28 @@ def _block(self, *, requested="", min_minor=None, max_minor=None):
         )
 
     def test_unset_skipped(self):
-        r = _run_bash(self._block(requested=""))
+        r = _run_bash(self._block(requested = ""))
         assert r.stdout.strip().endswith("BEST_PY=")
 
     def test_valid_in_range(self):
-        r = _run_bash(self._block(requested=sys.executable))
+        r = _run_bash(self._block(requested = sys.executable))
         assert f"BEST_PY={sys.executable}" in r.stdout
         assert "using requested Python" in r.stdout
 
     def test_nonexistent_skipped(self):
-        r = _run_bash(self._block(requested="/nonexistent/python3.99"))
+        r = _run_bash(self._block(requested = "/nonexistent/python3.99"))
         assert r.stdout.strip().endswith("BEST_PY=")
 
     def test_not_executable_skipped(self):
-        r = _run_bash(self._block(requested="/dev/null"))
+        r = _run_bash(self._block(requested = "/dev/null"))
         assert r.stdout.strip().endswith("BEST_PY=")
 
     def test_out_of_range(self):
         r = _run_bash(
             self._block(
-                requested=sys.executable,
-                min_minor=self._PY_MINOR + 10,
-                max_minor=self._PY_MINOR + 11,
+                requested = sys.executable,
+                min_minor = self._PY_MINOR + 10,
+                max_minor = self._PY_MINOR + 11,
             )
         )
         assert r.stdout.strip().endswith("BEST_PY=")
@@ -688,9 +679,9 @@ def test_out_of_range(self):
 class TestPS1Structural:
     """Static analysis of setup.ps1 -- no PowerShell runtime needed."""
 
-    @pytest.fixture(autouse=True)
+    @pytest.fixture(autouse = True)
     def _load(self):
-        self.content = SETUP_PS1.read_text(encoding="utf-8")
+        self.content = SETUP_PS1.read_text(encoding = "utf-8")
 
     def test_fn_step(self):
         assert re.search(r"^function step\s*\{", self.content, re.MULTILINE)
@@ -727,26 +718,25 @@ def test_env_vars_referenced(self):
 
 
 class TestCLISetup:
-
     def test_verbose_sets_env(self):
         from typer.testing import CliRunner
 
         from unsloth_cli.commands.studio import studio_app
 
         runner = CliRunner()
-        mock_result = MagicMock(returncode=0)
+        mock_result = MagicMock(returncode = 0)
         with (
             patch(
                 "unsloth_cli.commands.studio._find_setup_script",
-                return_value=Path("/f/setup.sh"),
+                return_value = Path("/f/setup.sh"),
             ),
             patch(
                 "unsloth_cli.commands.studio.subprocess.run",
-                return_value=mock_result,
+                return_value = mock_result,
             ) as mock_run,
             patch(
                 "unsloth_cli.commands.studio.platform.system",
-                return_value="Linux",
+                return_value = "Linux",
             ),
         ):
             result = runner.invoke(studio_app, ["setup", "--verbose"])
@@ -761,19 +751,19 @@ def test_no_verbose_env_none(self):
         from unsloth_cli.commands.studio import studio_app
 
         runner = CliRunner()
-        mock_result = MagicMock(returncode=0)
+        mock_result = MagicMock(returncode = 0)
         with (
             patch(
                 "unsloth_cli.commands.studio._find_setup_script",
-                return_value=Path("/f/setup.sh"),
+                return_value = Path("/f/setup.sh"),
             ),
             patch(
                 "unsloth_cli.commands.studio.subprocess.run",
-                return_value=mock_result,
+                return_value = mock_result,
             ) as mock_run,
             patch(
                 "unsloth_cli.commands.studio.platform.system",
-                return_value="Linux",
+                return_value = "Linux",
             ),
         ):
             result = runner.invoke(studio_app, ["setup"])
@@ -789,7 +779,7 @@ def test_script_not_found(self):
         runner = CliRunner()
         with patch(
             "unsloth_cli.commands.studio._find_setup_script",
-            return_value=None,
+            return_value = None,
         ):
             result = runner.invoke(studio_app, ["setup"])
         assert result.exit_code == 1
@@ -801,19 +791,19 @@ def test_nonzero_rc_propagated(self):
         from unsloth_cli.commands.studio import studio_app
 
         runner = CliRunner()
-        mock_result = MagicMock(returncode=42)
+        mock_result = MagicMock(returncode = 42)
         with (
             patch(
                 "unsloth_cli.commands.studio._find_setup_script",
-                return_value=Path("/f/setup.sh"),
+                return_value = Path("/f/setup.sh"),
             ),
             patch(
                 "unsloth_cli.commands.studio.subprocess.run",
-                return_value=mock_result,
+                return_value = mock_result,
             ),
             patch(
                 "unsloth_cli.commands.studio.platform.system",
-                return_value="Linux",
+                return_value = "Linux",
             ),
         ):
             result = runner.invoke(studio_app, ["setup"])

From 28981e5a64017aa4453e67407086ae883af3a7a7 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 24 Mar 2026 10:59:54 +0000
Subject: [PATCH 09/16] Add sandbox integration tests for PR #4494 UX fixes

Simulation harness (tests/simulate_pr4494.py) creates an isolated uv
venv, copies the real source files into it, and runs subprocess tests
for all three fixes with visual before/after demos and edge cases.

Standalone bash test (tests/test_try_quiet.sh) validates try_quiet
stderr redirect across 8 scenarios including broken-version contrast.

39 integration tests total (14 IPv6 + 15 try_quiet + 10 _step), all
existing 75 unit tests still pass.
---
 tests/simulate_pr4494.py | 502 +++++++++++++++++++++++++++++++++++++++
 tests/test_try_quiet.sh  | 159 +++++++++++++
 2 files changed, 661 insertions(+)
 create mode 100644 tests/simulate_pr4494.py
 create mode 100644 tests/test_try_quiet.sh

diff --git a/tests/simulate_pr4494.py b/tests/simulate_pr4494.py
new file mode 100644
index 0000000000..d49e7a4c6c
--- /dev/null
+++ b/tests/simulate_pr4494.py
@@ -0,0 +1,502 @@
+#!/usr/bin/env python3
+"""PR #4494 Simulation Harness -- sandbox integration tests for 3 UX fixes.
+
+Runs in a uv venv sandbox so it cannot break the workspace. Demonstrates
+before-vs-after for each fix and tests edge cases beyond the unit tests.
+
+Fix 1: IPv6 URL bracketing in startup banner
+Fix 2: try_quiet stderr redirect in setup.sh
+Fix 3: _step() label truncation in install_python_stack.py
+
+Usage:
+    python tests/simulate_pr4494.py              # all fixes, full sandbox
+    python tests/simulate_pr4494.py --fix 1      # single fix
+    python tests/simulate_pr4494.py --no-sandbox # use current Python directly
+    python tests/simulate_pr4494.py --keep-sandbox
+    python tests/simulate_pr4494.py --no-bash    # skip bash tests
+    python tests/simulate_pr4494.py --no-color   # disable ANSI colors
+    python tests/simulate_pr4494.py --verbose    # show subprocess output
+"""
+
+from __future__ import annotations
+
+import argparse
+import os
+import platform
+import shutil
+import subprocess
+import sys
+import tempfile
+import textwrap
+from pathlib import Path
+
+IS_WIN = platform.system() == "Windows"
+HAS_BASH = shutil.which("bash") is not None
+REPO = Path(__file__).resolve().parent.parent
+
+
+# ── Color helpers ─────────────────────────────────────────────────────
+
+
+class C:
+    """Thin ANSI color wrapper, toggled by --no-color / NO_COLOR."""
+    on = True
+
+    @staticmethod
+    def g(s: str) -> str:
+        return f"\033[92m{s}\033[0m" if C.on else s
+
+    @staticmethod
+    def r(s: str) -> str:
+        return f"\033[91m{s}\033[0m" if C.on else s
+
+    @staticmethod
+    def d(s: str) -> str:
+        return f"\033[38;5;245m{s}\033[0m" if C.on else s
+
+    @staticmethod
+    def b(s: str) -> str:
+        return f"\033[1m{s}\033[0m" if C.on else s
+
+    @staticmethod
+    def c(s: str) -> str:
+        return f"\033[96m{s}\033[0m" if C.on else s
+
+
+# ── Sandbox lifecycle ─────────────────────────────────────────────────
+
+
+def create_sandbox(verbose: bool) -> tuple[Path, Path]:
+    """Create isolated venv + importable package tree.  Returns (dir, python)."""
+    sdir = Path(tempfile.mkdtemp(prefix="sim_pr4494_"))
+    vdir = sdir / "venv"
+
+    uv = shutil.which("uv")
+    ok = False
+    if uv:
+        r = subprocess.run(
+            [uv, "venv", str(vdir), "--python", sys.executable],
+            capture_output=True, text=True,
+        )
+        ok = r.returncode == 0
+        if not ok and verbose:
+            print(f"  uv venv failed, falling back to stdlib venv: {r.stderr.strip()}")
+    if not ok:
+        subprocess.run(
+            [sys.executable, "-m", "venv", str(vdir)],
+            capture_output=True, text=True, check=True,
+        )
+
+    # Mirror the package layout so `import studio.backend.startup_banner` works
+    pkg = sdir / "studio" / "backend"
+    pkg.mkdir(parents=True)
+    (sdir / "studio" / "__init__.py").touch()
+    (pkg / "__init__.py").touch()
+    shutil.copy2(REPO / "studio" / "backend" / "startup_banner.py", pkg)
+    shutil.copy2(REPO / "studio" / "install_python_stack.py", sdir / "studio")
+
+    py = vdir / "Scripts" / "python.exe" if IS_WIN else vdir / "bin" / "python"
+    return sdir, py
+
+
+def cleanup_sandbox(sdir: Path, keep: bool) -> None:
+    if keep:
+        print(f"  Sandbox kept at: {sdir}")
+    else:
+        shutil.rmtree(sdir, ignore_errors=True)
+
+
+# ── Subprocess runner ─────────────────────────────────────────────────
+
+Result = tuple[str, bool, str]  # (name, passed, detail)
+
+
+def _run_script(python: Path, script: str, verbose: bool) -> list[Result]:
+    """Execute *script* in a subprocess, parse PASS:/FAIL: lines."""
+    try:
+        r = subprocess.run(
+            [str(python), "-c", script],
+            capture_output=True, text=True, timeout=60,
+        )
+    except subprocess.TimeoutExpired:
+        return [("timeout", False, "subprocess timed out after 60s")]
+    except Exception as exc:
+        return [("error", False, str(exc))]
+
+    results: list[Result] = []
+    for line in r.stdout.splitlines():
+        line = line.strip()
+        if line.startswith("PASS:"):
+            results.append((line[5:], True, ""))
+        elif line.startswith("FAIL:"):
+            parts = line[5:].split(":", 1)
+            results.append((parts[0], False, parts[1] if len(parts) > 1 else ""))
+    if verbose and r.stderr:
+        print(C.d("  [stderr] " + r.stderr[:500]))
+    if not results:
+        detail = f"no PASS/FAIL output; rc={r.returncode}"
+        if r.stdout:
+            detail += f"; stdout={r.stdout[:300]}"
+        if r.stderr:
+            detail += f"; stderr={r.stderr[:300]}"
+        return [("parse_error", False, detail)]
+    return results
+
+
+# ══════════════════════════════════════════════════════════════════════
+# Fix 1: IPv6 URL Bracketing
+# ══════════════════════════════════════════════════════════════════════
+
+FIX1_CASES: list[tuple[str, str, int, str, str]] = [
+    # (display_host, bind_host, port, expected_fragment, name)
+    ("2001:db8::1",          "::",      8888, "http://[2001:db8::1]:8888",          "1:std_ipv6"),
+    ("::1",                  "::",      8888, "http://[::1]:8888",                   "2:loopback_v6"),
+    ("2001:0db8:85a3::7334", "::",      8888, "http://[2001:0db8:85a3::7334]:8888", "3:long_ipv6"),
+    ("::ffff:192.168.1.1",   "::",      8888, "http://[::ffff:192.168.1.1]:8888",   "4:v4_mapped"),
+    ("fe80::1%eth0",         "::",      8888, "http://[fe80::1%eth0]:8888",          "5:link_local"),
+    ("192.168.1.100",        "0.0.0.0", 8888, "http://192.168.1.100:8888",          "6:ipv4"),
+    ("10.0.0.1",             "0.0.0.0", 9000, "http://10.0.0.1:9000",               "7:v4_alt_port"),
+    ("myhost.local",         "0.0.0.0", 8888, "http://myhost.local:8888",           "8:hostname"),
+    ("localhost",            "0.0.0.0", 8888, "http://localhost:8888",               "9:localhost"),
+    ("",                     "0.0.0.0", 8888, "http://:8888",                        "10:empty_host"),
+    ("[::1]",                "::",      8888, "http://[[::1]]:8888",                  "11:pre_bracketed"),
+    # Full-banner checks
+    ("2001:db8::1",          "::",      8888, "[2001:db8::1]",                       "B1:banner_v6_ext"),
+    ("192.168.1.5",          "0.0.0.0", 8888, "192.168.1.5",                         "B2:banner_v4_ext"),
+    ("::1",                  "::1",     8888, "[::1]",                                "B3:banner_v6_lo"),
+]
+
+
+def visual_demo_fix1() -> None:
+    print(f"\n  {C.b('Fix 1: IPv6 URL Bracketing')}")
+    print(f"  {C.d(chr(0x2500) * 52)}")
+    hdr = "  {:<34s}{:<30s}{:<30s}"
+    row = "  {:<34s}{:<30s}{:<30s}"
+    print(hdr.format("Host", "BEFORE (broken)", "AFTER (fixed)"))
+    demos = [
+        ("2001:db8::1",        8888),
+        ("::1",                8888),
+        ("::ffff:192.168.1.1", 8888),
+        ("192.168.1.100",      8888),
+        ("myhost.local",       8888),
+    ]
+    for host, port in demos:
+        broken = f"http://{host}:{port}"
+        fixed = f"http://[{host}]:{port}" if ":" in host else f"http://{host}:{port}"
+        tag = "" if broken != fixed else C.d(" (same)")
+        # strip ANSI for width calc in row
+        print(row.format(host, C.r(broken), C.g(fixed) + tag))
+    print()
+
+
+def _fix1_script(sandbox_dir: str) -> str:
+    return textwrap.dedent("""\
+        import sys, io
+        sys.path.insert(0, __SANDBOX__)
+
+        import studio.backend.startup_banner as banner
+        banner.stdout_supports_color = lambda: False
+
+        CASES = __CASES__
+        failed = 0
+        for display_host, bind_host, port, expected, name in CASES:
+            buf = io.StringIO()
+            old = sys.stdout
+            sys.stdout = buf
+            try:
+                banner.print_studio_access_banner(
+                    port=port, bind_host=bind_host, display_host=display_host,
+                )
+            finally:
+                sys.stdout = old
+            output = buf.getvalue()
+            if expected in output:
+                print("PASS:" + name)
+            else:
+                print("FAIL:" + name + ":expected " + repr(expected) + " not found in banner output")
+                failed += 1
+        sys.exit(1 if failed else 0)
+    """).replace("__SANDBOX__", repr(str(sandbox_dir))).replace("__CASES__", repr(FIX1_CASES))
+
+
+def test_fix1(python: Path, sandbox_dir: Path, verbose: bool) -> list[Result]:
+    visual_demo_fix1()
+    return _run_script(python, _fix1_script(str(sandbox_dir)), verbose)
+
+
+# ══════════════════════════════════════════════════════════════════════
+# Fix 2: try_quiet stderr redirect
+# ══════════════════════════════════════════════════════════════════════
+
+
+def visual_demo_fix2() -> None:
+    print(f"\n  {C.b('Fix 2: try_quiet stderr redirect')}")
+    print(f"  {C.d(chr(0x2500) * 52)}")
+    print(f"  {C.r('BEFORE')}: failure log appears on {C.r('STDOUT')} (mixed with step output)")
+    print(f"  {C.g('AFTER')}:  failure log appears on {C.g('STDERR')} (clean stream separation)")
+    print(f"  {C.d('          (step status line still goes to stdout as intended)')}")
+    print()
+
+
+def test_fix2(skip_bash: bool, verbose: bool) -> list[Result]:
+    visual_demo_fix2()
+    if skip_bash or not HAS_BASH:
+        reason = "skipped (--no-bash)" if skip_bash else "skipped (no bash on this platform)"
+        return [
+            ("1:success", True, reason),
+            ("2:fail_quiet", True, reason),
+            ("3:fail_verbose", True, reason),
+            ("4:multiline", True, reason),
+            ("5:exit_code", True, reason),
+            ("6:binary", True, reason),
+            ("7:success_verbose", True, reason),
+            ("8:broken_contrast", True, reason),
+        ]
+
+    script_path = REPO / "tests" / "test_try_quiet.sh"
+    if not script_path.exists():
+        return [("fix2", False, f"{script_path} not found")]
+
+    try:
+        r = subprocess.run(
+            ["bash", str(script_path)],
+            capture_output=True, text=True, timeout=30,
+        )
+    except subprocess.TimeoutExpired:
+        return [("fix2", False, "bash tests timed out")]
+
+    results: list[Result] = []
+    for line in r.stdout.splitlines():
+        line = line.strip()
+        if line.startswith("PASS:"):
+            results.append((line[5:], True, ""))
+        elif line.startswith("FAIL:"):
+            parts = line[5:].split(":", 1)
+            results.append((parts[0], False, parts[1] if len(parts) > 1 else ""))
+
+    if verbose:
+        if r.stderr:
+            print(C.d("  [bash stderr] " + r.stderr[:500]))
+        if r.returncode != 0 and not results:
+            print(C.r(f"  [bash rc={r.returncode}] stdout: {r.stdout[:300]}"))
+
+    if not results:
+        return [("fix2_parse", False, f"no output parsed; rc={r.returncode}")]
+    return results
+
+
+# ══════════════════════════════════════════════════════════════════════
+# Fix 3: _step() label truncation
+# ══════════════════════════════════════════════════════════════════════
+
+_COL = 15  # must match install_python_stack._COL
+
+FIX3_CASES: list[tuple[str, str, str, str]] = [
+    # (label, value, expected_output, name)
+    ("deps",           "ok",  "  deps"           + " " * 11 + "ok",  "1:short_4"),
+    ("error",          "bad", "  error"          + " " * 10 + "bad", "2:short_5"),
+    ("llama-quantize", "ok",  "  llama-quantize" + " " * 1  + "ok",  "3:len_14"),
+    ("",               "v",   "  "               + " " * 15 + "v",   "4:empty"),
+    ("x" * 15,         "v",   "  " + "x" * 15   + "v",              "5:exact_15"),
+    ("x" * 16,         "v",   "  " + "x" * 15   + "v",              "6:trunc_16"),
+    ("x" * 20,         "v",   "  " + "x" * 15   + "v",              "7:trunc_20"),
+    ("x" * 50,         "v",   "  " + "x" * 15   + "v",              "8:trunc_50"),
+    ("toolonglabel!!!!", "v", "  " + "toolonglabel!!!!"[:_COL] + "v", "9:readable"),
+]
+
+# Case 10: broken version contrast -- tested inline, not via real module
+_BROKEN_LABEL = "x" * 20
+_BROKEN_EXPECTED = "  " + _BROKEN_LABEL + "v"  # no truncation, no gap
+
+
+def visual_demo_fix3() -> None:
+    print(f"\n  {C.b('Fix 3: _step() Label Truncation')}")
+    print(f"  {C.d(chr(0x2500) * 52)}")
+    hdr = "  {:<20s}{:<32s}{:<32s}"
+    row = "  {:<20s}{:<32s}{:<32s}"
+    print(hdr.format("Label", "BEFORE (broken)", "AFTER (fixed)"))
+    demos = [
+        ("deps",    "ok"),
+        ("x" * 20,  "v"),
+        ("x" * 50,  "v"),
+    ]
+    for label, val in demos:
+        # Broken: no truncation, negative padding becomes empty
+        bpad = " " * max(0, _COL - len(label))
+        broken_out = f"  {label}{bpad}{val}"
+        # Fixed: truncate label, then pad
+        padded = label[:_COL]
+        fixed_out = f"  {padded}{' ' * (_COL - len(padded))}{val}"
+        tag = C.d(" (same)") if broken_out == fixed_out else C.d(" (truncated!)")
+        disp_label = repr(label) if len(label) <= 16 else f"'{'x' * 5}...' ({len(label)}c)"
+        print(row.format(
+            disp_label,
+            C.r(repr(broken_out)),
+            C.g(repr(fixed_out)) + tag,
+        ))
+    print()
+
+
+def _fix3_script(sandbox_dir: str) -> str:
+    return textwrap.dedent("""\
+        import sys, io
+        sys.path.insert(0, __SANDBOX__)
+
+        import studio.install_python_stack as ips
+        ips._HAS_COLOR = False
+
+        CASES = __CASES__
+        failed = 0
+        for label, value, expected, name in CASES:
+            buf = io.StringIO()
+            old = sys.stdout
+            sys.stdout = buf
+            try:
+                ips._step(label, value)
+            finally:
+                sys.stdout = old
+            actual = buf.getvalue().rstrip("\\n")
+            if actual == expected:
+                print("PASS:" + name)
+            else:
+                print("FAIL:" + name + ":expected " + repr(expected) + " got " + repr(actual))
+                failed += 1
+
+        # Case 10: broken version contrast (inline, not via module)
+        broken_label = "x" * 20
+        broken_padded = broken_label  # no [:_COL] truncation
+        broken_pad = " " * max(0, 15 - len(broken_padded))
+        broken_out = "  " + broken_padded + broken_pad + "v"
+        expected_broken = "  " + "x" * 20 + "v"
+
+        # Also get the fixed output for comparison
+        buf = io.StringIO()
+        old = sys.stdout
+        sys.stdout = buf
+        try:
+            ips._step(broken_label, "v")
+        finally:
+            sys.stdout = old
+        fixed_out = buf.getvalue().rstrip("\\n")
+
+        # The broken output should differ from the fixed output
+        if broken_out == expected_broken and broken_out != fixed_out:
+            print("PASS:10:broken_contrast")
+        else:
+            print("FAIL:10:broken_contrast:broken=" + repr(broken_out) + " fixed=" + repr(fixed_out))
+            failed += 1
+
+        sys.exit(1 if failed else 0)
+    """).replace("__SANDBOX__", repr(str(sandbox_dir))).replace("__CASES__", repr(FIX3_CASES))
+
+
+def test_fix3(python: Path, sandbox_dir: Path, verbose: bool) -> list[Result]:
+    visual_demo_fix3()
+    return _run_script(python, _fix3_script(str(sandbox_dir)), verbose)
+
+
+# ══════════════════════════════════════════════════════════════════════
+# Summary
+# ══════════════════════════════════════════════════════════════════════
+
+FIX_LABELS = {
+    1: "IPv6 URL bracketing",
+    2: "try_quiet stderr redirect",
+    3: "_step label truncation",
+}
+
+
+def print_summary(
+    results_by_fix: dict[int, list[Result]],
+    verbose: bool,
+) -> bool:
+    """Print pass/fail table.  Returns True if all passed."""
+    bar = chr(0x2550) * 56
+    thin = chr(0x2500) * 56
+    print(f"\n  {bar}")
+    print(f"  {C.b('PR #4494 Simulation Results')}")
+    print(f"  {bar}")
+
+    all_pass = True
+    grand_pass = 0
+    grand_total = 0
+
+    for fix_num in sorted(results_by_fix):
+        results = results_by_fix[fix_num]
+        n_pass = sum(1 for _, ok, _ in results if ok)
+        n_total = len(results)
+        grand_pass += n_pass
+        grand_total += n_total
+        ok = n_pass == n_total
+        if not ok:
+            all_pass = False
+        tag = C.g("[PASS]") if ok else C.r("[FAIL]")
+        label = FIX_LABELS.get(fix_num, f"Fix {fix_num}")
+        print(f"  Fix {fix_num}  {label:<34s} {tag}  {n_pass:>2}/{n_total}")
+
+        if verbose or not ok:
+            for name, passed, detail in results:
+                if not passed:
+                    print(f"         {C.r('FAIL')} {name}: {detail}")
+
+    print(f"  {thin}")
+    summary = f"Total: {grand_pass}/{grand_total} passed"
+    if all_pass:
+        print(f"  {C.g(summary)} -- {C.g('ALL PASSED')}")
+    else:
+        print(f"  {C.r(summary)} -- {C.r('SOME FAILED')}")
+    print(f"  {bar}")
+    return all_pass
+
+
+# ══════════════════════════════════════════════════════════════════════
+# Main
+# ══════════════════════════════════════════════════════════════════════
+
+
+def main() -> int:
+    ap = argparse.ArgumentParser(description="PR #4494 simulation harness")
+    ap.add_argument("--fix", type=int, choices=[1, 2, 3], help="run only this fix")
+    ap.add_argument("--no-sandbox", action="store_true", help="skip venv, use current Python")
+    ap.add_argument("--keep-sandbox", action="store_true", help="do not delete sandbox on exit")
+    ap.add_argument("--no-bash", action="store_true", help="skip bash tests (Fix 2)")
+    ap.add_argument("--no-color", action="store_true", help="disable ANSI colors")
+    ap.add_argument("--verbose", action="store_true", help="show subprocess output")
+    args = ap.parse_args()
+
+    if args.no_color or os.environ.get("NO_COLOR", "").strip():
+        C.on = False
+
+    fixes = [args.fix] if args.fix else [1, 2, 3]
+
+    # Sandbox setup
+    sandbox_dir: Path | None = None
+    if args.no_sandbox:
+        # Use current Python, put source on sys.path directly
+        sandbox_dir = REPO
+        python = Path(sys.executable)
+        print(C.d(f"  (no-sandbox mode, using {python})"))
+    else:
+        print(C.d("  Creating sandbox..."), end="", flush=True)
+        sandbox_dir, python = create_sandbox(args.verbose)
+        print(C.d(f" {sandbox_dir}"))
+
+    results_by_fix: dict[int, list[Result]] = {}
+
+    try:
+        if 1 in fixes:
+            results_by_fix[1] = test_fix1(python, sandbox_dir, args.verbose)
+        if 2 in fixes:
+            results_by_fix[2] = test_fix2(args.no_bash, args.verbose)
+        if 3 in fixes:
+            results_by_fix[3] = test_fix3(python, sandbox_dir, args.verbose)
+    finally:
+        if not args.no_sandbox and sandbox_dir and sandbox_dir != REPO:
+            cleanup_sandbox(sandbox_dir, args.keep_sandbox)
+
+    all_pass = print_summary(results_by_fix, args.verbose)
+    return 0 if all_pass else 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())
diff --git a/tests/test_try_quiet.sh b/tests/test_try_quiet.sh
new file mode 100644
index 0000000000..b2b295f558
--- /dev/null
+++ b/tests/test_try_quiet.sh
@@ -0,0 +1,159 @@
+#!/usr/bin/env bash
+# test_try_quiet.sh -- Standalone tests for try_quiet stderr redirect (Fix 2, PR #4494)
+#
+# Usage:  bash tests/test_try_quiet.sh
+#
+# Tests that the fixed try_quiet sends failure logs to stderr (not stdout),
+# and contrasts with the broken pre-fix behavior.
+set -uo pipefail
+
+PASS=0
+FAIL=0
+
+# ── Helpers ───────────────────────────────────────────────────────────
+
+# Minimal step() matching setup.sh signature (stdout, no color)
+step() {
+    local _label="$1" _value="$2"
+    echo "  $_label: $_value"
+}
+
+# Fixed try_quiet (PR #4494)
+try_quiet_fixed() {
+    local label="$1"; shift
+    local tmplog; tmplog=$(mktemp)
+    if "$@" > "$tmplog" 2>&1; then
+        rm -f "$tmplog"
+        return 0
+    else
+        local exit_code=$?
+        if [ "${UNSLOTH_VERBOSE:-0}" = "1" ]; then
+            step "error" "$label failed (exit code $exit_code)"
+            cat "$tmplog" >&2
+        fi
+        rm -f "$tmplog"
+        return $exit_code
+    fi
+}
+
+# Broken try_quiet (pre-fix: log dumps to stdout)
+try_quiet_broken() {
+    local label="$1"; shift
+    local tmplog; tmplog=$(mktemp)
+    if "$@" > "$tmplog" 2>&1; then
+        rm -f "$tmplog"
+        return 0
+    else
+        local exit_code=$?
+        if [ "${UNSLOTH_VERBOSE:-0}" = "1" ]; then
+            step "error" "$label failed (exit code $exit_code)"
+            cat "$tmplog"   # BUG: no >&2, leaks to stdout
+        fi
+        rm -f "$tmplog"
+        return $exit_code
+    fi
+}
+
+assert_eq() {
+    local name="$1" expected="$2" actual="$3"
+    if [ "$expected" = "$actual" ]; then
+        PASS=$((PASS + 1))
+        echo "PASS:$name"
+    else
+        FAIL=$((FAIL + 1))
+        echo "FAIL:$name:expected '$(echo "$expected" | head -c 60)' got '$(echo "$actual" | head -c 60)'"
+    fi
+}
+
+assert_contains() {
+    local name="$1" haystack="$2" needle="$3"
+    if printf '%s' "$haystack" | grep -qF "$needle"; then
+        PASS=$((PASS + 1))
+        echo "PASS:$name"
+    else
+        FAIL=$((FAIL + 1))
+        echo "FAIL:$name:should contain '$needle'"
+    fi
+}
+
+assert_not_contains() {
+    local name="$1" haystack="$2" needle="$3"
+    if printf '%s' "$haystack" | grep -qF "$needle"; then
+        FAIL=$((FAIL + 1))
+        echo "FAIL:$name:should NOT contain '$needle'"
+    else
+        PASS=$((PASS + 1))
+        echo "PASS:$name"
+    fi
+}
+
+tmpout=$(mktemp)
+tmperr=$(mktemp)
+trap 'rm -f "$tmpout" "$tmperr"' EXIT
+
+# ── Test 1: Success -- no output anywhere ─────────────────────────────
+rc=0
+UNSLOTH_VERBOSE=0 try_quiet_fixed "test" true > "$tmpout" 2> "$tmperr" || rc=$?
+assert_eq "1:success_rc" "0" "$rc"
+
+# ── Test 2: Failure, VERBOSE=0 -- silent ──────────────────────────────
+rc=0
+UNSLOTH_VERBOSE=0 try_quiet_fixed "test" false > "$tmpout" 2> "$tmperr" || rc=$?
+stdout_content=$(cat "$tmpout")
+stderr_content=$(cat "$tmperr")
+assert_eq "2:fail_quiet_stdout" "" "$stdout_content"
+assert_eq "2:fail_quiet_stderr" "" "$stderr_content"
+
+# ── Test 3: Failure, VERBOSE=1 -- log on stderr only ─────────────────
+rc=0
+UNSLOTH_VERBOSE=1 try_quiet_fixed "test" bash -c "echo logline_here; exit 1" > "$tmpout" 2> "$tmperr" || rc=$?
+stdout_content=$(cat "$tmpout")
+stderr_content=$(cat "$tmperr")
+assert_contains    "3:fail_verbose_status"  "$stdout_content" "failed"
+assert_contains    "3:fail_verbose_stderr"  "$stderr_content" "logline_here"
+assert_not_contains "3:fail_verbose_clean"  "$stdout_content" "logline_here"
+
+# ── Test 4: Multi-line failure, VERBOSE=1 -- all lines on stderr ──────
+rc=0
+UNSLOTH_VERBOSE=1 try_quiet_fixed "multi" bash -c "echo line_A; echo line_B; echo line_C; exit 1" > "$tmpout" 2> "$tmperr" || rc=$?
+stdout_content=$(cat "$tmpout")
+stderr_content=$(cat "$tmperr")
+assert_contains     "4:multi_stderr_A" "$stderr_content" "line_A"
+assert_contains     "4:multi_stderr_B" "$stderr_content" "line_B"
+assert_contains     "4:multi_stderr_C" "$stderr_content" "line_C"
+assert_not_contains "4:multi_stdout_A" "$stdout_content" "line_A"
+
+# ── Test 5: Exit code preservation ────────────────────────────────────
+rc=0
+UNSLOTH_VERBOSE=0 try_quiet_fixed "test" bash -c "exit 42" > "$tmpout" 2> "$tmperr" || rc=$?
+assert_eq "5:exit_code" "42" "$rc"
+
+# ── Test 6: Binary output in tmplog -- no crash ──────────────────────
+rc=0
+UNSLOTH_VERBOSE=1 try_quiet_fixed "bintest" bash -c "printf '\\x00\\xff\\xfe'; exit 3" > "$tmpout" 2> "$tmperr" || rc=$?
+assert_eq "6:binary_rc" "3" "$rc"
+
+# ── Test 7: Success, VERBOSE=1 -- no error output ────────────────────
+rc=0
+UNSLOTH_VERBOSE=1 try_quiet_fixed "test" true > "$tmpout" 2> "$tmperr" || rc=$?
+stdout_content=$(cat "$tmpout")
+stderr_content=$(cat "$tmperr")
+assert_eq "7:success_verbose_stdout" "" "$stdout_content"
+assert_eq "7:success_verbose_stderr" "" "$stderr_content"
+
+# ── Test 8: Broken version contrast -- log leaks to stdout ───────────
+rc=0
+UNSLOTH_VERBOSE=1 try_quiet_broken "test" bash -c "echo leaked_data; exit 1" > "$tmpout" 2> "$tmperr" || rc=$?
+stdout_content=$(cat "$tmpout")
+assert_contains "8:broken_leaks" "$stdout_content" "leaked_data"
+
+# ── Summary ──────────────────────────────────────────────────────────
+TOTAL=$((PASS + FAIL))
+echo ""
+if [ "$FAIL" -eq 0 ]; then
+    echo "test_try_quiet.sh: $PASS/$TOTAL passed -- ALL PASSED"
+    exit 0
+else
+    echo "test_try_quiet.sh: $PASS/$TOTAL passed, $FAIL FAILED"
+    exit 1
+fi

From 86490e77c9772b47798628b3598f752d95507de0 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 24 Mar 2026 11:00:30 +0000
Subject: [PATCH 10/16] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 tests/simulate_pr4494.py | 156 +++++++++++++++++++++++++--------------
 1 file changed, 99 insertions(+), 57 deletions(-)

diff --git a/tests/simulate_pr4494.py b/tests/simulate_pr4494.py
index d49e7a4c6c..db267aee3f 100644
--- a/tests/simulate_pr4494.py
+++ b/tests/simulate_pr4494.py
@@ -40,6 +40,7 @@
 
 class C:
     """Thin ANSI color wrapper, toggled by --no-color / NO_COLOR."""
+
     on = True
 
     @staticmethod
@@ -68,7 +69,7 @@ def c(s: str) -> str:
 
 def create_sandbox(verbose: bool) -> tuple[Path, Path]:
     """Create isolated venv + importable package tree.  Returns (dir, python)."""
-    sdir = Path(tempfile.mkdtemp(prefix="sim_pr4494_"))
+    sdir = Path(tempfile.mkdtemp(prefix = "sim_pr4494_"))
     vdir = sdir / "venv"
 
     uv = shutil.which("uv")
@@ -76,7 +77,8 @@ def create_sandbox(verbose: bool) -> tuple[Path, Path]:
     if uv:
         r = subprocess.run(
             [uv, "venv", str(vdir), "--python", sys.executable],
-            capture_output=True, text=True,
+            capture_output = True,
+            text = True,
         )
         ok = r.returncode == 0
         if not ok and verbose:
@@ -84,12 +86,14 @@ def create_sandbox(verbose: bool) -> tuple[Path, Path]:
     if not ok:
         subprocess.run(
             [sys.executable, "-m", "venv", str(vdir)],
-            capture_output=True, text=True, check=True,
+            capture_output = True,
+            text = True,
+            check = True,
         )
 
     # Mirror the package layout so `import studio.backend.startup_banner` works
     pkg = sdir / "studio" / "backend"
-    pkg.mkdir(parents=True)
+    pkg.mkdir(parents = True)
     (sdir / "studio" / "__init__.py").touch()
     (pkg / "__init__.py").touch()
     shutil.copy2(REPO / "studio" / "backend" / "startup_banner.py", pkg)
@@ -103,7 +107,7 @@ def cleanup_sandbox(sdir: Path, keep: bool) -> None:
     if keep:
         print(f"  Sandbox kept at: {sdir}")
     else:
-        shutil.rmtree(sdir, ignore_errors=True)
+        shutil.rmtree(sdir, ignore_errors = True)
 
 
 # ── Subprocess runner ─────────────────────────────────────────────────
@@ -116,7 +120,9 @@ def _run_script(python: Path, script: str, verbose: bool) -> list[Result]:
     try:
         r = subprocess.run(
             [str(python), "-c", script],
-            capture_output=True, text=True, timeout=60,
+            capture_output = True,
+            text = True,
+            timeout = 60,
         )
     except subprocess.TimeoutExpired:
         return [("timeout", False, "subprocess timed out after 60s")]
@@ -149,21 +155,33 @@ def _run_script(python: Path, script: str, verbose: bool) -> list[Result]:
 
 FIX1_CASES: list[tuple[str, str, int, str, str]] = [
     # (display_host, bind_host, port, expected_fragment, name)
-    ("2001:db8::1",          "::",      8888, "http://[2001:db8::1]:8888",          "1:std_ipv6"),
-    ("::1",                  "::",      8888, "http://[::1]:8888",                   "2:loopback_v6"),
-    ("2001:0db8:85a3::7334", "::",      8888, "http://[2001:0db8:85a3::7334]:8888", "3:long_ipv6"),
-    ("::ffff:192.168.1.1",   "::",      8888, "http://[::ffff:192.168.1.1]:8888",   "4:v4_mapped"),
-    ("fe80::1%eth0",         "::",      8888, "http://[fe80::1%eth0]:8888",          "5:link_local"),
-    ("192.168.1.100",        "0.0.0.0", 8888, "http://192.168.1.100:8888",          "6:ipv4"),
-    ("10.0.0.1",             "0.0.0.0", 9000, "http://10.0.0.1:9000",               "7:v4_alt_port"),
-    ("myhost.local",         "0.0.0.0", 8888, "http://myhost.local:8888",           "8:hostname"),
-    ("localhost",            "0.0.0.0", 8888, "http://localhost:8888",               "9:localhost"),
-    ("",                     "0.0.0.0", 8888, "http://:8888",                        "10:empty_host"),
-    ("[::1]",                "::",      8888, "http://[[::1]]:8888",                  "11:pre_bracketed"),
+    ("2001:db8::1", "::", 8888, "http://[2001:db8::1]:8888", "1:std_ipv6"),
+    ("::1", "::", 8888, "http://[::1]:8888", "2:loopback_v6"),
+    (
+        "2001:0db8:85a3::7334",
+        "::",
+        8888,
+        "http://[2001:0db8:85a3::7334]:8888",
+        "3:long_ipv6",
+    ),
+    (
+        "::ffff:192.168.1.1",
+        "::",
+        8888,
+        "http://[::ffff:192.168.1.1]:8888",
+        "4:v4_mapped",
+    ),
+    ("fe80::1%eth0", "::", 8888, "http://[fe80::1%eth0]:8888", "5:link_local"),
+    ("192.168.1.100", "0.0.0.0", 8888, "http://192.168.1.100:8888", "6:ipv4"),
+    ("10.0.0.1", "0.0.0.0", 9000, "http://10.0.0.1:9000", "7:v4_alt_port"),
+    ("myhost.local", "0.0.0.0", 8888, "http://myhost.local:8888", "8:hostname"),
+    ("localhost", "0.0.0.0", 8888, "http://localhost:8888", "9:localhost"),
+    ("", "0.0.0.0", 8888, "http://:8888", "10:empty_host"),
+    ("[::1]", "::", 8888, "http://[[::1]]:8888", "11:pre_bracketed"),
     # Full-banner checks
-    ("2001:db8::1",          "::",      8888, "[2001:db8::1]",                       "B1:banner_v6_ext"),
-    ("192.168.1.5",          "0.0.0.0", 8888, "192.168.1.5",                         "B2:banner_v4_ext"),
-    ("::1",                  "::1",     8888, "[::1]",                                "B3:banner_v6_lo"),
+    ("2001:db8::1", "::", 8888, "[2001:db8::1]", "B1:banner_v6_ext"),
+    ("192.168.1.5", "0.0.0.0", 8888, "192.168.1.5", "B2:banner_v4_ext"),
+    ("::1", "::1", 8888, "[::1]", "B3:banner_v6_lo"),
 ]
 
 
@@ -174,11 +192,11 @@ def visual_demo_fix1() -> None:
     row = "  {:<34s}{:<30s}{:<30s}"
     print(hdr.format("Host", "BEFORE (broken)", "AFTER (fixed)"))
     demos = [
-        ("2001:db8::1",        8888),
-        ("::1",                8888),
+        ("2001:db8::1", 8888),
+        ("::1", 8888),
         ("::ffff:192.168.1.1", 8888),
-        ("192.168.1.100",      8888),
-        ("myhost.local",       8888),
+        ("192.168.1.100", 8888),
+        ("myhost.local", 8888),
     ]
     for host, port in demos:
         broken = f"http://{host}:{port}"
@@ -190,7 +208,8 @@ def visual_demo_fix1() -> None:
 
 
 def _fix1_script(sandbox_dir: str) -> str:
-    return textwrap.dedent("""\
+    return (
+        textwrap.dedent("""\
         import sys, io
         sys.path.insert(0, __SANDBOX__)
 
@@ -216,7 +235,10 @@ def _fix1_script(sandbox_dir: str) -> str:
                 print("FAIL:" + name + ":expected " + repr(expected) + " not found in banner output")
                 failed += 1
         sys.exit(1 if failed else 0)
-    """).replace("__SANDBOX__", repr(str(sandbox_dir))).replace("__CASES__", repr(FIX1_CASES))
+    """)
+        .replace("__SANDBOX__", repr(str(sandbox_dir)))
+        .replace("__CASES__", repr(FIX1_CASES))
+    )
 
 
 def test_fix1(python: Path, sandbox_dir: Path, verbose: bool) -> list[Result]:
@@ -232,8 +254,12 @@ def test_fix1(python: Path, sandbox_dir: Path, verbose: bool) -> list[Result]:
 def visual_demo_fix2() -> None:
     print(f"\n  {C.b('Fix 2: try_quiet stderr redirect')}")
     print(f"  {C.d(chr(0x2500) * 52)}")
-    print(f"  {C.r('BEFORE')}: failure log appears on {C.r('STDOUT')} (mixed with step output)")
-    print(f"  {C.g('AFTER')}:  failure log appears on {C.g('STDERR')} (clean stream separation)")
+    print(
+        f"  {C.r('BEFORE')}: failure log appears on {C.r('STDOUT')} (mixed with step output)"
+    )
+    print(
+        f"  {C.g('AFTER')}:  failure log appears on {C.g('STDERR')} (clean stream separation)"
+    )
     print(f"  {C.d('          (step status line still goes to stdout as intended)')}")
     print()
 
@@ -241,7 +267,9 @@ def visual_demo_fix2() -> None:
 def test_fix2(skip_bash: bool, verbose: bool) -> list[Result]:
     visual_demo_fix2()
     if skip_bash or not HAS_BASH:
-        reason = "skipped (--no-bash)" if skip_bash else "skipped (no bash on this platform)"
+        reason = (
+            "skipped (--no-bash)" if skip_bash else "skipped (no bash on this platform)"
+        )
         return [
             ("1:success", True, reason),
             ("2:fail_quiet", True, reason),
@@ -260,7 +288,9 @@ def test_fix2(skip_bash: bool, verbose: bool) -> list[Result]:
     try:
         r = subprocess.run(
             ["bash", str(script_path)],
-            capture_output=True, text=True, timeout=30,
+            capture_output = True,
+            text = True,
+            timeout = 30,
         )
     except subprocess.TimeoutExpired:
         return [("fix2", False, "bash tests timed out")]
@@ -293,14 +323,14 @@ def test_fix2(skip_bash: bool, verbose: bool) -> list[Result]:
 
 FIX3_CASES: list[tuple[str, str, str, str]] = [
     # (label, value, expected_output, name)
-    ("deps",           "ok",  "  deps"           + " " * 11 + "ok",  "1:short_4"),
-    ("error",          "bad", "  error"          + " " * 10 + "bad", "2:short_5"),
-    ("llama-quantize", "ok",  "  llama-quantize" + " " * 1  + "ok",  "3:len_14"),
-    ("",               "v",   "  "               + " " * 15 + "v",   "4:empty"),
-    ("x" * 15,         "v",   "  " + "x" * 15   + "v",              "5:exact_15"),
-    ("x" * 16,         "v",   "  " + "x" * 15   + "v",              "6:trunc_16"),
-    ("x" * 20,         "v",   "  " + "x" * 15   + "v",              "7:trunc_20"),
-    ("x" * 50,         "v",   "  " + "x" * 15   + "v",              "8:trunc_50"),
+    ("deps", "ok", "  deps" + " " * 11 + "ok", "1:short_4"),
+    ("error", "bad", "  error" + " " * 10 + "bad", "2:short_5"),
+    ("llama-quantize", "ok", "  llama-quantize" + " " * 1 + "ok", "3:len_14"),
+    ("", "v", "  " + " " * 15 + "v", "4:empty"),
+    ("x" * 15, "v", "  " + "x" * 15 + "v", "5:exact_15"),
+    ("x" * 16, "v", "  " + "x" * 15 + "v", "6:trunc_16"),
+    ("x" * 20, "v", "  " + "x" * 15 + "v", "7:trunc_20"),
+    ("x" * 50, "v", "  " + "x" * 15 + "v", "8:trunc_50"),
     ("toolonglabel!!!!", "v", "  " + "toolonglabel!!!!"[:_COL] + "v", "9:readable"),
 ]
 
@@ -316,9 +346,9 @@ def visual_demo_fix3() -> None:
     row = "  {:<20s}{:<32s}{:<32s}"
     print(hdr.format("Label", "BEFORE (broken)", "AFTER (fixed)"))
     demos = [
-        ("deps",    "ok"),
-        ("x" * 20,  "v"),
-        ("x" * 50,  "v"),
+        ("deps", "ok"),
+        ("x" * 20, "v"),
+        ("x" * 50, "v"),
     ]
     for label, val in demos:
         # Broken: no truncation, negative padding becomes empty
@@ -328,17 +358,22 @@ def visual_demo_fix3() -> None:
         padded = label[:_COL]
         fixed_out = f"  {padded}{' ' * (_COL - len(padded))}{val}"
         tag = C.d(" (same)") if broken_out == fixed_out else C.d(" (truncated!)")
-        disp_label = repr(label) if len(label) <= 16 else f"'{'x' * 5}...' ({len(label)}c)"
-        print(row.format(
-            disp_label,
-            C.r(repr(broken_out)),
-            C.g(repr(fixed_out)) + tag,
-        ))
+        disp_label = (
+            repr(label) if len(label) <= 16 else f"'{'x' * 5}...' ({len(label)}c)"
+        )
+        print(
+            row.format(
+                disp_label,
+                C.r(repr(broken_out)),
+                C.g(repr(fixed_out)) + tag,
+            )
+        )
     print()
 
 
 def _fix3_script(sandbox_dir: str) -> str:
-    return textwrap.dedent("""\
+    return (
+        textwrap.dedent("""\
         import sys, io
         sys.path.insert(0, __SANDBOX__)
 
@@ -387,7 +422,10 @@ def _fix3_script(sandbox_dir: str) -> str:
             failed += 1
 
         sys.exit(1 if failed else 0)
-    """).replace("__SANDBOX__", repr(str(sandbox_dir))).replace("__CASES__", repr(FIX3_CASES))
+    """)
+        .replace("__SANDBOX__", repr(str(sandbox_dir)))
+        .replace("__CASES__", repr(FIX3_CASES))
+    )
 
 
 def test_fix3(python: Path, sandbox_dir: Path, verbose: bool) -> list[Result]:
@@ -455,13 +493,17 @@ def print_summary(
 
 
 def main() -> int:
-    ap = argparse.ArgumentParser(description="PR #4494 simulation harness")
-    ap.add_argument("--fix", type=int, choices=[1, 2, 3], help="run only this fix")
-    ap.add_argument("--no-sandbox", action="store_true", help="skip venv, use current Python")
-    ap.add_argument("--keep-sandbox", action="store_true", help="do not delete sandbox on exit")
-    ap.add_argument("--no-bash", action="store_true", help="skip bash tests (Fix 2)")
-    ap.add_argument("--no-color", action="store_true", help="disable ANSI colors")
-    ap.add_argument("--verbose", action="store_true", help="show subprocess output")
+    ap = argparse.ArgumentParser(description = "PR #4494 simulation harness")
+    ap.add_argument("--fix", type = int, choices = [1, 2, 3], help = "run only this fix")
+    ap.add_argument(
+        "--no-sandbox", action = "store_true", help = "skip venv, use current Python"
+    )
+    ap.add_argument(
+        "--keep-sandbox", action = "store_true", help = "do not delete sandbox on exit"
+    )
+    ap.add_argument("--no-bash", action = "store_true", help = "skip bash tests (Fix 2)")
+    ap.add_argument("--no-color", action = "store_true", help = "disable ANSI colors")
+    ap.add_argument("--verbose", action = "store_true", help = "show subprocess output")
     args = ap.parse_args()
 
     if args.no_color or os.environ.get("NO_COLOR", "").strip():
@@ -477,7 +519,7 @@ def main() -> int:
         python = Path(sys.executable)
         print(C.d(f"  (no-sandbox mode, using {python})"))
     else:
-        print(C.d("  Creating sandbox..."), end="", flush=True)
+        print(C.d("  Creating sandbox..."), end = "", flush = True)
         sandbox_dir, python = create_sandbox(args.verbose)
         print(C.d(f" {sandbox_dir}"))
 

From 630d6a931d5fabc75e5e4710813f8035a60b5a7a Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 24 Mar 2026 12:45:21 +0000
Subject: [PATCH 11/16] Truncate step() labels in setup.sh to match PS1 and
 Python

The %-15s printf format pads short labels but does not truncate long
ones.  Change to %-15.15s so labels wider than 15 chars are clipped,
matching the PowerShell .Substring(0,15) and Python label[:15] logic.
---
 studio/setup.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/studio/setup.sh b/studio/setup.sh
index b795be1530..7c8c4960fd 100755
--- a/studio/setup.sh
+++ b/studio/setup.sh
@@ -25,7 +25,7 @@ fi
 # ── Output helpers ──
 # Consistent column layout: 2-space indent, 15-char label (fits llama-quantize), then value.
 # Usage: step <label> <message> [color]   (color defaults to C_OK)
-step()    { printf "  ${C_DIM}%-15s${C_RST}${3:-$C_OK}%s${C_RST}\n" "$1" "$2"; }
+step()    { printf "  ${C_DIM}%-15.15s${C_RST}${3:-$C_OK}%s${C_RST}\n" "$1" "$2"; }
 substep() { printf "  ${C_DIM}%-15s%s${C_RST}\n" "" "$1"; }
 
 # ── Helper: run command quietly, show output only on failure ──

From 331edaea2b3535a89ffc660ca7b7603989f68e82 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 25 Mar 2026 03:36:11 +0000
Subject: [PATCH 12/16] Remove sandbox integration tests from PR

These test files are not part of the styling fix and should not
ship with this PR.
---
 tests/simulate_pr4494.py           | 544 -------------------
 tests/test_studio_setup_styling.py | 810 -----------------------------
 tests/test_try_quiet.sh            | 159 ------
 3 files changed, 1513 deletions(-)
 delete mode 100644 tests/simulate_pr4494.py
 delete mode 100644 tests/test_studio_setup_styling.py
 delete mode 100644 tests/test_try_quiet.sh

diff --git a/tests/simulate_pr4494.py b/tests/simulate_pr4494.py
deleted file mode 100644
index db267aee3f..0000000000
--- a/tests/simulate_pr4494.py
+++ /dev/null
@@ -1,544 +0,0 @@
-#!/usr/bin/env python3
-"""PR #4494 Simulation Harness -- sandbox integration tests for 3 UX fixes.
-
-Runs in a uv venv sandbox so it cannot break the workspace. Demonstrates
-before-vs-after for each fix and tests edge cases beyond the unit tests.
-
-Fix 1: IPv6 URL bracketing in startup banner
-Fix 2: try_quiet stderr redirect in setup.sh
-Fix 3: _step() label truncation in install_python_stack.py
-
-Usage:
-    python tests/simulate_pr4494.py              # all fixes, full sandbox
-    python tests/simulate_pr4494.py --fix 1      # single fix
-    python tests/simulate_pr4494.py --no-sandbox # use current Python directly
-    python tests/simulate_pr4494.py --keep-sandbox
-    python tests/simulate_pr4494.py --no-bash    # skip bash tests
-    python tests/simulate_pr4494.py --no-color   # disable ANSI colors
-    python tests/simulate_pr4494.py --verbose    # show subprocess output
-"""
-
-from __future__ import annotations
-
-import argparse
-import os
-import platform
-import shutil
-import subprocess
-import sys
-import tempfile
-import textwrap
-from pathlib import Path
-
-IS_WIN = platform.system() == "Windows"
-HAS_BASH = shutil.which("bash") is not None
-REPO = Path(__file__).resolve().parent.parent
-
-
-# ── Color helpers ─────────────────────────────────────────────────────
-
-
-class C:
-    """Thin ANSI color wrapper, toggled by --no-color / NO_COLOR."""
-
-    on = True
-
-    @staticmethod
-    def g(s: str) -> str:
-        return f"\033[92m{s}\033[0m" if C.on else s
-
-    @staticmethod
-    def r(s: str) -> str:
-        return f"\033[91m{s}\033[0m" if C.on else s
-
-    @staticmethod
-    def d(s: str) -> str:
-        return f"\033[38;5;245m{s}\033[0m" if C.on else s
-
-    @staticmethod
-    def b(s: str) -> str:
-        return f"\033[1m{s}\033[0m" if C.on else s
-
-    @staticmethod
-    def c(s: str) -> str:
-        return f"\033[96m{s}\033[0m" if C.on else s
-
-
-# ── Sandbox lifecycle ─────────────────────────────────────────────────
-
-
-def create_sandbox(verbose: bool) -> tuple[Path, Path]:
-    """Create isolated venv + importable package tree.  Returns (dir, python)."""
-    sdir = Path(tempfile.mkdtemp(prefix = "sim_pr4494_"))
-    vdir = sdir / "venv"
-
-    uv = shutil.which("uv")
-    ok = False
-    if uv:
-        r = subprocess.run(
-            [uv, "venv", str(vdir), "--python", sys.executable],
-            capture_output = True,
-            text = True,
-        )
-        ok = r.returncode == 0
-        if not ok and verbose:
-            print(f"  uv venv failed, falling back to stdlib venv: {r.stderr.strip()}")
-    if not ok:
-        subprocess.run(
-            [sys.executable, "-m", "venv", str(vdir)],
-            capture_output = True,
-            text = True,
-            check = True,
-        )
-
-    # Mirror the package layout so `import studio.backend.startup_banner` works
-    pkg = sdir / "studio" / "backend"
-    pkg.mkdir(parents = True)
-    (sdir / "studio" / "__init__.py").touch()
-    (pkg / "__init__.py").touch()
-    shutil.copy2(REPO / "studio" / "backend" / "startup_banner.py", pkg)
-    shutil.copy2(REPO / "studio" / "install_python_stack.py", sdir / "studio")
-
-    py = vdir / "Scripts" / "python.exe" if IS_WIN else vdir / "bin" / "python"
-    return sdir, py
-
-
-def cleanup_sandbox(sdir: Path, keep: bool) -> None:
-    if keep:
-        print(f"  Sandbox kept at: {sdir}")
-    else:
-        shutil.rmtree(sdir, ignore_errors = True)
-
-
-# ── Subprocess runner ─────────────────────────────────────────────────
-
-Result = tuple[str, bool, str]  # (name, passed, detail)
-
-
-def _run_script(python: Path, script: str, verbose: bool) -> list[Result]:
-    """Execute *script* in a subprocess, parse PASS:/FAIL: lines."""
-    try:
-        r = subprocess.run(
-            [str(python), "-c", script],
-            capture_output = True,
-            text = True,
-            timeout = 60,
-        )
-    except subprocess.TimeoutExpired:
-        return [("timeout", False, "subprocess timed out after 60s")]
-    except Exception as exc:
-        return [("error", False, str(exc))]
-
-    results: list[Result] = []
-    for line in r.stdout.splitlines():
-        line = line.strip()
-        if line.startswith("PASS:"):
-            results.append((line[5:], True, ""))
-        elif line.startswith("FAIL:"):
-            parts = line[5:].split(":", 1)
-            results.append((parts[0], False, parts[1] if len(parts) > 1 else ""))
-    if verbose and r.stderr:
-        print(C.d("  [stderr] " + r.stderr[:500]))
-    if not results:
-        detail = f"no PASS/FAIL output; rc={r.returncode}"
-        if r.stdout:
-            detail += f"; stdout={r.stdout[:300]}"
-        if r.stderr:
-            detail += f"; stderr={r.stderr[:300]}"
-        return [("parse_error", False, detail)]
-    return results
-
-
-# ══════════════════════════════════════════════════════════════════════
-# Fix 1: IPv6 URL Bracketing
-# ══════════════════════════════════════════════════════════════════════
-
-FIX1_CASES: list[tuple[str, str, int, str, str]] = [
-    # (display_host, bind_host, port, expected_fragment, name)
-    ("2001:db8::1", "::", 8888, "http://[2001:db8::1]:8888", "1:std_ipv6"),
-    ("::1", "::", 8888, "http://[::1]:8888", "2:loopback_v6"),
-    (
-        "2001:0db8:85a3::7334",
-        "::",
-        8888,
-        "http://[2001:0db8:85a3::7334]:8888",
-        "3:long_ipv6",
-    ),
-    (
-        "::ffff:192.168.1.1",
-        "::",
-        8888,
-        "http://[::ffff:192.168.1.1]:8888",
-        "4:v4_mapped",
-    ),
-    ("fe80::1%eth0", "::", 8888, "http://[fe80::1%eth0]:8888", "5:link_local"),
-    ("192.168.1.100", "0.0.0.0", 8888, "http://192.168.1.100:8888", "6:ipv4"),
-    ("10.0.0.1", "0.0.0.0", 9000, "http://10.0.0.1:9000", "7:v4_alt_port"),
-    ("myhost.local", "0.0.0.0", 8888, "http://myhost.local:8888", "8:hostname"),
-    ("localhost", "0.0.0.0", 8888, "http://localhost:8888", "9:localhost"),
-    ("", "0.0.0.0", 8888, "http://:8888", "10:empty_host"),
-    ("[::1]", "::", 8888, "http://[[::1]]:8888", "11:pre_bracketed"),
-    # Full-banner checks
-    ("2001:db8::1", "::", 8888, "[2001:db8::1]", "B1:banner_v6_ext"),
-    ("192.168.1.5", "0.0.0.0", 8888, "192.168.1.5", "B2:banner_v4_ext"),
-    ("::1", "::1", 8888, "[::1]", "B3:banner_v6_lo"),
-]
-
-
-def visual_demo_fix1() -> None:
-    print(f"\n  {C.b('Fix 1: IPv6 URL Bracketing')}")
-    print(f"  {C.d(chr(0x2500) * 52)}")
-    hdr = "  {:<34s}{:<30s}{:<30s}"
-    row = "  {:<34s}{:<30s}{:<30s}"
-    print(hdr.format("Host", "BEFORE (broken)", "AFTER (fixed)"))
-    demos = [
-        ("2001:db8::1", 8888),
-        ("::1", 8888),
-        ("::ffff:192.168.1.1", 8888),
-        ("192.168.1.100", 8888),
-        ("myhost.local", 8888),
-    ]
-    for host, port in demos:
-        broken = f"http://{host}:{port}"
-        fixed = f"http://[{host}]:{port}" if ":" in host else f"http://{host}:{port}"
-        tag = "" if broken != fixed else C.d(" (same)")
-        # strip ANSI for width calc in row
-        print(row.format(host, C.r(broken), C.g(fixed) + tag))
-    print()
-
-
-def _fix1_script(sandbox_dir: str) -> str:
-    return (
-        textwrap.dedent("""\
-        import sys, io
-        sys.path.insert(0, __SANDBOX__)
-
-        import studio.backend.startup_banner as banner
-        banner.stdout_supports_color = lambda: False
-
-        CASES = __CASES__
-        failed = 0
-        for display_host, bind_host, port, expected, name in CASES:
-            buf = io.StringIO()
-            old = sys.stdout
-            sys.stdout = buf
-            try:
-                banner.print_studio_access_banner(
-                    port=port, bind_host=bind_host, display_host=display_host,
-                )
-            finally:
-                sys.stdout = old
-            output = buf.getvalue()
-            if expected in output:
-                print("PASS:" + name)
-            else:
-                print("FAIL:" + name + ":expected " + repr(expected) + " not found in banner output")
-                failed += 1
-        sys.exit(1 if failed else 0)
-    """)
-        .replace("__SANDBOX__", repr(str(sandbox_dir)))
-        .replace("__CASES__", repr(FIX1_CASES))
-    )
-
-
-def test_fix1(python: Path, sandbox_dir: Path, verbose: bool) -> list[Result]:
-    visual_demo_fix1()
-    return _run_script(python, _fix1_script(str(sandbox_dir)), verbose)
-
-
-# ══════════════════════════════════════════════════════════════════════
-# Fix 2: try_quiet stderr redirect
-# ══════════════════════════════════════════════════════════════════════
-
-
-def visual_demo_fix2() -> None:
-    print(f"\n  {C.b('Fix 2: try_quiet stderr redirect')}")
-    print(f"  {C.d(chr(0x2500) * 52)}")
-    print(
-        f"  {C.r('BEFORE')}: failure log appears on {C.r('STDOUT')} (mixed with step output)"
-    )
-    print(
-        f"  {C.g('AFTER')}:  failure log appears on {C.g('STDERR')} (clean stream separation)"
-    )
-    print(f"  {C.d('          (step status line still goes to stdout as intended)')}")
-    print()
-
-
-def test_fix2(skip_bash: bool, verbose: bool) -> list[Result]:
-    visual_demo_fix2()
-    if skip_bash or not HAS_BASH:
-        reason = (
-            "skipped (--no-bash)" if skip_bash else "skipped (no bash on this platform)"
-        )
-        return [
-            ("1:success", True, reason),
-            ("2:fail_quiet", True, reason),
-            ("3:fail_verbose", True, reason),
-            ("4:multiline", True, reason),
-            ("5:exit_code", True, reason),
-            ("6:binary", True, reason),
-            ("7:success_verbose", True, reason),
-            ("8:broken_contrast", True, reason),
-        ]
-
-    script_path = REPO / "tests" / "test_try_quiet.sh"
-    if not script_path.exists():
-        return [("fix2", False, f"{script_path} not found")]
-
-    try:
-        r = subprocess.run(
-            ["bash", str(script_path)],
-            capture_output = True,
-            text = True,
-            timeout = 30,
-        )
-    except subprocess.TimeoutExpired:
-        return [("fix2", False, "bash tests timed out")]
-
-    results: list[Result] = []
-    for line in r.stdout.splitlines():
-        line = line.strip()
-        if line.startswith("PASS:"):
-            results.append((line[5:], True, ""))
-        elif line.startswith("FAIL:"):
-            parts = line[5:].split(":", 1)
-            results.append((parts[0], False, parts[1] if len(parts) > 1 else ""))
-
-    if verbose:
-        if r.stderr:
-            print(C.d("  [bash stderr] " + r.stderr[:500]))
-        if r.returncode != 0 and not results:
-            print(C.r(f"  [bash rc={r.returncode}] stdout: {r.stdout[:300]}"))
-
-    if not results:
-        return [("fix2_parse", False, f"no output parsed; rc={r.returncode}")]
-    return results
-
-
-# ══════════════════════════════════════════════════════════════════════
-# Fix 3: _step() label truncation
-# ══════════════════════════════════════════════════════════════════════
-
-_COL = 15  # must match install_python_stack._COL
-
-FIX3_CASES: list[tuple[str, str, str, str]] = [
-    # (label, value, expected_output, name)
-    ("deps", "ok", "  deps" + " " * 11 + "ok", "1:short_4"),
-    ("error", "bad", "  error" + " " * 10 + "bad", "2:short_5"),
-    ("llama-quantize", "ok", "  llama-quantize" + " " * 1 + "ok", "3:len_14"),
-    ("", "v", "  " + " " * 15 + "v", "4:empty"),
-    ("x" * 15, "v", "  " + "x" * 15 + "v", "5:exact_15"),
-    ("x" * 16, "v", "  " + "x" * 15 + "v", "6:trunc_16"),
-    ("x" * 20, "v", "  " + "x" * 15 + "v", "7:trunc_20"),
-    ("x" * 50, "v", "  " + "x" * 15 + "v", "8:trunc_50"),
-    ("toolonglabel!!!!", "v", "  " + "toolonglabel!!!!"[:_COL] + "v", "9:readable"),
-]
-
-# Case 10: broken version contrast -- tested inline, not via real module
-_BROKEN_LABEL = "x" * 20
-_BROKEN_EXPECTED = "  " + _BROKEN_LABEL + "v"  # no truncation, no gap
-
-
-def visual_demo_fix3() -> None:
-    print(f"\n  {C.b('Fix 3: _step() Label Truncation')}")
-    print(f"  {C.d(chr(0x2500) * 52)}")
-    hdr = "  {:<20s}{:<32s}{:<32s}"
-    row = "  {:<20s}{:<32s}{:<32s}"
-    print(hdr.format("Label", "BEFORE (broken)", "AFTER (fixed)"))
-    demos = [
-        ("deps", "ok"),
-        ("x" * 20, "v"),
-        ("x" * 50, "v"),
-    ]
-    for label, val in demos:
-        # Broken: no truncation, negative padding becomes empty
-        bpad = " " * max(0, _COL - len(label))
-        broken_out = f"  {label}{bpad}{val}"
-        # Fixed: truncate label, then pad
-        padded = label[:_COL]
-        fixed_out = f"  {padded}{' ' * (_COL - len(padded))}{val}"
-        tag = C.d(" (same)") if broken_out == fixed_out else C.d(" (truncated!)")
-        disp_label = (
-            repr(label) if len(label) <= 16 else f"'{'x' * 5}...' ({len(label)}c)"
-        )
-        print(
-            row.format(
-                disp_label,
-                C.r(repr(broken_out)),
-                C.g(repr(fixed_out)) + tag,
-            )
-        )
-    print()
-
-
-def _fix3_script(sandbox_dir: str) -> str:
-    return (
-        textwrap.dedent("""\
-        import sys, io
-        sys.path.insert(0, __SANDBOX__)
-
-        import studio.install_python_stack as ips
-        ips._HAS_COLOR = False
-
-        CASES = __CASES__
-        failed = 0
-        for label, value, expected, name in CASES:
-            buf = io.StringIO()
-            old = sys.stdout
-            sys.stdout = buf
-            try:
-                ips._step(label, value)
-            finally:
-                sys.stdout = old
-            actual = buf.getvalue().rstrip("\\n")
-            if actual == expected:
-                print("PASS:" + name)
-            else:
-                print("FAIL:" + name + ":expected " + repr(expected) + " got " + repr(actual))
-                failed += 1
-
-        # Case 10: broken version contrast (inline, not via module)
-        broken_label = "x" * 20
-        broken_padded = broken_label  # no [:_COL] truncation
-        broken_pad = " " * max(0, 15 - len(broken_padded))
-        broken_out = "  " + broken_padded + broken_pad + "v"
-        expected_broken = "  " + "x" * 20 + "v"
-
-        # Also get the fixed output for comparison
-        buf = io.StringIO()
-        old = sys.stdout
-        sys.stdout = buf
-        try:
-            ips._step(broken_label, "v")
-        finally:
-            sys.stdout = old
-        fixed_out = buf.getvalue().rstrip("\\n")
-
-        # The broken output should differ from the fixed output
-        if broken_out == expected_broken and broken_out != fixed_out:
-            print("PASS:10:broken_contrast")
-        else:
-            print("FAIL:10:broken_contrast:broken=" + repr(broken_out) + " fixed=" + repr(fixed_out))
-            failed += 1
-
-        sys.exit(1 if failed else 0)
-    """)
-        .replace("__SANDBOX__", repr(str(sandbox_dir)))
-        .replace("__CASES__", repr(FIX3_CASES))
-    )
-
-
-def test_fix3(python: Path, sandbox_dir: Path, verbose: bool) -> list[Result]:
-    visual_demo_fix3()
-    return _run_script(python, _fix3_script(str(sandbox_dir)), verbose)
-
-
-# ══════════════════════════════════════════════════════════════════════
-# Summary
-# ══════════════════════════════════════════════════════════════════════
-
-FIX_LABELS = {
-    1: "IPv6 URL bracketing",
-    2: "try_quiet stderr redirect",
-    3: "_step label truncation",
-}
-
-
-def print_summary(
-    results_by_fix: dict[int, list[Result]],
-    verbose: bool,
-) -> bool:
-    """Print pass/fail table.  Returns True if all passed."""
-    bar = chr(0x2550) * 56
-    thin = chr(0x2500) * 56
-    print(f"\n  {bar}")
-    print(f"  {C.b('PR #4494 Simulation Results')}")
-    print(f"  {bar}")
-
-    all_pass = True
-    grand_pass = 0
-    grand_total = 0
-
-    for fix_num in sorted(results_by_fix):
-        results = results_by_fix[fix_num]
-        n_pass = sum(1 for _, ok, _ in results if ok)
-        n_total = len(results)
-        grand_pass += n_pass
-        grand_total += n_total
-        ok = n_pass == n_total
-        if not ok:
-            all_pass = False
-        tag = C.g("[PASS]") if ok else C.r("[FAIL]")
-        label = FIX_LABELS.get(fix_num, f"Fix {fix_num}")
-        print(f"  Fix {fix_num}  {label:<34s} {tag}  {n_pass:>2}/{n_total}")
-
-        if verbose or not ok:
-            for name, passed, detail in results:
-                if not passed:
-                    print(f"         {C.r('FAIL')} {name}: {detail}")
-
-    print(f"  {thin}")
-    summary = f"Total: {grand_pass}/{grand_total} passed"
-    if all_pass:
-        print(f"  {C.g(summary)} -- {C.g('ALL PASSED')}")
-    else:
-        print(f"  {C.r(summary)} -- {C.r('SOME FAILED')}")
-    print(f"  {bar}")
-    return all_pass
-
-
-# ══════════════════════════════════════════════════════════════════════
-# Main
-# ══════════════════════════════════════════════════════════════════════
-
-
-def main() -> int:
-    ap = argparse.ArgumentParser(description = "PR #4494 simulation harness")
-    ap.add_argument("--fix", type = int, choices = [1, 2, 3], help = "run only this fix")
-    ap.add_argument(
-        "--no-sandbox", action = "store_true", help = "skip venv, use current Python"
-    )
-    ap.add_argument(
-        "--keep-sandbox", action = "store_true", help = "do not delete sandbox on exit"
-    )
-    ap.add_argument("--no-bash", action = "store_true", help = "skip bash tests (Fix 2)")
-    ap.add_argument("--no-color", action = "store_true", help = "disable ANSI colors")
-    ap.add_argument("--verbose", action = "store_true", help = "show subprocess output")
-    args = ap.parse_args()
-
-    if args.no_color or os.environ.get("NO_COLOR", "").strip():
-        C.on = False
-
-    fixes = [args.fix] if args.fix else [1, 2, 3]
-
-    # Sandbox setup
-    sandbox_dir: Path | None = None
-    if args.no_sandbox:
-        # Use current Python, put source on sys.path directly
-        sandbox_dir = REPO
-        python = Path(sys.executable)
-        print(C.d(f"  (no-sandbox mode, using {python})"))
-    else:
-        print(C.d("  Creating sandbox..."), end = "", flush = True)
-        sandbox_dir, python = create_sandbox(args.verbose)
-        print(C.d(f" {sandbox_dir}"))
-
-    results_by_fix: dict[int, list[Result]] = {}
-
-    try:
-        if 1 in fixes:
-            results_by_fix[1] = test_fix1(python, sandbox_dir, args.verbose)
-        if 2 in fixes:
-            results_by_fix[2] = test_fix2(args.no_bash, args.verbose)
-        if 3 in fixes:
-            results_by_fix[3] = test_fix3(python, sandbox_dir, args.verbose)
-    finally:
-        if not args.no_sandbox and sandbox_dir and sandbox_dir != REPO:
-            cleanup_sandbox(sandbox_dir, args.keep_sandbox)
-
-    all_pass = print_summary(results_by_fix, args.verbose)
-    return 0 if all_pass else 1
-
-
-if __name__ == "__main__":
-    sys.exit(main())
diff --git a/tests/test_studio_setup_styling.py b/tests/test_studio_setup_styling.py
deleted file mode 100644
index e34de4d316..0000000000
--- a/tests/test_studio_setup_styling.py
+++ /dev/null
@@ -1,810 +0,0 @@
-"""Tests for PR #4494 Studio setup log styling.
-
-Validates terminal output styling across:
-- studio/backend/startup_banner.py
-- studio/install_python_stack.py
-- studio/setup.sh  (bash subprocess tests)
-- studio/setup.ps1 (structural/static analysis only -- no pwsh on Linux)
-- unsloth_cli/commands/studio.py (CLI tests via typer CliRunner)
-"""
-
-from __future__ import annotations
-
-import os
-import re
-import subprocess
-import sys
-from pathlib import Path
-from unittest.mock import MagicMock, patch
-
-import pytest
-
-# Ensure repo root is importable even without pip install -e .
-_REPO = Path(__file__).resolve().parent.parent
-if str(_REPO) not in sys.path:
-    sys.path.insert(0, str(_REPO))
-
-SETUP_SH = _REPO / "studio" / "setup.sh"
-SETUP_PS1 = _REPO / "studio" / "setup.ps1"
-
-# ── Bash fragments inlined from setup.sh for subprocess tests ─────────
-
-_BASH_COLOR_INIT = r"""
-if [ -n "${NO_COLOR:-}" ]; then
-    C_TITLE= C_DIM= C_OK= C_WARN= C_ERR= C_RST=
-elif [ -t 1 ] || [ -n "${FORCE_COLOR:-}" ]; then
-    C_TITLE=$'\033[38;5;150m'
-    C_DIM=$'\033[38;5;245m'
-    C_OK=$'\033[38;5;108m'
-    C_WARN=$'\033[38;5;136m'
-    C_ERR=$'\033[91m'
-    C_RST=$'\033[0m'
-else
-    C_TITLE= C_DIM= C_OK= C_WARN= C_ERR= C_RST=
-fi
-"""
-
-_BASH_HELPERS = r"""
-step()    { printf "  ${C_DIM}%-15s${C_RST}${3:-$C_OK}%s${C_RST}\n" "$1" "$2"; }
-substep() { printf "  ${C_DIM}%-15s%s${C_RST}\n" "" "$1"; }
-"""
-
-_BASH_RUN_QUIET = r"""
-_run_quiet() {
-    local on_fail=$1
-    local label=$2
-    shift 2
-    local tmplog
-    tmplog=$(mktemp) || {
-        step "error" "Failed to create temporary file" "$C_ERR"
-        [ "$on_fail" = "exit" ] && exit 1 || return 1
-    }
-    if "$@" >"$tmplog" 2>&1; then
-        rm -f "$tmplog"
-        return 0
-    else
-        local exit_code=$?
-        step "error" "$label failed (exit code $exit_code)" "$C_ERR"
-        cat "$tmplog" >&2
-        rm -f "$tmplog"
-        if [ "$on_fail" = "exit" ]; then
-            exit "$exit_code"
-        else
-            return "$exit_code"
-        fi
-    fi
-}
-run_quiet() { _run_quiet exit "$@"; }
-run_quiet_no_exit() { _run_quiet return "$@"; }
-"""
-
-_BASH_TRY_QUIET = r"""
-try_quiet() {
-    local label="$1"; shift
-    local tmplog; tmplog=$(mktemp)
-    if "$@" > "$tmplog" 2>&1; then
-        rm -f "$tmplog"
-        return 0
-    else
-        local exit_code=$?
-        if [ "${UNSLOTH_VERBOSE:-0}" = "1" ]; then
-            step "error" "$label failed (exit code $exit_code)" "$C_ERR"
-            cat "$tmplog" >&2
-        fi
-        rm -f "$tmplog"
-        return $exit_code
-    fi
-}
-"""
-
-# Template for REQUESTED_PYTHON_VERSION tests (uses .replace() placeholders)
-_REQUESTED_PY_TPL = r"""
-C_DIM= C_OK= C_WARN= C_ERR= C_RST= C_TITLE=
-step()    { printf "  %-15s%s\n" "$1" "$2"; }
-substep() { printf "  %-15s%s\n" "" "$1"; }
-MIN_PY_MINOR=__MIN__
-MAX_PY_MINOR=__MAX__
-BEST_PY=""
-BEST_MINOR=0
-REQUESTED_PYTHON_VERSION="__REQ__"
-if [ -n "${REQUESTED_PYTHON_VERSION:-}" ] && [ -x "$REQUESTED_PYTHON_VERSION" ]; then
-    _req_ver=$("$REQUESTED_PYTHON_VERSION" --version 2>&1 | awk '{print $2}')
-    _req_major=$(echo "$_req_ver" | cut -d. -f1)
-    _req_minor=$(echo "$_req_ver" | cut -d. -f2)
-    if [ "$_req_major" -eq 3 ] 2>/dev/null && \
-       [ "$_req_minor" -ge "$MIN_PY_MINOR" ] 2>/dev/null && \
-       [ "$_req_minor" -le "$MAX_PY_MINOR" ] 2>/dev/null; then
-        BEST_PY="$REQUESTED_PYTHON_VERSION"
-        substep "using requested Python: $BEST_PY"
-    else
-        substep "ignoring requested Python $REQUESTED_PYTHON_VERSION ($_req_ver) -- outside range"
-    fi
-fi
-echo "BEST_PY=$BEST_PY"
-"""
-
-
-def _run_bash(
-    script: str, env_extra: dict[str, str] | None = None
-) -> subprocess.CompletedProcess[str]:
-    """Run a bash script fragment with a sanitised environment."""
-    env = {
-        k: v
-        for k, v in os.environ.items()
-        if k not in ("NO_COLOR", "FORCE_COLOR", "UNSLOTH_VERBOSE")
-    }
-    if env_extra:
-        env.update(env_extra)
-    return subprocess.run(
-        ["bash", "-c", script], capture_output = True, text = True, env = env
-    )
-
-
-@pytest.fixture(autouse = True)
-def _clean_env(monkeypatch):
-    """Remove colour / verbosity env vars before every test."""
-    monkeypatch.delenv("NO_COLOR", raising = False)
-    monkeypatch.delenv("FORCE_COLOR", raising = False)
-    monkeypatch.delenv("UNSLOTH_VERBOSE", raising = False)
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 1. startup_banner.py -- stdout_supports_color()  (9 tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestBannerSupportsColor:
-    """NO_COLOR > FORCE_COLOR > isatty."""
-
-    @staticmethod
-    def _call():
-        from studio.backend.startup_banner import stdout_supports_color
-
-        return stdout_supports_color()
-
-    def test_no_tty_no_env(self):
-        with patch("sys.stdout.isatty", return_value = False):
-            assert self._call() is False
-
-    def test_tty_returns_true(self):
-        with patch("sys.stdout.isatty", return_value = True):
-            assert self._call() is True
-
-    def test_no_color_disables_with_tty(self, monkeypatch):
-        monkeypatch.setenv("NO_COLOR", "1")
-        with patch("sys.stdout.isatty", return_value = True):
-            assert self._call() is False
-
-    def test_force_color_enables_no_tty(self, monkeypatch):
-        monkeypatch.setenv("FORCE_COLOR", "1")
-        with patch("sys.stdout.isatty", return_value = False):
-            assert self._call() is True
-
-    def test_force_color_zero_truthy(self, monkeypatch):
-        monkeypatch.setenv("FORCE_COLOR", "0")
-        with patch("sys.stdout.isatty", return_value = False):
-            assert self._call() is True
-
-    def test_force_color_empty_falsy(self, monkeypatch):
-        monkeypatch.setenv("FORCE_COLOR", "")
-        with patch("sys.stdout.isatty", return_value = False):
-            assert self._call() is False
-
-    def test_no_color_empty_no_disable(self, monkeypatch):
-        monkeypatch.setenv("NO_COLOR", "")
-        with patch("sys.stdout.isatty", return_value = True):
-            assert self._call() is True
-
-    def test_no_color_whitespace_no_disable(self, monkeypatch):
-        monkeypatch.setenv("NO_COLOR", "   ")
-        with patch("sys.stdout.isatty", return_value = True):
-            assert self._call() is True
-
-    def test_no_color_overrides_force(self, monkeypatch):
-        monkeypatch.setenv("NO_COLOR", "1")
-        monkeypatch.setenv("FORCE_COLOR", "1")
-        with patch("sys.stdout.isatty", return_value = True):
-            assert self._call() is False
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 2. startup_banner.py -- print_port_in_use_notice()  (3 tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestPortInUseNotice:
-    def test_contains_both_ports(self, capsys):
-        from studio.backend.startup_banner import print_port_in_use_notice
-
-        with patch(
-            "studio.backend.startup_banner.stdout_supports_color", return_value = False
-        ):
-            print_port_in_use_notice(8888, 8889)
-        out = capsys.readouterr().out
-        assert "8888" in out
-        assert "8889" in out
-
-    def test_no_color_no_ansi(self, capsys):
-        from studio.backend.startup_banner import print_port_in_use_notice
-
-        with patch(
-            "studio.backend.startup_banner.stdout_supports_color", return_value = False
-        ):
-            print_port_in_use_notice(8888, 8889)
-        assert "\033[" not in capsys.readouterr().out
-
-    def test_color_has_ansi(self, capsys):
-        from studio.backend.startup_banner import print_port_in_use_notice
-
-        with patch(
-            "studio.backend.startup_banner.stdout_supports_color", return_value = True
-        ):
-            print_port_in_use_notice(8888, 8889)
-        assert "\033[" in capsys.readouterr().out
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 3. startup_banner.py -- print_studio_access_banner()  (11 tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestStudioAccessBanner:
-    PORT = 8888
-
-    @staticmethod
-    def _call(*, bind_host, display_host, port = 8888, color = False):
-        from studio.backend.startup_banner import print_studio_access_banner
-
-        with patch(
-            "studio.backend.startup_banner.stdout_supports_color",
-            return_value = color,
-        ):
-            print_studio_access_banner(
-                port = port, bind_host = bind_host, display_host = display_host
-            )
-
-    @pytest.mark.parametrize(
-        "bind_host,display_host,expect_local,from_another,bound_addr,api_host",
-        [
-            ("0.0.0.0", "192.168.1.100", "127.0.0.1", True, False, "127.0.0.1"),
-            ("0.0.0.0", "127.0.0.1", "127.0.0.1", False, False, "127.0.0.1"),
-            ("0.0.0.0", "0.0.0.0", "127.0.0.1", False, False, "127.0.0.1"),
-            ("::", "2001:db8::1", "[::1]", True, False, "[::1]"),
-            ("::1", "::1", "[::1]", False, False, "[::1]"),
-            ("127.0.0.1", "127.0.0.1", "127.0.0.1", False, False, "127.0.0.1"),
-            ("localhost", "localhost", "127.0.0.1", False, False, "127.0.0.1"),
-            ("192.168.1.5", "192.168.1.5", "127.0.0.1", False, True, "192.168.1.5"),
-        ],
-        ids = [
-            "wildcard_v4_ext",
-            "wildcard_v4_loop",
-            "wildcard_v4_self",
-            "wildcard_v6_ext",
-            "loopback_v6",
-            "loopback_v4",
-            "localhost",
-            "specific_ip",
-        ],
-    )
-    def test_hosts(
-        self,
-        capsys,
-        bind_host,
-        display_host,
-        expect_local,
-        from_another,
-        bound_addr,
-        api_host,
-    ):
-        self._call(bind_host = bind_host, display_host = display_host, color = False)
-        out = capsys.readouterr().out
-        assert f"http://{expect_local}:{self.PORT}" in out
-        assert ("From another device" in out) == from_another
-        assert ("Bound address" in out) == bound_addr
-        assert f"http://{api_host}:{self.PORT}/api" in out
-
-    def test_color_on_has_ansi(self, capsys):
-        self._call(bind_host = "127.0.0.1", display_host = "127.0.0.1", color = True)
-        assert "\033[" in capsys.readouterr().out
-
-    def test_color_off_no_ansi(self, capsys):
-        self._call(bind_host = "127.0.0.1", display_host = "127.0.0.1", color = False)
-        assert "\033[" not in capsys.readouterr().out
-
-    def test_api_urls_present(self, capsys):
-        self._call(bind_host = "0.0.0.0", display_host = "10.0.0.1", color = False)
-        out = capsys.readouterr().out
-        assert "/api" in out
-        assert "/api/health" in out
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 4. install_python_stack.py -- Color detection  (5 tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestInstallColorDetect:
-    @staticmethod
-    def _mod():
-        import studio.install_python_stack as ips
-
-        return ips
-
-    def test_no_tty(self):
-        ips = self._mod()
-        with patch("sys.stdout.isatty", return_value = False):
-            assert ips._stdout_supports_color() is False
-
-    def test_tty_linux(self, monkeypatch):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "IS_WINDOWS", False)
-        with patch("sys.stdout.isatty", return_value = True):
-            assert ips._stdout_supports_color() is True
-
-    def test_force_color(self, monkeypatch):
-        monkeypatch.setenv("FORCE_COLOR", "1")
-        assert self._mod()._stdout_supports_color() is True
-
-    def test_no_color(self, monkeypatch):
-        monkeypatch.setenv("NO_COLOR", "1")
-        with patch("sys.stdout.isatty", return_value = True):
-            assert self._mod()._stdout_supports_color() is False
-
-    def test_windows_ctypes_fail(self, monkeypatch):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "IS_WINDOWS", True)
-        with patch("sys.stdout.isatty", return_value = True):
-            # ctypes.windll does not exist on Linux, so the except branch fires
-            assert ips._stdout_supports_color() is False
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 5. install_python_stack.py -- _step() formatting  (6 tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestInstallStep:
-    @staticmethod
-    def _mod():
-        import studio.install_python_stack as ips
-
-        return ips
-
-    def test_short_label_padding(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "_HAS_COLOR", False)
-        ips._step("deps", "ok")
-        out = capsys.readouterr().out
-        # "deps" (4 chars) + 11 spaces padding to fill 15-col layout
-        assert "  deps" + " " * 11 + "ok" in out
-
-    def test_exact_15_label(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "_HAS_COLOR", False)
-        label = "x" * 15
-        ips._step(label, "v")
-        out = capsys.readouterr().out
-        assert label + "v" in out
-
-    def test_long_label(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "_HAS_COLOR", False)
-        label = "x" * 20
-        ips._step(label, "v")
-        out = capsys.readouterr().out
-        # Label is clamped to _COL (15) chars; value appears right after
-        assert label[:15] + "v" in out
-        assert label not in out
-
-    def test_default_green(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "_HAS_COLOR", True)
-        ips._step("deps", "ok")
-        assert "\033[38;5;108m" in capsys.readouterr().out
-
-    def test_custom_color_fn(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "_HAS_COLOR", True)
-        ips._step("err", "fail", color_fn = ips._red)
-        assert "\033[91m" in capsys.readouterr().out
-
-    def test_no_color_no_ansi(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "_HAS_COLOR", False)
-        ips._step("deps", "ok")
-        assert "\033[" not in capsys.readouterr().out
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 6. install_python_stack.py -- _progress()  (4 tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestInstallProgress:
-    @staticmethod
-    def _mod():
-        import studio.install_python_stack as ips
-
-        return ips
-
-    def test_verbose_no_output(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "VERBOSE", True)
-        monkeypatch.setattr(ips, "_STEP", 0)
-        monkeypatch.setattr(ips, "_TOTAL", 5)
-        ips._progress("lbl")
-        assert capsys.readouterr().out == ""
-        assert ips._STEP == 1
-
-    def test_normal_has_cr(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "VERBOSE", False)
-        monkeypatch.setattr(ips, "_STEP", 0)
-        monkeypatch.setattr(ips, "_TOTAL", 5)
-        monkeypatch.setattr(ips, "_HAS_COLOR", False)
-        ips._progress("lbl")
-        out = capsys.readouterr().out
-        assert "\r" in out
-        assert not out.endswith("\n")
-
-    def test_final_step_newline(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "VERBOSE", False)
-        monkeypatch.setattr(ips, "_STEP", 4)
-        monkeypatch.setattr(ips, "_TOTAL", 5)
-        monkeypatch.setattr(ips, "_HAS_COLOR", False)
-        ips._progress("done")
-        assert capsys.readouterr().out.endswith("\n")
-
-    def test_bar_format(self, monkeypatch, capsys):
-        ips = self._mod()
-        monkeypatch.setattr(ips, "VERBOSE", False)
-        monkeypatch.setattr(ips, "_STEP", 0)
-        monkeypatch.setattr(ips, "_TOTAL", 10)
-        monkeypatch.setattr(ips, "_HAS_COLOR", False)
-        ips._progress("s1")
-        out = capsys.readouterr().out
-        assert re.search(r"\[=+-+\]", out)
-        assert " 1/10" in out
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 7. setup.sh -- Color initialization  (4 bash tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestBashColorInit:
-    _SCRIPT = _BASH_COLOR_INIT + '\necho "DIM=${C_DIM}END"'
-
-    def test_no_env_empty(self):
-        r = _run_bash(self._SCRIPT)
-        assert r.returncode == 0
-        assert "DIM=END" in r.stdout
-
-    def test_force_color_sets(self):
-        r = _run_bash(self._SCRIPT, {"FORCE_COLOR": "1"})
-        assert r.returncode == 0
-        # C_DIM is non-empty when FORCE_COLOR is set
-        assert "DIM=END" not in r.stdout
-
-    def test_no_color_clears(self):
-        r = _run_bash(self._SCRIPT, {"NO_COLOR": "1"})
-        assert r.returncode == 0
-        assert "DIM=END" in r.stdout
-
-    def test_no_color_beats_force(self):
-        r = _run_bash(self._SCRIPT, {"NO_COLOR": "1", "FORCE_COLOR": "1"})
-        assert r.returncode == 0
-        assert "DIM=END" in r.stdout
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 8. setup.sh -- step() and substep()  (5 bash tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestBashStep:
-    _PRE = _BASH_COLOR_INIT + _BASH_HELPERS
-
-    def test_step_padding(self):
-        r = _run_bash(self._PRE + '\nstep "deps" "installed"')
-        assert r.returncode == 0
-        assert "deps" in r.stdout and "installed" in r.stdout
-
-    def test_step_15_label(self):
-        r = _run_bash(self._PRE + '\nstep "aaaaaaaaaaaaaaa" "v"')
-        assert "aaaaaaaaaaaaaaa" in r.stdout and "v" in r.stdout
-
-    def test_step_long_label(self):
-        r = _run_bash(self._PRE + '\nstep "aaaaaaaaaaaaaaaaaaaa" "v"')
-        assert "aaaaaaaaaaaaaaaaaaaa" in r.stdout
-
-    def test_step_color(self):
-        r = _run_bash(self._PRE + '\nstep "deps" "ok"', {"FORCE_COLOR": "1"})
-        assert "\033[" in r.stdout
-
-    def test_substep(self):
-        r = _run_bash(self._PRE + '\nsubstep "building..."')
-        assert "building..." in r.stdout
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 9. setup.sh -- run_quiet / run_quiet_no_exit  (5 bash tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestBashRunQuiet:
-    _PRE = _BASH_COLOR_INIT + _BASH_HELPERS + _BASH_RUN_QUIET
-
-    def test_no_exit_success(self):
-        r = _run_bash(self._PRE + '\nrun_quiet_no_exit "ok" true; echo "RC=$?"')
-        assert "RC=0" in r.stdout
-
-    def test_no_exit_failure_rc(self):
-        r = _run_bash(
-            self._PRE
-            + "\nrun_quiet_no_exit \"cmd\" bash -c 'echo log_data; exit 42'"
-            + '\necho "RC=$?"'
-        )
-        assert "RC=42" in r.stdout
-
-    def test_no_exit_failure_shows_error(self):
-        r = _run_bash(
-            self._PRE
-            + "\nrun_quiet_no_exit \"my cmd\" bash -c 'echo log_data; exit 1'"
-            + "\ntrue"
-        )
-        assert "my cmd failed" in r.stdout
-        assert "log_data" in r.stderr
-
-    def test_run_quiet_failure_exits(self):
-        script = (
-            self._PRE
-            + """
-(
-    run_quiet "bad" false
-    echo "SHOULD_NOT_REACH"
-)
-echo "OUTER=$?"
-"""
-        )
-        r = _run_bash(script)
-        assert "SHOULD_NOT_REACH" not in r.stdout
-
-    def test_run_quiet_success(self):
-        r = _run_bash(self._PRE + '\nrun_quiet "ok" true; echo "RC=$?"')
-        assert "RC=0" in r.stdout
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 10. setup.sh -- try_quiet  (5 bash tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestBashTryQuiet:
-    _PRE = _BASH_COLOR_INIT + _BASH_HELPERS + _BASH_TRY_QUIET
-
-    def test_success_rc0(self):
-        r = _run_bash(self._PRE + '\ntry_quiet "ok" true; echo "RC=$?"')
-        assert "RC=0" in r.stdout
-
-    def test_fail_quiet_no_output(self):
-        r = _run_bash(
-            self._PRE
-            + "\ntry_quiet \"cmd\" bash -c 'echo secret; exit 1'"
-            + '\necho "RC=$?"'
-        )
-        assert "RC=1" in r.stdout
-        assert "secret" not in r.stdout
-        assert "error" not in r.stdout
-
-    def test_fail_verbose_shows(self):
-        r = _run_bash(
-            self._PRE
-            + "\ntry_quiet \"cmd\" bash -c 'echo visible; exit 1'"
-            + '\necho "RC=$?"',
-            {"UNSLOTH_VERBOSE": "1"},
-        )
-        assert "cmd failed" in r.stdout
-        assert "visible" in r.stderr
-
-    def test_preserves_exit_code(self):
-        r = _run_bash(
-            self._PRE + "\ntry_quiet \"cmd\" bash -c 'exit 42'" + '\necho "RC=$?"'
-        )
-        assert "RC=42" in r.stdout
-
-    def test_success_verbose_quiet(self):
-        r = _run_bash(
-            self._PRE + '\ntry_quiet "cmd" true; echo "RC=$?"',
-            {"UNSLOTH_VERBOSE": "1"},
-        )
-        assert "RC=0" in r.stdout
-        assert "error" not in r.stdout
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 11. setup.sh -- REQUESTED_PYTHON_VERSION  (5 bash tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestBashRequestedPython:
-    _PY_MINOR = sys.version_info.minor
-
-    def _block(self, *, requested = "", min_minor = None, max_minor = None):
-        if min_minor is None:
-            min_minor = self._PY_MINOR - 1
-        if max_minor is None:
-            max_minor = self._PY_MINOR + 1
-        return (
-            _REQUESTED_PY_TPL.replace("__MIN__", str(min_minor))
-            .replace("__MAX__", str(max_minor))
-            .replace("__REQ__", requested)
-        )
-
-    def test_unset_skipped(self):
-        r = _run_bash(self._block(requested = ""))
-        assert r.stdout.strip().endswith("BEST_PY=")
-
-    def test_valid_in_range(self):
-        r = _run_bash(self._block(requested = sys.executable))
-        assert f"BEST_PY={sys.executable}" in r.stdout
-        assert "using requested Python" in r.stdout
-
-    def test_nonexistent_skipped(self):
-        r = _run_bash(self._block(requested = "/nonexistent/python3.99"))
-        assert r.stdout.strip().endswith("BEST_PY=")
-
-    def test_not_executable_skipped(self):
-        r = _run_bash(self._block(requested = "/dev/null"))
-        assert r.stdout.strip().endswith("BEST_PY=")
-
-    def test_out_of_range(self):
-        r = _run_bash(
-            self._block(
-                requested = sys.executable,
-                min_minor = self._PY_MINOR + 10,
-                max_minor = self._PY_MINOR + 11,
-            )
-        )
-        assert r.stdout.strip().endswith("BEST_PY=")
-        assert "ignoring" in r.stdout
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 12. setup.ps1 -- Structural checks  (9 tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestPS1Structural:
-    """Static analysis of setup.ps1 -- no PowerShell runtime needed."""
-
-    @pytest.fixture(autouse = True)
-    def _load(self):
-        self.content = SETUP_PS1.read_text(encoding = "utf-8")
-
-    def test_fn_step(self):
-        assert re.search(r"^function step\s*\{", self.content, re.MULTILINE)
-
-    def test_fn_substep(self):
-        assert re.search(r"^function substep\s*\{", self.content, re.MULTILINE)
-
-    def test_fn_get_studio_ansi(self):
-        assert "function Get-StudioAnsi" in self.content
-
-    def test_fn_enable_vt(self):
-        assert "function Enable-StudioVirtualTerminal" in self.content
-
-    def test_ansi_title_code(self):
-        assert "38;5;150m" in self.content
-
-    def test_ansi_dim_code(self):
-        assert "38;5;245m" in self.content
-
-    def test_ansi_ok_code(self):
-        assert "38;5;108m" in self.content
-
-    def test_label_padding(self):
-        assert ".PadRight(15)" in self.content
-
-    def test_env_vars_referenced(self):
-        assert "NO_COLOR" in self.content
-        assert "UNSLOTH_VERBOSE" in self.content
-
-
-# ═══════════════════════════════════════════════════════════════════════
-# 13. studio.py CLI -- setup --verbose  (4 tests)
-# ═══════════════════════════════════════════════════════════════════════
-
-
-class TestCLISetup:
-    def test_verbose_sets_env(self):
-        from typer.testing import CliRunner
-
-        from unsloth_cli.commands.studio import studio_app
-
-        runner = CliRunner()
-        mock_result = MagicMock(returncode = 0)
-        with (
-            patch(
-                "unsloth_cli.commands.studio._find_setup_script",
-                return_value = Path("/f/setup.sh"),
-            ),
-            patch(
-                "unsloth_cli.commands.studio.subprocess.run",
-                return_value = mock_result,
-            ) as mock_run,
-            patch(
-                "unsloth_cli.commands.studio.platform.system",
-                return_value = "Linux",
-            ),
-        ):
-            result = runner.invoke(studio_app, ["setup", "--verbose"])
-        assert result.exit_code == 0
-        env_arg = mock_run.call_args.kwargs.get("env")
-        assert env_arg is not None
-        assert env_arg["UNSLOTH_VERBOSE"] == "1"
-
-    def test_no_verbose_env_none(self):
-        from typer.testing import CliRunner
-
-        from unsloth_cli.commands.studio import studio_app
-
-        runner = CliRunner()
-        mock_result = MagicMock(returncode = 0)
-        with (
-            patch(
-                "unsloth_cli.commands.studio._find_setup_script",
-                return_value = Path("/f/setup.sh"),
-            ),
-            patch(
-                "unsloth_cli.commands.studio.subprocess.run",
-                return_value = mock_result,
-            ) as mock_run,
-            patch(
-                "unsloth_cli.commands.studio.platform.system",
-                return_value = "Linux",
-            ),
-        ):
-            result = runner.invoke(studio_app, ["setup"])
-        assert result.exit_code == 0
-        env_arg = mock_run.call_args.kwargs.get("env")
-        assert env_arg is None
-
-    def test_script_not_found(self):
-        from typer.testing import CliRunner
-
-        from unsloth_cli.commands.studio import studio_app
-
-        runner = CliRunner()
-        with patch(
-            "unsloth_cli.commands.studio._find_setup_script",
-            return_value = None,
-        ):
-            result = runner.invoke(studio_app, ["setup"])
-        assert result.exit_code == 1
-        assert "Could not find" in result.output
-
-    def test_nonzero_rc_propagated(self):
-        from typer.testing import CliRunner
-
-        from unsloth_cli.commands.studio import studio_app
-
-        runner = CliRunner()
-        mock_result = MagicMock(returncode = 42)
-        with (
-            patch(
-                "unsloth_cli.commands.studio._find_setup_script",
-                return_value = Path("/f/setup.sh"),
-            ),
-            patch(
-                "unsloth_cli.commands.studio.subprocess.run",
-                return_value = mock_result,
-            ),
-            patch(
-                "unsloth_cli.commands.studio.platform.system",
-                return_value = "Linux",
-            ),
-        ):
-            result = runner.invoke(studio_app, ["setup"])
-        assert result.exit_code == 42
diff --git a/tests/test_try_quiet.sh b/tests/test_try_quiet.sh
deleted file mode 100644
index b2b295f558..0000000000
--- a/tests/test_try_quiet.sh
+++ /dev/null
@@ -1,159 +0,0 @@
-#!/usr/bin/env bash
-# test_try_quiet.sh -- Standalone tests for try_quiet stderr redirect (Fix 2, PR #4494)
-#
-# Usage:  bash tests/test_try_quiet.sh
-#
-# Tests that the fixed try_quiet sends failure logs to stderr (not stdout),
-# and contrasts with the broken pre-fix behavior.
-set -uo pipefail
-
-PASS=0
-FAIL=0
-
-# ── Helpers ───────────────────────────────────────────────────────────
-
-# Minimal step() matching setup.sh signature (stdout, no color)
-step() {
-    local _label="$1" _value="$2"
-    echo "  $_label: $_value"
-}
-
-# Fixed try_quiet (PR #4494)
-try_quiet_fixed() {
-    local label="$1"; shift
-    local tmplog; tmplog=$(mktemp)
-    if "$@" > "$tmplog" 2>&1; then
-        rm -f "$tmplog"
-        return 0
-    else
-        local exit_code=$?
-        if [ "${UNSLOTH_VERBOSE:-0}" = "1" ]; then
-            step "error" "$label failed (exit code $exit_code)"
-            cat "$tmplog" >&2
-        fi
-        rm -f "$tmplog"
-        return $exit_code
-    fi
-}
-
-# Broken try_quiet (pre-fix: log dumps to stdout)
-try_quiet_broken() {
-    local label="$1"; shift
-    local tmplog; tmplog=$(mktemp)
-    if "$@" > "$tmplog" 2>&1; then
-        rm -f "$tmplog"
-        return 0
-    else
-        local exit_code=$?
-        if [ "${UNSLOTH_VERBOSE:-0}" = "1" ]; then
-            step "error" "$label failed (exit code $exit_code)"
-            cat "$tmplog"   # BUG: no >&2, leaks to stdout
-        fi
-        rm -f "$tmplog"
-        return $exit_code
-    fi
-}
-
-assert_eq() {
-    local name="$1" expected="$2" actual="$3"
-    if [ "$expected" = "$actual" ]; then
-        PASS=$((PASS + 1))
-        echo "PASS:$name"
-    else
-        FAIL=$((FAIL + 1))
-        echo "FAIL:$name:expected '$(echo "$expected" | head -c 60)' got '$(echo "$actual" | head -c 60)'"
-    fi
-}
-
-assert_contains() {
-    local name="$1" haystack="$2" needle="$3"
-    if printf '%s' "$haystack" | grep -qF "$needle"; then
-        PASS=$((PASS + 1))
-        echo "PASS:$name"
-    else
-        FAIL=$((FAIL + 1))
-        echo "FAIL:$name:should contain '$needle'"
-    fi
-}
-
-assert_not_contains() {
-    local name="$1" haystack="$2" needle="$3"
-    if printf '%s' "$haystack" | grep -qF "$needle"; then
-        FAIL=$((FAIL + 1))
-        echo "FAIL:$name:should NOT contain '$needle'"
-    else
-        PASS=$((PASS + 1))
-        echo "PASS:$name"
-    fi
-}
-
-tmpout=$(mktemp)
-tmperr=$(mktemp)
-trap 'rm -f "$tmpout" "$tmperr"' EXIT
-
-# ── Test 1: Success -- no output anywhere ─────────────────────────────
-rc=0
-UNSLOTH_VERBOSE=0 try_quiet_fixed "test" true > "$tmpout" 2> "$tmperr" || rc=$?
-assert_eq "1:success_rc" "0" "$rc"
-
-# ── Test 2: Failure, VERBOSE=0 -- silent ──────────────────────────────
-rc=0
-UNSLOTH_VERBOSE=0 try_quiet_fixed "test" false > "$tmpout" 2> "$tmperr" || rc=$?
-stdout_content=$(cat "$tmpout")
-stderr_content=$(cat "$tmperr")
-assert_eq "2:fail_quiet_stdout" "" "$stdout_content"
-assert_eq "2:fail_quiet_stderr" "" "$stderr_content"
-
-# ── Test 3: Failure, VERBOSE=1 -- log on stderr only ─────────────────
-rc=0
-UNSLOTH_VERBOSE=1 try_quiet_fixed "test" bash -c "echo logline_here; exit 1" > "$tmpout" 2> "$tmperr" || rc=$?
-stdout_content=$(cat "$tmpout")
-stderr_content=$(cat "$tmperr")
-assert_contains    "3:fail_verbose_status"  "$stdout_content" "failed"
-assert_contains    "3:fail_verbose_stderr"  "$stderr_content" "logline_here"
-assert_not_contains "3:fail_verbose_clean"  "$stdout_content" "logline_here"
-
-# ── Test 4: Multi-line failure, VERBOSE=1 -- all lines on stderr ──────
-rc=0
-UNSLOTH_VERBOSE=1 try_quiet_fixed "multi" bash -c "echo line_A; echo line_B; echo line_C; exit 1" > "$tmpout" 2> "$tmperr" || rc=$?
-stdout_content=$(cat "$tmpout")
-stderr_content=$(cat "$tmperr")
-assert_contains     "4:multi_stderr_A" "$stderr_content" "line_A"
-assert_contains     "4:multi_stderr_B" "$stderr_content" "line_B"
-assert_contains     "4:multi_stderr_C" "$stderr_content" "line_C"
-assert_not_contains "4:multi_stdout_A" "$stdout_content" "line_A"
-
-# ── Test 5: Exit code preservation ────────────────────────────────────
-rc=0
-UNSLOTH_VERBOSE=0 try_quiet_fixed "test" bash -c "exit 42" > "$tmpout" 2> "$tmperr" || rc=$?
-assert_eq "5:exit_code" "42" "$rc"
-
-# ── Test 6: Binary output in tmplog -- no crash ──────────────────────
-rc=0
-UNSLOTH_VERBOSE=1 try_quiet_fixed "bintest" bash -c "printf '\\x00\\xff\\xfe'; exit 3" > "$tmpout" 2> "$tmperr" || rc=$?
-assert_eq "6:binary_rc" "3" "$rc"
-
-# ── Test 7: Success, VERBOSE=1 -- no error output ────────────────────
-rc=0
-UNSLOTH_VERBOSE=1 try_quiet_fixed "test" true > "$tmpout" 2> "$tmperr" || rc=$?
-stdout_content=$(cat "$tmpout")
-stderr_content=$(cat "$tmperr")
-assert_eq "7:success_verbose_stdout" "" "$stdout_content"
-assert_eq "7:success_verbose_stderr" "" "$stderr_content"
-
-# ── Test 8: Broken version contrast -- log leaks to stdout ───────────
-rc=0
-UNSLOTH_VERBOSE=1 try_quiet_broken "test" bash -c "echo leaked_data; exit 1" > "$tmpout" 2> "$tmperr" || rc=$?
-stdout_content=$(cat "$tmpout")
-assert_contains "8:broken_leaks" "$stdout_content" "leaked_data"
-
-# ── Summary ──────────────────────────────────────────────────────────
-TOTAL=$((PASS + FAIL))
-echo ""
-if [ "$FAIL" -eq 0 ]; then
-    echo "test_try_quiet.sh: $PASS/$TOTAL passed -- ALL PASSED"
-    exit 0
-else
-    echo "test_try_quiet.sh: $PASS/$TOTAL passed, $FAIL FAILED"
-    exit 1
-fi

From f3f7dbf4d927cb099c4dc6c9caf03122c0bdf3c0 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 25 Mar 2026 07:15:19 +0000
Subject: [PATCH 13/16] Show error output on failure instead of suppressing it

- install_python_stack.py: restore _red for patch_package_file
  warnings (was downgraded to _dim)
- setup.ps1: capture winget output and show on failure for CUDA,
  Node, Python, and OpenSSL installs (was piped to Out-Null)
- setup.ps1: always show git pull failure warning, not just in
  verbose mode
---
 studio/install_python_stack.py |  4 ++--
 studio/setup.ps1               | 25 ++++++++++++++++++-------
 2 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py
index c594bab747..0678e299b3 100644
--- a/studio/install_python_stack.py
+++ b/studio/install_python_stack.py
@@ -323,7 +323,7 @@ def patch_package_file(package_name: str, relative_path: str, url: str) -> None:
         text = True,
     )
     if result.returncode != 0:
-        _step(_LABEL, f"package {package_name} not found, skipping patch", _dim)
+        _step(_LABEL, f"package {package_name} not found, skipping patch", _red)
         return
 
     location = None
@@ -333,7 +333,7 @@ def patch_package_file(package_name: str, relative_path: str, url: str) -> None:
             break
 
     if not location:
-        _step(_LABEL, f"could not locate {package_name}", _dim)
+        _step(_LABEL, f"could not locate {package_name}", _red)
         return
 
     dest = Path(location) / relative_path
diff --git a/studio/setup.ps1 b/studio/setup.ps1
index 5fa3030d03..d5a183838e 100644
--- a/studio/setup.ps1
+++ b/studio/setup.ps1
@@ -732,15 +732,19 @@ if (-not $NvccPath) {
             if ($BestVersion) {
                 $prevEAPCuda = $ErrorActionPreference
                 $ErrorActionPreference = "Continue"
-                winget install --id=Nvidia.CUDA --version=$BestVersion -e --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
+                $wingetCudaLog = winget install --id=Nvidia.CUDA --version=$BestVersion -e --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-String
                 $ErrorActionPreference = $prevEAPCuda
+                if ($LASTEXITCODE -ne 0) { Write-Host $wingetCudaLog -ForegroundColor Red }
+                elseif ($script:UnslothVerbose) { Write-SetupVerboseDetail $wingetCudaLog "DarkGray" }
                 Refresh-Environment
                 $NvccPath = Find-Nvcc -MaxVersion $DriverMaxCuda
             } else {
                 step "cuda" "no winget package <= $DriverMaxCuda" "Yellow"
             }
         } else {
-            winget install --id=Nvidia.CUDA -e --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
+            $wingetCudaLog = winget install --id=Nvidia.CUDA -e --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-String
+            if ($LASTEXITCODE -ne 0) { Write-Host $wingetCudaLog -ForegroundColor Red }
+            elseif ($script:UnslothVerbose) { Write-SetupVerboseDetail $wingetCudaLog "DarkGray" }
             Refresh-Environment
             $NvccPath = Find-Nvcc
         }
@@ -865,7 +869,9 @@ if ($IsPipInstall) {
         }
         substep "installing Node LTS (winget)..."
         try {
-            winget install OpenJS.NodeJS.LTS --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
+            $wingetNodeLog = winget install OpenJS.NodeJS.LTS --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-String
+            if ($LASTEXITCODE -ne 0) { Write-Host $wingetNodeLog -ForegroundColor Red }
+            elseif ($script:UnslothVerbose) { Write-SetupVerboseDetail $wingetNodeLog "DarkGray" }
             Refresh-Environment
         } catch {
             step "error" "Node install failed -- https://nodejs.org/" "Red"
@@ -896,7 +902,9 @@ if ($HasPython) {
     substep "installing Python 3.12 (winget)..."
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
-        winget install -e --id Python.Python.3.12 --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
+        $wingetPyLog = winget install -e --id Python.Python.3.12 --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-String
+        if ($LASTEXITCODE -ne 0) { Write-Host $wingetPyLog -ForegroundColor Red }
+        elseif ($script:UnslothVerbose) { Write-SetupVerboseDetail $wingetPyLog "DarkGray" }
         Refresh-Environment
     }
     $HasPython = $null -ne (Get-Command python -ErrorAction SilentlyContinue)
@@ -1406,7 +1414,9 @@ if ($OpenSslRoot) {
     substep "OpenSSL dev (winget)..."
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
-        winget install -e --id ShiningLight.OpenSSL.Dev --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
+        $wingetSslLog = winget install -e --id ShiningLight.OpenSSL.Dev --accept-package-agreements --accept-source-agreements 2>&1 | Out-String
+        if ($LASTEXITCODE -ne 0) { Write-Host $wingetSslLog -ForegroundColor Red }
+        elseif ($script:UnslothVerbose) { Write-SetupVerboseDetail $wingetSslLog "DarkGray" }
         foreach ($root in $OpenSslRoots) {
             if (Test-Path (Join-Path $root 'include\openssl\ssl.h')) {
                 $OpenSslRoot = $root
@@ -1515,9 +1525,10 @@ if ((Test-Path $LlamaServerBin) -and -not $NeedRebuild) {
 
     if (Test-Path (Join-Path $LlamaCppDir ".git")) {
         substep "git pull llama.cpp..."
-        git -C $LlamaCppDir pull 2>&1 | Out-Null
-        if ($LASTEXITCODE -ne 0 -and $script:UnslothVerbose) {
+        $gitPullLog = git -C $LlamaCppDir pull 2>&1 | Out-String
+        if ($LASTEXITCODE -ne 0) {
             substep "git pull failed; using tree on disk" "Yellow"
+            if ($script:UnslothVerbose) { Write-SetupVerboseDetail $gitPullLog "Yellow" }
         }
     } else {
         substep "git clone llama.cpp..."

From 81a963a08f874fcb9627706a11de19adc2c3dd74 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 25 Mar 2026 09:01:52 +0000
Subject: [PATCH 14/16] Show winget error output for Git and CMake installs on
 failure

Same capture-and-print-on-failure pattern already used for
Node, Python, CUDA, and OpenSSL winget installs.
---
 studio/setup.ps1 | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/studio/setup.ps1 b/studio/setup.ps1
index d5a183838e..49f56a82b6 100644
--- a/studio/setup.ps1
+++ b/studio/setup.ps1
@@ -494,7 +494,9 @@ if (-not $HasGit) {
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
         try {
-            winget install Git.Git --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
+            $wingetGitLog = winget install Git.Git --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-String
+            if ($LASTEXITCODE -ne 0) { Write-Host $wingetGitLog -ForegroundColor Red }
+            elseif ($script:UnslothVerbose) { Write-SetupVerboseDetail $wingetGitLog "DarkGray" }
             Refresh-Environment
             $HasGit = $null -ne (Get-Command git -ErrorAction SilentlyContinue)
         } catch { }
@@ -515,7 +517,9 @@ if (-not $HasCmake) {
     $HasWinget = $null -ne (Get-Command winget -ErrorAction SilentlyContinue)
     if ($HasWinget) {
         try {
-            winget install Kitware.CMake --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-Null
+            $wingetCmakeLog = winget install Kitware.CMake --source winget --accept-package-agreements --accept-source-agreements 2>&1 | Out-String
+            if ($LASTEXITCODE -ne 0) { Write-Host $wingetCmakeLog -ForegroundColor Red }
+            elseif ($script:UnslothVerbose) { Write-SetupVerboseDetail $wingetCmakeLog "DarkGray" }
             Refresh-Environment
             $HasCmake = $null -ne (Get-Command cmake -ErrorAction SilentlyContinue)
         } catch { }

From 2a7c58124446aa78b7a8a73baadc41989e56bc89 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Thu, 26 Mar 2026 05:53:59 +0000
Subject: [PATCH 15/16] fix: preserve stderr for _run_quiet error messages in
 setup.sh

The step() helper writes to stdout, but _run_quiet's error header
was originally sent to stderr (>&2). Without the redirect, callers
that separate stdout/stderr would miss the failure headline while
still seeing the log body on stderr. Add >&2 to both step calls
inside _run_quiet to match main's behavior.
---
 studio/setup.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/studio/setup.sh b/studio/setup.sh
index 19ed37555b..1fa0521220 100755
--- a/studio/setup.sh
+++ b/studio/setup.sh
@@ -36,7 +36,7 @@ _run_quiet() {
 
     local tmplog
     tmplog=$(mktemp) || {
-        step "error" "Failed to create temporary file" "$C_ERR"
+        step "error" "Failed to create temporary file" "$C_ERR" >&2
         [ "$on_fail" = "exit" ] && exit 1 || return 1
     }
 
@@ -45,7 +45,7 @@ _run_quiet() {
         return 0
     else
         local exit_code=$?
-        step "error" "$label failed (exit code $exit_code)" "$C_ERR"
+        step "error" "$label failed (exit code $exit_code)" "$C_ERR" >&2
         cat "$tmplog" >&2
         rm -f "$tmplog"
 

From b15dc8fe40505af9f0137a43a82e1f9ed7a60041 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Thu, 26 Mar 2026 06:43:27 +0000
Subject: [PATCH 16/16] feat: add --verbose flag to setup and update commands

Wire UNSLOTH_VERBOSE=1 through _run_setup_script() so that
'unsloth studio update --verbose' (and the deprecated 'setup')
passes the flag to setup.sh / setup.ps1 / install_python_stack.py.
---
 unsloth_cli/commands/studio.py | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

diff --git a/unsloth_cli/commands/studio.py b/unsloth_cli/commands/studio.py
index a2f0873e22..f56f582c14 100644
--- a/unsloth_cli/commands/studio.py
+++ b/unsloth_cli/commands/studio.py
@@ -237,31 +237,41 @@ def stop():
 # ── unsloth studio setup / update ─────────────────────────────────────
 
 
-def _run_setup_script() -> None:
+def _run_setup_script(*, verbose: bool = False) -> None:
     """Find and run the studio setup/update script."""
     script = _find_setup_script()
     if not script:
         typer.echo("Error: Could not find setup script (setup.sh / setup.ps1).")
         raise typer.Exit(1)
 
+    env = {**os.environ, "UNSLOTH_VERBOSE": "1"} if verbose else None
+
     if platform.system() == "Windows":
         result = subprocess.run(
             ["powershell", "-ExecutionPolicy", "Bypass", "-File", str(script)],
+            env = env,
         )
     else:
-        result = subprocess.run(["bash", str(script)])
+        result = subprocess.run(["bash", str(script)], env = env)
 
     if result.returncode != 0:
         raise typer.Exit(result.returncode)
 
 
 @studio_app.command(hidden = True)
-def setup():
+def setup(
+    verbose: bool = typer.Option(
+        False,
+        "--verbose",
+        "-v",
+        help = "Full pip/build output during setup for troubleshooting.",
+    ),
+):
     """Deprecated: use 'unsloth studio update' or re-run install.sh."""
     typer.echo(
         "Note: 'unsloth studio setup' is deprecated. Use 'unsloth studio update' or re-run install.sh."
     )
-    _run_setup_script()
+    _run_setup_script(verbose = verbose)
 
 
 @studio_app.command()
@@ -272,6 +282,12 @@ def update(
     package: str = typer.Option(
         "unsloth", "--package", help = "Package name to install/update (for testing)"
     ),
+    verbose: bool = typer.Option(
+        False,
+        "--verbose",
+        "-v",
+        help = "Full pip/build output during update for troubleshooting.",
+    ),
 ):
     """Update Unsloth Studio dependencies and rebuild."""
     os.environ["STUDIO_LOCAL_INSTALL"] = "1" if local else "0"
@@ -281,7 +297,7 @@ def update(
         # have to guess from SCRIPT_DIR (which may be inside site-packages).
         repo_root = Path(__file__).resolve().parents[2]
         os.environ["STUDIO_LOCAL_REPO"] = str(repo_root)
-    _run_setup_script()
+    _run_setup_script(verbose = verbose)
 
 
 # ── unsloth studio reset-password ────────────────────────────────────