基于AI的一站式短剧生成平台,实现从剧本生成、角色设计、分镜制作到视频合成的全流程自动化。

Github地址

https://github.com/chatfire-AI/huobao-drama

功能特性

  • 角色管理

    • AI生成角色形象。
    • 批量角色生成。
    • 角色图片上传和管理。
  • 分镜制作

    • 自动生成分镜脚本。
    • 场景描述和镜头设计。
    • 分镜图片生成(文生图)。
    • 帧类型选择(首帧/关键帧/尾帧/分镜板)。
  • 视频生成

    • 图生视频自动生成。
    • 视频合成和剪辑。
    • 转场效果。
  • 资源管理

    • 素材库统一管理。
    • 本地存储支持。
    • 资源导入导出。
    • 任务进度追踪。

快速开始

  • 环境要求

    • Go:1.23+
    • Node.js:18+
    • npm:9+
    • FFmpeg:4.0+(必需)
    • SQLite:3.x(已内置)
  • 安装 FFmpeg

    • macOS:brew install ffmpeg
    • Ubuntu/Debian:sudo apt update && sudo apt install ffmpeg
    • Windows:从FFmpeg官网下载并配置环境变量
  • 配置文件

    • 复制并编辑configs/config.example.yamlconfigs/config.yaml

    • 配置文件示例:

      1
      2
      3
      4
      5
      6
      7
      8
      9
      10
      11
      12
      13
      14
      15
      16
      17
      18
      19
      20
      21
      22
      23
      24
      25
      26
      27
      28
      app:
      name:"Huobao Drama API"
      version:"1.0.0"
      debug:true

      server:
      port:5678
      host:"0.0.0.0"
      cors_origins:
      -"http://localhost:3012"
      read_timeout:600
      write_timeout:600

      database:
      type:"sqlite"
      path:"./data/drama_generator.db"
      max_idle:10
      max_open:100

      storage:
      type:"local"
      local_path:"./data/storage"
      base_url:"http://localhost:5678/static"

      ai:
      default_text_provider:"openai"
      default_image_provider:"openai"
      default_video_provider:"doubao"
  • 安装依赖

    1
    2
    3
    4
    5
    6
    git clone https://github.com/chatfire-AI/huobao-drama.git
    cd huobao-drama
    go mod download
    cd web
    npm install
    cd ..
  • 启动项目

    • 访问:http://localhost:5678
    • 前端地址:http://localhost:3012
    • 后端API:http://localhost:5678/api/v1
    • 开发模式

      1
      2
      3
      4
      5
      6
      # 终端1:启动后端服务
      go run main.go

      # 终端2:启动前端开发服务器
      cd web
      npm run dev
    • 单服务模式

      1
      2
      3
      4
      5
      6
      7
      # 1. 构建前端
      cd web
      npm run build
      cd ..

      # 2. 启动服务
      go run main.go

🐳 Docker 部署(推荐)

方式一:Docker Compose(推荐)

1
2
3
4
5
6
7
8
# 启动服务
docker-compose up -d

# 查看日志
docker-compose logs -f

# 停止服务
docker-compose down

方式二:Docker 命令

注意:Linux 用户需添加 --add-host=host.docker.internal:host-gateway 以访问宿主机服务

1
2
3
4
5
6
7
8
9
10
# 从 Docker Hub 运行
docker run -d \
--name huobao-drama \
-p 5678:5678 \
-v $(pwd)/data:/app/data \
--restart unless-stopped \
huobao/huobao-drama:latest

# 查看日志
docker logs -f huobao-drama

本地构建(可选):

1
2
docker build -t huobao-drama:latest .
docker run -d --name huobao-drama -p 5678:5678 -v $(pwd)/data:/app/data huobao-drama:latest

Docker 部署优势:

  • ✅ 开箱即用,内置默认配置
  • ✅ 环境一致性,避免依赖问题
  • ✅ 一键启动,无需安装 Go、Node.js、FFmpeg
  • ✅ 易于迁移和扩展
  • ✅ 自动健康检查和重启
  • ✅ 自动处理文件权限,无需手动配置

🔗 访问宿主机服务(Ollama/本地模型)

容器已配置支持访问宿主机服务,直接使用 http://host.docker.internal:端口号 即可。

配置步骤:

  1. 宿主机启动服务(监听所有接口)

    1
    export OLLAMA_HOST=0.0.0.0:11434 && ollama serve
  2. 前端 AI 服务配置

    • Base URL: http://host.docker.internal:11434/v1
    • Provider: openai
    • Model: qwen2.5:latest