code-puppy 0.0.361__py3-none-any.whl → 0.0.362__py3-none-any.whl

code_puppy/command_line/agent_menu.py

@@ -0,0 +1,325 @@
+"""Interactive terminal UI for selecting agents.
+
+Provides a split-panel interface for browsing and selecting agents
+with live preview of agent details.
+"""
+
+import sys
+import time
+from typing import List, Optional, Tuple
+
+from prompt_toolkit.application import Application
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.layout import Dimension, Layout, VSplit, Window
+from prompt_toolkit.layout.controls import FormattedTextControl
+from prompt_toolkit.widgets import Frame
+
+from code_puppy.agents import (
+    get_agent_descriptions,
+    get_available_agents,
+    get_current_agent,
+)
+from code_puppy.tools.command_runner import set_awaiting_user_input
+
+PAGE_SIZE = 10  # Agents per page
+
+
+def _get_agent_entries() -> List[Tuple[str, str, str]]:
+    """Get all agents with their display names and descriptions.
+
+    Returns:
+        List of tuples (agent_name, display_name, description) sorted by name.
+    """
+    available = get_available_agents()
+    descriptions = get_agent_descriptions()
+
+    entries = []
+    for name, display_name in available.items():
+        description = descriptions.get(name, "No description available")
+        entries.append((name, display_name, description))
+
+    # Sort alphabetically by agent name
+    entries.sort(key=lambda x: x[0].lower())
+    return entries
+
+
+def _render_menu_panel(
+    entries: List[Tuple[str, str, str]],
+    page: int,
+    selected_idx: int,
+    current_agent_name: str,
+) -> List:
+    """Render the left menu panel with pagination.
+
+    Args:
+        entries: List of (name, display_name, description) tuples
+        page: Current page number (0-indexed)
+        selected_idx: Currently selected index (global)
+        current_agent_name: Name of the current active agent
+
+    Returns:
+        List of (style, text) tuples for FormattedTextControl
+    """
+    lines = []
+    total_pages = (len(entries) + PAGE_SIZE - 1) // PAGE_SIZE if entries else 1
+    start_idx = page * PAGE_SIZE
+    end_idx = min(start_idx + PAGE_SIZE, len(entries))
+
+    lines.append(("bold", " Agents"))
+    lines.append(("fg:ansibrightblack", f" (Page {page + 1}/{total_pages})"))
+    lines.append(("", "\n\n"))
+
+    if not entries:
+        lines.append(("fg:yellow", "  No agents found."))
+        lines.append(("", "\n\n"))
+    else:
+        # Show agents for current page
+        for i in range(start_idx, end_idx):
+            name, display_name, _ = entries[i]
+            is_selected = i == selected_idx
+            is_current = name == current_agent_name
+
+            # Build the line
+            if is_selected:
+                lines.append(("fg:ansigreen", " ▶ "))
+                lines.append(("fg:ansigreen bold", display_name))
+            else:
+                lines.append(("", "   "))
+                lines.append(("", display_name))
+
+            # Add current marker
+            if is_current:
+                lines.append(("fg:ansicyan", " ← current"))
+
+            lines.append(("", "\n"))
+
+    # Navigation hints
+    lines.append(("", "\n"))
+    lines.append(("fg:ansibrightblack", "  ↑↓ "))
+    lines.append(("", "Navigate\n"))
+    lines.append(("fg:ansibrightblack", "  ←→ "))
+    lines.append(("", "Page\n"))
+    lines.append(("fg:green", "  Enter  "))
+    lines.append(("", "Select\n"))
+    lines.append(("fg:ansibrightred", "  Ctrl+C "))
+    lines.append(("", "Cancel"))
+
+    return lines
+
+
+def _render_preview_panel(
+    entry: Optional[Tuple[str, str, str]],
+    current_agent_name: str,
+) -> List:
+    """Render the right preview panel with agent details.
+
+    Args:
+        entry: Tuple of (name, display_name, description) or None
+        current_agent_name: Name of the current active agent
+
+    Returns:
+        List of (style, text) tuples for FormattedTextControl
+    """
+    lines = []
+
+    lines.append(("dim cyan", " AGENT DETAILS"))
+    lines.append(("", "\n\n"))
+
+    if not entry:
+        lines.append(("fg:yellow", "  No agent selected."))
+        lines.append(("", "\n"))
+        return lines
+
+    name, display_name, description = entry
+    is_current = name == current_agent_name
+
+    # Agent name (identifier)
+    lines.append(("bold", "  Name: "))
+    lines.append(("", name))
+    lines.append(("", "\n\n"))
+
+    # Display name
+    lines.append(("bold", "  Display Name: "))
+    lines.append(("fg:ansicyan", display_name))
+    lines.append(("", "\n\n"))
+
+    # Description
+    lines.append(("bold", "  Description:"))
+    lines.append(("", "\n"))
+
+    # Wrap description to fit panel
+    desc_lines = description.split("\n")
+    for desc_line in desc_lines:
+        # Word wrap long lines
+        words = desc_line.split()
+        current_line = "  "
+        for word in words:
+            if len(current_line) + len(word) + 1 > 60:
+                lines.append(("fg:ansibrightblack", current_line))
+                lines.append(("", "\n"))
+                current_line = "  " + word
+            else:
+                if current_line == "  ":
+                    current_line += word
+                else:
+                    current_line += " " + word
+        if current_line.strip():
+            lines.append(("fg:ansibrightblack", current_line))
+            lines.append(("", "\n"))
+
+    lines.append(("", "\n"))
+
+    # Current status
+    lines.append(("bold", "  Status: "))
+    if is_current:
+        lines.append(("fg:ansigreen bold", "✓ Currently Active"))
+    else:
+        lines.append(("fg:ansibrightblack", "Not active"))
+    lines.append(("", "\n"))
+
+    return lines
+
+
+async def interactive_agent_picker() -> Optional[str]:
+    """Show interactive terminal UI to select an agent.
+
+    Returns:
+        Agent name to switch to, or None if cancelled.
+    """
+    entries = _get_agent_entries()
+    current_agent = get_current_agent()
+    current_agent_name = current_agent.name if current_agent else ""
+
+    if not entries:
+        from code_puppy.messaging import emit_info
+
+        emit_info("No agents found.")
+        return None
+
+    # State
+    selected_idx = [0]  # Current selection (global index)
+    current_page = [0]  # Current page
+    result = [None]  # Selected agent name
+
+    total_pages = (len(entries) + PAGE_SIZE - 1) // PAGE_SIZE
+
+    def get_current_entry() -> Optional[Tuple[str, str, str]]:
+        if 0 <= selected_idx[0] < len(entries):
+            return entries[selected_idx[0]]
+        return None
+
+    # Build UI
+    menu_control = FormattedTextControl(text="")
+    preview_control = FormattedTextControl(text="")
+
+    def update_display():
+        """Update both panels."""
+        menu_control.text = _render_menu_panel(
+            entries, current_page[0], selected_idx[0], current_agent_name
+        )
+        preview_control.text = _render_preview_panel(
+            get_current_entry(), current_agent_name
+        )
+
+    menu_window = Window(
+        content=menu_control, wrap_lines=True, width=Dimension(weight=35)
+    )
+    preview_window = Window(
+        content=preview_control, wrap_lines=True, width=Dimension(weight=65)
+    )
+
+    menu_frame = Frame(menu_window, width=Dimension(weight=35), title="Agents")
+    preview_frame = Frame(preview_window, width=Dimension(weight=65), title="Preview")
+
+    root_container = VSplit(
+        [
+            menu_frame,
+            preview_frame,
+        ]
+    )
+
+    # Key bindings
+    kb = KeyBindings()
+
+    @kb.add("up")
+    def _(event):
+        if selected_idx[0] > 0:
+            selected_idx[0] -= 1
+            # Update page if needed
+            current_page[0] = selected_idx[0] // PAGE_SIZE
+            update_display()
+
+    @kb.add("down")
+    def _(event):
+        if selected_idx[0] < len(entries) - 1:
+            selected_idx[0] += 1
+            # Update page if needed
+            current_page[0] = selected_idx[0] // PAGE_SIZE
+            update_display()
+
+    @kb.add("left")
+    def _(event):
+        if current_page[0] > 0:
+            current_page[0] -= 1
+            selected_idx[0] = current_page[0] * PAGE_SIZE
+            update_display()
+
+    @kb.add("right")
+    def _(event):
+        if current_page[0] < total_pages - 1:
+            current_page[0] += 1
+            selected_idx[0] = current_page[0] * PAGE_SIZE
+            update_display()
+
+    @kb.add("enter")
+    def _(event):
+        entry = get_current_entry()
+        if entry:
+            result[0] = entry[0]  # Store agent name
+        event.app.exit()
+
+    @kb.add("c-c")
+    def _(event):
+        result[0] = None
+        event.app.exit()
+
+    layout = Layout(root_container)
+    app = Application(
+        layout=layout,
+        key_bindings=kb,
+        full_screen=False,
+        mouse_support=False,
+    )
+
+    set_awaiting_user_input(True)
+
+    # Enter alternate screen buffer once for entire session
+    sys.stdout.write("\033[?1049h")  # Enter alternate buffer
+    sys.stdout.write("\033[2J\033[H")  # Clear and home
+    sys.stdout.flush()
+    time.sleep(0.05)
+
+    try:
+        # Initial display
+        update_display()
+
+        # Clear the current buffer
+        sys.stdout.write("\033[2J\033[H")
+        sys.stdout.flush()
+
+        # Run application
+        await app.run_async()
+
+    finally:
+        # Exit alternate screen buffer once at end
+        sys.stdout.write("\033[?1049l")  # Exit alternate buffer
+        sys.stdout.flush()
+        # Reset awaiting input flag
+        set_awaiting_user_input(False)
+
+    # Clear exit message
+    from code_puppy.messaging import emit_info
+
+    emit_info("✓ Exited agent picker")
+
+    return result[0]

