diff --git a/app/services/llm.py b/app/services/llm.py
index 01bef0a..3e9ba16 100644
--- a/app/services/llm.py
+++ b/app/services/llm.py
@@ -439,14 +439,14 @@ def generate_script(
video_name=video_name,
video_path=compressed_video_path,
language=language,
- llm_provider_video="gemini",
+ llm_provider_video=config.app["video_llm_provider"],
progress_callback=progress_callback
)
if progress_callback:
progress_callback(60, "生成解说文案...") # 例如,在转录视频后
# 3. 编写解说文案
- script = writing_short_play(video_plot, video_name, "openai", count=300)
+ script = writing_short_play(video_plot, video_name, config.app["llm_provider"], count=300)
# 在关键步骤更新进度
if progress_callback:
@@ -454,7 +454,7 @@ def generate_script(
# 4. 文案匹配画面
if transcription != "":
- matched_script = screen_matching(huamian=transcription, wenan=script, llm_provider="openai")
+ matched_script = screen_matching(huamian=transcription, wenan=script, llm_provider=config.app["video_llm_provider"])
# 在关键步骤更新进度
if progress_callback:
progress_callback(80, "匹配成功")
@@ -769,58 +769,6 @@ def screen_matching(huamian: str, wenan: str, llm_provider: str):
- 请以严格的 JSON 格式返回数据，不要包含任何注释、标记或其他字符。数据应符合 JSON 语法，可以被 json.loads() 函数直接解析， 不要添加 ```json 或其他标记。
""" % (huamian, wenan)
-# prompt = """
-# 你是一位拥有10年丰富经验的影视解说创作专家。你的任务是根据提供的视频转录脚本和解说文案，创作一个引人入胜的解说脚本。请按照以下要求完成任务：
-#
-# 1. 输入数据：
-# - 视频转录脚本：包含时间戳、画面描述和人物台词
-# - 解说文案：需要你进行匹配和编排的内容
-# - 视频转录脚本和文案（由 XML 标记和 分隔）如下所示：
-# 视频转录脚本
-#
-# %s
-#
-# 文案：
-#
-# %s
-#
-#
-# 2. 输出要求：
-# - 格式：严格的JSON格式，可直接被json.loads()解析
-# - 结构：list[script]，其中script为字典类型
-# - script字段：
-# {
-# "picture": "画面描述",
-# "timestamp": "时间戳",
-# "narration": "解说文案",
-# "OST": true/false
-# }
-#
-# 3. 匹配规则：
-# a) 时间戳匹配：
-# - 根据文案内容选择最合适的画面时间段
-# - 避免时间重叠，确保画面不重复出现
-# - 适当合并或删减片段，不要完全照搬转录脚本
-# b) 画面描述：与转录脚本保持一致
-# c) 解说文案：
-# - 当OST为true时，narration为空字符串
-# - 当OST为false时，narration为解说文案，但是要确保文案字数不要超过 30字，若文案较长，则添加到下一个片段
-# d) OST（原声）：
-# - 按1:1比例穿插原声和解说片段
-# - 第一个片段必须是原声，时长不少于20秒
-# - 选择整个视频中最精彩的片段作为开场
-#
-# 4. 创作重点：
-# - 确保解说与画面高度匹配
-# - 巧妙安排原声和解说的交替，提升观众体验
-# - 创造一个引人入胜、节奏紧凑的解说脚本
-#
-# 5. 注意事项：
-# - 严格遵守JSON格式，不包含任何注释或额外标记
-# - 充分利用你的专业经验，创作出高质量、吸引人的解说内容
-#
-# 请基于以上要求，将提供的视频转录脚本和解说文案整合成一个专业、吸引人的解说脚本。你的创作将直接影响观众的观看体验，请发挥你的专业素养，创作出最佳效果。
-# """ % (huamian, wenan)
try:
response = _generate_response(prompt, llm_provider)
logger.success("匹配成功")
diff --git a/webui.py b/webui.py
index 79bd7a8..5e37dd7 100644
--- a/webui.py
+++ b/webui.py
@@ -124,7 +124,7 @@ def init_log():
_lvl = "DEBUG"
def format_record(record):
- # 获取日志记录中的文件全���径
+ # 获取日志记录中的文件全径
file_path = record["file"].path
# 将绝对路径转换为相对于项目根目录的路径
relative_path = os.path.relpath(file_path, root_dir)
@@ -290,7 +290,7 @@ with left_panel:
# 按创建时间降序排序
script_list.sort(key=lambda x: x["ctime"], reverse=True)
- # ��本文件 下拉框
+ # 本文件 下拉框
script_path = [(tr("Auto Generate"), ""), ]
for file in script_list:
display_name = file['file'].replace(root_dir, "")
@@ -385,12 +385,8 @@ with left_panel:
try:
with st.spinner("正在生成脚本..."):
- if not video_name:
- st.warning("视频名称不能为空")
- st.stop()
if not video_plot:
- st.warning("视频剧情不能为空")
- st.stop()
+ st.warning("视频剧情为空; 会极大影响生成效果！")
if params.video_clip_json_path == "" and params.video_origin_path != "":
update_progress(10, "压缩视频中...")
# 使用大模型生成视频脚本
@@ -756,6 +752,10 @@ with st.expander(tr("Video Check"), expanded=False):
# 开始按钮
start_button = st.button(tr("Generate Video"), use_container_width=True, type="primary")
if start_button:
+ # 重置日志容器和记录
+ log_container = st.empty()
+ log_records = []
+
config.save_config()
task_id = st.session_state.get('task_id')
if st.session_state.get('video_script_json_path') is not None:
@@ -778,16 +778,11 @@ if start_button:
scroll_to_bottom()
st.stop()
- log_container = st.empty()
- log_records = []
-
-
def log_received(msg):
with log_container:
log_records.append(msg)
st.code("\n".join(log_records))
-
logger.add(log_received)
st.toast(tr("生成视频"))