zeroquantz 0.1.0__py3-none-any.whl

zeroquantz/tui/styles/app.tcss

@@ -0,0 +1,424 @@
+/* ZeroQuantz TUI theme — OpenCode-style dark, cyan/green accents. */
+
+Screen {
+    background: #0b0e14;
+    color: #c9d1d9;
+}
+
+/* ---- welcome screen ---- */
+
+#welcome {
+    align: center middle;
+    height: 1fr;
+}
+
+.banner {
+    color: #c9d1d9;
+    text-style: bold;
+    content-align: center middle;
+    width: auto;
+    margin-bottom: 1;
+}
+
+.tagline {
+    color: #6e7681;
+    content-align: center middle;
+    width: auto;
+    margin-bottom: 1;
+}
+
+.status-line {
+    content-align: center middle;
+    width: auto;
+    margin-bottom: 1;
+}
+
+.prompt-line {
+    color: #58a6ff;
+    text-style: bold;
+    content-align: center middle;
+    width: auto;
+    margin-bottom: 1;
+}
+
+#menu {
+    width: auto;
+    max-width: 72;
+    height: auto;
+    background: transparent;
+    border: none;
+    padding: 0 2;
+}
+
+OptionList {
+    background: transparent;
+    border: none;
+}
+
+OptionList > .option-list--option-highlighted {
+    background: #172033;
+    text-style: bold;
+}
+
+OptionList:focus > .option-list--option-highlighted {
+    background: #1f2b40;
+    text-style: bold;
+}
+
+/* ---- workspace screen ---- */
+
+#topbar {
+    height: 1;
+    background: #11151c;
+    color: #8b949e;
+    padding: 0 1;
+}
+
+#body {
+    height: 1fr;
+}
+
+#conversation {
+    width: 1fr;
+    height: 1fr;
+    background: #0d1117;
+    padding: 0 1;
+    scrollbar-size-vertical: 1;
+}
+
+#sidebar {
+    width: 38;
+    height: 1fr;
+    background: #0d1117;
+    border-left: solid #21262d;
+    padding: 0 1;
+    display: block;
+}
+
+#sidebar.hidden {
+    display: none;
+}
+
+#progress {
+    height: auto;
+    color: #d29922;
+    padding: 0 1;
+    display: none;
+}
+
+#progress.active {
+    display: block;
+}
+
+#prompt {
+    dock: bottom;
+    border: tall #21262d;
+    background: #0d1117;
+    color: #c9d1d9;
+}
+
+#prompt:focus {
+    border: tall #58a6ff;
+}
+
+/* ---- wizard pages ---- */
+
+.page-title {
+    color: #58a6ff;
+    text-style: bold;
+    padding: 1 2 0 2;
+}
+
+.page-subtitle {
+    color: #6e7681;
+    padding: 0 2 1 2;
+}
+
+.page-hint {
+    color: #6e7681;
+    padding: 1 2;
+}
+
+#model-page {
+    padding: 1 2;
+    height: 1fr;
+}
+
+#model-input {
+    border: tall #21262d;
+    background: #0d1117;
+    margin: 0 0 1 0;
+}
+
+#model-input:focus {
+    border: tall #58a6ff;
+}
+
+#model-suggestions {
+    height: 1fr;
+    background: transparent;
+    border: none;
+    margin-bottom: 1;
+}
+
+#model-summary {
+    height: auto;
+    max-height: 12;
+    padding: 1 0;
+}
+
+#model-actions {
+    height: auto;
+    align: left middle;
+}
+
+#download-page {
+    padding: 1 2;
+    height: 1fr;
+}
+
+#dl-model {
+    height: auto;
+    margin-bottom: 1;
+}
+
+#dl-progress {
+    width: 100%;
+    margin: 1 0;
+}
+
+#quant-page {
+    padding: 1 2;
+    height: 1fr;
+}
+
+#quant-model {
+    padding: 0 0 1 0;
+    height: auto;
+}
+
+.ctl-row {
+    height: 1;
+    width: 1fr;
+}
+
+.ctl-label {
+    color: #8b949e;
+    width: auto;
+    padding: 0 1 0 2;
+}
+
+.ctl-value {
+    color: #3fb950;
+    text-style: bold;
+    width: auto;
+    padding: 0 1;
+}
+
+.chip {
+    width: auto;
+    height: 1;
+    padding: 0 1;
+    margin: 0 1 0 0;
+    color: #c9d1d9;
+}
+
+.chip:hover {
+    background: #161b22;
+}
+
+.chip:focus {
+    background: #161b22;
+    text-style: bold;
+}
+
+.chip.-selected {
+    background: #1f2b40;
+    color: #58a6ff;
+    text-style: bold;
+}
+
+#quant-table {
+    height: 1fr;
+    margin-top: 1;
+    background: #0d1117;
+}
+
+/* ---- plan + execute ---- */
+
+#plan-page,
+#exec-page {
+    padding: 1 2;
+    height: 1fr;
+}
+
+#deploy-chips {
+    height: 1;
+}
+
+#plan-scroll {
+    height: 1fr;
+}
+
+#plan-body {
+    height: auto;
+}
+
+#plan-status {
+    height: auto;
+}
+
+#exec-stages {
+    height: auto;
+    padding: 0 0 1 0;
+}
+
+#exec-progress {
+    width: 100%;
+    margin: 0 0 1 0;
+}
+
+#exec-log {
+    height: 1fr;
+    min-height: 6;
+    background: #0d1117;
+    border: round #21262d;
+    padding: 0 1;
+}
+
+DataTable > .datatable--cursor {
+    background: #1f2b40;
+    text-style: bold;
+}
+
+DataTable > .datatable--header {
+    color: #58a6ff;
+    text-style: bold;
+    background: #11151c;
+}
+
+#summary-page {
+    padding: 1 2;
+    height: 1fr;
+}
+
+#summary-scroll {
+    height: 1fr;
+}
+
+#summary-body {
+    height: auto;
+}
+
+#summary-log {
+    height: 9;
+    min-height: 3;
+    background: #0d1117;
+    border: round #21262d;
+    padding: 0 1;
+    margin-top: 1;
+}
+
+/* ---- settings ---- */
+
+#settings-page {
+    padding: 1 2;
+    height: 1fr;
+}
+
+#settings-tabs {
+    height: 1fr;
+}
+
+#dl-table,
+#q-table {
+    height: 1fr;
+    background: #0d1117;
+}
+
+#tok-status {
+    height: auto;
+    padding: 1 0;
+}
+
+/* ---- adjust page ---- */
+
+#adjust-page {
+    padding: 1 2;
+    height: 1fr;
+}
+
+#adjust-body {
+    height: auto;
+    padding: 1 0;
+}
+
+/* ---- confirm modal ---- */
+
+ConfirmScreen {
+    align: center middle;
+    background: #000000 60%;
+}
+
+#confirm-dialog {
+    width: 74;
+    max-width: 90%;
+    height: auto;
+    background: #0d1117;
+    border: round #f85149;
+    padding: 1 2;
+}
+
+#confirm-message {
+    height: auto;
+    padding-bottom: 1;
+}
+
+/* ---- token modal ---- */
+
+TokenScreen {
+    align: center middle;
+    background: #000000 60%;
+}
+
+#token-dialog {
+    width: 78;
+    max-width: 90%;
+    height: auto;
+    background: #0d1117;
+    border: round #58a6ff;
+    padding: 1 2;
+}
+
+.dialog-title {
+    color: #58a6ff;
+    text-style: bold;
+}
+
+.dialog-sub {
+    color: #6e7681;
+    padding: 0 0 1 0;
+}
+
+#token-status {
+    height: auto;
+    padding: 0 0 1 0;
+}
+
+#token-input {
+    border: tall #21262d;
+    background: #0b0e14;
+}
+
+#token-input:focus {
+    border: tall #58a6ff;
+}
+
+#token-actions {
+    height: auto;
+    padding-top: 1;
+}
+
+#token-actions Button {
+    margin: 0 2 0 0;
+}

