chore: further enhance console UI

Author: daltonnyx

AgentCrew/__init__.py

@@ -1 +1 @@
-__version__ = "0.7.7"
+__version__ = "0.7.8"

AgentCrew/modules/console/confirmation_handler.py

@@ -7,6 +7,7 @@
 
 from rich.text import Text
 from rich.panel import Panel
+from rich.box import HORIZONTALS
 from rich.console import Group
 import time
 
@@ -65,6 +66,7 @@ def display_tool_confirmation_request(self, tool_info, message_handler):
             Panel(
                 Group(*tool_texts_group),
                 title=header,
+                box=HORIZONTALS,
                 title_align="left",
                 border_style=RICH_STYLE_YELLOW,
             )

AgentCrew/modules/console/console_ui.py

@@ -17,6 +17,7 @@
 
 from .constants import (
     RICH_STYLE_GREEN,
+    RICH_STYLE_BLUE,
     RICH_STYLE_YELLOW,
     RICH_STYLE_GREEN_BOLD,
     RICH_STYLE_YELLOW_BOLD,
@@ -86,7 +87,7 @@ def listen(self, event: str, data: Any = None):
                 data
             )  # data is the thinking chunk
         elif event == "user_message_created":
-            self.console.print("\n")
+            pass
         elif event == "response_chunk":
             _, assistant_response = data
             if (
@@ -298,6 +299,14 @@ def _handle_terminal_resize(self, signum, frame):
             self.message_handler.agent.get_model(),
             self.message_handler.tool_manager.get_effective_yolo_mode(),
         )
+
+        self.display_handlers.print_divider("👤 YOU: ")
+
+        prompt = Text(
+            "  ",
+            style=RICH_STYLE_BLUE,
+        )
+        self.console.print(prompt, end="")
         self._is_resizing = False
 
     def start_streaming_response(self, agent_name: str):
@@ -424,7 +433,7 @@ def start(self):
                         )
                         self.input_handler.stop()
                         raise SystemExit(0)
