add cli 0.1.6

Author: frontboat

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "veotools"
-version = "0.1.5"
+version = "0.1.6"
 description = "A Python toolkit for AI-powered video generation and seamless stitching using Google's Veo models."
 readme = "readme.md"
 requires-python = ">=3.10"
@@ -52,3 +52,6 @@ allow-direct-references = true
 [project.optional-dependencies]
 mcp = ["mcp[cli]>=1.0.0"]
 
+[project.scripts]
+veo = "veotools.cli:main"
+

readme.md

@@ -61,6 +61,26 @@ final = veo.stitch_videos(
 )
 ```
 
+## CLI
+
+Install exposes the `veo` command. Use `-h/--help` on any subcommand.
+
+```bash
+# Basics
+veo preflight
+veo list-models --remote
+
+# Generate from text
+veo generate --prompt "cat riding a hat" --model veo-3.0-fast-generate-preview
+
+# Continue a video and stitch seamlessly
+veo continue --video dog.mp4 --prompt "the dog finds a treasure chest" --overlap 1.0
+
+# Help
+veo --help
+veo generate --help
+```
+
 ### Create a Story with Bridge
 
 ```python

veotools/__init__.py

@@ -35,7 +35,7 @@
     generate_cancel,
 )
 
-__version__ = "0.1.5"
+__version__ = "0.1.6"
 
 __all__ = [
     "VeoClient",

veotools/cli.py

@@ -0,0 +1,144 @@
+"""Veotools command-line interface (no extra deps).
+
+Usage examples:
+  veo preflight
+  veo list-models --remote
+  veo generate --prompt "cat riding a hat" --model veo-3.0-fast-generate-preview
+  veo continue --video dog.mp4 --prompt "the dog finds a treasure chest" --overlap 1.0
+"""
+
+from __future__ import annotations
+
+import argparse
+import json
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+import veotools as veo
+
+
+def _print_progress(message: str, percent: int):
+    bar_length = 24
+    filled = int(bar_length * percent / 100)
+    bar = "#" * filled + "-" * (bar_length - filled)
+    print(f"[{bar}] {percent:3d}% {message}", end="\r")
+    if percent >= 100:
+        print()
+
+
+def cmd_preflight(_: argparse.Namespace) -> int:
+    veo.init()
+    data = veo.preflight()
+    print(json.dumps(data, indent=2))
+    return 0
+
+
+def cmd_list_models(ns: argparse.Namespace) -> int:
+    veo.init()
+    data = veo.list_models(include_remote=ns.remote)
+    if ns.json:
+        print(json.dumps(data, indent=2))
+    else:
+        for m in data.get("models", []):
+            print(m.get("id"))
+    return 0
+
+
+def cmd_generate(ns: argparse.Namespace) -> int:
+    veo.init()
+    kwargs: Dict[str, Any] = {}
+    if ns.model:
+        kwargs["model"] = ns.model
+    if ns.image:
+        result = veo.generate_from_image(
+            image_path=Path(ns.image),
+            prompt=ns.prompt,
+            on_progress=_print_progress,
+        )
+    elif ns.video:
+        result = veo.generate_from_video(
+            video_path=Path(ns.video),
+            prompt=ns.prompt,
+            extract_at=ns.extract_at,
+            on_progress=_print_progress,
+            **kwargs,
+        )
+    else:
+        result = veo.generate_from_text(
+            ns.prompt,
+            on_progress=_print_progress,
+            **kwargs,
+        )
+    if ns.json:
+        print(json.dumps(result.to_dict(), indent=2))
+    else:
+        print(result.path)
+    return 0
+
+
+def cmd_continue(ns: argparse.Namespace) -> int:
+    veo.init()
+    # Generate continuation
+    gen = veo.generate_from_video(
+        video_path=Path(ns.video),
+        prompt=ns.prompt,
+        extract_at=ns.extract_at,
+        model=ns.model,
+        on_progress=_print_progress,
+    )
+    # Stitch with original
+    stitched = veo.stitch_videos([Path(ns.video), Path(gen.path)], overlap=ns.overlap)
+    if ns.json:
+        out = {
+            "generated": gen.to_dict(),
+            "stitched": stitched.to_dict(),
+        }
+        print(json.dumps(out, indent=2))
+    else:
+        print(stitched.path)
+    return 0
+
+
+def build_parser() -> argparse.ArgumentParser:
+    p = argparse.ArgumentParser(prog="veo", description="Veotools CLI")
+    sub = p.add_subparsers(dest="cmd", required=True)
+
+    s = sub.add_parser("preflight", help="Check environment and system prerequisites")
+    s.set_defaults(func=cmd_preflight)
+
+    s = sub.add_parser("list-models", help="List available models")
+    s.add_argument("--remote", action="store_true", help="Include remote discovery")
+    s.add_argument("--json", action="store_true", help="Output JSON")
+    s.set_defaults(func=cmd_list_models)
+
+    s = sub.add_parser("generate", help="Generate a video from text/image/video")
+    s.add_argument("--prompt", required=True)
+    s.add_argument("--model", help="Model ID (e.g., veo-3.0-fast-generate-preview)")
+    s.add_argument("--image", help="Path to input image")
+    s.add_argument("--video", help="Path to input video")
+    s.add_argument("--extract-at", type=float, default=-1.0, help="Time offset for video continuation")
+    s.add_argument("--json", action="store_true", help="Output JSON")
+    s.set_defaults(func=cmd_generate)
+
+    s = sub.add_parser("continue", help="Continue a video and stitch seamlessly")
+    s.add_argument("--video", required=True, help="Source video path")
+    s.add_argument("--prompt", required=True)
+    s.add_argument("--model", help="Model ID")
+    s.add_argument("--extract-at", type=float, default=-1.0)
+    s.add_argument("--overlap", type=float, default=1.0)
+    s.add_argument("--json", action="store_true")
+    s.set_defaults(func=cmd_continue)
+
+    return p
+
+
+def main(argv: Optional[list[str]] = None) -> int:
+    parser = build_parser()
+    ns = parser.parse_args(argv)
+    return ns.func(ns)
+
+
+if __name__ == "__main__":  # pragma: no cover
+    raise SystemExit(main())
+
+

veotools/core.py

@@ -34,8 +34,8 @@ def __init__(self, base_path: Optional[str] = None):
 
         Default resolution order for base path:
         1. VEO_OUTPUT_DIR environment variable (if set)
-        2. Project root alongside the package (../output)
-        3. Current working directory (./output) as a safe fallback
+        2. Current working directory (./output)
+        3. Package-adjacent directory (../output) as a last resort
         """
         resolved_base: Path
 
@@ -46,15 +46,21 @@ def __init__(self, base_path: Optional[str] = None):
         elif env_base:
             resolved_base = Path(env_base)
         else:
-            # 2) Attempt to place under project root (parent of the package dir)
+            # 2) Prefer CWD/output for installed packages (CLI/scripts)
+            cwd_candidate = Path.cwd() / "output"
             try:
-                package_root = Path(__file__).resolve().parents[2]
-                candidate = package_root / "output"
-                candidate.mkdir(parents=True, exist_ok=True)
-                resolved_base = candidate
+                cwd_candidate.mkdir(parents=True, exist_ok=True)
+                resolved_base = cwd_candidate
             except Exception:
-                # 3) Fallback to CWD/output
-                resolved_base = Path.cwd() / "output"
+                # 3) As a last resort, place beside the installed package
+                try:
+                    package_root = Path(__file__).resolve().parents[2]
+                    candidate = package_root / "output"
+                    candidate.mkdir(parents=True, exist_ok=True)
+                    resolved_base = candidate
+                except Exception:
+                    # Final fallback: user home
+                    resolved_base = Path.home() / "output"
 
         self.base_path = resolved_base
         self.base_path.mkdir(parents=True, exist_ok=True)