pulse-ai-cli 0.1.0__py3-none-any.whl

pulse_cli/tui/dashboard.py

@@ -0,0 +1,153 @@
+"""`pulse` (bare) — the TUI dashboard: animated logo + account/usage panels.
+
+Read-only: only GET /pages and GET /api/v1/billing/subscription are called.
+Network work runs in thread workers so the animation never stutters.
+"""
+from __future__ import annotations
+
+from textual.app import App, ComposeResult
+from textual.containers import Horizontal, Vertical, VerticalScroll
+from textual.widgets import Footer, Static
+
+from pulse_cli import __version__
+from pulse_cli import config as cfg
+from pulse_cli.client import PlatformClient, PulseClient
+from pulse_cli.errors import PulseCliError
+from pulse_cli.tui.logo import LOGO_COLS, LOGO_ROWS, render_logo
+
+FRAME_SECONDS = 1 / 12  # smooth but cheap
+
+_COMMANDS: list[tuple[str, str]] = [
+    ("pulse extract report.pdf", "convert a document to markdown + json"),
+    ("pulse extract ./invoices --no-wait", "submit a whole folder, fetch results later"),
+    ('pulse schema generate --prompt "vendor, total, date"', "draft an extraction schema with AI"),
+    ("pulse schema apply <extraction_id> --schema s.json", "pull structured fields from a document"),
+    ('pulse split <extraction_id> -t "intro:opening section"', "split a document into sections by topic"),
+    ("pulse tables <extraction_id>", "extract tables (html or json)"),
+    ("pulse jobs get <job_id> --watch", "watch or resume an async job"),
+]
+_CMD_PAD = max(len(c) for c, _ in _COMMANDS) + 3
+
+
+def _get_started_text() -> str:
+    lines = [f"[#E9F0F8]{c:<{_CMD_PAD}}[/][#85C4E4]{d}[/]" for c, d in _COMMANDS]
+    lines.append("")
+    lines.append(
+        "[#0E7BC9]tip[/]  [#CCDEE4]add [#E9F0F8]--json[/] to any command for raw output, "
+        "[#E9F0F8]-o <file>[/] to save, [#E9F0F8]-h[/] for help[/]"
+    )
+    return "\n".join(lines)
+
+
+class Logo(Static):
+    def on_mount(self) -> None:
+        self._frame = 0
+        self.update(render_logo(0))
+        self.set_interval(FRAME_SECONDS, self._tick)
+
+    def _tick(self) -> None:
+        self._frame += 1
+        self.update(render_logo(self._frame))
+
+
+class Dashboard(App):
+    TITLE = "Pulse"
+    BINDINGS = [("q", "quit", "Quit"), ("r", "refresh", "Refresh")]
+
+    CSS = f"""
+    Screen {{ background: #02182E; }}
+    #body {{ padding: 1 3 0 3; scrollbar-size-vertical: 1; }}
+    #header {{ height: {LOGO_ROWS}; margin-bottom: 1; }}
+    #logo {{ width: {LOGO_COLS}; height: {LOGO_ROWS}; }}
+    #brand {{ width: 1fr; height: {LOGO_ROWS}; padding-left: 5; align-vertical: middle; }}
+    #wordmark {{ color: #E9F0F8; text-style: bold; height: auto; }}
+    #tagline {{ color: #85C4E4; height: auto; margin-top: 1; }}
+    #panels {{ height: auto; margin-bottom: 1; }}
+    .panel {{
+        height: auto;
+        border: round #0E7BC9 50%;
+        border-title-color: #98E2FD;
+        border-title-style: bold;
+        padding: 1 2;
+        background: #02182E;
+    }}
+    #account {{ width: 1fr; margin-right: 2; }}
+    #usage {{ width: 1fr; }}
+    #commands {{ width: 100%; margin-bottom: 1; }}
+    """
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.conf = cfg.load()
+
+    def compose(self) -> ComposeResult:
+        with VerticalScroll(id="body"):
+            with Horizontal(id="header"):
+                yield Logo(id="logo")
+                with Vertical(id="brand"):
+                    yield Static("P U L S E", id="wordmark")
+                    yield Static(f"document extraction · cli v{__version__}", id="tagline")
+            with Horizontal(id="panels"):
+                account = Static("loading…", id="account", classes="panel")
+                account.border_title = "account"
+                yield account
+                usage = Static("loading…", id="usage", classes="panel")
+                usage.border_title = "usage"
+                yield usage
+            commands = Static(_get_started_text(), id="commands", classes="panel")
+            commands.border_title = "get started"
+            yield commands
+        yield Footer()
+
+    def on_mount(self) -> None:
+        self.action_refresh()
+
+    def action_refresh(self) -> None:
+        self.run_worker(self._load_account, thread=True, exclusive=True, group="account")
+        self.run_worker(self._load_usage, thread=True, exclusive=True, group="usage")
+
+    # --- data workers (thread) -------------------------------------------------
+
+    def _load_account(self) -> None:
+        lines = [
+            f"[#85C4E4]key[/]     [#E9F0F8]{cfg.redact(self.conf.api_key)}[/]  [#85C4E4]({self.conf.key_source})[/]",
+            f"[#85C4E4]engine[/]  [#E9F0F8]{self.conf.api_url}[/]",
+        ]
+        try:
+            with PulseClient(self.conf.api_key, self.conf.api_url) as client:
+                client.pages()  # cheap auth/connectivity probe
+            lines.append("[#85C4E4]status[/]  [green]● connected[/]")
+        except PulseCliError as exc:
+            lines.append(f"[#85C4E4]status[/]  [red]● {exc}[/]")
+        self.call_from_thread(self.query_one("#account", Static).update, "\n".join(lines))
+
+    def _load_usage(self) -> None:
+        lines: list[str] = []
+        try:
+            with PulseClient(self.conf.api_key, self.conf.api_url) as client:
+                pages = client.pages().get("total_pages_processed", 0)
+            lines.append(f"[#85C4E4]pages processed[/]  [#E9F0F8 bold]{pages:,}[/]")
+        except PulseCliError as exc:
+            lines.append(f"[red]{exc}[/]")
+
+        if self.conf.platform_url:
+            with PlatformClient(self.conf.api_key, self.conf.platform_url) as pc:
+                sub = pc.subscription()
+            if sub:
+                plan = sub.get("plan") or sub.get("plan_name") or "—"
+                status = sub.get("status") or "—"
+                lines.append(f"[#85C4E4]plan[/]             [#E9F0F8]{plan}[/]")
+                lines.append(f"[#85C4E4]subscription[/]     [#E9F0F8]{status}[/]")
+        else:
+            lines.append("[#85C4E4]plan[/]             [#CCDEE4]—[/]")
+        self.call_from_thread(self.query_one("#usage", Static).update, "\n".join(lines))
+
+
+def run_dashboard() -> None:
+    conf = cfg.load()
+    if not conf.api_key:
+        from pulse_cli.output import console
+
+        console.print("[muted]Not logged in. Run[/muted] [brand]pulse login[/brand] [muted]to get started.[/muted]")
+        raise SystemExit(1)
+    Dashboard().run()

