195 lines
7.9 KiB
Python
195 lines
7.9 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
import logging
|
|||
|
|
import os
|
|||
|
|
import subprocess
|
|||
|
|
|
|||
|
|
from Manim.Demos.ManimKit import SYS_PROMPT, generate_code_with_llm
|
|||
|
|
|
|||
|
|
# 更详细地控制日志输出
|
|||
|
|
logger = logging.getLogger('T0')
|
|||
|
|
logger.setLevel(logging.INFO)
|
|||
|
|
handler = logging.StreamHandler()
|
|||
|
|
handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
|
|||
|
|
logger.addHandler(handler)
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
try:
|
|||
|
|
# 首次生成唯一时间戳
|
|||
|
|
# timestamp = datetime.datetime.now().strftime('%Y%m%d_%H%M%S_%f')[:-3] # 保留到毫秒
|
|||
|
|
# 后续写死时间戳
|
|||
|
|
timestamp = "20250804_084910_884"
|
|||
|
|
|
|||
|
|
skeleton_basename = f"Skeleton/Skeleton_{timestamp}.py"
|
|||
|
|
scene_class_name = f"Skeleton_{timestamp}"
|
|||
|
|
|
|||
|
|
# 教师输入的提示词数组（简化版）
|
|||
|
|
teacher_prompts = [
|
|||
|
|
{
|
|||
|
|
"id": 1,
|
|||
|
|
"task": "生成片头标题动画",
|
|||
|
|
"requirements": [
|
|||
|
|
"黑色背景",
|
|||
|
|
"使用大标题文本'东师理想数学动画AI制作平台作品',字号=60，白色",
|
|||
|
|
"添加副标题'正弦与余弦的关系'，字号=36，蓝色",
|
|||
|
|
"主标题+副标题淡入2秒，停留3秒，淡出2秒",
|
|||
|
|
"最后清屏1秒"
|
|||
|
|
]
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"id": 2,
|
|||
|
|
"task": "生成可运行的 Manim 场景骨架",
|
|||
|
|
"requirements": [
|
|||
|
|
"黑色背景",
|
|||
|
|
"建立完整四象限坐标系（x∈[-3.5π,3.5π]，y∈[-1.5,1.5]）",
|
|||
|
|
"不画任何曲线，只显示坐标轴,坐标轴要记得带箭头"
|
|||
|
|
]
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"id": 3,
|
|||
|
|
"task": "在上一步基础上追加",
|
|||
|
|
"requirements": [
|
|||
|
|
"用绿色画 y = sin(x)",
|
|||
|
|
"用 MathTex 在 (-3π,1) 位置标 'sin(x)'",
|
|||
|
|
"动画：坐标轴淡入 1s → 曲线从左到右扫出 3s"
|
|||
|
|
]
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"id": 4,
|
|||
|
|
"task": "在上一步基础上追加",
|
|||
|
|
"requirements": [
|
|||
|
|
"用红色画 y = cos(x)",
|
|||
|
|
"用 MathTex 在 (3π,-1) 位置标 'cos(x)'",
|
|||
|
|
"动画：TransformFromCopy 把正弦变成余弦 3s",
|
|||
|
|
"最后在屏幕上方用黄色 MathTex 写出 cos(x) = sin(x + π/2)"
|
|||
|
|
]
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"id": 5,
|
|||
|
|
"task": "在上一步基础上追加",
|
|||
|
|
"requirements": [
|
|||
|
|
"黄色虚线竖线随 x 移动",
|
|||
|
|
"绿色圆点跟踪 sin(x)",
|
|||
|
|
"红色圆点跟踪 cos(x)",
|
|||
|
|
"ValueTracker 从 -3π 匀速扫到 3π，耗时 6s",
|
|||
|
|
"其余元素保持不变"
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
# 系统补充信息（教师无需关心）
|
|||
|
|
system_supplements = {
|
|||
|
|
"base_info": f"文件名：{skeleton_basename}，场景类：{scene_class_name}",
|
|||
|
|
"output_requirement": "输出：完整代码 + `if __name__ == \"__main__\":`",
|
|||
|
|
"continue_requirement": f"输出：完整替换 {skeleton_basename} 的新代码",
|
|||
|
|
"video_config": "帧率：30fps，输出目录：output/",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 设置保存路径
|
|||
|
|
skeleton_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), skeleton_basename)
|
|||
|
|
|
|||
|
|
# 显示所有可用任务
|
|||
|
|
print("可用任务列表:")
|
|||
|
|
for prompt_info in teacher_prompts:
|
|||
|
|
print(f"任务 {prompt_info['id']}: {prompt_info['task']}")
|
|||
|
|
for req in prompt_info['requirements']:
|
|||
|
|
print(f" - {req}")
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
# 让用户选择要执行的任务ID
|
|||
|
|
while True:
|
|||
|
|
try:
|
|||
|
|
selected_id = int(input("请输入要执行的任务ID (1-5): "))
|
|||
|
|
if 1 <= selected_id <= 5:
|
|||
|
|
break
|
|||
|
|
else:
|
|||
|
|
print("无效的ID，请输入1-5之间的数字。")
|
|||
|
|
except ValueError:
|
|||
|
|
print("请输入有效的数字。")
|
|||
|
|
|
|||
|
|
# 查找用户选择的任务
|
|||
|
|
selected_prompt = None
|
|||
|
|
for prompt_info in teacher_prompts:
|
|||
|
|
if prompt_info['id'] == selected_id:
|
|||
|
|
selected_prompt = prompt_info
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
# 处理用户选择的任务
|
|||
|
|
prompt_id = selected_prompt["id"]
|
|||
|
|
task = selected_prompt["task"]
|
|||
|
|
requirements = selected_prompt["requirements"]
|
|||
|
|
logger.info(f"处理任务 {prompt_id}: {task}...")
|
|||
|
|
|
|||
|
|
# 构建教师输入部分
|
|||
|
|
teacher_part = f"任务：{task}\n要求：\n"
|
|||
|
|
for req in requirements:
|
|||
|
|
teacher_part += f"- {req}\n"
|
|||
|
|
|
|||
|
|
# 构建完整提示词
|
|||
|
|
if prompt_id == 1:
|
|||
|
|
# 第一个任务，包含基础信息和输出要求
|
|||
|
|
full_prompt = f"""
|
|||
|
|
{teacher_part}
|
|||
|
|
{system_supplements['base_info']}
|
|||
|
|
{system_supplements['output_requirement']}
|
|||
|
|
重要配置要求:
|
|||
|
|
- 请设置输出目录为output/
|
|||
|
|
- 设置帧率为30fps
|
|||
|
|
- 确保生成MP4格式视频
|
|||
|
|
- 添加标题淡入淡出动画以确保生成视频而非静态图像
|
|||
|
|
"""
|
|||
|
|
else:
|
|||
|
|
# 后续任务，检查前置文件是否存在
|
|||
|
|
if os.path.exists(skeleton_path):
|
|||
|
|
# 构建继续要求
|
|||
|
|
full_prompt = f"""
|
|||
|
|
{teacher_part}
|
|||
|
|
{system_supplements['continue_requirement']}
|
|||
|
|
|
|||
|
|
{skeleton_basename} 内容：
|
|||
|
|
{open(skeleton_path, 'r', encoding='utf-8').read()}
|
|||
|
|
重要配置要求:
|
|||
|
|
- 请保持输出目录为output/
|
|||
|
|
- 保持帧率为30fps
|
|||
|
|
- 确保生成MP4格式视频
|
|||
|
|
"""
|
|||
|
|
else:
|
|||
|
|
logger.error(f"错误: 任务 {prompt_id} 依赖于前置任务生成的 {skeleton_basename} 文件，但该文件不存在。")
|
|||
|
|
logger.info("请先执行任务 1 生成基础文件，然后再执行后续任务。")
|
|||
|
|
exit(1)
|
|||
|
|
|
|||
|
|
# 格式化系统提示
|
|||
|
|
formatted_prompt = SYS_PROMPT.format(XuQiu=full_prompt,Preview=True)
|
|||
|
|
|
|||
|
|
# 调用函数生成代码
|
|||
|
|
generate_code_with_llm(prompt=formatted_prompt, save_path=skeleton_path)
|
|||
|
|
|
|||
|
|
logger.info(f"任务 {prompt_id} 处理完成！生成的文件为：{skeleton_basename}")
|
|||
|
|
|
|||
|
|
# 执行生成的Skeleton文件
|
|||
|
|
logger.info(f"正在执行 {skeleton_basename} ...")
|
|||
|
|
try:
|
|||
|
|
# 定义清晰的输出目录
|
|||
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
output_dir = os.path.join(current_dir, "output")
|
|||
|
|
os.makedirs(output_dir, exist_ok=True)
|
|||
|
|
|
|||
|
|
# 构建执行命令，根据preview参数决定是否添加-p选项
|
|||
|
|
command = ['python', skeleton_path, '-o', output_dir]
|
|||
|
|
|
|||
|
|
# 执行文件并通过命令行参数指定输出目录
|
|||
|
|
subprocess.run(
|
|||
|
|
command,
|
|||
|
|
check=True
|
|||
|
|
)
|
|||
|
|
logger.info(f"视频已生成到: {output_dir}\\videos\\1080p30\\Skeleton_{timestamp}.mp4")
|
|||
|
|
# TODO
|
|||
|
|
# 这个生成的视频路径对于研发来讲很重要，需要返回给前端，前端展示给用户查看效果
|
|||
|
|
|
|||
|
|
except subprocess.CalledProcessError as e:
|
|||
|
|
logger.error(f"执行文件时出错: {str(e)}")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"执行过程中发生未知错误: {str(e)}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"程序执行出错: {str(e)}")
|