优化可灵视频算力配置

This commit is contained in:
RockYang 2025-03-04 16:59:20 +08:00
parent 3958e99e4d
commit 7b9a7475a9
7 changed files with 177 additions and 273 deletions

View File

@ -144,15 +144,15 @@ type SystemConfig struct {
OrderPayTimeout int `json:"order_pay_timeout,omitempty"` //订单支付超时时间 OrderPayTimeout int `json:"order_pay_timeout,omitempty"` //订单支付超时时间
VipInfoText string `json:"vip_info_text,omitempty"` // 会员页面充值说明 VipInfoText string `json:"vip_info_text,omitempty"` // 会员页面充值说明
MjPower int `json:"mj_power,omitempty"` // MJ 绘画消耗算力 MjPower int `json:"mj_power,omitempty"` // MJ 绘画消耗算力
MjActionPower int `json:"mj_action_power,omitempty"` // MJ 操作(放大,变换)消耗算力 MjActionPower int `json:"mj_action_power,omitempty"` // MJ 操作(放大,变换)消耗算力
SdPower int `json:"sd_power,omitempty"` // SD 绘画消耗算力 SdPower int `json:"sd_power,omitempty"` // SD 绘画消耗算力
DallPower int `json:"dall_power,omitempty"` // DALL-E-3 绘图消耗算力 DallPower int `json:"dall_power,omitempty"` // DALL-E-3 绘图消耗算力
SunoPower int `json:"suno_power,omitempty"` // Suno 生成歌曲消耗算力 SunoPower int `json:"suno_power,omitempty"` // Suno 生成歌曲消耗算力
LumaPower int `json:"luma_power,omitempty"` // Luma 生成视频消耗算力 LumaPower int `json:"luma_power,omitempty"` // Luma 生成视频消耗算力
KeLingPower int `json:"keling_power,omitempty"` // 可灵生成视频消耗算力 KeLingPowers map[string]int `json:"keling_powers,omitempty"` // 可灵生成视频消耗算力
AdvanceVoicePower int `json:"advance_voice_power,omitempty"` // 高级语音对话消耗算力 AdvanceVoicePower int `json:"advance_voice_power,omitempty"` // 高级语音对话消耗算力
PromptPower int `json:"prompt_power,omitempty"` // 生成提示词消耗算力 PromptPower int `json:"prompt_power,omitempty"` // 生成提示词消耗算力
WechatCardURL string `json:"wechat_card_url,omitempty"` // 微信客服地址 WechatCardURL string `json:"wechat_card_url,omitempty"` // 微信客服地址
@ -171,6 +171,5 @@ type SystemConfig struct {
EmailWhiteList []string `json:"email_white_list"` // 邮箱白名单列表 EmailWhiteList []string `json:"email_white_list"` // 邮箱白名单列表
TranslateModelId int `json:"translate_model_id"` // 用来做提示词翻译的大模型 id TranslateModelId int `json:"translate_model_id"` // 用来做提示词翻译的大模型 id
MaxFileSize int `json:"max_file_size"` // 最大文件大小,单位MB MaxFileSize int `json:"max_file_size"` // 最大文件大小,单位MB
License string `json:"license"` // License
} }

View File

