agentcrew-ai 0.8.13__py3-none-any.whl → 0.9.1__py3-none-any.whl

AgentCrew/modules/console/{conversation_browser.py → conversation_browser/browser_ui.py}

@@ -1,10 +1,8 @@
-"""Conversation browser with split-panel interface.
-Provides Rich-based UI for listing and loading conversations with preview.
-"""
+"""Conversation browser UI rendering components."""
 
 from __future__ import annotations
 
-from typing import List, Dict, Any, Optional, Callable, Tuple, Tuple
+from typing import List, Dict, Any, Optional, Callable, Tuple
 from datetime import datetime
 
 from rich.console import Console, Group
@@ -14,10 +12,11 @@ from rich.text import Text
 from rich.layout import Layout
 from rich.rule import Rule
 from rich.box import ROUNDED
+from rich.live import Live
 
 from loguru import logger
 
-from .constants import (
+from ..constants import (
     RICH_STYLE_YELLOW,
     RICH_STYLE_YELLOW_BOLD,
     RICH_STYLE_BLUE,
@@ -28,8 +27,8 @@ from .constants import (
 )
 
 
-class ConversationBrowser:
-    """Interactive conversation browser with split-panel layout."""
+class ConversationBrowserUI:
+    """Handles UI rendering for the conversation browser."""
 
     def __init__(
         self,
@@ -38,28 +37,125 @@ class ConversationBrowser:
             Callable[[str], List[Dict[str, Any]]]
         ] = None,
     ):
-        """Initialize the conversation browser.
-
-        Args:
-            console: Rich console for rendering
-            get_conversation_history: Optional callback to fetch full conversation history
-        """
         self.console = console
         self.conversations: List[Dict[str, Any]] = []
+        self._all_conversations: List[Dict[str, Any]] = []
         self.selected_index = 0
         self.scroll_offset = 0
-        self.max_list_items = 50
-        self._running = False
         self._get_conversation_history = get_conversation_history
         self._preview_cache: Dict[str, Tuple[List[Dict[str, Any]], int]] = {}
-        self._g_pressed = False
+        self.selected_items: set[int] = set()
+        self._live: Optional[Live] = None
+        self._layout: Optional[Layout] = None
+        self._search_query: str = ""
+        self._search_mode: bool = False
+
+    @property
+    def max_list_items(self) -> int:
+        return self.console.height - 9
 
     def set_conversations(self, conversations: List[Dict[str, Any]]):
         """Set the conversations list to browse."""
+        self._all_conversations = conversations
         self.conversations = conversations
         self.selected_index = 0
         self.scroll_offset = 0
         self._preview_cache.clear()
+        self.selected_items.clear()
+        self._search_query = ""
+        self._search_mode = False
+
+    @property
+    def search_mode(self) -> bool:
+        return self._search_mode
+
+    @property
+    def search_query(self) -> str:
+        return self._search_query
+
+    def start_search_mode(self):
+        """Enter search mode, preserving previous search query."""
+        self._search_mode = True
+
+    def exit_search_mode(self, clear_filter: bool = False):
+        """Exit search mode."""
+        self._search_mode = False
+        if clear_filter:
+            self._search_query = ""
+            self.conversations = self._all_conversations
+            self.selected_index = 0
+            self.scroll_offset = 0
+            self.selected_items.clear()
+
+    def update_search_query(self, query: str):
+        """Update search query and filter conversations."""
+        self._search_query = query
+        self._filter_conversations()
+
+    def append_search_char(self, char: str):
+        """Append a character to search query."""
+        self._search_query += char
+        self._filter_conversations()
+
+    def backspace_search(self):
+        """Remove last character from search query."""
+        if self._search_query:
+            self._search_query = self._search_query[:-1]
+        self._filter_conversations()
+
+    def _filter_conversations(self):
+        """Filter conversations based on search query."""
+        if not self._search_query:
+            self.conversations = self._all_conversations
+        else:
+            query_lower = self._search_query.lower()
+            self.conversations = [
+                c
+                for c in self._all_conversations
+                if query_lower in c.get("title", "").lower()
+            ]
+        self.selected_index = 0
+        self.scroll_offset = 0
+        self.selected_items.clear()
+
+    def toggle_selection(self, index: Optional[int] = None) -> bool:
+        """Toggle selection state of an item. Returns True if state changed."""
+        idx = index if index is not None else self.selected_index
+        if idx < 0 or idx >= len(self.conversations):
+            return False
+        if idx in self.selected_items:
+            self.selected_items.discard(idx)
+        else:
+            self.selected_items.add(idx)
+        return True
+
+    def clear_selections(self):
+        """Clear all selected items."""
+        self.selected_items.clear()
+
+    def get_selected_conversation_ids(self) -> List[str]:
+        """Get IDs of all selected conversations."""
+        ids = []
+        for idx in sorted(self.selected_items):
+            if 0 <= idx < len(self.conversations):
+                convo_id = self.conversations[idx].get("id")
+                if convo_id:
+                    ids.append(convo_id)
+        return ids
+
+    def remove_conversations(self, indices: List[int]):
+        """Remove conversations at specified indices and update UI state."""
+        for idx in sorted(indices, reverse=True):
+            if 0 <= idx < len(self.conversations):
+                convo_id = self.conversations[idx].get("id")
+                if convo_id:
+                    self._preview_cache.pop(convo_id, None)
+                del self.conversations[idx]
+        self.selected_items.clear()
+        if self.selected_index >= len(self.conversations):
+            self.selected_index = max(0, len(self.conversations) - 1)
+        if self.scroll_offset > 0 and self.scroll_offset >= len(self.conversations):
+            self.scroll_offset = max(0, len(self.conversations) - self.max_list_items)
 
     def _format_timestamp(self, timestamp) -> str:
         """Format timestamp for display."""
@@ -87,8 +183,12 @@ class ConversationBrowser:
         header_table.add_column("right", justify="right", ratio=1)
 
         left_text = Text()
-        left_text.append("📚 ", style="bold")
+        left_text.append("\U0001f4da ", style="bold")
         left_text.append(f"{len(self.conversations)} ", style=RICH_STYLE_GREEN_BOLD)
+        if self._search_query:
+            left_text.append(
+                f"/ {len(self._all_conversations)} ", style=RICH_STYLE_GRAY
+            )
         left_text.append("conversations", style=RICH_STYLE_GRAY)
 
         center_text = Text()
@@ -134,7 +234,7 @@ class ConversationBrowser:
         )
         table.add_column("#", width=5, justify="right", no_wrap=True)
         table.add_column("Title", no_wrap=True, overflow="ellipsis")
