视频处理基础之gradio框架实现

这些函数是用于处理视频文件的Python代码片段，它们依赖于`ffmpeg`和`ffprobe`工具，这些工具是`FFmpeg`项目的一部分，用于处理视频和音频数据。下面是每个函数的用途和用法的总结：

1. `ffmpeg_installed()` 函数：
   - 用途：检查系统是否安装了`ffmpeg`工具。
   - 用法：调用此函数无需参数，它会返回一个布尔值，指示`ffmpeg`是否可用。

2. `video_is_playable(video_filepath: str)` 函数：
   - 用途：确定视频文件是否可以在浏览器中播放。
   - 用法：传入视频文件的路径作为字符串，函数会检查视频的容器格式和编解码器是否符合浏览器的播放标准（如`.mp4`容器使用`h264`编解码器，`.webm`容器使用`vp9`编解码器等）。如果视频符合标准，则返回`True`，否则返回`False`。

3. `convert_video_to_playable_mp4(video_path: str)` 函数：
   - 用途：将视频转换为浏览器可以播放的MP4格式。
   - 用法：传入视频文件的路径作为字符串，函数会尝试将视频转换为MP4格式。如果转换过程中出现错误，则返回原始视频路径。

4. `get_video_length(video_path: str | Path)` 函数：
   - 用途：获取视频文件的时长。
   - 用法：传入视频文件的路径，函数会使用`ffprobe`工具来获取视频的时长，并以秒为单位返回一个浮点数。

这些函数的实现依赖于外部命令行工具`ffmpeg`和`ffprobe`，因此在使用这些函数之前，需要确保这些工具已经正确安装在系统上。此外，代码中还包含了一些错误处理机制，以确保在处理视频文件时能够优雅地处理异常情况。例如，如果`ffmpeg`或`ffprobe`命令失败，或者视频文件的格式不符合预期，函数会返回原始视频路径或抛出异常。
IS_WASM = sys.platform == "emscripten"

def ffmpeg_installed() -> bool:
    if wasm_utils.IS_WASM:
        # TODO: Support ffmpeg in WASM
        return False

    return shutil.which("ffmpeg") is not None

def video_is_playable(video_filepath: str) -> bool:
    """Determines if a video is playable in the browser.

    A video is playable if it has a playable container and codec.
        .mp4 -> h264
        .webm -> vp9
        .ogg -> theora
    """
    from ffmpy import FFprobe, FFRuntimeError

    try:
        container = Path(video_filepath).suffix.lower()
        probe = FFprobe(
            global_options="-show_format -show_streams -select_streams v -print_format json",
            inputs={video_filepath: None},
        )
        output = probe.run(stderr=subprocess.PIPE, stdout=subprocess.PIPE)
        output = json.loads(output[0])
        video_codec = output["streams"][0]["codec_name"]
        return (container, video_codec) in [
            (".mp4", "h264"),
            (".ogg", "theora"),
            (".webm", "vp9"),
        ]
    # If anything goes wrong, assume the video can be played to not convert downstream
    except (FFRuntimeError, IndexError, KeyError):
        return True

def convert_video_to_playable_mp4(video_path: str) -> str:
    """Convert the video to mp4. If something goes wrong return the original video."""
    from ffmpy import FFmpeg, FFRuntimeError

    try:
        with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
            output_path = Path(video_path).with_suffix(".mp4")
            shutil.copy2(video_path, tmp_file.name)
            # ffmpeg will automatically use h264 codec (playable in browser) when converting to mp4
            ff = FFmpeg(
                inputs={str(tmp_file.name): None},
                outputs={str(output_path): None},
                global_options="-y -loglevel quiet",
            )
            ff.run()
    except FFRuntimeError as e:
        print(f"Error converting video to browser-playable format {str(e)}")
        output_path = video_path
    finally:
        # Remove temp file
        os.remove(tmp_file.name)  # type: ignore
    return str(output_path)

def get_video_length(video_path: str | Path):
    if wasm_utils.IS_WASM:
        raise wasm_utils.WasmUnsupportedError(
            "Video duration is not supported in the Wasm mode."
        )
    duration = subprocess.check_output(
        [
            "ffprobe",
            "-i",
            str(video_path),
            "-show_entries",
            "format=duration",
            "-v",
            "quiet",
            "-of",
            "csv={}".format("p=0"),
        ]
    )
    duration_str = duration.decode("utf-8").strip()
    duration_float = float(duration_str)

    return duration_float