-                    if user_input.strip() == "/list":
+                    elif user_input.strip() == "/list":
                         conversations = self.message_handler.list_conversations()
                         self.conversation_handler.update_cached_conversations(
                             conversations
@@ -526,8 +535,11 @@ def start(self):
                         self.command_handlers.handle_edit_config_command()
                         continue
 
+                    elif user_input.startswith("/voice"):
+                        self.input_handler._stop_input_thread()
+                        self.voice_recording = True
                     # Start loading animation while waiting for response
-                    if (
+                    elif (
                         not user_input.startswith("/")
                         or user_input.startswith("/file ")
                         or user_input.startswith("/consolidate ")
@@ -537,10 +549,6 @@ def start(self):
                     ):
                         self.start_loading_animation()
 
-                    if user_input.startswith("/voice"):
-                        self.input_handler._stop_input_thread()
-                        self.voice_recording = True
-
                     # Process user input and commands
                     should_exit, was_cleared = asyncio.run(
                         self.message_handler.process_user_input(user_input)

AgentCrew/modules/console/display_handlers.py

@@ -10,6 +10,7 @@
 from datetime import datetime
 from typing import Dict, Any, List
 from rich.console import Group
+from rich.box import HORIZONTALS, SQUARE
 from rich.markdown import Markdown
 from rich.text import Text
 from rich.panel import Panel
@@ -22,6 +23,7 @@
     RICH_STYLE_GRAY,
     RICH_STYLE_YELLOW_BOLD,
     RICH_STYLE_GREEN_BOLD,
+    RICH_STYLE_BLUE_BOLD,
     RICH_STYLE_FILE_ACCENT_BOLD,
     RICH_STYLE_WHITE,
     CODE_THEME,
@@ -75,11 +77,12 @@ def display_message(self, message: Text):
     def display_user_message(self, message: str):
         header = Text(
             "👤 YOU:",
-            style=RICH_STYLE_GREEN_BOLD,
+            style=RICH_STYLE_BLUE_BOLD,
         )
         user_panel = Panel(
             Text(message),
             title=header,
+            box=HORIZONTALS,
             title_align="left",
             border_style=RICH_STYLE_BLUE,
         )
@@ -93,6 +96,7 @@ def display_assistant_message(self, agent_name: str, message: str):
         assistant_panel = Panel(
             Markdown(message, code_theme=CODE_THEME),
             title=header,
+            box=HORIZONTALS,
             title_align="left",
             border_style=RICH_STYLE_GREEN,
         )
@@ -102,6 +106,14 @@ def display_divider(self):
         """Display a divider line."""
         pass
 
+    def print_divider(self, title=""):
+        """Display a divider line."""
+        title_length = len(title) + 1 if title else 0
+        self.console.print(
+            " ─" + title + ("─" * (self.console.width - title_length - 3)) + " ",
+            style=RICH_STYLE_BLUE,
+        )
+
     def display_debug_info(self, debug_info):
         """Display debug information."""
         self.console.print(Text("Current messages:", style=RICH_STYLE_YELLOW))
@@ -248,7 +260,7 @@ def display_token_usage(
             f"Total: {input_tokens + output_tokens:,} | Cost: ${total_cost:.4f} | Total: {session_cost:.4f}",
             style=RICH_STYLE_YELLOW,
         )
-        self.console.print(Panel(token_info))
+        self.console.print(Panel(token_info, box=HORIZONTALS))
         self.display_divider()
 
     def display_added_files(self):
@@ -263,10 +275,6 @@ def display_added_files(self):
     def print_welcome_message(self, version: str):
         """Print the welcome message for the chat."""
         welcome_messages = Group(
-            Text(
-                "🎮 Welcome to AgentCrew v" + version + " interactive chat!",
-                style=RICH_STYLE_YELLOW_BOLD,
-            ),
             Text("Press Ctrl+C twice to exit.", style=RICH_STYLE_GRAY),
             Text("Type '/exit' or '/quit' to end the session.", style=RICH_STYLE_GRAY),
             Text(
@@ -353,7 +361,16 @@ def print_welcome_message(self, version: str):
             ),
         )
 
-        self.console.print(Panel(welcome_messages))
+        self.console.print(
+            Panel(
+                welcome_messages,
+                box=SQUARE,
+                title=Text(
+                    "🎮 Welcome to AgentCrew v" + version + " interactive chat!",
+                    style=RICH_STYLE_YELLOW_BOLD,
+                ),
+            )
+        )
         self.display_divider()
 
     def print_prompt_prefix(

AgentCrew/modules/console/input_handler.py

@@ -179,20 +179,28 @@ def _(event):
                         self.message_handler.agent.get_model(),
                         self.message_handler.tool_manager.get_effective_yolo_mode(),
                     )
-                    prompt = Text("👤 YOU: ", style=RICH_STYLE_BLUE)
-                    self.console.print(prompt, end="")
                     time.sleep(0.2)
                     self.clear_buffer()
+                    prompt = Text("  ", style=RICH_STYLE_BLUE)
+                    self.console.print(prompt, end="")
 
         @kb.add(Keys.Backspace)
         def _(event):
             if not event.current_buffer.text:
                 prompt = Text(
-                    "👤 YOU: " if not self.is_message_processing else "",
+                    "  " if not self.is_message_processing else "",
                     style=RICH_STYLE_BLUE,
                 )
-                self.console.print("", end="\r")
-                self.console.print(prompt, end="")
+                if not self.is_message_processing:
+                    import sys
+
+                    sys.stdout.write("\x1b[1A")  # cursor up one line
+                    sys.stdout.write("\x1b[2K")
+                    sys.stdout.write("\r")
+                    self.display_handlers.print_divider("👤 YOU: ")
+                    self.console.print(prompt, end="")
+                else:
+                    self.console.print("", end="\r")
             else:
                 event.current_buffer.delete_before_cursor()
 
@@ -244,7 +252,9 @@ def _(event):
     def clear_buffer(self):
         if self._current_prompt_session:
             self._current_prompt_session.app.current_buffer.reset()
-            self._current_prompt_session.message = HTML("<ansiblue>👤 YOU:</ansiblue> ")
+            if not self.is_message_processing:
+                self.display_handlers.print_divider("👤 YOU: ")
+            self._current_prompt_session.message = HTML("  ")
             self._current_prompt_session.app.invalidate()
 
     def get_choice_input(self, message: str, values: list[str], default=None) -> str:
@@ -301,11 +311,9 @@ def _input_thread_worker(self):
                 )
                 self._current_prompt_session = session
 
-                prompt_text = (
-                    HTML("<ansiblue>👤 YOU:</ansiblue> ")
-                    if not self.is_message_processing
-                    else ""
-                )
+                if not self.is_message_processing:
+                    self.display_handlers.print_divider("👤 YOU: ")
+                prompt_text = HTML("  ") if not self.is_message_processing else ""
                 user_input = session.prompt(prompt_text)
 
                 self.message_handler.history_manager.reset_position()
@@ -393,6 +401,8 @@ def get_user_input(self):
                 if user_input is None:
                     continue
 
+                self.display_handlers.print_divider()
+
                 if user_input == "__EXIT__":
                     self.console.print(
                         Text(
@@ -417,7 +427,6 @@ def get_user_input(self):
                     )
                     return ""
                 else:
-                    self.display_handlers.display_divider()
                     return user_input
 
             except queue.Empty:

AgentCrew/modules/console/tool_display.py

@@ -8,6 +8,7 @@
 from typing import Dict
 from rich.console import Group
 from rich.panel import Panel
+from rich.box import HORIZONTALS
 from rich.text import Text
 
 from .constants import (
@@ -90,6 +91,7 @@ def display_tool_use(self, tool_use: Dict):
             Panel(
                 Group(*tool_texts_group),
                 title=header,
+                box=HORIZONTALS,
                 title_align="left",
             )
         )
@@ -124,7 +126,12 @@ def display_tool_result(self, data: Dict):
                 tool_texts_group.append(result_line)
 
         self.console.print(
-            Panel(Group(*tool_texts_group), title=header, title_align="left")
+            Panel(
+                Group(*tool_texts_group),
+                box=HORIZONTALS,
+                title=header,
+                title_align="left",
+            )
         )
 
     def display_tool_error(self, data: Dict):
@@ -145,6 +152,7 @@ def display_tool_error(self, data: Dict):
         self.console.print(
             Panel(
                 Group(*tool_texts_group),
+                box=HORIZONTALS,
                 title=header,
                 title_align="left",
                 border_style=RICH_STYLE_RED,

AgentCrew/modules/console/ui_effects.py

@@ -10,6 +10,7 @@
 import itertools
 import random
 from rich.live import Live
+from rich.box import HORIZONTALS
 from rich.markdown import Markdown
 from rich.panel import Panel
 
@@ -66,6 +67,10 @@ def _loading_animation(self, stop_event):
                 time.sleep(0.1)  # Control animation speed
             live.update("")  # Clear the live display when done
             live.stop()  # Stop the live display
+            import sys
+
+            sys.stdout.write("\x1b[1A")  # cursor up one line
+            sys.stdout.write("\x1b[2K")
 
     def start_loading_animation(self):
         """Start the loading animation."""
@@ -95,7 +100,9 @@ def start_streaming_response(self, agent_name: str):
 
         header = Text(f"🤖 {agent_name.upper()}:", style=RICH_STYLE_GREEN_BOLD)
 
-        live_panel = Panel("", title=header, border_style=RICH_STYLE_GREEN)
+        live_panel = Panel(
+            "", title=header, box=HORIZONTALS, border_style=RICH_STYLE_GREEN
+        )
 
         self.live = Live(
             live_panel,
@@ -151,6 +158,7 @@ def update_live_display(self, chunk: str):
             live_panel = Panel(
                 Markdown("\n".join(lines), code_theme=CODE_THEME),
                 title=header,
+                box=HORIZONTALS,
                 subtitle=subtitle,
                 title_align="left",
                 border_style=RICH_STYLE_GREEN,
@@ -195,6 +203,7 @@ def finish_response(self, response: str):
         assistant_panel = Panel(
             Markdown(markdown_formatted_response, code_theme=CODE_THEME),
             title=header,
+            box=HORIZONTALS,
             title_align="left",
             border_style=RICH_STYLE_GREEN,
         )

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentcrew-ai"
-version = "0.7.7"
+version = "0.7.8"
 requires-python = ">=3.12"
 classifiers = [
     "Programming Language :: Python :: 3",