open-data-sci 0.1.0__py3-none-any.whl

opendatasci/_tui/styles.tcss

@@ -0,0 +1,248 @@
+/* ─── OpenDataSci TUI Theme (Dark+) ─────────────────────────────────────────── */
+
+Screen {
+    background: #060a10;
+    color: #e6edf3;
+}
+
+AppHeader {
+    dock: top;
+    height: 5;
+    background: #0d1117;
+    border-bottom: solid #1a2030;
+}
+
+#header-logo {
+    width: 23;
+    height: 5;
+    padding-left: 2;
+    content-align: center middle;
+}
+
+#header-info {
+    height: 5;
+    padding: 1 3 1 2;
+    content-align: left middle;
+}
+
+#header-timer {
+    width: auto;
+    content-align: right middle;
+    padding-right: 1;
+}
+
+/* ─── Main layout ─────────────────────────────────────────────────────────── */
+
+#main {
+    height: 1fr;
+    layout: horizontal;
+}
+
+/* ─── Chat Pane ───────────────────────────────────────────────────────────── */
+
+ChatPane {
+    width: 100%;
+    layout: vertical;
+    background: #060a10;
+}
+
+#messages {
+    height: 1fr;
+    padding: 1 2;
+    scrollbar-color: #4a6080 #0d1520;
+    scrollbar-background: #0d1520;
+    scrollbar-corner-color: #0d1520;
+}
+
+#input-bar {
+    height: auto;
+    border-top: solid #1a2030;
+    padding: 1 2 1 2;
+    background: #060a10;
+}
+
+#completion-popup {
+    display: none;
+    height: auto;
+    background: #0d1117;
+    border: solid #30363d;
+    padding: 0 1;
+    margin-bottom: 1;
+}
+
+#completion-popup.active {
+    display: block;
+}
+
+#attachment-bar {
+    display: none;
+    height: auto;
+    padding: 0 1;
+    margin-bottom: 1;
+}
+
+#attachment-bar.active {
+    display: block;
+}
+
+#user-input {
+    background: #0d1117;
+    border: solid #1a2030;
+    color: #e6edf3;
+    padding: 0 1;
+    width: 100%;
+    height: 3;
+}
+
+#user-input:focus {
+    border: solid #79c0ff;
+}
+
+#user-input.awaiting-choice {
+    border: solid #c9963a;
+}
+
+
+/* ─── Workspace Panel ─────────────────────────────────────────────────────── */
+
+WorkspacePanel {
+    display: none;
+    height: auto;
+    max-height: 10;
+    border-top: solid #1a2030;
+    background: #0d1117;
+    padding: 0 2 1 2;
+}
+
+WorkspacePanel.active {
+    display: block;
+}
+
+/* ─── Message widgets ─────────────────────────────────────────────────────── */
+
+MessageBubble {
+    height: auto;
+    padding: 0;
+    margin-bottom: 1;
+}
+
+MessageBubble.user {
+    background: #0e1520;
+    border-left: thick #79c0ff;
+    padding: 0 2;
+}
+
+MessageBubble.thinking {
+    background: #1e2228;
+    border-left: thick #6e7681;
+    padding: 0 2;
+}
+
+MessageBubble.question {
+    background: #1c1408;
+    border-left: thick #c9963a;
+    padding: 0 2;
+}
+
+MessageBubble.agent {
+    padding: 0 0;
+}
+
+.msg-divider {
+    height: 1;
+    color: #1a2030;
+    margin: 0 0 1 0;
+}
+
+/* ─── Footer ──────────────────────────────────────────────────────────────── */
+
+Footer {
+    background: #0d1117;
+    color: #8b949e;
+    border-top: solid #1a2030;
+}
+
+Footer > .footer--key {
+    background: #1a2030;
+    color: #e6edf3;
+}
+
+Footer > .footer--highlight {
+    background: #79c0ff26;
+    color: #79c0ff;
+}
+
+/* ─── Markdown (agent response) ───────────────────────────────────────────── */
+
+Markdown {
+    background: transparent;
+    height: auto;
+    padding: 0;
+    margin: 0;
+}
+
+MarkdownH1 {
+    background: transparent;
+    border: none;
+    color: #79c0ff;
+    text-style: bold underline;
+    padding: 0 0 1 0;
+    margin: 0 0 1 0;
+    text-align: left;
+}
+
+MarkdownH2 {
+    background: transparent;
+    border: none;
+    color: #79c0ff;
+    text-style: bold;
+    padding: 0;
+    margin: 0 0 1 0;
+    text-align: left;
+}
+
+MarkdownH3, MarkdownH4, MarkdownH5, MarkdownH6 {
+    background: transparent;
+    border: none;
+    color: #8b949e;
+    text-style: bold;
+    padding: 0;
+    margin: 0 0 1 0;
+    text-align: left;
+}
+
+MarkdownParagraph {
+    background: transparent;
+    color: #e6edf3;
+    padding: 0;
+    margin: 0 0 1 0;
+}
+
+MarkdownFence {
+    background: #0d1117;
+    border: solid #1a2030;
+    padding: 0 1;
+    margin: 0 0 1 0;
+}
+
+MarkdownBlockQuote {
+    background: #0a1520;
+    border-left: thick #4caa5e;
+    padding: 0 1;
+    margin: 0 0 1 0;
+}
+
+MarkdownBulletList, MarkdownOrderedList {
+    background: transparent;
+    padding: 0;
+    margin: 0 0 1 0;
+}
+
+MarkdownHorizontalRule {
+    color: #1a2030;
+    margin: 1 0;
+}
+
+MarkdownTable {
+    margin: 0 0 1 0;
+}

