tui.py

from collections.abc import Generator

from prompt_toolkit import print_formatted_text
from prompt_toolkit.completion import CompleteEvent, Completer, Completion
from prompt_toolkit.document import Document
from prompt_toolkit.formatted_text import HTML

from openhands_cli import __version__
from openhands_cli.pt_style import get_cli_style

DEFAULT_STYLE = get_cli_style()

# Available commands with descriptions
COMMANDS = {
    "/exit": "Exit the application",
    "/help": "Display available commands",
    "/clear": "Clear the screen",
    "/status": "Display conversation details",
    "/new": "Create a new conversation",
}


class CommandCompleter(Completer):
    """Custom completer for commands with interactive dropdown."""

    def get_completions(
        self, document: Document, complete_event: CompleteEvent
    ) -> Generator[Completion, None, None]:
        text = document.text_before_cursor.lstrip()
        if text.startswith("/"):
            for command, description in COMMANDS.items():
                if command.startswith(text):
                    yield Completion(
                        command,
                        start_position=-len(text),
                        display_meta=description,
                        style="bg:ansidarkgray fg:gold",
                    )


def display_banner(session_id: str) -> None:
    print_formatted_text(
        HTML(r"""<gold>
     ___                    _   _                 _
    /  _ \ _ __   ___ _ __ | | | | __ _ _ __   __| |___
    | | | | '_ \ / _ \ '_ \| |_| |/ _` | '_ \ / _` / __|
    | |_| | |_) |  __/ | | |  _  | (_| | | | | (_| \__ \
    \___ /| .__/ \___|_| |_|_| |_|\__,_|_| |_|\__,_|___/
          |_|
    </gold>"""),
        style=DEFAULT_STYLE,
    )

    print_formatted_text(HTML(f"<grey>OpenHands CLI v{__version__}</grey>"))

    print_formatted_text("")
    print_formatted_text(HTML(f"<grey>Initialized conversation {session_id}</grey>"))
    print_formatted_text("")


def display_help() -> None:
    """Display help information about available commands."""
    print_formatted_text("")
    print_formatted_text(HTML("<gold>🤖 OpenHands CLI Help</gold>"))
    print_formatted_text(HTML("<grey>Available commands:</grey>"))
    print_formatted_text("")

    for command, description in COMMANDS.items():
        print_formatted_text(HTML(f"  <white>{command}</white> - {description}"))

    print_formatted_text("")
    print_formatted_text(HTML("<grey>Tips:</grey>"))
    print_formatted_text("  • Type / and press Tab to see command suggestions")
    print_formatted_text("  • Use arrow keys to navigate through suggestions")
    print_formatted_text("  • Press Enter to select a command")
    print_formatted_text("")