aloop 0.1.0__py3-none-any.whl

utils/tui/components.py

@@ -0,0 +1,306 @@
+"""Reusable UI components for the TUI."""
+
+from typing import Any, Dict, Optional
+
+from rich import box
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.panel import Panel
+from rich.text import Text
+
+from utils.tui.theme import Theme
+
+
+class Divider:
+    """A simple horizontal divider."""
+
+    def __init__(self, width: int = 60):
+        """Initialize divider.
+
+        Args:
+            width: Width of the divider in characters
+        """
+        self.width = width
+
+    def render(self, console: Console) -> None:
+        """Render the divider to the console."""
+        colors = Theme.get_colors()
+        console.print(Text("─" * self.width, style=colors.text_muted))
+
+
+class MessageDisplay:
+    """Display messages in Claude Code style - clean and minimal."""
+
+    def __init__(self, console: Console):
+        """Initialize message display.
+
+        Args:
+            console: Rich console instance
+        """
+        self.console = console
+
+    def user_message(self, message: str) -> None:
+        """Display a user message with > prefix.
+
+        Args:
+            message: User message text
+        """
+        colors = Theme.get_colors()
+        # User input with cyan > prefix
+        prefix = Text("> ", style=f"bold {colors.user_input}")
+        content = Text(message, style=colors.user_input)
+        self.console.print(Text.assemble(prefix, content))
+        self.console.print()
+
+    def assistant_message(self, message: str, use_markdown: bool = True) -> None:
+        """Display an assistant message.
+
+        Args:
+            message: Assistant message text
+            use_markdown: Whether to render as markdown
+        """
+        colors = Theme.get_colors()
+        if use_markdown:
+            md = Markdown(message)
+            self.console.print(md)
+        else:
+            self.console.print(Text(message, style=colors.assistant_output))
+        self.console.print()
+
+    def turn_divider(self, turn_number: Optional[int] = None) -> None:
+        """Display a divider between conversation turns.
+
+        Args:
+            turn_number: Optional turn number to display
+        """
+        colors = Theme.get_colors()
+        if turn_number is not None:
+            # Divider with turn number
+            left_line = "─" * 25
+            right_line = "─" * 25
+            turn_text = f" Turn {turn_number} "
+            self.console.print(Text(f"{left_line}{turn_text}{right_line}", style=colors.text_muted))
+        else:
+            # Simple divider
+            self.console.print(Text("─" * 60, style=colors.text_muted))
+        self.console.print()
+
+
+class ToolCallDisplay:
+    """Display tool calls with a clean, professional look."""
+
+    def __init__(self, console: Console):
+        """Initialize tool call display.
+
+        Args:
+            console: Rich console instance
+        """
+        self.console = console
+
+    def show_call(
+        self,
+        tool_name: str,
+        arguments: Dict[str, Any],
+        result: Optional[str] = None,
+        success: bool = True,
+        duration: Optional[float] = None,
+        size: Optional[str] = None,
+    ) -> None:
+        """Display a tool call with its arguments and optional result.
+
+        Args:
+            tool_name: Name of the tool
+            arguments: Tool arguments
+            result: Optional result summary
+            success: Whether the call succeeded
+            duration: Optional duration in seconds
+            size: Optional result size string
+        """
+        colors = Theme.get_colors()
+
+        # Build content
+        lines = []
+
+        # Arguments
+        for key, value in arguments.items():
+            value_str = str(value)
+            if len(value_str) > 60:
+                value_str = value_str[:57] + "..."
+            lines.append(f"  [dim]{key}:[/dim] {value_str}")
+
+        # Status line if result provided
+        if result is not None:
+            lines.append("  " + "─" * 50)
+            status_icon = "✓" if success else "✗"
+            status_color = colors.success if success else colors.error
+            status_parts = [f"[{status_color}]{status_icon}[/{status_color}]"]
+            status_parts.append("Success" if success else "Failed")
+            if size:
+                status_parts.append(f"({size}")
+                if duration:
+                    status_parts[-1] += f", {duration:.1f}s)"
+                else:
+                    status_parts[-1] += ")"
+            elif duration:
+                status_parts.append(f"({duration:.1f}s)")
+            lines.append("  " + " ".join(status_parts))
+
+        content = "\n".join(lines) if lines else ""
+
+        # Create panel
+        panel = Panel(
+            content,
+            title=f"[{colors.tool_accent}]Tool: {tool_name}[/{colors.tool_accent}]",
+            title_align="left",
+            border_style=colors.text_muted,
+            box=box.ROUNDED,
+            padding=(0, 1),
+        )
+        self.console.print(panel)
+
+
+class ThinkingDisplay:
+    """Display thinking/reasoning content."""
+
+    def __init__(self, console: Console, max_preview: int = 300):
+        """Initialize thinking display.
+
+        Args:
+            console: Rich console instance
+            max_preview: Maximum characters to show in preview
+        """
+        self.console = console
+        self.max_preview = max_preview
+
+    def show(
+        self,
+        thinking: str,
+        duration: Optional[float] = None,
+        expanded: bool = False,
+    ) -> None:
+        """Display thinking content.
+
+        Args:
+            thinking: Thinking text
+            duration: Optional duration in seconds
+            expanded: Whether to show full content or preview
+        """
+        if not thinking:
+            return
+
+        colors = Theme.get_colors()
+
+        # Truncate if not expanded
+        if not expanded and len(thinking) > self.max_preview:
+            display_text = thinking[: self.max_preview]
+            display_text += f"\n[dim]... ({len(thinking) - self.max_preview} more chars)[/dim]"
+        else:
+            display_text = thinking
+
+        # Build title
+        title = f"[{colors.thinking_accent}]Thinking[/{colors.thinking_accent}]"
+
+        # Build subtitle with duration
+        subtitle = None
+        if duration:
+            subtitle = f"[dim]Duration: {duration:.1f}s[/dim]"
+
+        panel = Panel(
+            display_text,
+            title=title,
+            subtitle=subtitle,
+            title_align="left",
+            subtitle_align="right",
+            border_style=colors.text_muted,
+            box=box.ROUNDED,
+            padding=(0, 1),
+        )
+        self.console.print(panel)
+
+
+class MemoryStatsDisplay:
+    """Display memory statistics with visual progress bars."""
+
+    def __init__(self, console: Console):
+        """Initialize memory stats display.
+
+        Args:
+            console: Rich console instance
+        """
+        self.console = console
+
+    def _make_progress_bar(
+        self,
+        current: int,
+        total: int,
+        width: int = 16,
+        filled_char: str = "█",
+        empty_char: str = "░",
+    ) -> str:
+        """Create a text-based progress bar.
+
+        Args:
+            current: Current value
+            total: Maximum value
+            width: Width in characters
+            filled_char: Character for filled portion
+            empty_char: Character for empty portion
+
+        Returns:
+            Progress bar string
+        """
+        ratio = 0 if total == 0 else min(current / total, 1.0)
+        filled = int(width * ratio)
+        empty = width - filled
+        return filled_char * filled + empty_char * empty
+
+    def show(self, stats: Dict[str, Any], context_limit: int = 60000) -> None:
+        """Display memory statistics.
+
+        Args:
+            stats: Statistics dictionary from memory manager
+            context_limit: Maximum context window size
+        """
+        colors = Theme.get_colors()
+
+        # Calculate values
+        input_tokens = stats.get("total_input_tokens", 0)
+        output_tokens = stats.get("total_output_tokens", 0)
+        current_tokens = stats.get("current_tokens", 0)
+        compression_count = stats.get("compression_count", 0)
+        net_savings = stats.get("net_savings", 0)
+        total_cost = stats.get("total_cost", 0)
+
+        # Build content
+        lines = []
+        lines.append("  [bold]Tokens[/bold]")
+
+        # Input tokens bar
+        input_bar = self._make_progress_bar(input_tokens, context_limit)
+        lines.append(f"  ├─ Input:    {input_bar}  {input_tokens:,}")
+
+        # Output tokens bar
+        output_bar = self._make_progress_bar(output_tokens, context_limit)
+        lines.append(f"  ├─ Output:   {output_bar}  {output_tokens:,}")
+
+        # Context usage
+        lines.append(f"  └─ Context:  {current_tokens:,} / {context_limit // 1000}K")
+        lines.append("")
+
+        # Summary line
+        savings_str = f"+{net_savings:,}" if net_savings >= 0 else f"{net_savings:,}"
+        lines.append(
+            f"  Cost: ${total_cost:.4f}  │  Compressions: {compression_count}  │  Saved: {savings_str}"
+        )
+
+        content = "\n".join(lines)
+
+        panel = Panel(
+            content,
+            title=f"[{colors.primary}]Memory Statistics[/{colors.primary}]",
+            title_align="left",
+            border_style=colors.text_muted,
+            box=box.ROUNDED,
+            padding=(0, 1),
+        )
+        self.console.print(panel)