opendatasci/_tui/styles_visible.tcss

@@ -0,0 +1,245 @@
+/* ─── OpenDataSci TUI Theme (Visible / Color-blind safe) ────────────────────── */
+/* Uses the Okabe-Ito color palette, which is safe for most forms of          */
+/* color vision deficiency (deuteranopia, protanopia, tritanopia).            */
+
+Screen {
+    background: #060a10;
+    color: #f5f5f5;
+}
+
+/* ─── Header ──────────────────────────────────────────────────────────────── */
+
+AppHeader {
+    dock: top;
+    height: 5;
+    background: #0d1117;
+    border-bottom: solid #1e2d40;
+}
+
+#header-info {
+    height: 5;
+    padding: 1 3;
+    content-align: left middle;
+}
+
+#header-timer {
+    width: auto;
+    content-align: right middle;
+    padding-right: 1;
+}
+
+/* ─── Main layout ─────────────────────────────────────────────────────────── */
+
+#main {
+    height: 1fr;
+    layout: horizontal;
+}
+
+/* ─── Chat Pane ───────────────────────────────────────────────────────────── */
+
+ChatPane {
+    width: 100%;
+    layout: vertical;
+    background: #060a10;
+}
+
+#messages {
+    height: 1fr;
+    padding: 1 2;
+    scrollbar-color: #4a6080 #0d1520;
+    scrollbar-background: #0d1520;
+    scrollbar-corner-color: #0d1520;
+}
+
+#input-bar {
+    height: auto;
+    border-top: solid #1e2d40;
+    padding: 1 2 1 2;
+    background: #060a10;
+}
+
+#completion-popup {
+    display: none;
+    height: auto;
+    background: #0d1117;
+    border: solid #1e2d40;
+    padding: 0 1;
+    margin-bottom: 1;
+}
+
+#completion-popup.active {
+    display: block;
+}
+
+#attachment-bar {
+    display: none;
+    height: auto;
+    padding: 0 1;
+    margin-bottom: 1;
+}
+
+#attachment-bar.active {
+    display: block;
+}
+
+#user-input {
+    background: #0d1117;
+    border: solid #1e2d40;
+    color: #f5f5f5;
+    padding: 0 1;
+    width: 100%;
+    height: 3;
+}
+
+#user-input:focus {
+    border: solid #56b4e9;
+}
+
+#user-input.awaiting-choice {
+    border: solid #e69f00;
+}
+
+
+/* ─── Workspace Panel ─────────────────────────────────────────────────────── */
+
+WorkspacePanel {
+    display: none;
+    height: auto;
+    max-height: 10;
+    border-top: solid #1e2d40;
+    background: #0d1117;
+    padding: 0 2 1 2;
+}
+
+WorkspacePanel.active {
+    display: block;
+}
+
+/* ─── Message widgets ─────────────────────────────────────────────────────── */
+
+MessageBubble {
+    height: auto;
+    padding: 0;
+    margin-bottom: 1;
+}
+
+MessageBubble.user {
+    background: #0a1520;
+    border-left: thick #56b4e9;
+    padding: 0 2;
+}
+
+MessageBubble.thinking {
+    background: #1e2228;
+    border-left: thick #8090a0;
+    padding: 0 2;
+}
+
+MessageBubble.question {
+    background: #1c1608;
+    border-left: thick #e69f00;
+    padding: 0 2;
+}
+
+MessageBubble.agent {
+    padding: 0 0;
+}
+
+.msg-divider {
+    height: 1;
+    color: #1e2d40;
+    margin: 0 0 1 0;
+}
+
+/* ─── Footer ──────────────────────────────────────────────────────────────── */
+
+Footer {
+    background: #0d1117;
+    color: #a0b0c0;
+    border-top: solid #1e2d40;
+}
+
+Footer > .footer--key {
+    background: #1e2d40;
+    color: #f5f5f5;
+}
+
+Footer > .footer--highlight {
+    background: #56b4e926;
+    color: #56b4e9;
+}
+
+/* ─── Markdown (agent response) ───────────────────────────────────────────── */
+
+Markdown {
+    background: transparent;
+    height: auto;
+    padding: 0;
+    margin: 0;
+}
+
+MarkdownH1 {
+    background: transparent;
+    border: none;
+    color: #56b4e9;
+    text-style: bold underline;
+    padding: 0 0 1 0;
+    margin: 0 0 1 0;
+    text-align: left;
+}
+
+MarkdownH2 {
+    background: transparent;
+    border: none;
+    color: #56b4e9;
+    text-style: bold;
+    padding: 0;
+    margin: 0 0 1 0;
+    text-align: left;
+}
+
+MarkdownH3, MarkdownH4, MarkdownH5, MarkdownH6 {
+    background: transparent;
+    border: none;
+    color: #a0b0c0;
+    text-style: bold;
+    padding: 0;
+    margin: 0 0 1 0;
+    text-align: left;
+}
+
+MarkdownParagraph {
+    background: transparent;
+    color: #f5f5f5;
+    padding: 0;
+    margin: 0 0 1 0;
+}
+
+MarkdownFence {
+    background: #0d1117;
+    border: solid #1e2d40;
+    padding: 0 1;
+    margin: 0 0 1 0;
+}
+
+MarkdownBlockQuote {
+    background: #0a1520;
+    border-left: thick #009e73;
+    padding: 0 1;
+    margin: 0 0 1 0;
+}
+
+MarkdownBulletList, MarkdownOrderedList {
+    background: transparent;
+    padding: 0;
+    margin: 0 0 1 0;
+}
+
+MarkdownHorizontalRule {
+    color: #1e2d40;
+    margin: 1 0;
+}
+
+MarkdownTable {
+    margin: 0 0 1 0;
+}

