vibe it baby

Author: johnnygreco

examples/cli_back_demo.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python
+"""
+Demonstration of the 'back' functionality in Data Designer CLI.
+
+This example shows how to use the BACK sentinel value to implement
+navigation between prompts in a wizard-style interface.
+"""
+
+from data_designer.cli.interactive import BACK, prompt_text_input, select_with_arrows
+
+
+def simple_wizard_with_back() -> None:
+    """Demonstrate a simple wizard that supports going back."""
+    print("=== Simple Wizard with Back Support ===\n")
+
+    # Step 1: Get name
+    step = 1
+    name = None
+    age = None
+    color = None
+
+    while True:
+        if step == 1:
+            # First prompt - no back option
+            name = prompt_text_input("What is your name?", default="Alice", allow_back=False)
+            if name is None:  # Cancelled
+                print("Wizard cancelled!")
+                return
+            step = 2
+
+        elif step == 2:
+            # Second prompt - can go back to step 1
+            age = prompt_text_input("What is your age?", default="25", allow_back=True)
+            if age is None:  # Cancelled
+                print("Wizard cancelled!")
+                return
+            elif age is BACK:  # Go back
+                step = 1
+                continue
+            step = 3
+
+        elif step == 3:
+            # Third prompt - can go back to step 2
+            color_options = {
+                "red": "Red - Bold and vibrant",
+                "blue": "Blue - Calm and serene",
+                "green": "Green - Fresh and natural",
+            }
+            color = select_with_arrows(color_options, "Select your favorite color", allow_back=True)
+            if color is None:  # Cancelled
+                print("Wizard cancelled!")
+                return
+            elif color is BACK:  # Go back
+                step = 2
+                continue
+            step = 4
+
+        elif step == 4:
+            # Final step - show results
+            print("\n=== Results ===")
+            print(f"Name: {name}")
+            print(f"Age: {age}")
+            print(f"Favorite Color: {color}")
+            return
+
+
+if __name__ == "__main__":
+    simple_wizard_with_back()

pyproject.toml

@@ -29,8 +29,10 @@ dependencies = [
   "pygments>=2.19.2",
   "pyyaml>=6.0.1",
   "python-json-logger==2.0.7",
+  "prompt-toolkit>=3.0.0",
   "requests<3,>=2.32.2",
   "rich>=13.7.1",
+  "typer>=0.12.0",
   "anyascii>=0.3.3,<1.0",
   "boto3==1.35.74",
   "datasets>=4.0.0",
@@ -52,6 +54,9 @@ dependencies = [
   "ruff==0.12.3",
 ]
 
+[project.scripts]
+data-designer = "data_designer.cli:main"
+
 [dependency-groups]
 dev = [
   "jsonpath-ng==1.5.3",

src/data_designer/cli/__init__.py

@@ -0,0 +1,40 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+import typer
+
+# Initialize Typer app with custom configuration
+app = typer.Typer(
+    name="data-designer",
+    help="Data Designer CLI - Configure model providers and models for synthetic data generation",
+    add_completion=False,
+    no_args_is_help=True,
+    rich_markup_mode="rich",
+)
+
+# Import and register command groups
+# We import here to avoid circular dependencies
+from data_designer.cli.commands import list as list_cmd
+from data_designer.cli.commands import models, providers, validate
+
+# Create config subcommand group
+config_app = typer.Typer(
+    name="config",
+    help="Manage configuration files",
+    no_args_is_help=True,
+)
+config_app.command(name="providers", help="Configure model providers interactively")(providers.providers_command)
+config_app.command(name="models", help="Configure models interactively")(models.models_command)
+config_app.command(name="list", help="List current configurations")(list_cmd.list_command)
+config_app.command(name="validate", help="Validate configuration files")(validate.validate_command)
+
+app.add_typer(config_app, name="config")
+
+
+def main() -> None:
+    """Main entry point for the CLI."""
+    app()
+
+
+if __name__ == "__main__":
+    main()

src/data_designer/cli/commands/__init__.py

@@ -0,0 +1,2 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0