给视频增加字幕，根据视频宽度和字幕的长度对比，判断是否要换行
单个字幕
import os
from moviepy.editor import VideoFileClip, CompositeVideoClip, ImageClip
from PIL import Image, ImageFont, ImageDraw
import numpy as np
def wrap_text(text, font, max_width):
"""
自动换行函数：根据指定的最大宽度对文本进行折行
:param text: 字幕文本
:param font: PIL字体对象
:param max_width: 每行最大宽度
:return: 换行后的文本列表
"""
lines = []
current_line = ""
for char in text: # 中文按字符处理
test_line = current_line + char
line_width = font.getbbox(test_line)[2] # 使用 getbbox(文本边界) 获取宽度
if line_width <= max_width:
current_line = test_line
else:
lines.append(current_line)
current_line = char
if current_line: # 添加最后一行
lines.append(current_line)
return lines
def generate_subtitle_image(text, font_path, font_size, video_width, video_height, white_space):
"""
生成带有字幕的图像，自动换行
:param text: 字幕文本
:param font_path: 字体路径
:param font_size: 字体大小
:param video_width: 视频宽度
:param video_height: 视频高度
:param white_space: 视频左右留白
:return: 带有字幕的图像（numpy数组）
"""
font = ImageFont.truetype(font_path, font_size)
max_width = video_width - white_space # 留左右边距
# 自动换行
wrapped_text = wrap_text(text, font, max_width)
# 创建字幕图像
image = Image.new("RGBA", (video_width, video_height), (255, 255, 255, 0)) # 透明背景
draw = ImageDraw.Draw(image)
# 绘制字幕：居中显示，每行水平居中，底部显示
line_height = font.getbbox("测试")[3] # 使用 getbbox 获取行高
y_position = video_height - line_height * len(wrapped_text) - 20 # 字幕距底部20像素
for line in wrapped_text:
text_width = font.getbbox(line)[2]
x_position = (video_width - text_width) // 2 # 居中
draw.text((x_position, y_position), line, font=font, fill="white")
y_position += line_height
return np.array(image)
def output(subtitle_image):
# 创建字幕ImageClip
subtitle_clip = ImageClip(subtitle_image, duration=video_clip.duration)
# 合成视频
composite_clip = CompositeVideoClip([video_clip, subtitle_clip])
# 输出视频
composite_clip.write_videofile(output_path, codec="libx264", fps=24)
if __name__ == '__main__':
# 视频路径
video_path = "1127.mp4"
output_path = "output_with_112750长.mp4"
# 加载视频并获取宽高
video_clip = VideoFileClip(video_path)
video_width, video_height = video_clip.w, video_clip.h
print(f"视频宽度: {video_width}px, 视频高度: {video_height}px")
# 字幕文本
subtitle_text = "sdakjslfkdfgdfsfagfasfgsagf"
print(f"字幕原始文本长度: {len(subtitle_text)}")
# 字幕样式
custom_font_path = r"C:\Windows\Fonts\msyh.ttc" # 替换为实际字体路径
if not os.path.exists(custom_font_path):
print(f"字体文件不存在：{custom_font_path}")
else:
print(f"字体文件存在：{custom_font_path}")
font_size = 50
white_space = 20 #字幕左右留白
# 生成字幕图像
subtitle_image = generate_subtitle_image(
subtitle_text, custom_font_path, font_size, video_width, video_height, white_space
)
output(subtitle_image)
多段字幕
from moviepy.editor import VideoFileClip, CompositeVideoClip, ImageClip
from PIL import Image, ImageFont, ImageDraw
import numpy as np
# 自动换行函数
def wrap_text(text, font, max_width):
lines = []
current_line = ""
for char in text:
test_line = current_line + char
line_width = font.getbbox(test_line)[2]
if line_width <= max_width:
current_line = test_line
else:
if current_line:
lines.append(current_line)
current_line = char
if current_line:
lines.append(current_line)
return lines
# 计算每段字幕的宽高，并生成图像
def generate_subtitle_image(text, font_path, font_size, video_width, line_spacing=10):
font = ImageFont.truetype(font_path, font_size)
max_width = video_width - 40 # Subtract some padding
wrapped_text = wrap_text(text, font, max_width)
# 计算每行的高度，基于行数和行距来计算总高度
line_height = font.getbbox("测试")[3]
image_height = line_height * len(wrapped_text) + line_spacing * (len(wrapped_text) - 1) + 20 # 额外的顶部和底部填充
image = Image.new("RGBA", (video_width, image_height), (255, 255, 255, 0))
draw = ImageDraw.Draw(image)
y_position = 10
for line in wrapped_text:
text_width = font.getbbox(line)[2]
x_position = (video_width - text_width) // 2 # 居中显示
draw.text((x_position, y_position), line, font=font, fill="white")
y_position += line_height + line_spacing
return np.array(image), image_height
# 动态调整字体大小（如果需要）
def adjust_font_size(text, font_path, max_width, max_height, initial_font_size=40):
font_size = initial_font_size
font = ImageFont.truetype(font_path, font_size)
# 不断减小字体，直到文本适配屏幕宽度和高度
while True:
wrapped_text = wrap_text(text, font, max_width)
line_height = font.getbbox("测试")[3]
total_height = line_height * len(wrapped_text) + (len(wrapped_text) - 1) * 10 # 行高和行间距
if total_height <= max_height and all(font.getbbox(line)[2] <= max_width for line in wrapped_text):
break
font_size -= 1 # 字体过大时减小字体
font = ImageFont.truetype(font_path, font_size)
return font, wrapped_text, line_height
def output(subtitle_clips):
# 合成视频
composite_clip = CompositeVideoClip([video_clip] + subtitle_clips)
composite_clip.write_videofile(output_path, codec="libx264", fps=24)
def letter_generation(subtitles):
# 生成字幕片段
subtitle_clips = []
for subtitle in subtitles:
# 调整字体大小以适配视频宽度和高度
font, wrapped_text, line_height = adjust_font_size(
subtitle["text"], custom_font_path, video_width - 40, video_height // 4, initial_font_size
)
# 生成字幕图像
subtitle_image, subtitle_height = generate_subtitle_image(
subtitle["text"], custom_font_path, font.size, video_width, line_spacing
)
# 设置字幕位置并考虑底部留白（底部空隙 20 像素）
subtitle_clip = (
ImageClip(subtitle_image, duration=subtitle["end"] - subtitle["start"])
.set_start(subtitle["start"])
.set_position(("center", video_height - subtitle_height - 30)) # 给字幕底部增加 30 像素空隙
)
subtitle_clips.append(subtitle_clip)
output(subtitle_clips)
if __name__ == '__main__':
# 主逻辑
video_path = "1127.mp4"
output_path = "output_with_subtitles1.mp4"
# 视频片段
video_clip = VideoFileClip(video_path)
video_width, video_height = video_clip.w, video_clip.h
# 字体配置
custom_font_path = r"C:\Windows\Fonts\msyh.ttc" # 替换为实际字体路径
initial_font_size = 40
line_spacing = 10 # 设置行距
# 字幕列表，每段字幕包含文本、开始时间、结束时间
subtitles = [
{"text": "这是第一段字幕，显示时间是0到5秒adfa", "start": 0, "end": 5},
{"text": "这是第二段字幕，显示时间是5到10秒，这段字幕会自动换行af。", "start": 5, "end": 10},
{"text": "第三段字幕更短，显示时间是10到15秒asda。", "start": 10, "end": 15},
]
letter_generation(subtitles)