@ -48,6 +48,7 @@ func (h *ConfigHandler) Update(c *gin.Context) {
} }
if err := c.ShouldBindJSON(&data); err != nil { if err := c.ShouldBindJSON(&data); err != nil {
logger.Errorf("Update config failed: %v", err)
resp.ERROR(c, types.InvalidArgs) resp.ERROR(c, types.InvalidArgs)
return return
} }
@ -131,14 +132,6 @@ func (h *ConfigHandler) Active(c *gin.Context) {
resp.ERROR(c, err.Error()) resp.ERROR(c, err.Error())
return return
} }
h.App.SysConfig.License = data.License
// 将 license 写入数据库
err = h.DB.Model(&model.Config{}).Where("marker", "system").UpdateColumn("config_json", utils.JsonEncode(h.App.SysConfig)).Error
if err != nil {
resp.ERROR(c, err.Error())
return
}
resp.SUCCESS(c) resp.SUCCESS(c)
@ -153,7 +146,6 @@ func (h *ConfigHandler) GetLicense(c *gin.Context) {
// FixData 修复数据 // FixData 修复数据
func (h *ConfigHandler) FixData(c *gin.Context) { func (h *ConfigHandler) FixData(c *gin.Context) {
resp.ERROR(c, "当前升级版本没有数据需要修正!") resp.ERROR(c, "当前升级版本没有数据需要修正!")
return
//var fixed bool //var fixed bool
//version := "data_fix_4.1.4" //version := "data_fix_4.1.4"
//err := h.levelDB.Get(version, &fixed) //err := h.levelDB.Get(version, &fixed)

View File

@ -281,6 +281,8 @@ func (h *VideoHandler) Remove(c *gin.Context) {
// 删除文件 // 删除文件
_ = h.uploader.GetUploadHandler().Delete(job.CoverURL) _ = h.uploader.GetUploadHandler().Delete(job.CoverURL)
_ = h.uploader.GetUploadHandler().Delete(job.VideoURL) _ = h.uploader.GetUploadHandler().Delete(job.VideoURL)
resp.SUCCESS(c)
} }
func (h *VideoHandler) Publish(c *gin.Context) { func (h *VideoHandler) Publish(c *gin.Context) {

View File

@ -13,9 +13,6 @@ import (
"geekai/core" "geekai/core"
"geekai/core/types" "geekai/core/types"
"geekai/store" "geekai/store"
"geekai/store/model"
"geekai/utils"
"gorm.io/gorm"
"strings" "strings"
"time" "time"
@ -25,27 +22,17 @@ import (
type LicenseService struct { type LicenseService struct {
config types.ApiConfig config types.ApiConfig
db *gorm.DB
levelDB *store.LevelDB levelDB *store.LevelDB
license *types.License license *types.License
urlWhiteList []string urlWhiteList []string
machineId string machineId string
} }
func NewLicenseService(server *core.AppServer, levelDB *store.LevelDB, db *gorm.DB) *LicenseService { func NewLicenseService(server *core.AppServer, levelDB *store.LevelDB) *LicenseService {
var license types.License var license types.License
var machineId string var machineId string
err := levelDB.Get(types.LicenseKey, &license) err := levelDB.Get(types.LicenseKey, &license)
logger.Infof("License: %+v", server.SysConfig) logger.Infof("License: %+v", server.SysConfig)
if err != nil {
var cfg model.Config
db.Where("marker", "system").First(&cfg)
var sysConfig types.SystemConfig
if err := utils.JsonDecode(cfg.Config, &sysConfig); err == nil {
license.Key = sysConfig.License
license.IsActive = true
}
}
info, err := host.Info() info, err := host.Info()
if err == nil { if err == nil {
machineId = info.HostID machineId = info.HostID

View File

@ -519,7 +519,7 @@ func (s *Service) KeLingCreate(task types.VideoTask) (KeLingRespVo, error) {
// 2. 构建API请求参数 // 2. 构建API请求参数
payload := map[string]interface{}{ payload := map[string]interface{}{
"model": params.Model, "model_name": params.Model,
"prompt": task.Prompt, "prompt": task.Prompt,
"negative_prompt": params.NegPrompt, "negative_prompt": params.NegPrompt,
"cfg_scale": params.CfgScale, "cfg_scale": params.CfgScale,

View File

@ -21,19 +21,11 @@
<el-row :gutter="10"> <el-row :gutter="10">
<el-col :span="8" v-for="item in rates" :key="item.value"> <el-col :span="8" v-for="item in rates" :key="item.value">
<div <div
class="flex-col items-center" class="flex-col items-center"
:class=" :class="item.value === params.aspect_ratio ? 'grid-content active' : 'grid-content'"
item.value === params.aspect_ratio @click="changeRate(item)"
? 'grid-content active'
: 'grid-content'
"
@click="changeRate(item)"
> >
<el-image <el-image class="icon proportion" :src="item.img" fit="cover"></el-image>
class="icon proportion"
:src="item.img"
fit="cover"
></el-image>
<div class="texts">{{ item.text }}</div> <div class="texts">{{ item.text }}</div>
</div> </div>
</el-col> </el-col>
@ -43,9 +35,8 @@
<!-- 模型选择 --> <!-- 模型选择 -->
<div class="param-line"> <div class="param-line">
<el-form-item label="模型选择"> <el-form-item label="模型选择">
<el-select v-model="params.model" placeholder="请选择模型"> <el-select v-model="params.model" placeholder="请选择模型" @change="updateModelPower">
<el-option label="默认模型" value="default" /> <el-option v-for="item in models" :key="item.value" :label="item.text" :value="item.value" />
<el-option label="动漫风格" value="anime" />
</el-select> </el-select>
</el-form-item> </el-form-item>
</div> </div>
@ -53,7 +44,7 @@
<!-- 视频时长 --> <!-- 视频时长 -->
<div class="param-line"> <div class="param-line">
<el-form-item label="视频时长"> <el-form-item label="视频时长">
<el-select v-model="params.duration" placeholder="请选择时长"> <el-select v-model="params.duration" placeholder="请选择时长" @change="updateModelPower">
<el-option label="5秒" value="5" /> <el-option label="5秒" value="5" />
<el-option label="10秒" value="10" /> <el-option label="10秒" value="10" />
</el-select> </el-select>
@ -63,7 +54,7 @@
<!-- 生成模式 --> <!-- 生成模式 -->
<div class="param-line"> <div class="param-line">
<el-form-item label="生成模式"> <el-form-item label="生成模式">
<el-select v-model="params.mode" placeholder="请选择模式"> <el-select v-model="params.mode" placeholder="请选择模式" @change="updateModelPower">
<el-option label="标准模式" value="std" /> <el-option label="标准模式" value="std" />
<el-option label="专业模式" value="pro" /> <el-option label="专业模式" value="pro" />
</el-select> </el-select>
@ -73,17 +64,12 @@
<!-- 创意程度 --> <!-- 创意程度 -->
<div class="param-line"> <div class="param-line">
<el-form-item label="创意程度"> <el-form-item label="创意程度">
<el-slider <el-slider v-model="params.cfg_scale" :min="0" :max="1" :step="0.1" />
v-model="params.cfg_scale"
:min="0"
:max="1"
:step="0.1"
/>
</el-form-item> </el-form-item>
</div> </div>
<!-- 运镜控制 --> <!-- 运镜控制 -->
<div class="param-line"> <div class="param-line" v-if="showCameraControl">
<div class="param-line pt"> <div class="param-line pt">
<span>运镜控制</span> <span>运镜控制</span>
<el-tooltip content="生成画面的运镜效果" placement="right"> <el-tooltip content="生成画面的运镜效果" placement="right">
@ -94,11 +80,8 @@
</div> </div>
<!-- 添加运镜类型选择 --> <!-- 添加运镜类型选择 -->
<el-form-item label="运镜类型"> <div class="param-line pt">
<el-select <el-select v-model="params.camera_control.type" placeholder="请选择运镜类型">
v-model="params.camera_control.type"
placeholder="请选择运镜类型"
>
<el-option label="请选择" value="" /> <el-option label="请选择" value="" />
<el-option label="简单运镜" value="simple" /> <el-option label="简单运镜" value="simple" />
<el-option label="下移拉远" value="down_back" /> <el-option label="下移拉远" value="down_back" />
@ -106,54 +89,27 @@
<el-option label="右旋推进" value="right_turn_forward" /> <el-option label="右旋推进" value="right_turn_forward" />
<el-option label="左旋推进" value="left_turn_forward" /> <el-option label="左旋推进" value="left_turn_forward" />
</el-select> </el-select>
</el-form-item> </div>
<!-- 仅在simple模式下显示详细配置 --> <!-- 仅在simple模式下显示详细配置 -->
<div <div class="camera-control mt-2" v-if="params.camera_control.type === 'simple'">
class="camera-control"
v-if="params.camera_control.type === 'simple'"
>
<el-form-item label="水平移动"> <el-form-item label="水平移动">
<el-slider <el-slider v-model="params.camera_control.config.horizontal" :min="-10" :max="10" />
v-model="params.camera_control.config.horizontal"
:min="-10"
:max="10"
/>
</el-form-item> </el-form-item>
<el-form-item label="垂直移动"> <el-form-item label="垂直移动">
<el-slider <el-slider v-model="params.camera_control.config.vertical" :min="-10" :max="10" />
v-model="params.camera_control.config.vertical"
:min="-10"
:max="10"
/>
</el-form-item> </el-form-item>
<el-form-item label="左右旋转"> <el-form-item label="左右旋转">
<el-slider <el-slider v-model="params.camera_control.config.pan" :min="-10" :max="10" />
v-model="params.camera_control.config.pan"
:min="-10"
:max="10"
/>
</el-form-item> </el-form-item>
<el-form-item label="上下旋转"> <el-form-item label="上下旋转">
<el-slider <el-slider v-model="params.camera_control.config.tilt" :min="-10" :max="10" />
v-model="params.camera_control.config.tilt"
:min="-10"
:max="10"
/>
</el-form-item> </el-form-item>
<el-form-item label="横向翻转"> <el-form-item label="横向翻转">
<el-slider <el-slider v-model="params.camera_control.config.roll" :min="-10" :max="10" />
v-model="params.camera_control.config.roll"
:min="-10"
:max="10"
/>
</el-form-item> </el-form-item>
<el-form-item label="镜头缩放"> <el-form-item label="镜头缩放">
<el-slider <el-slider v-model="params.camera_control.config.zoom" :min="-10" :max="10" />
v-model="params.camera_control.config.zoom"
:min="-10"
:max="10"
/>
</el-form-item> </el-form-item>
</div> </div>
</div> </div>
@ -165,19 +121,12 @@
<div class="main-content task-list-inner"> <div class="main-content task-list-inner">
<!-- 任务类型选择 --> <!-- 任务类型选择 -->
<div class="param-line"> <div class="param-line">
<el-tabs <el-tabs v-model="params.task_type" @tab-change="tabChange" class="title-tabs">
v-model="params.task_type"
@tab-change="tabChange"
class="title-tabs"
>
<el-tab-pane label="文生视频" name="text2video"> <el-tab-pane label="文生视频" name="text2video">
<div class="text">使用文字描述想要生成视频的内容</div> <div class="text">使用文字描述想要生成视频的内容</div>
</el-tab-pane> </el-tab-pane>
<el-tab-pane label="图生视频" name="image2video"> <el-tab-pane label="图生视频" name="image2video">
<div class="text"> <div class="text">以某张图片为底稿参考来创作视频生成类似风格或类型视频支持 PNG /JPG/JPEG 格式图片</div>
以某张图片为底稿参考来创作视频生成类似风格或类型视频支持 PNG
/JPG/JPEG
</div>
</el-tab-pane> </el-tab-pane>
</el-tabs> </el-tabs>
</div> </div>
@ -186,20 +135,14 @@
<div class="generation-area"> <div class="generation-area">
<div v-if="params.task_type === 'text2video'" class="text2video"> <div v-if="params.task_type === 'text2video'" class="text2video">
<el-input <el-input
v-model="params.prompt" v-model="params.prompt"
type="textarea" type="textarea"
maxlength="500" maxlength="500"
:autosize="{ minRows: 4, maxRows: 6 }" :autosize="{ minRows: 4, maxRows: 6 }"
placeholder="请在此输入视频提示词,您也可以点击下面的提示词助手生成视频提示词" placeholder="请在此输入视频提示词,您也可以点击下面的提示词助手生成视频提示词"
/> />
<el-row class="text-info"> <el-row class="text-info">
<el-button <el-button class="generate-btn" @click="generatePrompt" :loading="isGenerating" size="small" color="#5865f2">
class="generate-btn"
@click="generatePrompt"
:loading="isGenerating"
size="small"
color="#5865f2"
>
<i class="iconfont icon-chuangzuo"></i> <i class="iconfont icon-chuangzuo"></i>
生成专业视频提示词 生成专业视频提示词
</el-button> </el-button>
@ -210,35 +153,15 @@
<div class="image-upload img-inline"> <div class="image-upload img-inline">
<div class="upload-box img-uploader"> <div class="upload-box img-uploader">
<h4>起始帧</h4> <h4>起始帧</h4>
<el-upload <el-upload class="uploader img-uploader" :auto-upload="true" :show-file-list="false" :http-request="uploadStartImage" accept=".jpg,.png,.jpeg">
class="uploader img-uploader" <img v-if="params.image" :src="params.image" class="preview" />
:auto-upload="true"
:show-file-list="false"
:http-request="uploadStartImage"
accept=".jpg,.png,.jpeg"
>
<img
v-if="params.image"
:src="params.image"
class="preview"
/>
<el-icon v-else class="upload-icon"><Plus /></el-icon> <el-icon v-else class="upload-icon"><Plus /></el-icon>
</el-upload> </el-upload>
</div> </div>
<div class="upload-box img-uploader"> <div class="upload-box img-uploader">
<h4>结束帧</h4> <h4>结束帧</h4>
<el-upload <el-upload class="uploader" :auto-upload="true" :show-file-list="false" :http-request="uploadEndImage" accept=".jpg,.png,.jpeg">
class="uploader" <img v-if="params.image_tail" :src="params.image_tail" class="preview" />
:auto-upload="true"
:show-file-list="false"
:http-request="uploadEndImage"
accept=".jpg,.png,.jpeg"
>
<img
v-if="params.image_tail"
:src="params.image_tail"
class="preview"
/>
<el-icon v-else class="upload-icon"><Plus /></el-icon> <el-icon v-else class="upload-icon"><Plus /></el-icon>
</el-upload> </el-upload>
</div> </div>
@ -247,10 +170,7 @@
<div class="flex-row justify-between items-center"> <div class="flex-row justify-between items-center">
<div class="flex-row justify-start items-center"> <div class="flex-row justify-start items-center">
<span>提示词</span> <span>提示词</span>
<el-tooltip <el-tooltip content="输入你想要的内容,用逗号分割" placement="right">
content="输入你想要的内容,用逗号分割"
placement="right"
>
<el-icon> <el-icon>
<InfoFilled /> <InfoFilled />
</el-icon> </el-icon>
@ -259,12 +179,7 @@
</div> </div>
</div> </div>
<div class="param-line pt"> <div class="param-line pt">
<el-input <el-input v-model="params.prompt" type="textarea" :autosize="{ minRows: 4, maxRows: 6 }" placeholder="描述视频画面细节" />
v-model="params.prompt"
type="textarea"
:autosize="{ minRows: 4, maxRows: 6 }"
placeholder="描述视频画面细节"
/>
</div> </div>
</div> </div>
@ -273,10 +188,7 @@
<div class="flex-row justify-between items-center"> <div class="flex-row justify-between items-center">
<div class="flex-row justify-start items-center"> <div class="flex-row justify-start items-center">
<span>不希望出现的内容可选</span> <span>不希望出现的内容可选</span>
<el-tooltip <el-tooltip content="不想出现在图片上的元素(例如:树,建筑)" placement="right">
content="不想出现在图片上的元素(例如:树,建筑)"
placement="right"
>
<el-icon> <el-icon>
<InfoFilled /> <InfoFilled />
</el-icon> </el-icon>
@ -286,31 +198,26 @@
</div> </div>
<div class="param-line pt"> <div class="param-line pt">
<el-input <el-input
v-model="params.negative_prompt" v-model="params.negative_prompt"
type="textarea" type="textarea"
:autosize="{ minRows: 4, maxRows: 6 }" :autosize="{ minRows: 4, maxRows: 6 }"
placeholder="请在此输入你不希望出现在视频上的内容" placeholder="请在此输入你不希望出现在视频上的内容"
/> />
</div> </div>
<!-- 算力显示 --> <!-- 算力显示 -->
<el-row class="text-info"> <el-row class="text-info">
<el-text type="primary" <el-text type="primary"
>每次生成视频消耗 >每次生成视频消耗 <el-text type="warning">{{ powerCost }}算力;</el-text> </el-text
<el-text type="warning">{{ powerCost }}算力;</el-text> </el-text
>&nbsp;&nbsp; >&nbsp;&nbsp;
<el-text type="primary" <el-text type="primary"
>当前可用算力<el-text type="warning">{{ >当前可用算力<el-text type="warning">{{ availablePower }}</el-text></el-text
availablePower
}}</el-text></el-text
> >
</el-row> </el-row>
<!-- 生成按钮 --> <!-- 生成按钮 -->
<div class="submit-btn"> <div class="submit-btn">
<el-button type="primary" :dark="false" @click="generate" round <el-button type="primary" :dark="false" @click="generate" round>立即生成</el-button>
>立即生成</el-button
>
</div> </div>
</div> </div>
@ -333,67 +240,52 @@
<h2 class="record-title pt">创作记录</h2> <h2 class="record-title pt">创作记录</h2>
<!-- 已完成的任务 --> <!-- 已完成的任务 -->
<v3-waterfall <v3-waterfall
:key="waterfallKey" :key="waterfallKey"
:list="finishedTasks" :list="finishedTasks"
@scrollReachBottom="fetchTasks" @scrollReachBottom="fetchTasks"
:gap="20" :gap="20"
:bottomGap="20" :bottomGap="20"
:colWidth="300" :colWidth="300"
:distanceToScroll="100" :distanceToScroll="100"
:isLoading="loading" :isLoading="loading"
:isOver="isOver" :isOver="isOver"
class="task-waterfall" class="task-waterfall"
> >
<template #default="slotProp"> <template #default="slotProp">
<div <div
class="job-item-box" class="job-item-box"
:class="{ :class="{
processing: slotProp.item.progress < 100, processing: slotProp.item.progress < 100,
error: slotProp.item.progress === 101 error: slotProp.item.progress === 101,
}" }"
> >
<video <video
v-if="slotProp.item.progress === 100" v-if="slotProp.item.progress === 100"
class="preview" class="preview"
:src="slotProp.item.video_url" :src="slotProp.item.video_url"
@click="previewVideo(slotProp.item)" @click="previewVideo(slotProp.item)"
controls controls
></video> ></video>
<div v-else class="status-overlay"> <div v-else class="status-overlay">
<div <div v-if="slotProp.item.progress === 101" class="error-status">
v-if="slotProp.item.progress === 101"
class="error-status"
>
<el-icon><CloseBold /></el-icon> <el-icon><CloseBold /></el-icon>
任务失败 任务失败
</div> </div>
<div v-else class="processing-status"> <div v-else class="processing-status">
<el-progress <el-progress :percentage="slotProp.item.progress" :stroke-width="12" status="success" />
:percentage="slotProp.item.progress"
:stroke-width="12"
status="success"
/>
</div> </div>
</div> </div>
<div class="tools"> <div class="tools">
<el-button <el-button v-if="slotProp.item.progress === 100" @click="downloadVideo(slotProp.item)">
v-if="slotProp.item.progress === 100"
@click="downloadVideo(slotProp.item)"
>
<el-icon><Download /></el-icon> <el-icon><Download /></el-icon>
</el-button> </el-button>
<el-button type="danger" @click="deleteTask(slotProp.item)"> <el-button type="danger" @click="deleteTask(slotProp.item)">
<el-icon><Delete /></el-icon> <el-icon><Delete /></el-icon>
</el-button> </el-button>
<div class="show-prompt"> <div class="show-prompt">
<el-popover <el-popover placement="left" title="提示词" :width="240" trigger="hover">
placement="left"
title="提示词"
:width="240"
trigger="hover"
>
<template #reference> <template #reference>
<el-icon class="chromefilled"> <el-icon class="chromefilled">
<ChromeFilled /> <ChromeFilled />
@ -403,10 +295,7 @@
<template #default> <template #default>
<div class="mj-list-item-prompt"> <div class="mj-list-item-prompt">
<span>{{ slotProp.item.prompt }}</span> <span>{{ slotProp.item.prompt }}</span>
<el-icon <el-icon class="copy-prompt-mj" :data-clipboard-text="slotProp.item.prompt">
class="copy-prompt-mj"
:data-clipboard-text="slotProp.item.prompt"
>
<DocumentCopy /> <DocumentCopy />
</el-icon> </el-icon>
</div> </div>
@ -430,12 +319,7 @@
<!-- 视频预览对话框 --> <!-- 视频预览对话框 -->
<el-dialog v-model="previewVisible" title="视频预览" width="80%"> <el-dialog v-model="previewVisible" title="视频预览" width="80%">
<video <video v-if="currentVideo" :src="currentVideo" controls style="width: 100%"></video>
v-if="currentVideo"
:src="currentVideo"
controls
style="width: 100%"
></video>
</el-dialog> </el-dialog>
</div> </div>
</template> </template>
@ -443,31 +327,18 @@
<script setup> <script setup>
import TaskList from "@/components/TaskList.vue"; import TaskList from "@/components/TaskList.vue";
import { ref, reactive, onMounted, onUnmounted, watch } from "vue"; import { ref, reactive, onMounted, onUnmounted, watch } from "vue";
import { import { Plus, Delete, InfoFilled, ChromeFilled, DocumentCopy, Download, CloseBold } from "@element-plus/icons-vue";
Plus,
Delete,
InfoFilled,
ChromeFilled,
DocumentCopy,
Download,
CloseBold
} from "@element-plus/icons-vue";
import { httpGet, httpPost, httpDownload } from "@/utils/http"; import { httpGet, httpPost, httpDownload } from "@/utils/http";
import { ElMessage, ElMessageBox } from "element-plus"; import { ElMessage, ElMessageBox } from "element-plus";
import { checkSession } from "@/store/cache"; import { checkSession, getSystemInfo } from "@/store/cache";
import { import { closeLoading, showLoading, showMessageError, showMessageOK } from "@/utils/dialog";
closeLoading,
showLoading,
showMessageError,
showMessageOK
} from "@/utils/dialog";
import { replaceImg } from "@/utils/libs"; import { replaceImg } from "@/utils/libs";
// //
const params = reactive({ const params = reactive({
task_type: "text2video", task_type: "text2video",
model: "default", model: "kling-v1-6",
prompt: "", prompt: "",
negative_prompt: "", negative_prompt: "",
cfg_scale: 0.7, cfg_scale: 0.7,
@ -482,28 +353,42 @@ const params = reactive({
pan: 0, pan: 0,
tilt: 0, tilt: 0,
roll: 0, roll: 0,
zoom: 0 zoom: 0,
} },
}, },
image: "", image: "",
image_tail: "" image_tail: "",
}); });
const rates = [ const rates = [
{css: "square", value: "1:1", text: "1:1", img: "/images/mj/rate_1_1.png"}, { css: "square", value: "1:1", text: "1:1", img: "/images/mj/rate_1_1.png" },
{ {
css: "size16-9", css: "size16-9",
value: "16:9", value: "16:9",
text: "16:9", text: "16:9",
img: "/images/mj/rate_16_9.png" img: "/images/mj/rate_16_9.png",
}, },
{ {
css: "size9-16", css: "size9-16",
value: "9:16", value: "9:16",
text: "9:16", text: "9:16",
img: "/images/mj/rate_9_16.png" img: "/images/mj/rate_9_16.png",
} },
]; ];
const models = ref([
{
text: "可灵 1.6",
value: "kling-v1-6",
},
{
text: "可灵 1.5",
value: "kling-v1-5",
},
{
text: "可灵 1.0",
value: "kling-v1",
},
]);
// //
const changeRate = (item) => { const changeRate = (item) => {
@ -525,6 +410,8 @@ const previewVisible = ref(false);
const currentVideo = ref(""); const currentVideo = ref("");
const isOver = ref(false); const isOver = ref(false);
const pullTask = ref(true); const pullTask = ref(true);
const showCameraControl = ref(false); //
const keLingPowers = ref({});
// //
@ -547,6 +434,12 @@ const uploadStartImage = async (file) => {
} }
}; };
//
const updateModelPower = () => {
showCameraControl.value = params.model === "kling-v1-5" && params.mode === "pro";
powerCost.value = keLingPowers.value[`${params.model}_${params.mode}_${params.duration}`];
};
const uploadEndImage = async (file) => { const uploadEndImage = async (file) => {
const formData = new FormData(); const formData = new FormData();
formData.append("file", file.file); formData.append("file", file.file);
@ -565,7 +458,7 @@ const generatePrompt = async () => {
} }
isGenerating.value = true; isGenerating.value = true;
try { try {
const res = await httpPost("/api/prompt/video", {prompt: params.prompt}); const res = await httpPost("/api/prompt/video", { prompt: params.prompt });
params.prompt = res.data; params.prompt = res.data;
} catch (e) { } catch (e) {
showMessageError("生成失败: " + e.message); showMessageError("生成失败: " + e.message);
@ -593,8 +486,9 @@ const generate = async () => {
try { try {
await httpPost("/api/video/keling/create", params); await httpPost("/api/video/keling/create", params);
showMessageOK("任务创建成功"); showMessageOK("任务创建成功");
// runningTasks.value.push({
await fetchTasks(); progress: 0,
});
// //
pullTask.value = true; pullTask.value = true;
} catch (e) { } catch (e) {
@ -617,31 +511,24 @@ const fetchTasks = async () => {
page: currentPage.value, page: currentPage.value,
page_size: pageSize.value, page_size: pageSize.value,
type: "keling", type: "keling",
task_type: taskFilter.value === "all" ? "" : taskFilter.value task_type: taskFilter.value === "all" ? "" : taskFilter.value,
}); });
// //
const data = res.data || {}; const data = res.data || {};
const newRunning = data.items.filter( runningTasks.value = data.items.filter((task) => task.progress < 100 && task.progress !== 101);
(task) => task.progress < 100 && task.progress !== 101
);
runningTasks.value = [...runningTasks.value, ...newRunning];
// //
if (newRunning.length === 0) { if (runningTasks.value.length === 0) {
pullTask.value = false; pullTask.value = false;
} }
const newFinished = data.items.filter((task) => task.progress >= 100); finishedTasks.value = data.items.filter((task) => task.progress >= 100);
finishedTasks.value = [...finishedTasks.value, ...newFinished];
// // // //
waterfallKey.value = Date.now(); waterfallKey.value = Date.now();
total.value = data.total; total.value = data.total;
const shouldLoadNextPage = const shouldLoadNextPage = runningTasks.value.length > 0 || (runningTasks.value.length === 0 && finishedTasks.value.length < total.value);
runningTasks.value.length > 0 ||
(runningTasks.value.length === 0 &&
finishedTasks.value.length < total.value);
if (shouldLoadNextPage) { if (shouldLoadNextPage) {
currentPage.value++; currentPage.value++;
@ -654,7 +541,7 @@ const fetchTasks = async () => {
console.log("[Pagination] isOver:", isOver.value, { console.log("[Pagination] isOver:", isOver.value, {
running: runningTasks.value.length, running: runningTasks.value.length,
finished: finishedTasks.value.length, finished: finishedTasks.value.length,
total: total.value total: total.value,
}); });
waterfallKey.value = Date.now(); waterfallKey.value = Date.now();
@ -694,8 +581,10 @@ const downloadVideo = async (task) => {
const deleteTask = async (task) => { const deleteTask = async (task) => {
try { try {
await ElMessageBox.confirm("确定要删除该任务吗?"); await ElMessageBox.confirm("确定要删除该任务吗?");
await httpGet("/api/video/remove", {id: task.id}); await httpGet("/api/video/remove", { id: task.id });
showMessageOK("删除成功"); showMessageOK("删除成功");
isOver.value = false;
loading.value = false;
await fetchTasks(); await fetchTasks();
} catch (e) { } catch (e) {
if (e !== "cancel") { if (e !== "cancel") {
@ -707,22 +596,27 @@ const deleteTask = async (task) => {
// //
onMounted(async () => { onMounted(async () => {
checkSession() checkSession()
.then(async () => { .then(async (u) => {
isLogin.value = true; isLogin.value = true;
console.log("mounted-isLogin-可以继续", isLogin.value); availablePower.value = u.power;
console.log("mounted-isLogin-可以继续", isLogin.value);
await fetchTasks(); await fetchTasks();
setInterval(() => { setInterval(() => {
if (pullTask.value) { if (pullTask.value) {
fetchTasks(); fetchTasks();
} }
}, 5000) }, 5000);
}) })
.catch(() => { .catch(() => {});
});
}); //
onUnmounted(() => { getSystemInfo().then((res) => {
keLingPowers.value = res.data.keling_powers;
updateModelPower();
});
}); });
onUnmounted(() => {});
// //
watch([runningTasks, finishedTasks], () => { watch([runningTasks, finishedTasks], () => {
if (checkAllCompleted()) { if (checkAllCompleted()) {

View File

@ -243,8 +243,24 @@
<el-form-item label="Luma 算力" prop="luma_power"> <el-form-item label="Luma 算力" prop="luma_power">
<el-input v-model.number="system['luma_power']" placeholder="使用 Luma 生成一段视频消耗算力" /> <el-input v-model.number="system['luma_power']" placeholder="使用 Luma 生成一段视频消耗算力" />
</el-form-item> </el-form-item>
<el-form-item label="可灵算力" prop="keling_power"> <el-form-item>
<el-input v-model.number="system['keling_power']" placeholder="使用快手可灵生成一段视频消耗算力" /> <template #label>
<div class="label-title">
可灵算力
<el-tooltip effect="dark" content="可灵每个模型价格不一样具体请参考https://api.geekai.pro/models" raw-content placement="right">
<el-icon>
<InfoFilled />
</el-icon>
</el-tooltip>
</div>
</template>
<el-row :gutter="20" v-if="system['keling_powers']">
<el-col :span="6" v-for="[key] in Object.entries(system['keling_powers'])" :key="key">
<el-form-item :label="key" label-position="left">
<el-input v-model.number="system['keling_powers'][key]" size="small" />
</el-form-item>
</el-col>
</el-row>
</el-form-item> </el-form-item>
<el-form-item> <el-form-item>
<template #label> <template #label>
@ -412,6 +428,20 @@ onMounted(() => {
httpGet("/api/admin/config/get?key=system") httpGet("/api/admin/config/get?key=system")
.then((res) => { .then((res) => {
system.value = res.data; system.value = res.data;
system.value.keling_powers = system.value.keling_powers || {
"kling-v1-6_std_5": 240,
"kling-v1-6_std_10": 480,
"kling-v1-6_pro_5": 420,
"kling-v1-6_pro_10": 840,
"kling-v1-5_std_5": 240,
"kling-v1-5_std_10": 480,
"kling-v1-5_pro_5": 420,
"kling-v1-5_pro_10": 840,
"kling-v1_std_5": 120,
"kling-v1_std_10": 240,
"kling-v1_pro_5": 420,
"kling-v1_pro_10": 840,
};
configBak.value = copyObj(system.value); configBak.value = copyObj(system.value);
}) })
.catch((e) => { .catch((e) => {