使用Python和Whisper库生成字幕

在Python中，我们可以使用开源的Whisper库来处理字幕文件。Whisper库支持多种字幕格式，包括SRT、ASS和SSA等。下面是一个简单的示例，演示如何使用Python和Whisper库生成SRT字幕文件。
首先，确保你已经安装了Python和Whisper库。你可以使用pip命令来安装Whisper库：

pip install whisper

接下来，创建一个Python脚本，例如generate_srt_captions.py，并在脚本中编写以下代码：

import whisper
# 输入视频文件路径和输出字幕文件路径
input_video = 'input.mp4'
output_srt = 'output.srt'
# 加载视频文件并提取音频轨道
video = whisper.Video(input_video)
audio = video.audio
# 创建字幕对象并添加时间戳和文本
captions = []
start_time = 0
end_time = audio.length
caption_text = 'Hello, world!'
captions.append((start_time, end_time, caption_text))
# 保存字幕文件
with open(output_srt, 'w', encoding='utf-8') as file:
for caption in captions:
start, end, text = caption
file.write(f'{start:.2f}
')  # 时间戳格式为00:00:00.00
file.write(f'{end:.2f}
')  # 时间戳格式为00:00:00.00
file.write(f'{text}
')  # 字幕文本，每个字幕占一行

这个脚本会从视频中提取音频轨道，并创建一个字幕对象。然后，我们将一个时间戳、结束时间和一个文本添加到字幕对象中。最后，我们将字幕保存为SRT文件。请注意，这个示例只是为了演示如何生成字幕文件的基本概念，实际上你可以根据自己的需求调整脚本中的代码。
现在，你可以运行这个脚本来生成字幕文件。打开终端或命令提示符，导航到脚本所在的目录，并运行以下命令：

python generate_srt_captions.py

这将生成一个名为output.srt的字幕文件。你可以将这个字幕文件与视频一起播放，以便在视频中添加字幕。请注意，这只是一个简单的示例，你可能需要根据你的实际需求进行调整和扩展。