zeroquantz/tui/widgets/__init__.py

@@ -0,0 +1,9 @@
+"""ZeroQuantz TUI widgets."""
+
+from __future__ import annotations
+
+from zeroquantz.tui.widgets.chip import Chip
+from zeroquantz.tui.widgets.sidebar import Sidebar
+from zeroquantz.tui.widgets.status_bar import StatusBar
+
+__all__ = ["Chip", "Sidebar", "StatusBar"]

zeroquantz/tui/widgets/chip.py

@@ -0,0 +1,36 @@
+"""A small, clickable (and keyboard-focusable) chip for inline segmented controls."""
+
+from __future__ import annotations
+
+from textual.binding import Binding
+from textual.message import Message
+from textual.widgets import Static
+
+
+class Chip(Static):
+    """A selectable inline control. Click it, or focus it and press Enter/Space."""
+
+    can_focus = True
+    BINDINGS = [Binding("enter,space", "select", "Select", show=False)]
+
+    class Selected(Message):
+        def __init__(self, group: str, value: str) -> None:
+            self.group = group
+            self.value = value
+            super().__init__()
+
+    def __init__(self, label: str, *, group: str, value: str, selected: bool = False) -> None:
+        super().__init__(label)
+        self.chip_group = group
+        self.chip_value = value
+        self.add_class("chip")
+        self.set_class(selected, "-selected")
+
+    def set_selected(self, selected: bool) -> None:
+        self.set_class(selected, "-selected")
+
+    def on_click(self) -> None:
+        self.post_message(self.Selected(self.chip_group, self.chip_value))
+
+    def action_select(self) -> None:
+        self.post_message(self.Selected(self.chip_group, self.chip_value))

