ds-agent-cli 0.1.0 → 0.1.1

package/ds_agent/__init__.py

@@ -1,6 +1,6 @@
 from pathlib import Path
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 # Allow ds_agent.* imports to resolve modules currently stored under src/.
 _src_dir = Path(__file__).resolve().parent.parent / "src"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ds-agent-cli",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "DS-Agent CLI wrapper for npm global install",
   "license": "MIT",
   "publishConfig": {
@@ -25,4 +25,4 @@
     "python",
     "ml"
   ]
-}
+}

package/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_packages, setup
 
 setup(
     name="ds-agent",
-    version="0.1.0",
+    version="0.1.1",
     description="Data Science Agent CLI",
     packages=find_packages(),
     include_package_data=True,

package/src/cli.py

@@ -9,6 +9,8 @@ from rich.panel import Panel
 from rich.progress import Progress, SpinnerColumn, TextColumn, BarColumn
 from rich.live import Live
 from rich import print as rprint
+from rich import box
+from rich.text import Text
 from pathlib import Path
 import json
 import sys
@@ -16,7 +18,7 @@ import os
 import io
 import contextlib
 import importlib.util
-from typing import Optional
+from typing import Optional, Tuple
 
 # Add src to path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__)))
@@ -40,35 +42,85 @@ app.add_typer(sessions_app, name="sessions")
 console = Console()
 
 
+def _hex_to_rgb(hex_color: str) -> Tuple[int, int, int]:
+    value = hex_color.lstrip("#")
+    return int(value[0:2], 16), int(value[2:4], 16), int(value[4:6], 16)
+
+
+def _rgb_to_hex(r: int, g: int, b: int) -> str:
+    return f"#{r:02x}{g:02x}{b:02x}"
+
+
+def _gradient_text(content: str, start_color: str, end_color: str, *, bold: bool = True) -> Text:
+    """Render per-character color interpolation for a more futuristic terminal banner."""
+    start_r, start_g, start_b = _hex_to_rgb(start_color)
+    end_r, end_g, end_b = _hex_to_rgb(end_color)
+
+    visible_count = sum(1 for ch in content if ch != "\n")
+    steps = max(visible_count - 1, 1)
+    current = 0
+
+    out = Text()
+    for ch in content:
+        if ch == "\n":
+            out.append("\n")
+            continue
+
+        t = current / steps
+        r = int(start_r + (end_r - start_r) * t)
+        g = int(start_g + (end_g - start_g) * t)
+        b = int(start_b + (end_b - start_b) * t)
+        weight = "bold " if bold else ""
+        out.append(ch, style=f"{weight}{_rgb_to_hex(r, g, b)}")
+        current += 1
+
+    return out
+
+
 def _print_cli_home():
     """Render a first-run friendly home screen for users running the CLI without arguments."""
-    banner = r"""[bold cyan]
+    banner = r"""
   ____  ____         _    ____ _____ _   _ _____
  |  _ \/ ___|       / \  / ___| ____| \ | |_   _|
  | | | \___ \      / _ \| |  _|  _| |  \| | | |
  | |_| |___) |    / ___ \ |_| | |___| |\  | | |
  |____/|____/    /_/   \_\____|_____|_| \_| |_|
