Skip to content

Commit 1df4ecc

Browse files
committed
Spruce up the startup banner
Signed-off-by: Brianna Major <brianna.major@kitware.com>
1 parent 083eded commit 1df4ecc

1 file changed

Lines changed: 15 additions & 3 deletions

File tree

src/vtk_prompt/vtk_prompt_ui.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
"""
1818

1919
import sys
20+
import textwrap
2021
from typing import Any
2122

2223
import vtk
@@ -261,11 +262,22 @@ def start(self) -> None:
261262
self.server.start()
262263

263264

265+
def _print_startup_banner() -> None:
266+
"""Print a styled startup banner with ANSI colors."""
267+
banner = textwrap.dedent("""\
268+
\033[36m╭──────────────────────────────────────────────────────────────────────╮
269+
\033[1mVTK Prompt UI\033[0m\033[36m │
270+
\033[97mEnter your API token in the application settings.\033[36m │
271+
\033[97mSupported providers: OpenAI, Anthropic, Google Gemini, NVIDIA NIM\033[36m │
272+
\033[97mFor local Ollama, use custom base URL and model configuration.\033[36m │
273+
╰──────────────────────────────────────────────────────────────────────╯\033[0m
274+
""")
275+
print(banner)
276+
277+
264278
def main() -> None:
265279
"""Start the trame app."""
266-
print("VTK Prompt UI - Enter your API token in the application settings.")
267-
print("Supported providers: OpenAI, Anthropic, Google Gemini, NVIDIA NIM")
268-
print("For local Ollama, use custom base URL and model configuration.")
280+
_print_startup_banner()
269281

270282
# Check for custom prompt file in CLI arguments
271283
custom_prompt_file = None

0 commit comments

Comments
 (0)