File size: 1,609 Bytes
ed6bec6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
# generator/cli.py
import argparse
from pathlib import Path
from .run_generator import main as generate_main
from .validator import validate_jsonl
from .batch_generator import generate_batches
def cli():
parser = argparse.ArgumentParser(prog="glyphicgen", description="Glyphic dataset generator and tools")
subparsers = parser.add_subparsers(dest="command", required=True)
# generate
gen_parser = subparsers.add_parser("generate", help="Generate a Glyphic training dataset")
gen_parser.add_argument("--output", type=Path, help="Output JSONL path (overrides config)")
# validate
val_parser = subparsers.add_parser("validate", help="Validate a Glyphic JSONL dataset")
val_parser.add_argument("path", type=Path, help="Path to JSONL dataset")
# batch
batch_parser = subparsers.add_parser("batch", help="Generate multiple datasets with varying configs")
batch_parser.add_argument("--count", type=int, default=5, help="Number of datasets to generate")
batch_parser.add_argument("--outdir", type=Path, default=Path("training/batches"), help="Output directory")
args = parser.parse_args()
if args.command == "generate":
# reuse existing generator; optional override
if args.output:
from .config import OUTPUT_JSONL
# monkey-patch for now
globals()["OUTPUT_JSONL"] = args.output
generate_main()
elif args.command == "validate":
validate_jsonl(args.path)
elif args.command == "batch":
generate_batches(args.count, args.outdir)
if __name__ == "__main__":
cli()
|