Function render_manim

functions/function_calling/video.py:14–58

(
    script_path: str,
    quality: str = "h",
    preview: bool = True,
)

    return [node.name for node in ast.walk(tree) if isinstance(node, ast.ClassDef)]

def render_manim(
    script_path: str,
    quality: str = "h",
    preview: bool = True,
) -> Path:
    output_dir = Path.cwd() / "media" 
    print("Clearing media folder:", output_dir)
    shutil.rmtree(output_dir, ignore_errors=True)
    script_path = Path(script_path).resolve()
    if not script_path.exists():
        raise FileNotFoundError(script_path, " does not exist.")
    scene_name = _get_class_names(str(script_path))[0] 
    cmd = [
        sys.executable, "-m", "manim",
        f"-pq{quality}",
    ]

    if preview:
        cmd.insert(3, "-p")

    cmd.extend([str(script_path), scene_name])

    print("Running:", " ".join(cmd))
    try:
        subprocess.run(cmd, check=True, capture_output=True, text=True)
    except subprocess.CalledProcessError as e:
        print("Manim rendering failed:")
        print("stdout:", e.stdout)
        print("stderr:", e.stderr)
        error_info = f"Error rendering {scene_name} from {script_path}: {e.stderr}"
        # shutil.rmtree(output_dir, ignore_errors=True)
        raise RuntimeError(error_info)

    # Find valid mp4 files where no parent directory contains a partial_movie_files folder
    video_file = None
    for mp4_file in (Path.cwd() / "media" / "videos").parent.rglob("*.mp4"):
        if mp4_file.name == f"{scene_name}.mp4":
            video_file = mp4_file
            break

    target_path = script_path.parent / video_file.name
    print(f"Copying video to {target_path}")
    shutil.copy2(video_file, target_path)
    shutil.rmtree(output_dir, ignore_errors=True)
    return target_path

def concat_videos(video_paths: list[Path]) -> Path:
    if not video_paths: