mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-12 19:52:48 +00:00
366 lines
12 KiB
Python
366 lines
12 KiB
Python
import traceback
|
||
|
||
# import pysrt
|
||
from typing import Optional
|
||
from typing import List
|
||
from loguru import logger
|
||
from moviepy import *
|
||
from PIL import ImageFont
|
||
from contextlib import contextmanager
|
||
from moviepy import (
|
||
VideoFileClip,
|
||
AudioFileClip,
|
||
TextClip,
|
||
CompositeVideoClip,
|
||
CompositeAudioClip
|
||
)
|
||
|
||
|
||
from app.models.schema import VideoAspect, SubtitlePosition
|
||
|
||
|
||
def wrap_text(text, max_width, font, fontsize=60):
|
||
"""
|
||
文本自动换行处理
|
||
Args:
|
||
text: 待处理的文本
|
||
max_width: 最大宽度
|
||
font: 字体文件路径
|
||
fontsize: 字体大小
|
||
|
||
Returns:
|
||
tuple: (换行后的文本, 文本高度)
|
||
"""
|
||
# 创建字体对象
|
||
font = ImageFont.truetype(font, fontsize)
|
||
|
||
def get_text_size(inner_text):
|
||
inner_text = inner_text.strip()
|
||
left, top, right, bottom = font.getbbox(inner_text)
|
||
return right - left, bottom - top
|
||
|
||
width, height = get_text_size(text)
|
||
if width <= max_width:
|
||
return text, height
|
||
|
||
logger.debug(f"换行文本, 最大宽度: {max_width}, 文本宽度: {width}, 文本: {text}")
|
||
|
||
processed = True
|
||
|
||
_wrapped_lines_ = []
|
||
words = text.split(" ")
|
||
_txt_ = ""
|
||
for word in words:
|
||
_before = _txt_
|
||
_txt_ += f"{word} "
|
||
_width, _height = get_text_size(_txt_)
|
||
if _width <= max_width:
|
||
continue
|
||
else:
|
||
if _txt_.strip() == word.strip():
|
||
processed = False
|
||
break
|
||
_wrapped_lines_.append(_before)
|
||
_txt_ = f"{word} "
|
||
_wrapped_lines_.append(_txt_)
|
||
if processed:
|
||
_wrapped_lines_ = [line.strip() for line in _wrapped_lines_]
|
||
result = "\n".join(_wrapped_lines_).strip()
|
||
height = len(_wrapped_lines_) * height
|
||
# logger.warning(f"wrapped text: {result}")
|
||
return result, height
|
||
|
||
_wrapped_lines_ = []
|
||
chars = list(text)
|
||
_txt_ = ""
|
||
for word in chars:
|
||
_txt_ += word
|
||
_width, _height = get_text_size(_txt_)
|
||
if _width <= max_width:
|
||
continue
|
||
else:
|
||
_wrapped_lines_.append(_txt_)
|
||
_txt_ = ""
|
||
_wrapped_lines_.append(_txt_)
|
||
result = "\n".join(_wrapped_lines_).strip()
|
||
height = len(_wrapped_lines_) * height
|
||
logger.debug(f"换行文本: {result}")
|
||
return result, height
|
||
|
||
|
||
@contextmanager
|
||
def manage_clip(clip):
|
||
"""
|
||
视频片段资源管理器
|
||
Args:
|
||
clip: 视频片段对象
|
||
|
||
Yields:
|
||
VideoFileClip: 视频片段对象
|
||
"""
|
||
try:
|
||
yield clip
|
||
finally:
|
||
clip.close()
|
||
del clip
|
||
|
||
|
||
def resize_video_with_padding(clip, target_width: int, target_height: int):
|
||
"""
|
||
调整视频尺寸并添加黑边
|
||
Args:
|
||
clip: 视频片段
|
||
target_width: 目标宽度
|
||
target_height: 目标高度
|
||
|
||
Returns:
|
||
CompositeVideoClip: 调整尺寸后的视频
|
||
"""
|
||
clip_ratio = clip.w / clip.h
|
||
target_ratio = target_width / target_height
|
||
|
||
if clip_ratio == target_ratio:
|
||
return clip.resize((target_width, target_height))
|
||
|
||
if clip_ratio > target_ratio:
|
||
scale_factor = target_width / clip.w
|
||
else:
|
||
scale_factor = target_height / clip.h
|
||
|
||
new_width = int(clip.w * scale_factor)
|
||
new_height = int(clip.h * scale_factor)
|
||
clip_resized = clip.resize(newsize=(new_width, new_height))
|
||
|
||
background = ColorClip(
|
||
size=(target_width, target_height),
|
||
color=(0, 0, 0)
|
||
).set_duration(clip.duration)
|
||
|
||
return CompositeVideoClip([
|
||
background,
|
||
clip_resized.set_position("center")
|
||
])
|
||
|
||
|
||
def loop_audio_clip(audio_clip: AudioFileClip, target_duration: float) -> AudioFileClip:
|
||
"""
|
||
循环音频片段直到达到目标时长
|
||
|
||
参数:
|
||
audio_clip: 原始音频片段
|
||
target_duration: 目标时长（秒）
|
||
返回:
|
||
循环后的音频片段
|
||
"""
|
||
# 计算需要循环的次数
|
||
loops_needed = int(target_duration / audio_clip.duration) + 1
|
||
|
||
# 创建足够长的音频
|
||
extended_audio = audio_clip
|
||
for _ in range(loops_needed - 1):
|
||
extended_audio = CompositeAudioClip([
|
||
extended_audio,
|
||
audio_clip.set_start(extended_audio.duration)
|
||
])
|
||
|
||
# 裁剪到目标时长
|
||
return extended_audio.subclip(0, target_duration)
|
||
|
||
|
||
def calculate_subtitle_position(position, video_height: int, text_height: int = 0) -> tuple:
|
||
"""
|
||
计算字幕在视频中的具体位置
|
||
|
||
Args:
|
||
position: 位置配置，可以是 SubtitlePosition 枚举值或表示距顶部百分比的浮点数
|
||
video_height: 视频高度
|
||
text_height: 字幕文本高度
|
||
|
||
Returns:
|
||
tuple: (x, y) 坐标
|
||
"""
|
||
margin = 50 # 字幕距离边缘的边距
|
||
|
||
if isinstance(position, (int, float)):
|
||
# 百分比位置
|
||
return ('center', int(video_height * position))
|
||
|
||
# 预设位置
|
||
if position == SubtitlePosition.TOP:
|
||
return ('center', margin)
|
||
elif position == SubtitlePosition.CENTER:
|
||
return ('center', video_height // 2)
|
||
elif position == SubtitlePosition.BOTTOM:
|
||
return ('center', video_height - margin - text_height)
|
||
|
||
# 默认底部
|
||
return ('center', video_height - margin - text_height)
|
||
|
||
|
||
def generate_video_v3(
|
||
video_path: str,
|
||
subtitle_style: dict,
|
||
volume_config: dict,
|
||
subtitle_path: Optional[str] = None,
|
||
bgm_path: Optional[str] = None,
|
||
narration_path: Optional[str] = None,
|
||
output_path: str = "output.mp4",
|
||
font_path: Optional[str] = None
|
||
) -> None:
|
||
"""
|
||
合并视频素材，包括视频、字幕、BGM和解说音频
|
||
|
||
参数:
|
||
video_path: 原视频文件路径
|
||
subtitle_path: SRT字幕文件路径（可选）
|
||
bgm_path: 背景音乐文件路径（可选）
|
||
narration_path: 解说音频文件路径（可选）
|
||
output_path: 输出文件路径
|
||
volume_config: 音量配置字典，可包含以下键：
|
||
- original: 原声音量（0-1），默认1.0
|
||
- bgm: BGM音量（0-1），默认0.3
|
||
- narration: 解说音量（0-1），默认1.0
|
||
subtitle_style: 字幕样式配置字典，可包含以下键：
|
||
- font: 字体名称
|
||
- fontsize: 字体大小
|
||
- color: 字体颜色
|
||
- stroke_color: 描边颜色
|
||
- stroke_width: 描边宽度
|
||
- bg_color: 背景色
|
||
- position: 位置支持 SubtitlePosition 枚举值或 0-1 之间的浮点数（表示距顶部的百分比）
|
||
- method: 文字渲染方法
|
||
font_path: 字体文件路径（.ttf/.otf 等格式）
|
||
"""
|
||
# 检查视频文件是否存在
|
||
if not os.path.exists(video_path):
|
||
raise FileNotFoundError(f"视频文件不存在: {video_path}")
|
||
|
||
# 加载视频
|
||
video = VideoFileClip(video_path)
|
||
subtitle_clips = []
|
||
|
||
# 处理字幕（如果提供）
|
||
if subtitle_path:
|
||
if os.path.exists(subtitle_path):
|
||
# 检查字体文件
|
||
if font_path and not os.path.exists(font_path):
|
||
logger.warning(f"警告：字体文件不存在: {font_path}")
|
||
|
||
try:
|
||
subs = pysrt.open(subtitle_path)
|
||
logger.info(f"读取到 {len(subs)} 条字幕")
|
||
|
||
for index, sub in enumerate(subs):
|
||
start_time = sub.start.ordinal / 1000
|
||
end_time = sub.end.ordinal / 1000
|
||
|
||
try:
|
||
# 检查字幕文本是否为空
|
||
if not sub.text or sub.text.strip() == '':
|
||
logger.info(f"警告：第 {index + 1} 条字幕内容为空，已跳过")
|
||
continue
|
||
|
||
# 处理字幕文本：确保是字符串，并处理可能的列表情况
|
||
if isinstance(sub.text, (list, tuple)):
|
||
subtitle_text = ' '.join(str(item) for item in sub.text if item is not None)
|
||
else:
|
||
subtitle_text = str(sub.text)
|
||
|
||
subtitle_text = subtitle_text.strip()
|
||
|
||
if not subtitle_text:
|
||
logger.info(f"警告：第 {index + 1} 条字幕处理后为空，已跳过")
|
||
continue
|
||
|
||
# 创建临时 TextClip 来获取文本高度
|
||
temp_clip = TextClip(
|
||
subtitle_text,
|
||
font=font_path,
|
||
fontsize=subtitle_style['fontsize'],
|
||
color=subtitle_style['color']
|
||
)
|
||
text_height = temp_clip.h
|
||
temp_clip.close()
|
||
|
||
# 计算字幕位置
|
||
position = calculate_subtitle_position(
|
||
subtitle_style['position'],
|
||
video.h,
|
||
text_height
|
||
)
|
||
|
||
# 创建最终的 TextClip
|
||
text_clip = (TextClip(
|
||
subtitle_text,
|
||
font=font_path,
|
||
fontsize=subtitle_style['fontsize'],
|
||
color=subtitle_style['color']
|
||
)
|
||
.set_position(position)
|
||
.set_duration(end_time - start_time)
|
||
.set_start(start_time))
|
||
subtitle_clips.append(text_clip)
|
||
|
||
except Exception as e:
|
||
logger.error(f"警告：创建第 {index + 1} 条字幕时出错: {traceback.format_exc()}")
|
||
|
||
logger.info(f"成功创建 {len(subtitle_clips)} 条字幕剪辑")
|
||
except Exception as e:
|
||
logger.info(f"警告：处理字幕文件时出错: {str(e)}")
|
||
else:
|
||
logger.info(f"提示：字幕文件不存在: {subtitle_path}")
|
||
|
||
# 合并音频
|
||
audio_clips = []
|
||
|
||
# 添加原声（设置音量）
|
||
logger.debug(f"音量配置: {volume_config}")
|
||
if video.audio is not None:
|
||
original_audio = video.audio.volumex(volume_config['original'])
|
||
audio_clips.append(original_audio)
|
||
|
||
# 添加BGM（如果提供）
|
||
if bgm_path:
|
||
bgm = AudioFileClip(bgm_path)
|
||
if bgm.duration < video.duration:
|
||
bgm = loop_audio_clip(bgm, video.duration)
|
||
else:
|
||
bgm = bgm.subclip(0, video.duration)
|
||
bgm = bgm.volumex(volume_config['bgm'])
|
||
audio_clips.append(bgm)
|
||
|
||
# 添加解说音频（如果提供）
|
||
if narration_path:
|
||
narration = AudioFileClip(narration_path).volumex(volume_config['narration'])
|
||
audio_clips.append(narration)
|
||
|
||
# 合成最终视频（包含字幕）
|
||
if subtitle_clips:
|
||
final_video = CompositeVideoClip([video] + subtitle_clips, size=video.size)
|
||
else:
|
||
logger.info("警告：没有字幕被添加到视频中")
|
||
final_video = video
|
||
|
||
if audio_clips:
|
||
final_audio = CompositeAudioClip(audio_clips)
|
||
final_video = final_video.set_audio(final_audio)
|
||
|
||
# 导出视频
|
||
logger.info("开始导出视频...") # 调试信息
|
||
final_video.write_videofile(
|
||
output_path,
|
||
codec='libx264',
|
||
audio_codec='aac',
|
||
fps=video.fps
|
||
)
|
||
logger.info(f"视频已导出到: {output_path}") # 调试信息
|
||
|
||
# 清理资源
|
||
video.close()
|
||
for clip in subtitle_clips:
|
||
clip.close()
|
||
if bgm_path:
|
||
bgm.close()
|
||
if narration_path:
|
||
narration.close()
|