@oh-story/dsh 0.1.4 → 0.1.6

package/lib/video-recap/skills/video-cut/scripts/cut_cli.py

@@ -0,0 +1,320 @@
+"""Command-line orchestration for the video-cut skill."""
+
+import json
+
+
+from pathlib import Path
+
+from lib import CONFIG, get_video_duration, log
+
+from cut_contract import (
+    _write_edited_source_meta,
+    load_clip_plan,
+    normalize_clip_plan,
+    normalize_multi_source_clip_plan,
+    parse_duration_seconds,
+    should_reuse_edited_source,
+    value_fingerprint,
+)
+from cut_render import (
+    build_edited_source_video,
+    update_delivery_qc,
+    write_cut_delivery_qc,
+)
+from media_geometry import _select_output_geometry
+from narration_mapping import (
+    lint_mapped_narration,
+    map_narration_to_clips,
+    update_cut_qc,
+)
+from sentence_boundaries import (
+    _combine_boundary_windows,
+    _load_sentence_boundary_windows,
+    _load_silence_for_source,
+    _load_source_speech_spans,
+    enforce_clip_sentence_boundaries,
+    snap_clip_ends_to_lines,
+    snap_clip_starts_to_lines,
+    snap_clips_off_shot_changes,
+    snap_multi_source_clips,
+)
+
+
+def main():
+    import argparse
+
+    parser = argparse.ArgumentParser(
+        description="video-cut: build an edited source video from an agent clip plan and map narration onto the cut timeline."
+    )
+    parser.add_argument("video", help="source video path")
+    parser.add_argument(
+        "--work-dir",
+        required=True,
+        help="dir holding clip_plan.json (and optionally narration.json)",
+    )
+    parser.add_argument(
+        "--clip-plan",
+        default=None,
+        help="clip plan json (default: <work-dir>/clip_plan.json)",
+    )
+    parser.add_argument(
+        "--sources-manifest",
+        default=None,
+        help="multi-source manifest json mapping source_id values to source media",
+    )
+    parser.add_argument(
+        "--narration",
+        default=None,
+        help="narration json to map (default: <work-dir>/narration.json)",
+    )
+    parser.add_argument(
+        "--target-duration",
+        default=None,
+        help="target output duration, e.g. 10m / 600 / 00:10:00",
+    )
+    parser.add_argument(
+        "--clip-padding",
+        type=float,
+        default=None,
+        help="seconds to pad each clip on both ends (default: CLIP_PADDING env, else 0)",
+    )
+    parser.add_argument(
+        "--allow-overlap",
+        action="store_true",
+        help="allow overlapping/duplicate source ranges",
+    )
+    parser.add_argument(
+        "--normalize-only",
+        action="store_true",
+        help="only normalize the clip plan -> clip_plan_validated.json (no render/map); "
+        "lets validate lint the SAME padded/pruned plan the mapper uses",
+    )
+    parser.add_argument(
+        "--no-narration-map",
+        action="store_true",
+        help="render edited_source.mp4 but do NOT map narration.json onto the cut "
+        "(cut-first/narrate-second: narration is authored in OUTPUT time, no mapping)",
+    )
+    parser.add_argument(
+        "--allow-sparse-cut",
+        action="store_true",
+        help="do not block on heavy narration drop / sparse output (e.g. an intentional montage)",
+    )
+    parser.add_argument(
+        "--allow-duration-drift",
+        action="store_true",
+        help="do not block when validated clip duration is far from --target-duration",
+    )
+    args = parser.parse_args()
+
+    # CLIP_PADDING is declared in every skill's CONFIG, but video-cut is the only place that
+    # implements padding — and it used to read the CLI flag alone, so setting the env var did
+    # nothing at all while `clip_padding_source: "env"` reported otherwise. CLI still wins.
+    clip_padding = (
+        args.clip_padding
+        if args.clip_padding is not None
+        else float(CONFIG.get("clip_padding", 0.0) or 0.0)
+    )
+
+    work_dir = Path(args.work_dir)
+    work_dir.mkdir(parents=True, exist_ok=True)
+    clip_plan_path = (
+        Path(args.clip_plan) if args.clip_plan else work_dir / "clip_plan.json"
+    )
+    raw_plan = load_clip_plan(clip_plan_path)
+
+    target_seconds = (
+        parse_duration_seconds(args.target_duration) if args.target_duration else None
+    )
+    sources_manifest = (
+        json.loads(Path(args.sources_manifest).read_text(encoding="utf-8"))
+        if args.sources_manifest
+        else None
+    )
+    if sources_manifest is not None:
+        validated_plan = normalize_multi_source_clip_plan(
+            raw_plan,
+            sources_manifest,
+            target_duration=target_seconds,
+            clip_padding=clip_padding,
+            allow_overlap=args.allow_overlap,
+        )
+        video_duration = None
+    else:
+        video_duration = get_video_duration(args.video)
+        validated_plan = normalize_clip_plan(
+            raw_plan,
+            video_duration,
+            target_duration=target_seconds,
+            clip_padding=clip_padding,
+            allow_overlap=args.allow_overlap,
+        )
+
+    # Keep boundaries off the original footage's hard cuts (avoids 闪烁 at the edit point).
+    # This visual-only pass runs FIRST. The sentence/quiet pass below is the final authority:
+    # a prettier edit point must never move the final boundary back inside a spoken sentence.
+    if sources_manifest is None and CONFIG.get("scene_cut_snap", True):
+        validated_plan = snap_clips_off_shot_changes(
+            validated_plan,
+            args.video,
+            video_duration=video_duration,
+            margin=CONFIG.get("scene_cut_snap_margin", 0.5),
+            threshold=CONFIG.get("scene_cut_detect_threshold", 0.4),
+        )
+
+    if sources_manifest is None:
+        silence_periods = _load_silence_for_source(work_dir, None)
+        sentence_boundaries = _load_sentence_boundary_windows(work_dir)
+        safe_boundaries = _combine_boundary_windows(
+            silence_periods, sentence_boundaries
+        )
+        if CONFIG.get("snap_clip_line_end", True):
+            validated_plan = snap_clip_starts_to_lines(
+                validated_plan,
+                safe_boundaries,
+                video_duration,
+                CONFIG.get("clip_start_snap_max_prepend", 1.8),
+                max_trim=CONFIG.get("clip_start_snap_max_trim", 0.35),
+            )
+            validated_plan = snap_clip_ends_to_lines(
+                validated_plan,
+                safe_boundaries,
+                video_duration,
+                CONFIG.get("clip_snap_max_extend", 2.0),
+            )
+        validated_plan = enforce_clip_sentence_boundaries(
+            validated_plan,
+            safe_boundaries,
+            _load_source_speech_spans(work_dir),
+            video_duration,
+        )
+
+    # Multi-source: snap each clip against ITS OWN source's pauses/shot-changes (single-source
+    # snaps above can't, since silence_periods.json and args.video are per-project, not per-source).
+    if sources_manifest is not None:
+        validated_plan = snap_multi_source_clips(
+            validated_plan,
+            validated_plan.get("sources", {}),
+            work_dir,
+            line_max_extend=CONFIG.get("clip_snap_max_extend", 2.0),
+            scene_margin=CONFIG.get("scene_cut_snap_margin", 0.5),
+            scene_threshold=CONFIG.get("scene_cut_detect_threshold", 0.4),
+            do_line_snap=CONFIG.get("snap_clip_line_end", True),
+            do_scene_snap=CONFIG.get("scene_cut_snap", True),
+            start_max_prepend=CONFIG.get("clip_start_snap_max_prepend", 1.8),
+            start_max_trim=CONFIG.get("clip_start_snap_max_trim", 0.35),
+        )
+
+    if isinstance(validated_plan, dict):
+        validated_plan["raw_plan_fingerprint"] = value_fingerprint(raw_plan)
+        validated_plan.setdefault("qc", {})["join_fade_ms"] = round(
+            max(0.0, float(CONFIG.get("clip_join_audio_fade_ms", 30.0) or 0.0)), 3
+        )
+        source_paths = []
+        for clip in validated_plan.get("clips", []):
+            source_path = str(clip.get("source_path") or "")
+            if source_path and source_path not in source_paths:
+                source_paths.append(source_path)
+        if not source_paths:
+            source_paths = [str(args.video)]
+        _, _, _, geometry_qc = _select_output_geometry(
+            source_paths, validated_plan.get("clips", [])
+        )
+        validated_plan["qc"]["output_geometry"] = geometry_qc
+        validated_plan["qc"]["output_geometry_reason"] = geometry_qc.get("reason")
+        allow_duration_drift = bool(args.allow_duration_drift or args.allow_sparse_cut)
+        drift_source = (
+            "--allow-duration-drift"
+            if args.allow_duration_drift
+            else ("--allow-sparse-cut" if args.allow_sparse_cut else None)
+        )
+        update_cut_qc(
+            validated_plan,
+            allow_duration_drift=allow_duration_drift,
+            duration_drift_allowed_by=drift_source,
+        )
+        update_delivery_qc(
+            validated_plan,
+            source_paths=source_paths,
+            output_path=work_dir / "edited_source.mp4",
+        )
+    (work_dir / "clip_plan_validated.json").write_text(
+        json.dumps(validated_plan, ensure_ascii=False, indent=2), encoding="utf-8"
+    )
+    if (validated_plan.get("qc") or {}).get("blocking"):
+        raise SystemExit(
+            "clip_plan QC blocking: fix unsafe sentence boundaries or target-duration drift. "
+            "Only duration drift can be explicitly accepted with --allow-duration-drift; "
+            "sentence truncation is never allowed. See clip_plan_validated.json['qc']."
+        )
+    if args.normalize_only:
+        # normalize-only produces planned delivery facts in clip_plan_validated.json, but no
+        # rendered/reused media exists in this run, so remove any stale final delivery artifact.
+        (work_dir / "cut_delivery_qc.json").unlink(missing_ok=True)
+        print(
+            json.dumps(
+                {
+                    "status": "normalized",
+                    "clips": len(validated_plan["clips"]),
+                    "total_duration": validated_plan["total_duration"],
+                },
+                ensure_ascii=False,
+            )
+        )
+        return
+
+    edited_source_path = work_dir / "edited_source.mp4"
+    if should_reuse_edited_source(edited_source_path, validated_plan, args.video):
+        log(f"复用剪辑源视频: {edited_source_path}")
+        update_delivery_qc(
+            validated_plan,
+            source_paths=source_paths,
+            output_path=edited_source_path,
+            rendered=True,
+        )
+        write_cut_delivery_qc(work_dir, validated_plan)
+        _write_edited_source_meta(edited_source_path, validated_plan, args.video)
+        (work_dir / "clip_plan_validated.json").write_text(
+            json.dumps(validated_plan, ensure_ascii=False, indent=2), encoding="utf-8"
+        )
+    else:
+        build_edited_source_video(
+            args.video, validated_plan, work_dir, edited_source_path
+        )
+        (work_dir / "clip_plan_validated.json").write_text(
+            json.dumps(validated_plan, ensure_ascii=False, indent=2), encoding="utf-8"
+        )
+
+    narration_path = (
+        Path(args.narration) if args.narration else work_dir / "narration.json"
+    )
+    if narration_path.exists() and not args.no_narration_map:
+        narration = json.loads(narration_path.read_text(encoding="utf-8"))
+        mapped = map_narration_to_clips(narration, validated_plan)
+        if not mapped:
+            raise SystemExit("narration 没有落入 clip_plan 片段内的有效解说")
+        (work_dir / "narration_mapped.json").write_text(
+            json.dumps(mapped, ensure_ascii=False, indent=2), encoding="utf-8"
+        )
+        log(f"映射解说 {len(mapped)} 段 → narration_mapped.json")
+        report = lint_mapped_narration(
+            mapped, len(narration), validated_plan["total_duration"]
+        )
+        (work_dir / "narration_mapped_lint.json").write_text(
+            json.dumps(report, ensure_ascii=False, indent=2), encoding="utf-8"
+        )
+        for w in report["warnings"]:
+            log(f"  ⚠️ 剪后解说同步: {w['message']} [{w['code']}]")
+        if report.get("clamped_count"):
+            raise SystemExit(
+                "剪后解说有句子被 clip 边界裁断。移动 clip_plan 边界或重写整句后重跑；"
+                "--allow-sparse-cut 不能跳过语句完整性门禁。详见 narration_mapped_lint.json"
+            )
+        if report.get("blocking") and not args.allow_sparse_cut:
+            raise SystemExit(
+                "剪后解说与保留片段对不上：丢弃过多或成片过稀疏。改 narration.json / clip_plan.json "
+                "让解说落在保留片段内后重跑，或加 --allow-sparse-cut 接受当前映射。详见 narration_mapped_lint.json"
+            )
+    log(
+        f"剪辑模式: {len(validated_plan['clips'])} 个片段 → {validated_plan['total_duration']:.1f}s"
+    )