utils/tui/input_handler.py

@@ -0,0 +1,393 @@
+"""Enhanced input handling with auto-completion and keyboard shortcuts."""
+
+from typing import Callable, List, Optional
+
+from prompt_toolkit import PromptSession
+from prompt_toolkit.completion import CompleteEvent, Completer, Completion
+from prompt_toolkit.history import FileHistory, InMemoryHistory
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.keys import Keys
+from prompt_toolkit.styles import Style
+
+from utils.tui.command_registry import CommandRegistry
+from utils.tui.theme import Theme
+
+
+def _normalize_command_tree(
+    commands: list[str] | None,
+    command_subcommands: dict[str, dict[str, str]] | None,
+) -> tuple[list[str], dict[str, dict[str, str]]]:
+    top_commands: list[str] = []
+    seen_top: set[str] = set()
+    subcommands: dict[str, dict[str, str]] = {}
+
+    def _add_top(cmd: str) -> None:
+        if cmd in seen_top:
+            return
+        seen_top.add(cmd)
+        top_commands.append(cmd)
+
+    for raw in commands or []:
+        parts = [p for p in raw.strip().split(" ") if p]
+        if not parts:
+            continue
+        top = parts[0]
+        _add_top(top)
+        if len(parts) > 1:
+            sub = " ".join(parts[1:])
+            subcommands.setdefault(top, {}).setdefault(sub, "")
+
+    for top, subs in (command_subcommands or {}).items():
+        _add_top(top)
+        subcommands.setdefault(top, {}).update(subs)
+
+    return top_commands, subcommands
+
+
+class CommandCompleter(Completer):
+    """Auto-completer for slash commands and file paths."""
+
+    def __init__(
+        self,
+        commands: Optional[List[str]] = None,
+        help_texts: Optional[dict[str, str]] = None,
+        command_subcommands: Optional[dict[str, dict[str, str]]] = None,
+        display_texts: Optional[dict[str, str]] = None,
+    ):
+        """Initialize completer.
+
+        Args:
+            commands: List of available commands (without leading /)
+            help_texts: Optional help text per command (same keys as commands)
+            command_subcommands: Optional mapping like {"model": {"edit": "..."}}.
+            display_texts: Optional display text (with leading / and args hints).
+        """
+        default_commands = [
+            "help",
+            "reset",
+            "stats",
+            "resume",
+            "theme",
+            "verbose",
+            "compact",
+            "model",
+            "exit",
+            "quit",
+        ]
+        self.commands, self.command_subcommands = _normalize_command_tree(
+            commands or default_commands,
+            command_subcommands,
+        )
+        self.help_texts = help_texts or {}
+        self.display_texts = display_texts or {}
+
+    def get_completions(self, document, complete_event):
+        """Get completions for the current input.
+
+        Args:
+            document: Current document
+            complete_event: Completion event
+
+        Yields:
+            Completion objects
+        """
+        text = document.text_before_cursor
+
+        # Complete commands starting with /
+        if text.startswith("/"):
+            cmd_text = text[1:]  # Remove leading /
+            if " " in cmd_text:
+                base, _, rest = cmd_text.partition(" ")
+                if base in self.command_subcommands and " " not in rest:
+                    matching = [
+                        sub for sub in self.command_subcommands[base] if sub.startswith(rest)
+                    ]
+                    for sub in matching:
+                        key = f"{base} {sub}".strip()
+                        display = self.display_texts.get(key, f"/{base} {sub}")
+                        yield Completion(
+                            sub,
+                            start_position=-len(rest),
+                            display=display,
+                            display_meta=self._get_command_help(key),
+                        )
+                return
+
+            matching_commands = [cmd for cmd in self.commands if cmd.startswith(cmd_text)]
+            for cmd in matching_commands:
+                yield Completion(
+                    cmd,
+                    start_position=-len(cmd_text),
+                    display=self.display_texts.get(cmd, f"/{cmd}"),
+                    display_meta=self._get_command_help(cmd),
+                )
+
+    def _get_command_help(self, cmd: str) -> str:
+        """Get help text for a command.
+
+        Args:
+            cmd: Command name
+
+        Returns:
+            Help text
+        """
+        help_texts = {
+            "help": "Show available commands",
+            "reset": "Clear conversation memory",
+            "stats": "Show token/memory stats",
+            "resume": "List and resume a previous session",
+            "theme": "Switch color theme",
+            "verbose": "Toggle verbose output",
+            "compact": "Compress conversation memory",
+            "model": "Manage models",
+            "model edit": "Edit `.aloop/models.yaml` (auto-reload on save)",
+            "exit": "Exit interactive mode",
+            "quit": "Same as /exit",
+        }
+        if cmd in self.help_texts:
+            return self.help_texts[cmd]
+        if " " in cmd:
+            base, _, rest = cmd.partition(" ")
+            sub_help = self.command_subcommands.get(base, {}).get(rest)
+            if sub_help:
+                return sub_help
+        return help_texts.get(cmd, "")
+
+
+class InputHandler:
+    """Enhanced input handler with completion, history, and shortcuts."""
+
+    def __init__(
+        self,
+        history_file: Optional[str] = None,
+        commands: Optional[List[str]] = None,
+        command_help: Optional[dict[str, str]] = None,
+        command_subcommands: Optional[dict[str, dict[str, str]]] = None,
+        command_registry: CommandRegistry | None = None,
+    ):
+        """Initialize input handler.
+
+        Args:
+            history_file: Path to history file (None for in-memory)
+            commands: List of available commands
+        """
+        # Set up history
+        history = FileHistory(history_file) if history_file else InMemoryHistory()
+
+        display_texts: dict[str, str] | None = None
+        if command_registry is not None:
+            commands = [c.name for c in command_registry.commands]
+            command_help = command_registry.to_help_map()
+            command_subcommands = command_registry.to_subcommand_map()
+            display_texts = command_registry.to_display_map()
+
+        # Set up completer
+        self.completer = CommandCompleter(
+            commands,
+            help_texts=command_help,
+            command_subcommands=command_subcommands,
+            display_texts=display_texts,
+        )
+
+        # Set up key bindings
+        self.key_bindings = self._create_key_bindings()
+
+        # Callback handlers
+        self._on_clear_screen: Optional[Callable[[], None]] = None
+        self._on_toggle_thinking: Optional[Callable[[], None]] = None
+        self._on_show_stats: Optional[Callable[[], None]] = None
+
+        def bottom_toolbar():
+            text = self.session.default_buffer.text
+            suggestions = self._get_command_suggestions(text)
+            if not suggestions:
+                return ""
+
+            fragments: list[tuple[str, str]] = []
+            fragments.append(("class:toolbar.hint", "Commands: "))
+            for i, (display, help_text) in enumerate(suggestions[:6]):
+                if i:
+                    fragments.append(("class:toolbar.hint", "  "))
+                fragments.append(("class:toolbar.cmd", display))
+                if help_text:
+                    fragments.append(("class:toolbar.hint", f" — {help_text}"))
+            return fragments
+
+        # Create prompt session with auto-complete while typing.
+        # The completer itself is responsible for returning results only for slash commands.
+        self.session: PromptSession = PromptSession(
+            history=history,
+            completer=self.completer,
+            key_bindings=self.key_bindings,
+            complete_while_typing=True,
+            enable_history_search=True,
+            bottom_toolbar=bottom_toolbar,
+        )
+
+        def _on_text_insert(_buffer) -> None:
+            # Best-effort: show completion menu right after typing "/" at the beginning.
+            buf = self.session.default_buffer
+            if buf.text == "/" and buf.cursor_position == 1:
+                buf.start_completion(
+                    select_first=False,
+                    complete_event=CompleteEvent(text_inserted=True),
+                )
+
+        self.session.default_buffer.on_text_insert += _on_text_insert
+
+        def _on_text_changed(_buffer) -> None:
+            # Codex-style: when the input starts with "/", keep the completion menu in sync
+            # with every keystroke. (Some terminals don't refresh completion state reliably
+            # unless we explicitly trigger it.)
+            buf = self.session.default_buffer
+            if buf.text.startswith("/"):
+                buf.start_completion(
+                    select_first=False,
+                    complete_event=CompleteEvent(text_inserted=True),
+                )
+            else:
+                buf.cancel_completion()
+
+        self.session.default_buffer.on_text_changed += _on_text_changed
+
+    def _get_command_suggestions(self, text: str) -> list[tuple[str, str]]:
+        if not text.startswith("/"):
+            return []
+        cmd_text = text[1:]
+        if " " in cmd_text:
+            base, _, rest = cmd_text.partition(" ")
+            if base in self.completer.command_subcommands and " " not in rest:
+                matches = [
+                    f"{base} {sub}"
+                    for sub in self.completer.command_subcommands[base]
+                    if sub.startswith(rest)
+                ]
+                return [
+                    (
+                        self.completer.display_texts.get(cmd, f"/{cmd}"),
+                        self.completer._get_command_help(cmd),
+                    )
+                    for cmd in matches
+                ]
+            return []
+
+        matches = [cmd for cmd in self.completer.commands if cmd.startswith(cmd_text)]
+        return [
+            (
+                self.completer.display_texts.get(cmd, f"/{cmd}"),
+                self.completer._get_command_help(cmd),
+            )
+            for cmd in matches
+        ]
+
+    def _create_key_bindings(self) -> KeyBindings:
+        """Create custom key bindings.
+
+        Returns:
+            KeyBindings instance
+        """
+        kb = KeyBindings()
+
+        @kb.add("/", eager=True)
+        def slash_command(event):
+            """Insert '/' and, when starting a command, show suggestions immediately."""
+            buffer = event.current_buffer
+            at_start = buffer.text == "" and buffer.cursor_position == 0
+            buffer.insert_text("/")
+            if at_start:
+                buffer.start_completion(select_first=False)
+
+        @kb.add(Keys.ControlL)
+        def clear_screen(event):
+            """Clear the screen."""
+            event.app.renderer.clear()
+            if self._on_clear_screen:
+                self._on_clear_screen()
+
+        @kb.add(Keys.ControlT)
+        def toggle_thinking(event):
+            """Toggle thinking display."""
+            if self._on_toggle_thinking:
+                self._on_toggle_thinking()
+
+        @kb.add(Keys.ControlS)
+        def show_stats(event):
+            """Show quick stats."""
+            if self._on_show_stats:
+                self._on_show_stats()
+
+        return kb
+
+    def set_callbacks(
+        self,
+        on_clear_screen: Optional[Callable[[], None]] = None,
+        on_toggle_thinking: Optional[Callable[[], None]] = None,
+        on_show_stats: Optional[Callable[[], None]] = None,
+    ) -> None:
+        """Set callback handlers for keyboard shortcuts.
+
+        Args:
+            on_clear_screen: Callback for Ctrl+L
+            on_toggle_thinking: Callback for Ctrl+T
+            on_show_stats: Callback for Ctrl+S
+        """
+        self._on_clear_screen = on_clear_screen
+        self._on_toggle_thinking = on_toggle_thinking
+        self._on_show_stats = on_show_stats
+
+    def get_style(self) -> Style:
+        """Get prompt_toolkit style based on current theme.
+
+        Returns:
+            Style instance
+        """
+        colors = Theme.get_colors()
+        return Style.from_dict(
+            {
+                "prompt": f"{colors.user_input} bold",
+                "": colors.text_primary,
+                "completion-menu": f"bg:{colors.bg_secondary} {colors.text_primary}",
+                "completion-menu.completion": f"bg:{colors.bg_secondary} {colors.text_primary}",
+                "completion-menu.completion.current": f"bg:{colors.primary} #000000",
+                "completion-menu.meta.completion": f"bg:{colors.bg_secondary} {colors.text_muted}",
+                "completion-menu.meta.completion.current": f"bg:{colors.primary} #000000",
+                "toolbar.hint": colors.text_muted,
+                "toolbar.cmd": f"{colors.primary} bold",
+                "scrollbar.background": colors.bg_secondary,
+                "scrollbar.button": colors.text_muted,
+            }
+        )
+
+    async def prompt_async(self, prompt_text: str = "> ") -> str:
+        """Get input from user asynchronously.
+
+        Args:
+            prompt_text: Prompt text to display
+
+        Returns:
+            User input string
+        """
+        style = self.get_style()
+
+        result = await self.session.prompt_async(
+            [("class:prompt", prompt_text)],
+            style=style,
+        )
+        return result.strip()
+
+    def prompt(self, prompt_text: str = "> ") -> str:
+        """Get input from user synchronously.
+
+        Args:
+            prompt_text: Prompt text to display
+
+        Returns:
+            User input string
+        """
+        style = self.get_style()
+
+        result = self.session.prompt(
+            [("class:prompt", prompt_text)],
+            style=style,
+        )
+        return result.strip()