zeroquantz/tui/widgets/sidebar.py

@@ -0,0 +1,107 @@
+"""Collapsible side panel: Model / Hardware / Goal / Plan / Session at a glance."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from rich.console import Group
+from rich.rule import Rule
+from rich.table import Table
+from rich.text import Text
+from textual.widgets import Static
+
+from zeroquantz.utils import units
+
+if TYPE_CHECKING:
+    from zeroquantz.core.context import AppContext
+
+_HEAD = "bold cyan"
+_MUTED = "grey62"
+
+
+def _section(title: str, rows: list[tuple[str, str]]) -> Group:
+    grid = Table.grid(padding=(0, 1))
+    grid.add_column(style=_MUTED, no_wrap=True)
+    grid.add_column(style="white")
+    for k, v in rows:
+        grid.add_row(k, v)
+    return Group(Text(title, style=_HEAD), grid, Text(""))
+
+
+class Sidebar(Static):
+    """Reactive summary of the current session state."""
+
+    def update_context(self, ctx: AppContext) -> None:
+        sections: list = []
+
+        # Model
+        mp = ctx.session.model_profile
+        if mp is not None:
+            sections.append(
+                _section(
+                    "Model",
+                    [
+                        ("id", mp.model_id),
+                        ("arch", mp.architecture or "?"),
+                        ("params", units.humanize_params(mp.parameter_count)),
+                        ("dtype", mp.dtype),
+                        ("size", f"{mp.estimated_weight_size_gb:.1f} GB"),
+                    ],
+                )
+            )
+        else:
+            sections.append(_section("Model", [("", "none loaded")]))
+
+        # Goal
+        goal = ctx.session.goal
+        goal_rows = [("objective", str(goal.objective))]
+        if goal.max_vram_gb:
+            goal_rows.append(("max VRAM", f"{goal.max_vram_gb:g} GB"))
+        if goal.runtime:
+            goal_rows.append(("runtime", goal.runtime))
+        if goal.max_quality_loss:
+            goal_rows.append(("max loss", f"{goal.max_quality_loss*100:g}%"))
+        sections.append(_section("Goal", goal_rows))
+
+        # Hardware
+        hw = ctx.hardware
+        sections.append(
+            _section(
+                "Hardware",
+                [
+                    ("gpu", hw.gpu_name),
+                    ("vram", f"{hw.total_vram_gb:g} GB" if hw.total_vram_gb else "n/a"),
+                    ("sm", hw.sm or "-"),
+                    ("cuda", hw.cuda_version or ("yes" if hw.cuda_available else "no")),
+                ],
+            )
+        )
+
+        # Plan
+        plan = ctx.session.plan
+        if plan is not None:
+            sections.append(
+                _section(
+                    "Active plan",
+                    [
+                        ("scheme", f"{plan.low_precision}/{plan.high_precision}"),
+                        ("size", f"{plan.estimated_size_gb:.1f} GB"),
+                        ("vram", f"{plan.estimated_vram_gb:.1f} GB"),
+                        ("fits", "yes" if plan.fits_budget else "no"),
+                    ],
+                )
+            )
+
+        # Session
+        sections.append(
+            _section(
+                "Session",
+                [
+                    ("name", ctx.session.name),
+                    ("ops", str(len(ctx.session.operation_log))),
+                    ("selected", ctx.session.selected_method or "-"),
+                ],
+            )
+        )
+
+        self.update(Group(Rule(style="grey30"), *sections))

zeroquantz/tui/widgets/status_bar.py

@@ -0,0 +1,43 @@
+"""Top status bar: model + goal on the left, GPU + VRAM on the right."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from rich.table import Table
+from rich.text import Text
+from textual.widgets import Static
+
+if TYPE_CHECKING:
+    from zeroquantz.core.context import AppContext
+
+
+class StatusBar(Static):
+    """A one-line context header, updated as the session changes."""
+
+    def update_context(self, ctx: AppContext) -> None:
+        session = ctx.session
+        model = session.model_id or "no model"
+        goal = session.goal
+        goal_bits = [str(goal.objective)]
+        if goal.max_vram_gb:
+            goal_bits.append(f"≤{goal.max_vram_gb:g}GB")
+        if goal.runtime:
+            goal_bits.append(goal.runtime)
+        left = Text.assemble(
+            ("Model ", "grey62"),
+            (model, "bold #58a6ff"),
+            ("   Goal ", "grey62"),
+            (" · ".join(goal_bits), "white"),
+        )
+        vram = f"{ctx.hardware.total_vram_gb:g} GB" if ctx.hardware.total_vram_gb else "n/a"
+        right = Text.assemble(
+            (ctx.hardware.gpu_name, "white"),
+            ("  VRAM ", "grey62"),
+            (vram, "bold #3fb950" if ctx.hardware.cuda_available else "bold #d29922"),
+        )
+        grid = Table.grid(expand=True)
+        grid.add_column(justify="left", ratio=1)
+        grid.add_column(justify="right")
+        grid.add_row(left, right)
+        self.update(grid)

