lxbfYeaa/ContentGeneration
Template
1
0
Fork 0
forked from LiveCarta/ContentGeneration
Code Pull Requests Activity

Add skip-audio mode and resilient merge handling

Browse Source
This commit is contained in:
Artsiom Siamashka
2026-04-02 13:25:54 +02:00
parent a0a66264d2
commit 08ebab6348
3 changed files with 56 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
run_video_pipeline.py
View File

@@ -40,6 +40,7 @@ def parse_args() -> argparse.Namespace:
parser.add_argument("--output", type=Path, default=DEFAULT_OUTPUT)
parser.add_argument("--seed", type=int, default=1)
parser.add_argument("--skip-generate", action="store_true")
parser.add_argument("--skip-audio-generate", action="store_true")
parser.add_argument("--skip-merge", action="store_true")
parser.add_argument("--skip-concat", action="store_true")
parser.add_argument("--skip-s3-upload", action="store_true")
@@ -123,6 +124,16 @@ def main() -> int:
LOGGER.error("Reel script was not generated at %s", args.reel_script)
return 1
if not args.skip_generate and not args.skip_audio_generate:
run_step(
"Generate Audios",
_with_log_level([
sys.executable,
str(SCRIPT_DIR / "generate_audios.py"),
], args.log_level),
cwd=args.base_dir,
)
if not args.skip_generate:
run_step(
"Generate Videos",
@@ -145,18 +156,22 @@ def main() -> int:
)
if not args.skip_merge:
merge_cmd = [
sys.executable,
str(SCRIPT_DIR / "merge_audio_video.py"),
"--videos-dir",
str(args.videos_dir),
"--audios-dir",
str(args.audios_dir),
"--output-dir",
str(args.merged_dir),
]
if args.skip_audio_generate:
merge_cmd.append("--allow-missing-audio")
run_step(
"Merge Audio + Video",
_with_log_level([
sys.executable,
str(SCRIPT_DIR / "merge_audio_video.py"),
"--videos-dir",
str(args.videos_dir),
"--audios-dir",
str(args.audios_dir),
"--output-dir",
str(args.merged_dir),
], args.log_level),
_with_log_level(merge_cmd, args.log_level),
)
if not args.skip_concat: