attacklm 0.7.0__tar.gz → 0.8.0__tar.gz

{attacklm-0.7.0 → attacklm-0.8.0}/CHANGELOG.md

@@ -4,6 +4,57 @@ All notable changes to AttackLM are documented in this file. Versions follow [Se
 
 ---
 
+## [0.8.0] — 2026-06-30 — Unified CLI with subcommands
+
+### Added
+- **Unified CLI**: New `attacklm` command with argparse subcommands replaces 21 hyphenated commands
+  - `attacklm train [--dataset all] [--hpo]` — consolidates `attacklm-train`, `attacklm-train-all`, `attacklm-train-lora`, `attacklm-hpo`
+  - `attacklm init [--extract-only|--buckets-only|--attribute-only|--clone-only]` — consolidates `attacklm-init`, `attacklm-extract`, `attacklm-buckets`, `attacklm-attribute`, `attacklm-clone`
+  - `attacklm build [--merge-only|--gguf-only|--register-ollama]` — consolidates `attacklm-build`, `attacklm-merge`, `attacklm-gguf`, `attacklm-register-ollama`
+  - `attacklm eval [--collect-ref|--score|--compare|--golden]` — consolidates `attacklm-eval`, `attacklm-collect-ref`, `attacklm-score`, `attacklm-compare`, `attacklm-golden`
+  - `attacklm balance` — replaces `attacklm-balance`
+  - `attacklm infer` — replaces `attacklm-infer`
+  - `attacklm gui` — replaces `attacklm-gui`
+  - `attacklm demo` — replaces `attacklm-demo`
+- **`attacklm --version`**: Shows version number
+- **Evaluation screen in GUI**: New `EvalFormScreen` with buttons for all eval subcommands (Retention, Collect Ref, Score, Compare, Golden)
+- **Build screen in GUI**: Consolidated buttons for Full Build, Merge Only, GGUF Only
+
+### Changed
+- **GUI main menu**: Consolidated from 8 buttons to 6 (Train, Init, Balance, Build, Infer, Eval) — removed separate Extract/Merge/Pipeline buttons
+- **GUI train form**: Now calls `attacklm train` instead of `attacklm-train`
+- **GUI command forms**: All commands now use `attacklm <subcommand>` format instead of `attacklm-<command>`
+- **Deprecation warnings**: All old hyphenated commands print a deprecation notice before delegating to the new subcommand
+
+### Deprecated
+- All 21 hyphenated commands (`attacklm-train`, `attacklm-train-all`, etc.) still work but print deprecation warnings. They will be removed in v0.9.0.
+
+---
+
+## [0.7.1] — 2026-06-30 — GUI, VRAM fixes, dataset caching
+
+### Added
+- **attacklm-gui**: Terminal-based TUI wrapper (Textual) for all CLI commands
+  - Form-based training config with 40+ params in 5 tabs (Basic, LoRA, GaLore, Advanced, Hardware)
+  - Live training monitor with loss sparkline, VRAM gauge, progress bar, log viewer
+  - One-click screens for extract, balance, infer, merge, build, pipeline, init
+  - 5 built-in presets (3B Q-GaLore, 3B LoRA, 7B Q-GaLore, 7B QLoRA, etc.)
+  - Pause/resume/quit controls during training
+  - Works in terminal-only environments (no X11/WSLg required)
+- **Dataset caching**: Tokenized datasets saved to disk after first run, reloaded in ~1s on subsequent runs (keyed by dataset + model + max_length + packing)
+- **config.json fix**: Training runs now preserve HF model architecture fields (model_type, architectures, hidden_size, etc.) for GGUF conversion compatibility
+
+### Fixed
+- **VRAM fragmentation**: Removed `device_map="auto"` for single-GPU training — was splitting model across GPU+CPU, creating meta tensors, and leaving VRAM fragmented. Model now loads directly to GPU.
+- **Spectrum crash on 7B**: Skip meta-device tensors in SNR computation (caused by CPU-offloaded layers from device_map="auto")
+- **VRAM parsing**: Added regex for compact VRAM format (`VRAM X/Y GB (A alloc + C cache)`) in training output parser
+
+### Changed
+- Moved attacklm-gui into the main AttackLM repo under `attacklm-gui/`
+- 17 GUI tests passing
+
+---
+
 ## [0.7.0] — 2026-06-29 — Defensive data extraction, test fixes, script cleanup
 
 ### Added

{attacklm-0.7.0 → attacklm-0.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: attacklm
-Version: 0.7.0
+Version: 0.8.0
 Summary: QLoRA-fine-tuned Qwen2.5-Coder for offensive/defensive security and AI red-teaming
 Project-URL: Homepage, https://github.com/Veedubin/AttackLM
 Project-URL: Repository, https://github.com/Veedubin/AttackLM
@@ -205,6 +205,27 @@ AttackLM — QLoRA fine-tuning pipeline for a MITRE ATT&CK-grounded security AI
 
 ---
 
+## GUI (New in v0.7.1)
+
+AttackLM now includes a terminal-based GUI for all commands. No more memorizing 40+ CLI flags.
+
+```bash
+pip install attacklm-gui
+attacklm-gui
+```
+
+Features:
+- **Training form** with 40+ params in 5 tabs (Basic, LoRA, GaLore, Advanced, Hardware)
+- **Live training monitor** with loss sparkline, VRAM gauge, progress bar, and log viewer
+- **One-click screens** for extract, balance, infer, merge, build, pipeline, and init
+- **5 built-in presets** (3B Q-GaLore, 3B LoRA, 7B Q-GaLore, 7B QLoRA, etc.)
+- **Pause/resume/quit** controls during training
+- Works in **terminal-only** environments — no X11, no browser, no GPU required. Works over SSH and WSL.
+
+The CLI still works exactly as before. The GUI is a thin wrapper that constructs and runs CLI commands.
+
+---
+
 ## Install
 The recommended way to install the full CUDA training stack:
 
@@ -346,29 +367,62 @@ This layout allows the pipeline to deterministically extract data from upstream
 ---
 
 ## CLI Reference
+
+> **v0.8.0**: The 21 hyphenated commands are consolidated into a single
+> `attacklm` command with subcommands. The old commands still work but
+> print a deprecation warning. They will be removed in v0.9.0.
+
+### Unified Command (v0.8.0+)
+
 | Command | Description |
 |---------|-------------|
-| `attacklm-init` | One-shot init: clone $\rightarrow$ extract $\rightarrow$ attribute $\rightarrow$ buckets |
-| `attacklm-train` | Train a single QLoRA adapter |
-| `attacklm-train-all` | Orchestrated training for all buckets / HPO |
-| `attacklm-balance` | Build a balanced subset of buckets |
-| `attacklm-hpo` | Coordinate-descent HPO sweep |
-| `attacklm-infer` | Smoke-test inference |
-| `attacklm-merge` | Merge LoRA adapter into base model |
-| `attacklm-gguf` | Convert merged model to GGUF (llama.cpp) |
-| `attacklm-build` | One-shot merge $\rightarrow$ GGUF $\rightarrow$ install |
-| `attacklm-demo` | Multi-agent orchestrator demo |
-| `attacklm-extract` | Extract data from cloned repositories |
-| `attacklm-buckets` | Organize data into bucket structure |
-| `attacklm-attribute` | Add source and license to JSONL records |
-| `attacklm-clone` | Clone upstream data repositories |
-| `attacklm-train-lora` | Direct LoRA training on single dataset |
-| `attacklm-eval` | Run retention evaluation suite |
-| `attacklm-collect-ref` | Collect reference model outputs |
-| `attacklm-score` | Score candidate models against reference |
-| `attacklm-compare` | Compare multiple candidate model scores |
-| `attacklm-golden` | Execute golden vector regression gates |
-| `attacklm-register-ollama`| Register GGUF model with Ollama |
+| `attacklm train` | Train a model (QLoRA, GaLore, Q-GaLore, Spectrum, PiSSA) |
+| `attacklm train --dataset all` | Train all buckets (replaces `attacklm-train-all`) |
+| `attacklm train --hpo` | Run HPO sweep (replaces `attacklm-hpo`) |
+| `attacklm init` | One-shot init: clone → extract → attribute → buckets |
+| `attacklm init --extract-only` | Extract data only (replaces `attacklm-extract`) |
+| `attacklm init --buckets-only` | Organize into buckets only (replaces `attacklm-buckets`) |
+| `attacklm init --attribute-only` | Add attribution only (replaces `attacklm-attribute`) |
+| `attacklm init --clone-only` | Clone repos only (replaces `attacklm-clone`) |
+| `attacklm balance` | Build a balanced subset of buckets |
+| `attacklm build` | One-shot merge → GGUF → install to LM Studio |
+| `attacklm build --merge-only` | Merge adapter only (replaces `attacklm-merge`) |
+| `attacklm build --gguf-only` | Convert to GGUF only (replaces `attacklm-gguf`) |
+| `attacklm build --register-ollama` | Register GGUF with Ollama (replaces `attacklm-register-ollama`) |
+| `attacklm infer` | Smoke-test inference |
+| `attacklm eval` | Run retention evaluation suite |
+| `attacklm eval --collect-ref` | Collect reference model outputs |
+| `attacklm eval --score` | Score candidate models against reference |
+| `attacklm eval --compare` | Compare multiple candidate model scores |
+| `attacklm eval --golden` | Execute golden vector regression gates |
+| `attacklm gui` | Terminal GUI for all commands |
+| `attacklm demo` | Multi-agent orchestrator demo |
+
+### Legacy Commands (Deprecated, removed in v0.9.0)
+
+| Old Command | New Command |
+|-------------|-------------|
+| `attacklm-train` | `attacklm train` |
+| `attacklm-train-all` | `attacklm train --dataset all` |
+| `attacklm-train-lora` | `attacklm train` |
+| `attacklm-hpo` | `attacklm train --hpo` |
+| `attacklm-init` | `attacklm init` |
+| `attacklm-extract` | `attacklm init --extract-only` |
+| `attacklm-buckets` | `attacklm init --buckets-only` |
+| `attacklm-attribute` | `attacklm init --attribute-only` |
+| `attacklm-clone` | `attacklm init --clone-only` |
+| `attacklm-balance` | `attacklm balance` |
+| `attacklm-build` | `attacklm build` |
+| `attacklm-merge` | `attacklm build --merge-only` |
+| `attacklm-gguf` | `attacklm build --gguf-only` |
+| `attacklm-register-ollama` | `attacklm build --register-ollama` |
+| `attacklm-infer` | `attacklm infer` |
+| `attacklm-eval` | `attacklm eval` |
+| `attacklm-collect-ref` | `attacklm eval --collect-ref` |
+| `attacklm-score` | `attacklm eval --score` |
+| `attacklm-compare` | `attacklm eval --compare` |
+| `attacklm-golden` | `attacklm eval --golden` |
+| `attacklm-demo` | `attacklm demo` |
 
 ---
 

{attacklm-0.7.0 → attacklm-0.8.0}/README.md

@@ -7,6 +7,27 @@ AttackLM — QLoRA fine-tuning pipeline for a MITRE ATT&CK-grounded security AI
 
 ---
 
+## GUI (New in v0.7.1)
+
+AttackLM now includes a terminal-based GUI for all commands. No more memorizing 40+ CLI flags.
+
+```bash
+pip install attacklm-gui
+attacklm-gui
+```
+
+Features:
+- **Training form** with 40+ params in 5 tabs (Basic, LoRA, GaLore, Advanced, Hardware)
+- **Live training monitor** with loss sparkline, VRAM gauge, progress bar, and log viewer
+- **One-click screens** for extract, balance, infer, merge, build, pipeline, and init
+- **5 built-in presets** (3B Q-GaLore, 3B LoRA, 7B Q-GaLore, 7B QLoRA, etc.)
+- **Pause/resume/quit** controls during training
+- Works in **terminal-only** environments — no X11, no browser, no GPU required. Works over SSH and WSL.
+
+The CLI still works exactly as before. The GUI is a thin wrapper that constructs and runs CLI commands.
+
+---
+
 ## Install
 The recommended way to install the full CUDA training stack:
 
@@ -148,29 +169,62 @@ This layout allows the pipeline to deterministically extract data from upstream
 ---
 
 ## CLI Reference
+
+> **v0.8.0**: The 21 hyphenated commands are consolidated into a single
+> `attacklm` command with subcommands. The old commands still work but
+> print a deprecation warning. They will be removed in v0.9.0.
+
+### Unified Command (v0.8.0+)
+
 | Command | Description |
 |---------|-------------|
-| `attacklm-init` | One-shot init: clone $\rightarrow$ extract $\rightarrow$ attribute $\rightarrow$ buckets |
-| `attacklm-train` | Train a single QLoRA adapter |
-| `attacklm-train-all` | Orchestrated training for all buckets / HPO |
-| `attacklm-balance` | Build a balanced subset of buckets |
-| `attacklm-hpo` | Coordinate-descent HPO sweep |
-| `attacklm-infer` | Smoke-test inference |
-| `attacklm-merge` | Merge LoRA adapter into base model |
-| `attacklm-gguf` | Convert merged model to GGUF (llama.cpp) |
-| `attacklm-build` | One-shot merge $\rightarrow$ GGUF $\rightarrow$ install |
-| `attacklm-demo` | Multi-agent orchestrator demo |
-| `attacklm-extract` | Extract data from cloned repositories |
-| `attacklm-buckets` | Organize data into bucket structure |
-| `attacklm-attribute` | Add source and license to JSONL records |
-| `attacklm-clone` | Clone upstream data repositories |
-| `attacklm-train-lora` | Direct LoRA training on single dataset |
-| `attacklm-eval` | Run retention evaluation suite |
-| `attacklm-collect-ref` | Collect reference model outputs |
-| `attacklm-score` | Score candidate models against reference |
-| `attacklm-compare` | Compare multiple candidate model scores |
-| `attacklm-golden` | Execute golden vector regression gates |
-| `attacklm-register-ollama`| Register GGUF model with Ollama |
+| `attacklm train` | Train a model (QLoRA, GaLore, Q-GaLore, Spectrum, PiSSA) |
+| `attacklm train --dataset all` | Train all buckets (replaces `attacklm-train-all`) |
+| `attacklm train --hpo` | Run HPO sweep (replaces `attacklm-hpo`) |
+| `attacklm init` | One-shot init: clone → extract → attribute → buckets |
+| `attacklm init --extract-only` | Extract data only (replaces `attacklm-extract`) |
+| `attacklm init --buckets-only` | Organize into buckets only (replaces `attacklm-buckets`) |
+| `attacklm init --attribute-only` | Add attribution only (replaces `attacklm-attribute`) |
+| `attacklm init --clone-only` | Clone repos only (replaces `attacklm-clone`) |
+| `attacklm balance` | Build a balanced subset of buckets |
+| `attacklm build` | One-shot merge → GGUF → install to LM Studio |
+| `attacklm build --merge-only` | Merge adapter only (replaces `attacklm-merge`) |
+| `attacklm build --gguf-only` | Convert to GGUF only (replaces `attacklm-gguf`) |
+| `attacklm build --register-ollama` | Register GGUF with Ollama (replaces `attacklm-register-ollama`) |
+| `attacklm infer` | Smoke-test inference |
+| `attacklm eval` | Run retention evaluation suite |
+| `attacklm eval --collect-ref` | Collect reference model outputs |
+| `attacklm eval --score` | Score candidate models against reference |
+| `attacklm eval --compare` | Compare multiple candidate model scores |
+| `attacklm eval --golden` | Execute golden vector regression gates |
+| `attacklm gui` | Terminal GUI for all commands |
+| `attacklm demo` | Multi-agent orchestrator demo |
+
+### Legacy Commands (Deprecated, removed in v0.9.0)
+
+| Old Command | New Command |
+|-------------|-------------|
+| `attacklm-train` | `attacklm train` |
+| `attacklm-train-all` | `attacklm train --dataset all` |
+| `attacklm-train-lora` | `attacklm train` |
+| `attacklm-hpo` | `attacklm train --hpo` |
+| `attacklm-init` | `attacklm init` |
+| `attacklm-extract` | `attacklm init --extract-only` |
+| `attacklm-buckets` | `attacklm init --buckets-only` |
+| `attacklm-attribute` | `attacklm init --attribute-only` |
+| `attacklm-clone` | `attacklm init --clone-only` |
+| `attacklm-balance` | `attacklm balance` |
+| `attacklm-build` | `attacklm build` |
+| `attacklm-merge` | `attacklm build --merge-only` |
+| `attacklm-gguf` | `attacklm build --gguf-only` |
+| `attacklm-register-ollama` | `attacklm build --register-ollama` |
+| `attacklm-infer` | `attacklm infer` |
+| `attacklm-eval` | `attacklm eval` |
+| `attacklm-collect-ref` | `attacklm eval --collect-ref` |
+| `attacklm-score` | `attacklm eval --score` |
+| `attacklm-compare` | `attacklm eval --compare` |
+| `attacklm-golden` | `attacklm eval --golden` |
+| `attacklm-demo` | `attacklm demo` |
 
 ---
 

attacklm-0.8.0/attacklm-gui/README.md

@@ -0,0 +1,145 @@
+# AttackLM GUI
+
+Terminal-based GUI wrapper for the [AttackLM](https://github.com/Veedubin/AttackLM) CLI training tools. Built with [Textual](https://textual.textualize.io/).
+
+## Why?
+
+AttackLM has a unified CLI (`attacklm train`, `attacklm init`, etc.) with 8 subcommands. `attacklm train` alone has 40+ parameters. Remembering all the flags is painful. The GUI provides:
+
+- **Form-based parameter input** with tabs, presets, and validation
+- **Live training monitoring** with loss charts, VRAM gauges, and log streaming
+- **One-click commands** for extract, balance, infer, merge, build, and more
+- **Works everywhere** — terminal-only, no X11 required (WSL, SSH, headless)
+
+The unified CLI (`attacklm <subcommand>`) still works as before. The GUI is a thin wrapper that constructs and runs CLI commands.
+
+## Installation
+
+```bash
+pip install attacklm-gui
+```
+
+Or from source:
+
+```bash
+git clone https://github.com/Veedubin/AttackLM.git
+cd AttackLM-Models/attacklm-gui
+pip install -e .
+```
+
+Requires `attacklm` (the CLI package) and `textual>=2.0`.
+
+## Usage
+
+```bash
+attacklm-gui
+```
+
+### Main Menu
+
+```
+┌──────────────────────────────────────┐
+│         AttackLM GUI v0.1.0          │
+│                                      │
+│  🏋️  Train Model                     │
+│  🚀  Init Dataset                    │
+│  ⚖️  Balance Dataset                 │
+│  📦  Build & Install                 │
+│  🧠  Run Inference                   │
+│  📊  Evaluate                        │
+└──────────────────────────────────────┘
+```
+
+### Training Form
+
+The training form organizes 40+ parameters into tabs:
+
+| Tab | Parameters |
+|-----|-----------|
+| **Basic** | dataset, output, base model, epochs, batch size, max length |
+| **LoRA** | rank, alpha, dropout, DoRA, RSLoRA, target modules, PiSSA |
+| **GaLore** | Q-GaLore, rank, 32-bit, Spectrum layer freezing |
+| **Advanced** | eval split, early stopping, save steps, gradient accum, optimizer, packing, live LR |
+| **Hardware** | precision (BF16/FP16/FP32), multi-GPU, MoE safe, auto-tune |
+
+**Presets** save and load complete configurations. Built-in presets:
+- `3B Q-GaLore Spectrum` — full-parameter training on 16GB GPU
+- `3B Q-GaLore Rank 128` — higher quality variant
+- `3B LoRA Default` — standard QLoRA
+- `7B Q-GaLore` — for 24GB GPUs
+- `7B QLoRA Default` — standard QLoRA for 7B
+
+### Live Training Monitor
+
+```
+┌─────────────────────────────────────────────────────┐
+│  Training: attacklm-3b-qgalore-spectrum             │
+│  Epoch 12/30  |  Step 1896  |  Elapsed: 1h 23m     │
+├──────────────────────┬──────────────────────────────┤
+│  Loss: 1.1143 ▂▃▅▆▇  │  VRAM: 5.9/15.6 GB (62%)   │
+│  Eval Loss: 1.176    │  alloc: 5.9  cache: 4.8     │
+│  Trend: ↓ -0.0116    │  ████████░░░░               │
+├──────────────────────┴──────────────────────────────┤
+│  Log Output (scrolling)                             │
+│  Epoch 12/20  58% | loss 1.1143 | 2,844 tok/s ...  │
+├─────────────────────────────────────────────────────┤
+│  [P]ause  [S]top at checkpoint  [Q]uit              │
+└─────────────────────────────────────────────────────┘
+```
+
+Controls:
+- **Pause/Resume** — SIGSTOP/SIGCONT the training process
+- **Stop at Checkpoint** — graceful stop after current eval
+- **Quit** — kill training immediately
+
+### Other Commands
+
+| Screen | What it does |
+|--------|-------------|
+| **Init** | Clone repos, extract data, organize buckets |
+| **Balance** | Balance a dataset with cap size |
+| **Inference** | Run a trained model with a prompt |
+| **Build** | Merge → GGUF → install to LM Studio |
+| **Eval** | Retention eval, collect-ref, score, compare, golden |
+| **Pipeline** | Run a YAML training pipeline |
+
+## Keyboard Shortcuts
+
+| Key | Action |
+|-----|--------|
+| `q` | Quit |
+| `m` | Return to main menu |
+| `Tab` | Next field |
+| `Enter` | Activate button |
+| `Esc` | Go back |
+
+## Architecture
+
+```
+attacklm-gui/
+├── src/attacklm_gui/
+│   ├── app.py              # Textual App entry point
+│   ├── cli.py              # CLI entry point
+│   ├── runner.py           # Subprocess manager + output parser
+│   ├── presets.py          # Save/load training presets
+│   ├── screens/
+│   │   ├── main_menu.py    # Command launcher (6 buttons)
+│   │   ├── train_form.py   # Training parameter form
+│   │   ├── train_live.py   # Live training monitor
+│   │   └── command_forms.py # Init, Balance, Infer, Build, Eval, Pipeline screens
+│   └── widgets/
+│       └── __init__.py
+└── tests/
+    └── test_gui.py
+```
+
+The GUI never touches AttackLM's internal logic. It constructs CLI command strings and runs them via `asyncio.create_subprocess_exec`. All training, extraction, and inference logic stays in `scripts/*.py`.
+
+## Requirements
+
+- Python 3.10+
+- Linux (any flavor) or WSL
+- Terminal with 256-color support
+- AttackLM installed (`pip install attacklm[all]`)
+
+No X11, no GPU, no browser required. Works over SSH.

attacklm-0.8.0/attacklm-gui/pyproject.toml

@@ -0,0 +1,19 @@
+[project]
+name = "attacklm-gui"
+version = "0.1.0"
+description = "Terminal GUI wrapper for AttackLM CLI training tools"
+requires-python = ">=3.10"
+dependencies = [
+    "textual>=2.0",
+    "attacklm",
+]
+
+[project.scripts]
+attacklm-gui = "attacklm_gui.cli:main"
+
+[build-system]
+requires = ["setuptools>=75"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+where = ["src"]

attacklm-0.8.0/attacklm-gui/src/attacklm_gui/__init__.py

@@ -0,0 +1,3 @@
+"""AttackLM GUI — Terminal-based wrapper for the AttackLM CLI."""
+
+__version__ = "0.1.0"

attacklm-0.8.0/attacklm-gui/src/attacklm_gui/app.py

@@ -0,0 +1,58 @@
+"""Main Textual application for AttackLM GUI."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from textual.app import App, ComposeResult
+from textual.binding import Binding
+from textual.widgets import Header, Footer
+
+from attacklm_gui.presets import ensure_builtin_presets
+from attacklm_gui.screens.main_menu import MainMenuScreen
+
+
+class AttackLMApp(App):
+    """Terminal GUI wrapper for AttackLM CLI tools."""
+
+    TITLE = "AttackLM GUI"
+    SUB_TITLE = "v0.1.0"
+    CSS_PATH = None  # We'll use inline CSS for now
+
+    BINDINGS = [
+        Binding("q", "quit", "Quit", show=True),
+        Binding("m", "main_menu", "Main Menu", show=True),
+        Binding("?", "help", "Help", show=False),
+    ]
+
+    def __init__(self) -> None:
+        super().__init__()
+        self.attacklm_dir = self._find_attacklm_dir()
+
+    def _find_attacklm_dir(self) -> Path:
+        """Find the AttackLM project directory."""
+        # Try common locations
+        candidates = [
+            Path.cwd(),
+            Path.home() / "Projects" / "reverse_engineering" / "AttackLM",
+            Path.home() / "AttackLM",
+        ]
+        for c in candidates:
+            if (c / "pyproject.toml").exists() and (
+                c / "scripts" / "train_template.py"
+            ).exists():
+                return c
+        return Path.cwd()
+
+    def compose(self) -> ComposeResult:
+        yield Header()
+        yield Footer()
+
+    def on_mount(self) -> None:
+        ensure_builtin_presets()
+        self.push_screen(MainMenuScreen())
+
+    def action_main_menu(self) -> None:
+        """Return to main menu."""
+        while len(self.screen_stack) > 1:
+            self.pop_screen()

attacklm-0.8.0/attacklm-gui/src/attacklm_gui/cli.py

@@ -0,0 +1,17 @@
+"""CLI entry point for attacklm-gui."""
+
+import sys
+from pathlib import Path
+
+
+def main() -> int:
+    """Launch the AttackLM TUI."""
+    from attacklm_gui.app import AttackLMApp
+
+    app = AttackLMApp()
+    app.run()
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())