-        table.add_column("Date", width=16, justify="right", no_wrap=True)
+        table.add_column("Date", width=10, justify="right", no_wrap=True)
 
         visible_count = min(
             self.max_list_items, len(self.conversations) - self.scroll_offset
@@ -143,31 +243,53 @@ class ConversationBrowser:
         for i in range(visible_count):
             idx = self.scroll_offset + i
             convo = self.conversations[idx]
-            is_selected = idx == self.selected_index
+            is_cursor = idx == self.selected_index
+            is_marked = idx in self.selected_items
 
             index_text = f"{idx + 1}"
-            title = convo.get("title", "Untitled")
+            title = convo.get("title", "Untitled").replace("\n", " ")
             timestamp = self._format_timestamp(convo.get("timestamp"))
 
-            if is_selected:
+            mark_indicator = "\u25cf " if is_marked else "  "
+            cursor_indicator = "\u25b8" if is_cursor else " "
+
+            if is_cursor and is_marked:
+                table.add_row(
+                    Text(index_text, style="bold magenta"),
+                    Text(
+                        f"{mark_indicator}{cursor_indicator}{title}",
+                        style="bold magenta",
+                    ),
+                    Text(timestamp, style="magenta"),
+                )
+            elif is_cursor:
                 table.add_row(
                     Text(index_text, style=RICH_STYLE_GREEN_BOLD),
-                    Text(f"▸ {title}", style=RICH_STYLE_GREEN_BOLD),
+                    Text(
+                        f"{mark_indicator}{cursor_indicator}{title}",
+                        style=RICH_STYLE_GREEN_BOLD,
+                    ),
                     Text(timestamp, style=RICH_STYLE_GREEN),
                 )
+            elif is_marked:
+                table.add_row(
+                    Text(index_text, style="magenta"),
+                    Text(f"{mark_indicator} {title}", style="magenta"),
+                    Text(timestamp, style="magenta"),
+                )
             else:
                 table.add_row(
                     Text(index_text, style=RICH_STYLE_GRAY),
-                    Text(f"  {title}", style=RICH_STYLE_BLUE),
+                    Text(f"{mark_indicator} {title}", style=RICH_STYLE_BLUE),
                     Text(timestamp, style=RICH_STYLE_GRAY),
                 )
 
         scroll_parts = []
         if self.scroll_offset > 0:
-            scroll_parts.append(f"↑{self.scroll_offset}")
+            scroll_parts.append(f"\u2191{self.scroll_offset}")
         remaining = len(self.conversations) - self.scroll_offset - visible_count
         if remaining > 0:
-            scroll_parts.append(f"↓{remaining}")
+            scroll_parts.append(f"\u2193{remaining}")
 
         subtitle = None
         if scroll_parts:
@@ -266,7 +388,7 @@ class ConversationBrowser:
         preview_lines = []
 
         title = convo.get("title", "Untitled")
-        preview_lines.append(Text(f"📌 {title}", style=RICH_STYLE_YELLOW_BOLD))
+        preview_lines.append(Text(f"\U0001f4cc {title}", style=RICH_STYLE_YELLOW_BOLD))
 
         convo_id = convo.get("id", "unknown")
         timestamp = self._format_timestamp(convo.get("timestamp"))
@@ -275,7 +397,7 @@ class ConversationBrowser:
         meta_table.add_column("key", style=RICH_STYLE_GRAY)
         meta_table.add_column("value", style=RICH_STYLE_WHITE)
 
-        display_id = convo_id[:24] + "…" if len(convo_id) > 24 else convo_id
+        display_id = convo_id[:24] + "\u2026" if len(convo_id) > 24 else convo_id
         meta_table.add_row("ID:", display_id)
         meta_table.add_row("Created:", timestamp)
 
@@ -297,13 +419,13 @@ class ConversationBrowser:
                 max_content_len = 120
                 content_display = content.replace("\n", " ").strip()
                 if len(content_display) > max_content_len:
-                    content_display = content_display[:max_content_len] + "…"
+                    content_display = content_display[:max_content_len] + "\u2026"
 
                 preview_lines.append(Text(""))
 
                 if role == "user":
                     user_header = Text()
-                    user_header.append("👤 ", style="bold")
+                    user_header.append("\U0001f464 ", style="bold")
                     user_header.append("User", style=RICH_STYLE_BLUE)
                     preview_lines.append(user_header)
                     preview_lines.append(
@@ -311,7 +433,7 @@ class ConversationBrowser:
                     )
                 else:
                     assistant_header = Text()
-                    assistant_header.append("🤖 ", style="bold")
+                    assistant_header.append("\U0001f916 ", style="bold")
                     assistant_header.append("Assistant", style=RICH_STYLE_GREEN)
                     preview_lines.append(assistant_header)
                     preview_lines.append(
@@ -326,7 +448,9 @@ class ConversationBrowser:
                 preview_lines.append(Text(""))
                 preview_lines.append(Rule(style=RICH_STYLE_GRAY))
                 preview_lines.append(
-                    Text(f"  … and {remaining} more messages", style=RICH_STYLE_GRAY)
+                    Text(
+                        f"  \u2026 and {remaining} more messages", style=RICH_STYLE_GRAY
+                    )
                 )
         else:
             basic_preview = convo.get("preview", "No preview available")
@@ -342,6 +466,9 @@ class ConversationBrowser:
 
     def _create_help_panel(self) -> Panel:
         """Create the help panel with keyboard shortcuts."""
+        if self._search_mode:
+            return self._create_search_bar()
+
         help_table = Table(
             show_header=False,
             box=None,
@@ -353,9 +480,9 @@ class ConversationBrowser:
         help_table.add_column("section3", justify="right", ratio=1)
 
         nav_text = Text()
-        nav_text.append("↑/k ", style=RICH_STYLE_GREEN_BOLD)
+        nav_text.append("\u2191/k ", style=RICH_STYLE_GREEN_BOLD)
         nav_text.append("Up  ", style=RICH_STYLE_GRAY)
-        nav_text.append("↓/j ", style=RICH_STYLE_GREEN_BOLD)
+        nav_text.append("\u2193/j ", style=RICH_STYLE_GREEN_BOLD)
         nav_text.append("Down  ", style=RICH_STYLE_GRAY)
         nav_text.append("gg ", style=RICH_STYLE_GREEN_BOLD)
         nav_text.append("Top  ", style=RICH_STYLE_GRAY)
@@ -365,14 +492,20 @@ class ConversationBrowser:
         action_text = Text()
         action_text.append("Enter/l ", style=RICH_STYLE_GREEN_BOLD)
         action_text.append("Load  ", style=RICH_STYLE_GRAY)
-        action_text.append("Esc/q ", style=RICH_STYLE_GREEN_BOLD)
-        action_text.append("Exit", style=RICH_STYLE_GRAY)
+        action_text.append("v ", style=RICH_STYLE_GREEN_BOLD)
+        action_text.append("Select  ", style=RICH_STYLE_GRAY)
+        action_text.append("dd ", style=RICH_STYLE_GREEN_BOLD)
+        action_text.append("Delete", style=RICH_STYLE_GRAY)
 
         page_text = Text()
-        page_text.append("PgUp/Ctrl+U ", style=RICH_STYLE_GREEN_BOLD)
-        page_text.append("Page Up  ", style=RICH_STYLE_GRAY)
-        page_text.append("PgDn/Ctrl+D ", style=RICH_STYLE_GREEN_BOLD)
-        page_text.append("Page Down", style=RICH_STYLE_GRAY)
+        page_text.append("/ ", style=RICH_STYLE_GREEN_BOLD)
+        page_text.append("Search  ", style=RICH_STYLE_GRAY)
+        page_text.append("Esc/q ", style=RICH_STYLE_GREEN_BOLD)
+        page_text.append("Exit", style=RICH_STYLE_GRAY)
+        if self.selected_items:
+            page_text.append(
+                f"  ({len(self.selected_items)} selected)", style="magenta"
+            )
 
         help_table.add_row(nav_text, action_text, page_text)
 
@@ -382,8 +515,38 @@ class ConversationBrowser:
             box=ROUNDED,
         )
 
-    def _render(self):
-        """Render the split-panel interface."""
+    def _create_search_bar(self) -> Panel:
+        """Create the search bar panel."""
+        search_text = Text()
+        search_text.append("/ ", style=RICH_STYLE_GREEN_BOLD)
+        search_text.append(self._search_query, style=RICH_STYLE_WHITE)
+        search_text.append("\u2588", style="blink bold cyan")
+
+        help_text = Text()
+        help_text.append("  Enter ", style=RICH_STYLE_GREEN_BOLD)
+        help_text.append("Confirm  ", style=RICH_STYLE_GRAY)
+        help_text.append("Esc ", style=RICH_STYLE_GREEN_BOLD)
+        help_text.append("Cancel", style=RICH_STYLE_GRAY)
+
+        search_table = Table(
+            show_header=False,
+            box=None,
+            padding=0,
+            expand=True,
+        )
+        search_table.add_column("search", justify="left", ratio=2)
+        search_table.add_column("help", justify="right", ratio=1)
+        search_table.add_row(search_text, help_text)
+
+        return Panel(
+            search_table,
+            border_style="cyan",
+            box=ROUNDED,
+            title=Text("Search ", style=RICH_STYLE_YELLOW_BOLD),
+        )
+
+    def _create_layout(self) -> Layout:
+        """Create the layout structure."""
         layout = Layout()
         layout.split_column(
             Layout(name="header", size=3),
@@ -395,16 +558,52 @@ class ConversationBrowser:
             Layout(name="list", ratio=1, minimum_size=40),
             Layout(name="preview", ratio=1, minimum_size=40),
         )
-
-        layout["header"].update(self._create_header())
-        layout["list"].update(self._create_list_panel())
-        layout["preview"].update(self._create_preview_panel())
-        layout["help"].update(self._create_help_panel())
-
+        return layout
+
+    def _update_layout(self):
+        """Update layout panels with current content."""
+        if self._layout is None:
+            return
+        self._layout["header"].update(self._create_header())
+        self._layout["list"].update(self._create_list_panel())
+        self._layout["preview"].update(self._create_preview_panel())
+        self._layout["help"].update(self._create_help_panel())
+
+    def start_live(self):
+        """Start live display mode."""
         self.console.clear()
-        self.console.print(layout)
-
-    def _handle_navigation(self, direction: str) -> bool:
+        self._layout = self._create_layout()
+        self._update_layout()
+        self._live = Live(
+            self._layout,
+            console=self.console,
+            refresh_per_second=10,
+            screen=True,
+        )
+        self._live.start()
+
+    def stop_live(self):
+        """Stop live display mode."""
+        if self._live:
+            self._live.stop()
+            self._live = None
+        self._layout = None
+
+    def render(self):
+        """Update the display with current state."""
+        if self._live and self._layout:
+            self._update_layout()
+            self._live.refresh()
+        else:
+            layout = self._create_layout()
+            layout["header"].update(self._create_header())
+            layout["list"].update(self._create_list_panel())
+            layout["preview"].update(self._create_preview_panel())
+            layout["help"].update(self._create_help_panel())
+            self.console.clear()
+            self.console.print(layout)
+
+    def handle_navigation(self, direction: str) -> bool:
         """Handle navigation (up/down/top/bottom). Returns True if selection changed."""
         if not self.conversations:
             return False
@@ -442,116 +641,3 @@ class ConversationBrowser:
     def get_selected_conversation_index(self) -> int:
         """Get the 1-based index of the currently selected conversation."""
         return self.selected_index + 1
-
-    def show(self) -> Optional[str]:
-        """Show the interactive conversation browser.
-
-        Returns:
-            The ID of the selected conversation, or None if cancelled.
-        """
-        if not self.conversations:
-            self.console.print(
-                Text("No conversations available.", style=RICH_STYLE_YELLOW)
-            )
-            return None
-
-        self._running = True
-        self._g_pressed = False
-        selected_id = None
-
-        from prompt_toolkit import PromptSession
-        from prompt_toolkit.key_binding import KeyBindings
-        from prompt_toolkit.keys import Keys
-
-        self._render()
-
-        kb = KeyBindings()
-
-        @kb.add(Keys.Up)
-        @kb.add("k")
-        def _(event):
-            self._g_pressed = False
-            if self._handle_navigation("up"):
-                self._render()
-
-        @kb.add(Keys.Down)
-        @kb.add("j")
-        def _(event):
-            self._g_pressed = False
-            if self._handle_navigation("down"):
-                self._render()
-
-        @kb.add(Keys.ControlP)
-        def _(event):
-            self._g_pressed = False
-            if self._handle_navigation("up"):
-                self._render()
-
-        @kb.add(Keys.ControlN)
-        def _(event):
-            self._g_pressed = False
-            if self._handle_navigation("down"):
-                self._render()
-
-        @kb.add("g")
-        def _(event):
-            if self._g_pressed:
-                self._g_pressed = False
-                if self._handle_navigation("top"):
-                    self._render()
-            else:
-                self._g_pressed = True
-
-        @kb.add("G")
-        def _(event):
-            self._g_pressed = False
-            if self._handle_navigation("bottom"):
-                self._render()
-
-        @kb.add(Keys.ControlU)
-        @kb.add(Keys.PageUp)
-        def _(event):
-            self._g_pressed = False
-            if self._handle_navigation("page_up"):
-                self._render()
-
-        @kb.add(Keys.ControlD)
-        @kb.add(Keys.PageDown)
-        def _(event):
-            self._g_pressed = False
-            if self._handle_navigation("page_down"):
-                self._render()
-
-        @kb.add(Keys.Enter)
-        @kb.add("l")
-        def _(event):
-            nonlocal selected_id
-            self._g_pressed = False
-            selected_id = self.get_selected_conversation_id()
-            event.app.exit()
-
-        @kb.add(Keys.Escape)
-        @kb.add("q")
-        def _(event):
-            self._g_pressed = False
-            event.app.exit()
-
-        @kb.add(Keys.ControlC)
-        def _(event):
-            self._g_pressed = False
-            event.app.exit()
-
-        @kb.add(Keys.Any)
-        def _(event):
-            self._g_pressed = False
-
-        try:
-            session = PromptSession(key_bindings=kb)
-            session.prompt("")
-        except (KeyboardInterrupt, EOFError):
-            pass
-        except Exception as e:
-            logger.error(f"Error in conversation browser: {e}")
-
-        self._running = False
-        return selected_id

AgentCrew/modules/console/conversation_handler.py

@@ -4,7 +4,7 @@ Manages conversation loading, listing, and display functionality.
 """
 
 from __future__ import annotations
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Optional
 from rich.text import Text
 
 from .constants import RICH_STYLE_YELLOW, RICH_STYLE_RED
@@ -20,10 +20,16 @@ class ConversationHandler:
 
     def __init__(self, console_ui: ConsoleUI):
         """Initialize the conversation handler."""
+        self._console_ui = console_ui
         self.console = console_ui.console
         self.display_handlers = console_ui.display_handlers
         self._cached_conversations = []
 
+    @property
+    def _message_handler(self):
+        """Get message handler from console UI."""
+        return self._console_ui.message_handler
+
     def handle_load_conversation(self, load_arg: str, message_handler):
         """
         Handle loading a conversation by number or ID.
@@ -92,3 +98,30 @@ class ConversationHandler:
     def get_cached_conversations(self):
         """Get the cached conversations list."""
         return self._cached_conversations
+
+    def get_conversation_history(
+        self, conversation_id: str
+    ) -> Optional[List[Dict[str, Any]]]:
+        """Get conversation history for preview in browser."""
+        if self._message_handler.persistent_service:
+            return self._message_handler.persistent_service.get_conversation_history(
+                conversation_id
+            )
+        return None
+
+    def delete_conversations(self, conversation_ids: List[str]) -> bool:
+        """Delete conversations by their IDs.
+
+        Args:
+            conversation_ids: List of conversation IDs to delete
+
+        Returns:
+            True if all deletions were successful
+        """
+        if not conversation_ids:
+            return False
+        success = True
+        for convo_id in conversation_ids:
+            if not self._message_handler.delete_conversation_by_id(convo_id):
+                success = False
+        return success