zeroquantz/utils/__init__.py

@@ -0,0 +1,8 @@
+"""Small, dependency-light helpers shared across ZeroQuantz."""
+
+from __future__ import annotations
+
+from zeroquantz.utils import units
+from zeroquantz.utils.paths import ZeroQuantzPaths, paths
+
+__all__ = ["ZeroQuantzPaths", "paths", "units"]

zeroquantz/utils/config.py

@@ -0,0 +1,46 @@
+"""Load ``~/.zeroquantz/config.toml`` into a typed :class:`AppConfig`.
+
+Precedence (low -> high) is defaults < config file < session < CLI flags. This
+module only handles the file layer; callers apply the higher-precedence layers.
+"""
+
+from __future__ import annotations
+
+import tomllib
+from pathlib import Path
+from typing import Any
+
+from pydantic import BaseModel
+
+from zeroquantz.core.exceptions import ConfigError
+from zeroquantz.utils.paths import paths
+
+
+class BenchmarkConfig(BaseModel):
+    warmup_runs: int = 3
+    runs: int = 10
+
+
+class AppConfig(BaseModel):
+    default_runtime: str | None = None
+    objective: str = "balanced"
+    cache_dir: str | None = None
+    benchmark: BenchmarkConfig = BenchmarkConfig()
+
+
+def load_config(path: str | Path | None = None) -> AppConfig:
+    config_path = Path(path) if path is not None else paths().config_file
+    if not config_path.exists():
+        return AppConfig()
+    try:
+        data: dict[str, Any] = tomllib.loads(config_path.read_text(encoding="utf-8"))
+    except (OSError, tomllib.TOMLDecodeError) as exc:
+        raise ConfigError(
+            f"Could not read config at {config_path}.", detail=str(exc)
+        ) from exc
+    try:
+        return AppConfig.model_validate(data)
+    except ValueError as exc:
+        raise ConfigError(
+            f"Invalid configuration in {config_path}.", detail=str(exc)
+        ) from exc