cobalt-cli-linux 0.1.1__tar.gz → 0.1.3__tar.gz

{cobalt_cli_linux-0.1.1 → cobalt_cli_linux-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cobalt-cli-linux
-Version: 0.1.1
+Version: 0.1.3
 Summary: Agentic Groq-powered CLI assistant for Debian Linux
 Author: Cobalt
 License-Expression: MIT
@@ -77,6 +77,8 @@ cobalt --interactive
 
 This launches a fullscreen TUI that feels similar to a professional agent console, with the message stream and a compact prompt area instead of a raw shell prompt.
 
+In the interface, press `Escape` to unfocus the prompt. While unfocused, `n` starts a new chat and `d` deletes the selected chat; click the prompt or type a character to focus it again. Long prompts wrap inside the prompt box.
+
 You can also pass arguments directly:
 
 ```bash

{cobalt_cli_linux-0.1.1 → cobalt_cli_linux-0.1.3}/README.md

@@ -52,6 +52,8 @@ cobalt --interactive
 
 This launches a fullscreen TUI that feels similar to a professional agent console, with the message stream and a compact prompt area instead of a raw shell prompt.
 
+In the interface, press `Escape` to unfocus the prompt. While unfocused, `n` starts a new chat and `d` deletes the selected chat; click the prompt or type a character to focus it again. Long prompts wrap inside the prompt box.
+
 You can also pass arguments directly:
 
 ```bash

{cobalt_cli_linux-0.1.1 → cobalt_cli_linux-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "cobalt-cli-linux"
-version = "0.1.1"
+version = "0.1.3"
 description = "Agentic Groq-powered CLI assistant for Debian Linux"
 readme = "README.md"
 requires-python = ">=3.10"

{cobalt_cli_linux-0.1.1 → cobalt_cli_linux-0.1.3}/src/cobalt_cli_linux/tui.py

@@ -8,6 +8,14 @@ from pathlib import Path
 from .agent import CobaltAgent
 from .history import ChatStore
 
+COBALT_CLI_ART = (
+    "   ____ ___  ____    _    _   _____    ____ _     ___ ",
+    "  / ___/ _ \\| __ )  / \\  | | |_   _|  / ___| |   |_ _|",
+    " | |  | | | |  _ \\ / _ \\ | |   | |   | |   | |    | | ",
+    " | |__| |_| | |_) / ___ \\| |___| |   | |___| |___ | | ",
+    "  \\____\\___/|____/_/   \\_\\_____|_|    \\____|_____|___|",
+)
+
 
 class CobaltTUI:
     def __init__(self, settings) -> None:
@@ -22,15 +30,7 @@ class CobaltTUI:
             self.current_chat = self.chat_store.load_chat(self.selected_chat_path)
         self.history = self.current_chat
         self.saved_chats = self.chat_files
-        self.ascii_lines = self._load_ascii_art()
-
-    def _load_ascii_art(self) -> list[str]:
-        ascii_path = Path(__file__).with_name("ASCII.txt")
-        if not ascii_path.exists():
-            ascii_path = Path(__file__).resolve().parents[2] / "ASCII.txt"
-        if not ascii_path.exists():
-            return ["COBALT"]
-        return [line.rstrip() for line in ascii_path.read_text(encoding="utf-8").splitlines() if line.strip()]
+        self.input_focused = True
 
     def _refresh_chat_files(self) -> None:
         self.chat_files = self.chat_store.list_chats()
@@ -58,8 +58,38 @@ class CobaltTUI:
         title = " ".join(words[:4]).strip()
         return title.title() if title else "New chat"
 
-    def _should_handle_shortcut(self, ch: int) -> bool:
-        return ch in (14, 4)
+    def _should_handle_shortcut(self, ch: int, input_focused: bool = False) -> bool:
+        return not input_focused and ch in (14, 4, ord("n"), ord("d"))
+
+    def _input_box_geometry(self, main_x: int, main_y: int, main_w: int, main_h: int, input_text: str) -> tuple[int, int, int, int, list[str]]:
+        box_width = max(20, main_w - 4)
+        text_width = max(10, box_width - 6)
+        input_lines = textwrap.wrap(
+            input_text,
+            width=text_width,
+            replace_whitespace=False,
+            drop_whitespace=False,
+            break_long_words=True,
+            break_on_hyphens=False,
+        ) or [""]
+        max_lines = max(1, min(6, main_h - 6))
+        input_lines = input_lines[-max_lines:]
+        box_height = len(input_lines) + 3
+        box_x = main_x + 2
+        box_y = main_y + main_h - box_height - 1
+        return box_x, box_y, box_width, box_height, input_lines
+
+    def _draw_input_box(self, stdscr, box_x: int, box_y: int, box_width: int, box_height: int, input_lines: list[str], focused: bool) -> None:
+        rule = "-" * max(0, box_width - 2)
+        top_label = " Ask AI "
+        top = f"-{top_label}{rule}"
+        bottom_label = " [Enter to submit] "
+        bottom = f"-{'-' * max(0, box_width - len(bottom_label) - 2)}{bottom_label}-"
+        stdscr.addstr(box_y, box_x, top[:box_width])
+        for index, line in enumerate(input_lines):
+            prompt = "> " if index == 0 else "  "
+            stdscr.addstr(box_y + 1 + index, box_x, f"{prompt}{line}"[:box_width])
+        stdscr.addstr(box_y + box_height - 1, box_x, bottom[:box_width], curses.A_BOLD if focused else curses.A_DIM)
 
     def _compose_lines(self, text: str, width: int) -> list[str]:
         lines: list[str] = []
@@ -93,8 +123,12 @@ class CobaltTUI:
 
         sidebar_w = min(26, max(20, width // 4))
         sidebar_x = 1
-        sidebar_y = 1
-        sidebar_h = max(8, height - 4)
+        for index, line in enumerate(COBALT_CLI_ART):
+            x = max(0, (width - len(line)) // 2)
+            stdscr.addstr(index, x, line[: max(0, width - x)])
+
+        sidebar_y = 6
+        sidebar_h = max(8, height - 9)
         self._draw_box(stdscr, sidebar_y, sidebar_x, sidebar_h, sidebar_w, " Chats ")
 
         chat_files = self.chat_store.list_chats()[-8:]
@@ -112,20 +146,12 @@ class CobaltTUI:
 
         main_x = sidebar_x + sidebar_w + 2
         main_w = max(20, width - main_x - 2)
-        main_y = 1
-        main_h = max(8, height - 4)
+        main_y = 6
+        main_h = max(8, height - 9)
         self._draw_box(stdscr, main_y, main_x, main_h, main_w, f" {self.history.title or 'conversation'} ")
 
-        if not self.history.messages:
-            for idx, line in enumerate(self.ascii_lines):
-                y = main_y + 2 + idx
-                x = main_x + max(0, (main_w - len(line)) // 2)
-                stdscr.addstr(y, x, line[: max(0, main_w - 2)])
-            prompt = f"> {input_text}" if input_text else "> "
-            prompt_y = main_y + 2 + len(self.ascii_lines) + 2
-            prompt_x = main_x + max(0, (main_w - len(prompt)) // 2)
-            stdscr.addstr(prompt_y, prompt_x, prompt[: max(0, main_w - 2)], curses.A_REVERSE)
-        else:
+        box_x, box_y, box_width, box_height, input_lines = self._input_box_geometry(main_x, main_y, main_w, main_h, input_text)
+        if self.history.messages:
             rows: list[str] = []
             for message in self.history.latest(18):
                 role = message.get("role", "user").upper()
@@ -135,16 +161,15 @@ class CobaltTUI:
                     rows.append(f"{prefix}{line}" if prefix and line else line)
                     prefix = ""
 
-            visible = rows[-max(4, main_h - 4):]
+            visible = rows[-max(1, box_y - main_y - 2):]
             for idx, line in enumerate(visible[: main_h - 4]):
                 y = main_y + 1 + idx
+                if y >= box_y:
+                    break
                 clipped = line[: max(0, main_w - 4)]
                 stdscr.addstr(y, main_x + 2, clipped)
 
-            prompt = f"> {input_text}" if input_text else "> "
-            prompt_y = main_y + main_h - 3
-            prompt_x = main_x + max(0, (main_w - len(prompt)) // 2)
-            stdscr.addstr(prompt_y, prompt_x, prompt[: max(0, main_w - 2)], curses.A_REVERSE)
+        self._draw_input_box(stdscr, box_x, box_y, box_width, box_height, input_lines, self.input_focused)
 
         if status:
             status_line = status[: max(0, width - 1)]
@@ -165,17 +190,22 @@ class CobaltTUI:
 
         agent = CobaltAgent(settings=self.settings)
         agent.history = self.history
-        agent.process_stream(input_text.strip(), on_chunk=on_chunk)
+        response = agent.process_stream(input_text.strip(), on_chunk=on_chunk)
         self.history = agent.history
         self.current_chat = self.history
         self.saved_chats = self.chat_store.list_chats()
+        if response and not buffer:
+            buffer = response
+        if not buffer:
+            raise RuntimeError("The assistant returned an empty response")
+        self._render(stdscr, "", "")
         return buffer
 
     def _handle_mouse_click(self, x: int, y: int) -> bool:
         sidebar_w = min(26, max(20, curses.COLS // 4))
         sidebar_x = 1
-        sidebar_y = 1
-        sidebar_h = max(8, curses.LINES - 4)
+        sidebar_y = 6
+        sidebar_h = max(8, curses.LINES - 9)
 
         if sidebar_x <= x < sidebar_x + sidebar_w and sidebar_y <= y < sidebar_y + sidebar_h:
             chat_files = self.chat_store.list_chats()[-8:]
@@ -190,6 +220,16 @@ class CobaltTUI:
                 return True
         return False
 
+    def _click_is_in_input(self, x: int, y: int, input_text: str) -> bool:
+        height, width = curses.LINES, curses.COLS
+        sidebar_w = min(26, max(20, width // 4))
+        main_x = 3 + sidebar_w
+        main_y = 6
+        main_w = max(20, width - main_x - 2)
+        main_h = max(8, height - 9)
+        box_x, box_y, box_width, box_height, _ = self._input_box_geometry(main_x, main_y, main_w, main_h, input_text)
+        return box_x <= x < box_x + box_width and box_y <= y < box_y + box_height
+
     def run(self) -> int:
         def _main(stdscr):
             curses.curs_set(0)
@@ -202,6 +242,7 @@ class CobaltTUI:
             curses.noecho()
             curses.mousemask(curses.BUTTON1_PRESSED | curses.BUTTON1_RELEASED | curses.BUTTON1_CLICKED)
             input_text = ""
+            self.input_focused = True
             status = ""
 
             while True:
@@ -210,25 +251,32 @@ class CobaltTUI:
 
                 if ch == curses.KEY_MOUSE:
                     _, x, y, _, button_state = curses.getmouse()
+                    if button_state & (curses.BUTTON1_PRESSED | curses.BUTTON1_CLICKED | curses.BUTTON1_RELEASED) and self._click_is_in_input(x, y, input_text):
+                        self.input_focused = True
+                        continue
                     if button_state & (curses.BUTTON1_PRESSED | curses.BUTTON1_CLICKED | curses.BUTTON1_RELEASED) and self._handle_mouse_click(x, y):
-                        status = "Chat selected"
+                        status = ""
                         input_text = ""
+                        self.input_focused = False
                     continue
 
                 if ch == 3:
                     break
-                if self._should_handle_shortcut(ch):
-                    if ch == 14:
+                if ch == 27:
+                    self.input_focused = False
+                    continue
+                if self._should_handle_shortcut(ch, self.input_focused):
+                    if ch in (14, ord("n")):
                         new_chat = self.chat_store.new_chat(title="New chat")
                         self.current_chat = new_chat
                         self.history = self.current_chat
                         self.selected_chat_path = str(self.current_chat.path)
                         self.chat_files = self.chat_store.list_chats()
                         input_text = ""
-                        status = "New chat started"
+                        status = ""
                         continue
 
-                    if ch == 4 and self.selected_chat_path:
+                    if ch in (4, ord("d")) and self.selected_chat_path:
                         target = Path(self.selected_chat_path)
                         if target.exists():
                             self.chat_store.delete_chat(target)
@@ -242,7 +290,7 @@ class CobaltTUI:
                                 self.history = self.current_chat
                                 self.selected_chat_path = str(self.current_chat.path)
                             input_text = ""
-                            status = "Chat deleted"
+                            status = ""
                             continue
                 if ch in (10, 13, curses.KEY_ENTER):
                     if input_text.strip():
@@ -251,7 +299,7 @@ class CobaltTUI:
                                 self.history.title = self._suggest_chat_title(input_text.strip())
                                 self.history.save()
                             self._stream_response(stdscr, input_text)
-                            status = "Response received"
+                            status = ""
                         except (OSError, RuntimeError, ValueError) as exc:
                             self.history.add("assistant", f"Error: {exc}")
                             status = ""
@@ -263,9 +311,8 @@ class CobaltTUI:
                 if ch in (curses.KEY_RESIZE,):
                     continue
                 if 32 <= ch <= 126:
+                    self.input_focused = True
                     input_text += chr(ch)
-                elif ch == 27:
-                    break
 
             return 0
 

{cobalt_cli_linux-0.1.1 → cobalt_cli_linux-0.1.3}/src/cobalt_cli_linux.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cobalt-cli-linux
-Version: 0.1.1
+Version: 0.1.3
 Summary: Agentic Groq-powered CLI assistant for Debian Linux
 Author: Cobalt
 License-Expression: MIT
@@ -77,6 +77,8 @@ cobalt --interactive
 
 This launches a fullscreen TUI that feels similar to a professional agent console, with the message stream and a compact prompt area instead of a raw shell prompt.
 
+In the interface, press `Escape` to unfocus the prompt. While unfocused, `n` starts a new chat and `d` deletes the selected chat; click the prompt or type a character to focus it again. Long prompts wrap inside the prompt box.
+
 You can also pass arguments directly:
 
 ```bash

{cobalt_cli_linux-0.1.1 → cobalt_cli_linux-0.1.3}/tests/test_cli.py

@@ -51,9 +51,11 @@ def test_tui_shortcuts_ignore_typed_letters() -> None:
 
     assert tui._should_handle_shortcut(14) is True
     assert tui._should_handle_shortcut(4) is True
+    assert tui._should_handle_shortcut(ord("n")) is True
+    assert tui._should_handle_shortcut(ord("d")) is True
     assert tui._should_handle_shortcut(ord("q")) is False
-    assert tui._should_handle_shortcut(ord("n")) is False
-    assert tui._should_handle_shortcut(ord("d")) is False
+    assert tui._should_handle_shortcut(ord("n"), True) is False
+    assert tui._should_handle_shortcut(ord("d"), True) is False
 
 
 def test_q_is_not_a_shortcut_when_typing() -> None:
@@ -124,6 +126,12 @@ def test_stream_error_reads_response_before_inspecting_body(monkeypatch) -> None
         raise AssertionError("Expected streamed API error")
 
 
+def test_tui_empty_chat_does_not_load_prompt_art_as_body_content(tmp_path: Path) -> None:
+    tui = CobaltTUI(settings=Settings())
+
+    assert not hasattr(tui, "ascii_lines")
+
+
 def test_write_file_tool_creates_and_verifies_file(tmp_path: Path) -> None:
     agent = CobaltAgent(settings=Settings(), history=ConversationHistory(tmp_path / "history.json"))
     target = tmp_path / "test.txt"