-[/bold cyan]"""
-    console.print(banner)
-    console.print("[dim]The open data science agent CLI[/dim]\n")
-
-    table = Table(show_header=False, box=None, pad_edge=False)
-    table.add_column("Command", style="bold white", width=52, no_wrap=True)
-    table.add_column("What it does", style="white")
-
-    table.add_row("ds-agent quickstart", "Show examples for all major workflows")
-    table.add_row("ds-agent analyze <file>", "Run end-to-end automated analysis")
-    table.add_row("ds-agent chat [file]", "Start interactive session with memory")
-    table.add_row("ds-agent report <file>", "Generate full HTML profiling report")
-    table.add_row("ds-agent train <file> --target <col>", "Train baseline ML models")
-    table.add_row("ds-agent tune <file> --target <col>", "Hyperparameter optimization")
-    table.add_row("ds-agent plot <file>", "Generate charts and visualizations")
-    table.add_row("ds-agent forecast <file> --time <col> --target <col>", "Time-series forecasting")
-    table.add_row("ds-agent sessions list", "List and manage saved sessions")
-    table.add_row("ds-agent --help", "Full command reference")
+""".strip("\n")
+
+    hero_content = Text()
+    hero_content.append_text(_gradient_text(banner, "#ff8d63", "#66f0ff", bold=True))
+    hero_content.append("\n\n")
+    hero_content.append_text(_gradient_text("A stylish, one-command data science command center", "#ffd4be", "#9ef7ff", bold=False))
+
+    hero = Panel(
+        hero_content,
+        border_style="#7bd7ff",
+        box=box.ROUNDED,
+        padding=(1, 2),
+        title="[bold #ff9b72]DS-AGENT[/bold #ff9b72]",
+        subtitle="[bold #7bd7ff]Neon CLI[/bold #7bd7ff]"
+    )
+    console.print(hero)
+
+    quick = Table(box=box.SIMPLE_HEAVY, show_header=True, header_style="bold #7bd7ff")
+    quick.add_column("Command", style="bold white", width=54, no_wrap=True)
+    quick.add_column("Use Case", style="#d0c9bc")
+
+    quick.add_row("ds-agent quickstart", "Copy-paste examples for every core workflow")
+    quick.add_row("ds-agent analyze <file>", "Run complete ML workflow with one command")
+    quick.add_row("ds-agent chat [file]", "Interactive assistant with session memory")
+    quick.add_row("ds-agent report <file>", "Generate beautiful HTML profiling reports")
+    quick.add_row("ds-agent train <file> --target <col>", "Train baseline models quickly")
+    quick.add_row("ds-agent tune <file> --target <col>", "Optimize models with Optuna")
+    quick.add_row("ds-agent plot <file>", "Create plots and visual diagnostics")
+    quick.add_row("ds-agent forecast <file> --time <col> --target <col>", "Time-series forecasting")
+    quick.add_row("ds-agent sessions list", "View, resume, or clean saved sessions")
+    quick.add_row("ds-agent --help", "Full command and options reference")
 
-    console.print(table)
-    console.print("\n[dim]Tip: run [bold]ds-agent quickstart[/bold] for copy-paste commands.[/dim]")
+    console.print()
+    console.print(quick)
+    console.print("\n[bold #7bd7ff]Start here:[/bold #7bd7ff] [white]ds-agent quickstart[/white]")
+    console.print("[dim]Tip: add --help to any command for detailed usage, examples, and flags.[/dim]")
 
 
 @app.callback(invoke_without_command=True)
@@ -93,9 +145,15 @@ def root_callback(
 @app.command()
 def quickstart():
     """Show beginner-friendly command guide with copy-paste examples."""
-    guide = Table(title="⚡ Quickstart Guide")
-    guide.add_column("Workflow", style="cyan", width=24, no_wrap=True)
-    guide.add_column("Command", style="white", no_wrap=True)
+    console.print(Panel.fit(
+        "[bold #7bd7ff]Quickstart Command Deck[/bold #7bd7ff]\n[dim]Run these exactly as shown and replace file paths.[/dim]",
+        border_style="#7bd7ff",
+        box=box.ROUNDED
+    ))
+
+    guide = Table(box=box.SIMPLE_HEAVY, show_header=True, header_style="bold #7bd7ff")
+    guide.add_column("Workflow", style="bold white", width=24, no_wrap=True)
+    guide.add_column("Command", style="#d0c9bc", no_wrap=True)
 
     guide.add_row("Analyze", "ds-agent analyze data.csv --target price --task \"predict price\"")
     guide.add_row("Chat", "ds-agent chat data.csv")
@@ -110,7 +168,8 @@ def quickstart():
 
     console.print()
     console.print(guide)
-    console.print("\n[dim]Use [bold]ds-agent <command> --help[/bold] for detailed options and examples.[/dim]")
+    console.print("\n[bold #7bd7ff]Need details?[/bold #7bd7ff] [white]ds-agent <command> --help[/white]")
+    console.print("[dim]Example: ds-agent report --help | ds-agent forecast --help[/dim]")
 
 
 def _load_module_from_src(module_name: str, relative_path: str):