code_puppy/command_line/core_commands.py

@@ -6,6 +6,7 @@ discovered by the command registry system.
 
 import os
 
+from code_puppy.command_line.agent_menu import interactive_agent_picker
 from code_puppy.command_line.command_registry import register_command
 from code_puppy.command_line.model_picker_completion import update_model_in_input
 from code_puppy.command_line.motd import print_motd
@@ -398,115 +399,6 @@ def handle_agent_command(command: str) -> bool:
         return True
 
 
-async def interactive_agent_picker() -> str | None:
-    """Show an interactive arrow-key selector to pick an agent (async version).
-
-    Returns:
-        The selected agent name, or None if cancelled
-    """
-    import sys
-    import time
-
-    from rich.console import Console
-    from rich.panel import Panel
-    from rich.text import Text
-
-    from code_puppy.agents import (
-        get_agent_descriptions,
-        get_available_agents,
-        get_current_agent,
-    )
-    from code_puppy.tools.command_runner import set_awaiting_user_input
-    from code_puppy.tools.common import arrow_select_async
-
-    # Load available agents
-    available_agents = get_available_agents()
-    descriptions = get_agent_descriptions()
-    current_agent = get_current_agent()
-
-    # Build choices with current agent indicator and keep track of agent names
-    choices = []
-    agent_names = list(available_agents.keys())
-    for agent_name in agent_names:
-        display_name = available_agents[agent_name]
-        if agent_name == current_agent.name:
-            choices.append(f"✓ {agent_name} - {display_name} (current)")
-        else:
-            choices.append(f"  {agent_name} - {display_name}")
-
-    # Create preview callback to show agent description
-    def get_preview(index: int) -> str:
-        """Get the description for the agent at the given index."""
-        agent_name = agent_names[index]
-        description = descriptions.get(agent_name, "No description available")
-        return description
-
-    # Create panel content
-    panel_content = Text()
-    panel_content.append("🐶 Select an agent to use\n", style="bold cyan")
-    panel_content.append("Current agent: ", style="dim")
-    panel_content.append(f"{current_agent.name}", style="bold green")
-    panel_content.append(" - ", style="dim")
-    panel_content.append(current_agent.display_name, style="bold green")
-    panel_content.append("\n", style="dim")
-    panel_content.append(current_agent.description, style="dim italic")
-
-    # Display panel
-    panel = Panel(
-        panel_content,
-        title="[bold white]Agent Selection[/bold white]",
-        border_style="cyan",
-        padding=(1, 2),
-    )
-
-    # Pause spinners BEFORE showing panel
-    set_awaiting_user_input(True)
-    time.sleep(0.3)  # Let spinners fully stop
-
-    local_console = Console()
-    emit_info("")
-    local_console.print(panel)
-    emit_info("")
-
-    # Flush output before prompt_toolkit takes control
-    sys.stdout.flush()
-    sys.stderr.flush()
-    time.sleep(0.1)
-
-    selected_agent = None
-
-    try:
-        # Final flush
-        sys.stdout.flush()
-
-        # Show arrow-key selector with preview (async version)
-        choice = await arrow_select_async(
-            "💭 Which agent would you like to use?",
-            choices,
-            preview_callback=get_preview,
-        )
-
-        # Extract agent name from choice (remove prefix and suffix)
-        if choice:
-            # Remove the "✓ " or "  " prefix and extract agent name (before " - ")
-            choice_stripped = choice.strip().lstrip("✓").strip()
-            # Split on " - " and take the first part (agent name)
-            agent_name = choice_stripped.split(" - ")[0].strip()
-            # Remove " (current)" suffix if present
-            if agent_name.endswith(" (current)"):
-                agent_name = agent_name[:-10].strip()
-            selected_agent = agent_name
-
-    except (KeyboardInterrupt, EOFError):
-        emit_error("Cancelled by user")
-        selected_agent = None
-
-    finally:
-        set_awaiting_user_input(False)
-
-    return selected_agent
-
-
 async def interactive_model_picker() -> str | None:
     """Show an interactive arrow-key selector to pick a model (async version).
 

{code_puppy-0.0.361.dist-info → code_puppy-0.0.362.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.361
+Version: 0.0.362
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy

{code_puppy-0.0.361.dist-info → code_puppy-0.0.362.dist-info}/RECORD

@@ -67,6 +67,7 @@ code_puppy/api/routers/sessions.py,sha256=GqYRT7IJYPpEdTseLF3FIpbvvD86lIqwwPswL3
 code_puppy/api/templates/terminal.html,sha256=9alh6tTbLyXPDjBvkXw8nEWPXB-m_LIceGGRYpSLuyo,13125
 code_puppy/command_line/__init__.py,sha256=y7WeRemfYppk8KVbCGeAIiTuiOszIURCDjOMZv_YRmU,45
 code_puppy/command_line/add_model_menu.py,sha256=CpURhxPvUhLHLBV_uwH1ODfJ-WAcGklvlsjEf5Vfvg4,43255
+code_puppy/command_line/agent_menu.py,sha256=A63k3jkr9htUmzR4v2ORWcyO-COzvbTwaIgDz_AnVK8,9468
 code_puppy/command_line/attachments.py,sha256=4Q5I2Es4j0ltnz5wjw2z0QXMsiMJvEfWRkPf_lJeITM,13093
 code_puppy/command_line/autosave_menu.py,sha256=de7nOmFmEH6x5T7C95U8N8xgxxeF-l5lgaJzGJsF3ZY,19824
 code_puppy/command_line/clipboard.py,sha256=oe9bfAX5RnT81FiYrDmhvHaePS1tAT-NFG1fSXubSD4,16869
@@ -74,7 +75,7 @@ code_puppy/command_line/colors_menu.py,sha256=LoFVfJ-Mo-Eq9hnb2Rj5mn7oBCnadAGr-8
 code_puppy/command_line/command_handler.py,sha256=CY9F27eovZJK_kpU1YmbroYLWGTCuouCOQ-TXfDp-nw,10916
 code_puppy/command_line/command_registry.py,sha256=qFySsw1g8dol3kgi0p6cXrIDlP11_OhOoaQ5nAadWXg,4416
 code_puppy/command_line/config_commands.py,sha256=qS9Cm758DPz2QGvHLhAV4Tp_Xfgo3PyoCoLDusbnmCw,25742
-code_puppy/command_line/core_commands.py,sha256=q5yi_xjPSoCjX8ET4PnMVsZlGwJ2PlEXVnao_7_1rJk,30619
+code_puppy/command_line/core_commands.py,sha256=OF3u5MKmj78zEvX15k5BdcU9CbVTEJJYKz24HX_Dmtk,27088
 code_puppy/command_line/diff_menu.py,sha256=_Gr9SP9fbItk-08dya9WTAR53s_PlyAvEnbt-8VWKPk,24141
 code_puppy/command_line/file_path_completion.py,sha256=gw8NpIxa6GOpczUJRyh7VNZwoXKKn-yvCqit7h2y6Gg,2931
 code_puppy/command_line/load_context_completion.py,sha256=a3JvLDeLLSYxVgTjAdqWzS4spjv6ccCrK2LKZgVJ1IM,2202
@@ -206,10 +207,10 @@ code_puppy/tools/browser/chromium_terminal_manager.py,sha256=w1thQ_ACb6oV45L93TS
 code_puppy/tools/browser/terminal_command_tools.py,sha256=9byOZku-dwvTtCl532xt7Lumed_jTn0sLvUe_X75XCQ,19068
 code_puppy/tools/browser/terminal_screenshot_tools.py,sha256=J_21YO_495NvYgNFu9KQP6VYg2K_f8CtSdZuF94Yhnw,18448
 code_puppy/tools/browser/terminal_tools.py,sha256=F5LjVH3udSCFHmqC3O1UJLoLozZFZsEdX42jOmkqkW0,17853
-code_puppy-0.0.361.data/data/code_puppy/models.json,sha256=FMQdE_yvP_8y0xxt3K918UkFL9cZMYAqW1SfXcQkU_k,3105
-code_puppy-0.0.361.data/data/code_puppy/models_dev_api.json,sha256=wHjkj-IM_fx1oHki6-GqtOoCrRMR0ScK0f-Iz0UEcy8,548187
-code_puppy-0.0.361.dist-info/METADATA,sha256=AkNpl1g19rE0ZEPI-0MWepw8wFq_zLHhe8o_dVlYaRQ,27614
-code_puppy-0.0.361.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-code_puppy-0.0.361.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
-code_puppy-0.0.361.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.361.dist-info/RECORD,,
+code_puppy-0.0.362.data/data/code_puppy/models.json,sha256=FMQdE_yvP_8y0xxt3K918UkFL9cZMYAqW1SfXcQkU_k,3105
+code_puppy-0.0.362.data/data/code_puppy/models_dev_api.json,sha256=wHjkj-IM_fx1oHki6-GqtOoCrRMR0ScK0f-Iz0UEcy8,548187
+code_puppy-0.0.362.dist-info/METADATA,sha256=3o2P9erqXHD7H0pAfDy_DTolwLVEkDatTTPDzHS2hY8,27614
+code_puppy-0.0.362.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+code_puppy-0.0.362.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
+code_puppy-0.0.362.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.362.dist-info/RECORD,,