pulse_cli/tui/logo.py

@@ -0,0 +1,76 @@
+"""Pulse logo as terminal art: the 8×8 rounded-square grid from logo-icon.svg.
+
+The brand mark is an 8×8 grid of squares at 10% opacity with eight full-opacity
+cells tracing a heartbeat waveform. We render each cell as a 2-char block and
+animate a glow that travels along the waveform left-to-right, easing through the
+brand palette (navy -> pulse blue -> sky).
+"""
+from __future__ import annotations
+
+from rich.text import Text
+
+GRID = 8
+
+# Waveform cells in (row, col), ordered left-to-right (decoded from logo-icon.svg).
+HEARTBEAT: list[tuple[int, int]] = [
+    (4, 0), (4, 1), (3, 2), (2, 3), (3, 4), (4, 5), (5, 6), (4, 7),
+]
+_WAVE_INDEX = {cell: i for i, cell in enumerate(HEARTBEAT)}
+
+# Palette stops for the traveling glow (dim -> peak).
+_DIM_BG = "#0A2238"      # 10%-opacity squares (navy-tinted)
+_WAVE_BASE = "#0E7BC9"   # brand pulse blue (resting waveform)
+_GLOW = ["#0E7BC9", "#4A9ED6", "#85C4E4", "#98E2FD"]  # base -> peak
+
+
+def _lerp(a: str, b: str, t: float) -> str:
+    """Linear hex-color interpolation, t in [0,1]."""
+    av = [int(a[i : i + 2], 16) for i in (1, 3, 5)]
+    bv = [int(b[i : i + 2], 16) for i in (1, 3, 5)]
+    return "#{:02x}{:02x}{:02x}".format(*(round(x + (y - x) * t) for x, y in zip(av, bv)))
+
+
+def wave_color(index: int, frame: int, period: int = 24) -> str:
+    """Color of waveform cell ``index`` at animation ``frame``.
+
+    A glow peak sweeps the 8 waveform cells once per ``period`` frames, with a
+    short trailing falloff behind it.
+    """
+    pos = (frame % period) / period * (len(HEARTBEAT) + 4) - 2  # overshoot so it fully exits
+    dist = pos - index
+    if 0 <= dist < 3:  # trailing glow
+        t = 1.0 - dist / 3
+    elif -1 < dist < 0:  # leading edge
+        t = 1.0 + dist
+    else:
+        t = 0.0
+    if t <= 0:
+        return _WAVE_BASE
+    # Ease through the glow ramp to the peak color.
+    return _lerp(_WAVE_BASE, _GLOW[-1], t)
+
+
+# Rendered size: cells are "██" (2 cols x 1 row ≈ square in a terminal);
+# gaps are 2 cols / 1 row, so cell and gap strides are square on both axes.
+LOGO_COLS = GRID * 4 - 2  # 30
+LOGO_ROWS = GRID * 2 - 1  # 15
+
+
+def render_logo(frame: int = 0) -> Text:
+    """Render one animation frame of the logo grid as a rich Text.
+
+    Pure full-block glyphs (no half-blocks), so it renders identically in any
+    terminal/font. A 2-space gap between columns and a blank line between
+    rows keep the mark square overall (terminal cells are ~1:2 wide:tall).
+    """
+    out = Text()
+    for r in range(GRID):
+        for c in range(GRID):
+            idx = _WAVE_INDEX.get((r, c))
+            color = _DIM_BG if idx is None else wave_color(idx, frame)
+            out.append("██", style=color)
+            if c < GRID - 1:
+                out.append("  ")
+        if r < GRID - 1:
+            out.append("\n\n")
+    return out