二次开发指南
项目结构
aivideo_v2/
├── main.py # FastAPI 后端主入口
├── start.bat # Windows 一键启动
├── video-platform/ # 核心 AI 平台
│ ├── bin/ # FFmpeg / FFprobe 内置二进制
│ ├── models/project.py # 数据模型(枚举 + Project/Scene/VideoConfig)
│ ├── agents/ # Agent 业务层
│ │ ├── script_writer.py # 剧本创作
│ │ ├── image_creator.py # 文生图
│ │ ├── video_creator.py # 图生视频
│ │ └── video_composer.py # 视频合成
│ ├── services/ # 基础设施服务层
│ │ ├── db.py # SQLite DAO(六表 CRUD)
│ │ ├── generation.py # GenerationManager 管线编排
│ │ ├── auth_service.py # 认证(PBKDF2/token/SVG 验证码)
│ │ ├── llm_client.py # LLM 客户端
│ │ ├── comfyui_img.py # ComfyUI 文生图客户端
│ │ ├── comfyui_vid.py # ComfyUI 图生视频客户端
│ │ └── voice_actor.py # IndexTTS 配音
│ └── api/ # 路由(auth/project/asset/tutorial/settings)
├── sqlite/lvf.db # 唯一数据源
├── output/ # 生成产物
└── video-frontend/ # Vue 3 前端扩展 Agent 管线
- 在
video-platform/agents/下创建新的 Agent 类; - 在
services/generation.py的run()中编排新步骤; - 在
main.py中挂载对应接口(如需); - 在前端创作页
StepFlow中添加对应步骤展示。
# agents/video_reviewer.py 示例
class VideoReviewer:
"""视频质量审核 Agent"""
def review(self, project):
# 调用 AI 模型评估视频质量
return project前端扩展
- 组件:
video-frontend/src/components/下创建新组件; - API:
src/api/下按模块封装调用; - 状态:
src/stores/(Pinia)管理; - 路由:
src/router/index.ts注册。
扩展方向:历史项目列表、分镜微调可视化编辑、视频模板、在线剪辑等。
编码规范
- Python:PEP 8、4 空格缩进、最大行宽 100、Type Hints;
- Vue:组合式 API +
<script setup>,组件名 PascalCase; - Git 提交:
<type>(<scope>): <subject>,type ∈ feat/fix/docs/style/refactor/test/chore。