基于 OpenAI 接口生成 SRT 字幕,并通过 FFmpeg 将字幕封装到视频中。
建议先创建并激活一个 conda 环境,再安装运行所需依赖。
conda create -n livecap python=3.12 -y
conda activate livecap
conda install -c conda-forge ffmpeg -y安装完成后可执行以下命令确认 FFmpeg 可用:
ffmpeg -versionpip install openai如果你需要运行仓库中的测试,也建议在同一环境中执行以上安装步骤。
from agent import SubtitleAgent
agent = SubtitleAgent(api_key="YOUR_OPENAI_API_KEY")
output_path = agent.run_once(
["第一句文案", "第二句文案"],
in_video="src.mp4",
out_video="sub_out.mp4",
srt_path="sub.srt",
)
print(output_path)python -m unittest -v测试依赖:
- 已安装
ffmpeg - 已安装
openai