opendatasci/_tui/theme.py

@@ -0,0 +1,113 @@
+"""Color theme definitions for OpenDataSci TUI."""
+
+# Default dark theme — deeper backgrounds, muted accents
+DARK: dict[str, str] = {
+    "accent": "#79c0ff",  # soft blue
+    "success": "#4caa5e",  # muted sage-green
+    "error": "#c97a74",  # muted terracotta-red
+    "warning": "#c9963a",  # muted amber
+    "thinking": "#9f75d6",  # muted lavender
+    "text_primary": "#e6edf3",
+    "text_secondary": "#8b949e",
+    "text_muted": "#6e7681",
+    "text_dim": "#21262d",
+    "separator": "#1a2030",
+    "logo": "#c9a86c",
+    "timer": "bright_green",
+    "tool_running": "#58a6ff",
+    "tool_done": "#3fb950",
+}
+
+# Color-blind safe theme — Okabe-Ito palette
+VISIBLE: dict[str, str] = {
+    "accent": "#56b4e9",  # sky blue
+    "success": "#009e73",  # bluish green / teal
+    "error": "#d55e00",  # vermilion
+    "warning": "#e69f00",  # orange
+    "thinking": "#cc79a7",  # reddish purple
+    "text_primary": "#f5f5f5",
+    "text_secondary": "#a0b0c0",
+    "text_muted": "#8090a0",
+    "text_dim": "#3d4a5c",
+    "separator": "#1a2030",
+    "logo": "#e69f00",
+    "timer": "cyan",
+    "tool_running": "#56b4e9",
+    "tool_done": "#009e73",
+}
+
+# Light theme — light background, dark text (GitHub-inspired)
+LIGHT: dict[str, str] = {
+    "accent": "#0969da",
+    "success": "#1a7f37",
+    "error": "#cf222e",
+    "warning": "#9a6700",
+    "thinking": "#8250df",
+    "text_primary": "#1f2328",
+    "text_secondary": "#656d76",
+    "text_muted": "#8c959f",
+    "text_dim": "#d0d7de",
+    "separator": "#d8dee4",
+    "logo": "#bf8700",
+    "timer": "blue",
+    "tool_running": "#0969da",
+    "tool_done": "#1a7f37",
+}
+
+# Solarized Dark — Ethan Schoonover palette
+SOLARIZED: dict[str, str] = {
+    "accent": "#268bd2",  # blue
+    "success": "#859900",  # green
+    "error": "#dc322f",  # red
+    "warning": "#b58900",  # yellow
+    "thinking": "#6c71c4",  # violet
+    "text_primary": "#fdf6e3",  # base3
+    "text_secondary": "#93a1a1",  # base1
+    "text_muted": "#586e75",  # base01
+    "text_dim": "#073642",  # base02
+    "separator": "#002b36",  # base03
+    "logo": "#cb4b16",  # orange
+    "timer": "bright_cyan",
+    "tool_running": "#268bd2",
+    "tool_done": "#2aa198",  # cyan
+}
+
+# Dracula — popular dark palette
+DRACULA: dict[str, str] = {
+    "accent": "#bd93f9",  # purple
+    "success": "#50fa7b",  # green
+    "error": "#ff5555",  # red
+    "warning": "#ffb86c",  # orange
+    "thinking": "#ff79c6",  # pink
+    "text_primary": "#f8f8f2",  # foreground
+    "text_secondary": "#6272a4",  # comment
+    "text_muted": "#44475a",
+    "text_dim": "#282a36",
+    "separator": "#21222c",
+    "logo": "#f1fa8c",  # yellow
+    "timer": "bright_magenta",
+    "tool_running": "#8be9fd",  # cyan
+    "tool_done": "#50fa7b",
+}
+
+# Registry of selectable themes. Keys are the names users pass to --theme
+# and to the /themes command.
+THEMES: dict[str, dict[str, str]] = {
+    "default": DARK,
+    "accessible": VISIBLE,
+    "light": LIGHT,
+    "solarized": SOLARIZED,
+    "dracula": DRACULA,
+}
+
+THEME_DESCRIPTIONS: dict[str, str] = {
+    "default": "Dark background with muted accents (built-in default)",
+    "accessible": "Okabe-Ito palette — colour-blind safe",
+    "light": "Light background with dark text",
+    "solarized": "Solarized Dark by Ethan Schoonover",
+    "dracula": "Dracula — vivid pastels on near-black",
+}
+
+# Mutated at startup by OpenDataSciApp based on --theme flag.
+active: dict[str, str] = dict(DARK)
+active_name: str = "default"

