From 3b83279855eb2f05ee7720abbc40aa2a6b21f632 Mon Sep 17 00:00:00 2001 From: binary-husky Date: Sat, 20 Jan 2024 22:17:51 +0800 Subject: [PATCH] anim generation bug fix #896 --- crazy_functions/数学动画生成manim.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/crazy_functions/数学动画生成manim.py b/crazy_functions/数学动画生成manim.py index 30ea430..0acf559 100644 --- a/crazy_functions/数学动画生成manim.py +++ b/crazy_functions/数学动画生成manim.py @@ -1,6 +1,7 @@ -from toolbox import CatchException, update_ui, gen_time_str -from .crazy_utils import request_gpt_model_in_new_thread_with_ui_alive -from .crazy_utils import input_clipping +import os +from toolbox import CatchException, update_ui, gen_time_str, promote_file_to_downloadzone +from crazy_functions.crazy_utils import request_gpt_model_in_new_thread_with_ui_alive +from crazy_functions.crazy_utils import input_clipping def inspect_dependency(chatbot, history): # 尝试导入依赖，如果缺少依赖，则给出安装建议 @@ -95,6 +96,8 @@ def 动画生成(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt res = eval_manim(code) chatbot.append(("生成的视频文件路径", res)) + if os.path.exists(res): + promote_file_to_downloadzone(res, chatbot=chatbot) yield from update_ui(chatbot=chatbot, history=history) # 刷新界面 # 界面更新 # 在这里放一些网上搜集的demo，辅助gpt生成代码