记录 | 基于MaxKB的文字生成视频

前言

参考文章：如何利用智谱全模态免费模型，生成大家都喜欢的图、文、视并茂的文章！

自己的感想
本文记录了创建文字生成视频的函数库的过程。如果想复现本文，需要你逐一按需看完下面三篇记录。
1、记录 | Docker的windows版安装
2、记录 | 基于Docker Desktop的MaxKB安装
3、记录 | MaxKB创建本地AI智能问答系统

一、安装SDK

打开Docker Destop后，进入Container下的MaxKB中，在Exec中输入安装命令。

pip install zhipuai

二、创建视频函数库

先创建函数，在函数中添加参数prompt。这个参数是创建视频所需要的文字描述。详情可以通过https://bigmodel.cn/dev/api/videomodel/cogvideox 查看。
接着写入一段Python代码，如下所示。

from zhipuai import ZhipuAI
import time

def generate_and_get_video_url(prompt):
    #替换为你自己的key
    api_key = "XXXXXXXXXXXXXXXXXXXXXX"
    model = "cogvideox-flash"
    # 创建ZhipuAI客户端
    client = ZhipuAI(api_key=api_key)
    
    # 生成视频
    generation_response = client.videos.generations(
        model=model,
        prompt=prompt
    )
    request_id = generation_response.id  # 使用属性访问
    
    # 检查视频生成状态并等待完成
    while True:
        retrieval_response = client.videos.retrieve_videos_result(
            id=request_id
        )
        
        task_status = retrieval_response.task_status  # 使用属性访问
        
        if task_status == "SUCCESS":
            # 获取视频URL
            if retrieval_response.video_result:
                video_url = retrieval_response.video_result[0].url  # 使用属性访问
                #return video_url
                return f'<video controls width=500 height=300 src="{video_url}" frameborder="0" scrolling="no" allowfullscreen="true" alt="占位视频"></video>'
            else:
                print("视频生成成功，但未找到视频 URL")
                return None
        elif task_status == "FAILED":
            print("视频生成失败")
            return None
        else:
            # 如果任务还在处理中，等待一段时间后再次检查
            time.sleep(5)
            print("视频正在生成中，稍后再试...")

三、调试

打开已经创建好的函数库，点击调试。在调试中输入prompt参数：

哈尔滨的中央大街上，人声鼎沸，热闹非凡，灯光闪烁，远处天空上烟花绚丽多彩

等待几秒后，可以获得连接：

复制上面的https链接并打开浏览器进行查看，如下图所示。

更新时间

• 2025-02-02：创建。