opendatasci/_tui/tools_display.py

@@ -0,0 +1,86 @@
+"""Tool display metadata for the TUI renderer.
+
+Centralises ToolDisplay declarations and the global REGISTRY so the TUI
+and streaming layer can look up icons, labels, and summary arguments without
+knowing anything about tool internals.
+
+Usage::
+
+    from opendatasci._tui.tools_display import REGISTRY, ToolDisplay
+
+    display = REGISTRY.get("execute_python_code", ToolDisplay(label="execute_python_code"))
+    print(display.icon, display.label)  # 🐍 Code
+"""
+
+import types
+from dataclasses import dataclass
+
+
+@dataclass(frozen=True)
+class ToolDisplay:
+    """Display metadata for a single tool.
+
+    Attributes:
+        label:          Human-readable tool name shown in the TUI
+                        (e.g. ``"Python"``, ``"Web Search"``).
+        icon:           Optional single emoji rendered before the label
+                        (e.g. ``"🐍"``).
+        summary_arg:    Name of the tool argument whose value is used as the
+                        short summary line in the ephemeral block.  ``None``
+                        means no summary (the block shows only the label and
+                        communication).
+        display:        When ``True`` (default) an ephemeral block is shown
+                        while the tool runs and kept as a breadcrumb afterwards.
+                        Set to ``False`` for silent internal tools.
+    """
+
+    label: str
+    icon: str = ""
+    summary_arg: str | None = None
+    display: bool = True
+
+
+# Global registry: canonical tool name → ToolDisplay.
+# _registry is the mutable backing dict; REGISTRY is the public read-only view.
+_registry: dict[str, ToolDisplay] = {}
+REGISTRY: types.MappingProxyType[str, ToolDisplay] = types.MappingProxyType(_registry)
+
+
+def register(tool_name: str, display: ToolDisplay) -> None:
+    """Associate *display* with *tool_name* in the global registry."""
+    _registry[tool_name] = display
+
+
+# ── Registrations ─────────────────────────────────────────────────────────────
+# String literals are used for tool names so this module stays free of imports
+# from opendatasci.tools.factory (which creates a circular dependency).
+
+register("execute_python_code", ToolDisplay(label="Code", icon="🐍", summary_arg="summary"))
+register("execute_cli_command", ToolDisplay(label="Command", icon="⌨️", summary_arg="summary"))
+register("list_python_libs", ToolDisplay(label="Checking available libraries", icon="📦"))
+register("enter_plan_mode", ToolDisplay(label="Planning", icon="🎯"))
+register("exit_plan_mode", ToolDisplay(label="Done planning", icon="✅"))
+register("load_skill", ToolDisplay(label="Loading skill", icon="🧠", summary_arg="summary"))
+register("spawn_workers", ToolDisplay(label="Spawning workers", icon="⚙️"))
+register(
+    "read_dataset_info",
+    ToolDisplay(label="Reading dataset info", icon="📚", summary_arg="summary"),
+)
+register(
+    "update_dataset_info",
+    ToolDisplay(label="Updating dataset notes", icon="📝", display=False),
+)
+register("profile_dataset", ToolDisplay(label="Profiling dataset", icon="📊"))
+register("list_workspace_files", ToolDisplay(label="Listing workspace files", icon="📁"))
+register(
+    "web_search",
+    ToolDisplay(label="Searching the web", icon="🌐", summary_arg="summary"),
+)
+register(
+    "fetch_url",
+    ToolDisplay(label="Fetching content", icon="🔗", summary_arg="summary"),
+)
+register("ask_user_mcq", ToolDisplay(label="Question", icon="💬", display=False))
+register("enter_self_review_mode", ToolDisplay(label="Reviewing progress so far", icon="🔍"))
+register("exit_self_review_mode", ToolDisplay(label="Done reviewing progress", icon="✅"))
+register("verify_python_code", ToolDisplay(label="Reviewing code", icon="🔎"))