344 lines
8.7 KiB
TypeScript
344 lines
8.7 KiB
TypeScript
import { BookTaskDetailModel } from '@/define/db/model/bookTaskDetail'
|
||
import {
|
||
BookBackTaskStatus,
|
||
BookBackTaskType,
|
||
BookTaskStatus,
|
||
BookType,
|
||
TaskExecuteType,
|
||
BookRepalceDataType
|
||
} from '../../enum/bookEnum'
|
||
import { MJAction } from '../../enum/bookEnum'
|
||
import { BookTaskDetail } from './bookTaskDetail'
|
||
import { ImageCategory } from '@/define/data/imageData'
|
||
import { ImageGenerateMode } from '@/define/data/mjData'
|
||
|
||
declare namespace Book {
|
||
//#region 小说相关
|
||
|
||
/**
|
||
* 小说数据类型
|
||
*/
|
||
type SelectBook = {
|
||
id?: string
|
||
no?: number
|
||
name?: string
|
||
bookFolderPath?: string
|
||
type?: BookType
|
||
oldVideoPath?: string
|
||
srtPath?: string
|
||
audioPath?: string
|
||
draftDepend?: string // 草稿依赖
|
||
imageFolder?: string
|
||
draftSrtStyle?: string | null // 草稿字幕样式
|
||
backgroundMusic?: string | null // 背景音乐ID
|
||
friendlyReminder?: string | null // 友情提示
|
||
subtitlePosition?: string
|
||
watermarkPosition?: string
|
||
updateTime?: Date
|
||
createTime?: Date
|
||
version?: string
|
||
imageStyle?: string[] | null // 软件内置的样式
|
||
autoAnalyzeCharacter?: string | null // 自动分析角色设置
|
||
customizeImageStyle?: string[] | null // 自定义的样式
|
||
videoConfig?: string | null // 合成视频设置
|
||
prefixPrompt?: string | null // 前缀
|
||
suffixPrompt?: string | null // 后缀
|
||
}
|
||
|
||
/**
|
||
* 查询小说的条件
|
||
*/
|
||
type QueryBookCondition = {
|
||
id?: string
|
||
name?: string
|
||
page?: number
|
||
pageSize?: number
|
||
type?: BookType
|
||
}
|
||
|
||
/**
|
||
* 查询小说的返回类型
|
||
*/
|
||
type QueryBookConditionResponse = {
|
||
resultBooks: SelectBook[]
|
||
bookLength: number
|
||
}
|
||
|
||
//#endregion
|
||
|
||
//#region 小说任务相关
|
||
|
||
/**
|
||
* 小说批次任务类型
|
||
*/
|
||
type SelectBookTask = {
|
||
no?: number
|
||
id?: string
|
||
bookId?: string
|
||
name?: string
|
||
generateVideoPath?: string
|
||
srtPath?: string
|
||
audioPath?: string
|
||
draftDepend?: string
|
||
draftSrtStyle?: string | null // 草稿字幕样式
|
||
backgroundMusic?: string | null // 背景音乐ID
|
||
friendlyReminder?: string | null // 友情提示
|
||
imageFolder?: string
|
||
customizeImageStyle?: string[]
|
||
cacheImageList?: string[]
|
||
prefixPrompt?: string
|
||
suffixPrompt?: string
|
||
styleList?: BookStyle[] | DefineBookStyle[] // 样式列表,
|
||
status?: BookTaskStatus
|
||
errorMsg?: string
|
||
version?: string
|
||
imageCategory?: ImageCategory // 出图方式
|
||
imageStyle?: string[] | null // 软件内置的样式
|
||
autoAnalyzeCharacter?: string | null // 自动分析角色设置
|
||
customizeImageStyle?: string[] | null // 自定义的样式
|
||
videoConfig?: string | null // 合成视频设置
|
||
prefixPrompt?: string | null // 前缀
|
||
suffixPrompt?: string | null // 后缀
|
||
isAuto?: boolean // 是否标记全自动
|
||
subImageFolder?: string[] | null // 子图片文件夹地址，多个
|
||
openVideoGenerate?: boolean // 是否开启视频生成
|
||
createTime?: Date
|
||
updateTime?: Date
|
||
}
|
||
|
||
/**
|
||
* 查询小说批次任务的条件
|
||
*/
|
||
type QueryBookTaskCondition = {
|
||
id?: string
|
||
no?: number
|
||
name?: string
|
||
bookId?: string
|
||
bookTaskId?: string
|
||
page?: number
|
||
pageSize?: number
|
||
}
|
||
|
||
/**
|
||
* 查询小说人物的返回数据类型
|
||
*/
|
||
type QueryBookTaskConditionResponse = {
|
||
bookTasks: SelectBookTask[]
|
||
bookTaskLength: number
|
||
}
|
||
|
||
//#endregion
|
||
|
||
//#region 小说任务详细数据相关
|
||
|
||
type SelectBookTaskDetail = {
|
||
id?: string
|
||
no?: number
|
||
name?: string
|
||
bookId?: string
|
||
bookTaskId?: string
|
||
videoPath?: string // 视频地址
|
||
generateVideoPath?: string // 生成的视频地址
|
||
audioPath?: string // 音频地址
|
||
draftDepend?: string // 草稿依赖
|
||
word?: string // 文案
|
||
oldImage?: string // 旧图片（用于SD的图生图）
|
||
afterGpt?: string // GPT生成的文案
|
||
startTime?: number // 开始时间
|
||
endTime?: number // 结束时间
|
||
timeLimit?: string // 事件实现（0 -- 3000）
|
||
subValue?: BookTaskDetail.CopywritingSubValue[] | string // 包含的字幕数据
|
||
characterTags?: string[] // 角色标签
|
||
sceneTags?: string[] // 场景标签
|
||
styleTags?: string[] // 风格标签
|
||
promptSort?: string // 提示词排序
|
||
gptPrompt?: string // GPT提示词
|
||
mjMessage?: MJMessage // MJ消息
|
||
videoMessage?: BookTaskDetail.VideoMessage // 视频消息
|
||
outImagePath?: string // 输出图片地址
|
||
subImagePath?: string[] // 子图片地址
|
||
imageLock?: boolean // 图片锁
|
||
reversePrompt?: ReversePrompt[] // 反推的提示词数据
|
||
prompt?: string // 提示
|
||
adetailer?: boolean // 是否开启修脸
|
||
sdConifg?: SDConfig // SD配置
|
||
subtitlePosition?: string // 字幕位置
|
||
status?: BookTaskStatus
|
||
createTime?: Date
|
||
updateTime?: Date
|
||
}
|
||
|
||
type QueryBookTaskDetailCondition = {
|
||
id?: string
|
||
name?: string
|
||
bookId?: string
|
||
bookTaskId?: string
|
||
page?: number
|
||
pageSize?: number
|
||
}
|
||
|
||
//#endregion
|
||
|
||
// 添加批次任务
|
||
type AddBookTask = {
|
||
copyBookTask: boolean // 是否复制旧的小说任务
|
||
count: number // 批次数量
|
||
prefixPrompt?: string // 前缀提示
|
||
suffixPrompt?: string // 后缀提示词
|
||
selectBookTask?: string // 选择的旧的小说任务
|
||
selectTaskDataCategory?: string[] // 选择复制的数据类型
|
||
selectBookId: string // 选择的小说ID
|
||
}
|
||
|
||
// 返回的MJ消息
|
||
type MJMessage = {
|
||
id?: string
|
||
mjApiUrl?: string
|
||
progress: number
|
||
category: ImageGenerateMode
|
||
imageClick?: string // 图片点击(显示的小的)
|
||
imageShow?: string // 图片实际的地址
|
||
messageId?: string // 消息ID（可以是MJ中的，也可以是API中的）
|
||
action?: MJAction // 动作（生图，反推之类）
|
||
status: string // 状态
|
||
message?: string // 消息
|
||
}
|
||
|
||
type WebuiConfig = {
|
||
sampler_name: string // 采样器名称
|
||
negative_prompt: string // 负面提示
|
||
batch_size: number // 批次大小
|
||
steps: number // 步数
|
||
cfg_scale: number // 提示词相关性
|
||
denoising_strength: number // 降噪强度
|
||
width: number // 宽度
|
||
height: number // 高度
|
||
seed: number // 种子
|
||
init_images: string // 初始图片(垫图的图片地址)
|
||
id: string
|
||
}
|
||
|
||
type SDConfig = {
|
||
checkpoints: string // 大模型
|
||
api: string // api地址
|
||
model: string // 生图方式
|
||
webuiConfig: WebuiConfig
|
||
id: string
|
||
}
|
||
|
||
type ReversePrompt = {
|
||
id?: string
|
||
bookTaskDetailId?: string
|
||
prompt?: string
|
||
promptCN?: string
|
||
isSelect?: boolean
|
||
}
|
||
|
||
// 查询小说任务的
|
||
type QueryBookBackTaskCondition = {
|
||
id: string
|
||
bookId: string
|
||
bookTaskId: string
|
||
name: string
|
||
type: string
|
||
status: string
|
||
executeType: string
|
||
count: number
|
||
}
|
||
|
||
// 更新小说任务状态
|
||
type UpdateBookTaskListStatus = {
|
||
id: string
|
||
status: BookBackTaskStatus
|
||
errorMessage?: string
|
||
startTime?: number
|
||
endTime?: number
|
||
}
|
||
|
||
// 小说人物
|
||
type BookTask = {
|
||
id: string
|
||
bookId: string
|
||
bookTaskId: string
|
||
bookTaskDetailId: string
|
||
name: string // 任务名称，小说名+批次名+分镜名
|
||
type: BookBackTaskType
|
||
status: BookBackTaskStatus
|
||
errorMessage?: string
|
||
executeType: TaskExecuteType // 任务执行类型，手动还是自动
|
||
createTime: Date
|
||
updateTime: Date
|
||
startTime?: number
|
||
endTime?: number
|
||
}
|
||
|
||
// 获取小说文案的参数
|
||
type GetVideoFrameTextParams = {
|
||
id: string
|
||
type: SubtitleSavePositionType
|
||
videoPath: string
|
||
subtitlePosition?: string
|
||
}
|
||
|
||
// 小说风格
|
||
type BookStyle = {
|
||
id?: string
|
||
label?: string
|
||
key?: string
|
||
value?: string
|
||
children?: string
|
||
type?: string
|
||
prompt?: string
|
||
image_url?: string
|
||
cref_cw?: number
|
||
lora?: string
|
||
chinese_prompt?: string
|
||
lora_weight?: number
|
||
show_image?: string
|
||
isShow?: boolean
|
||
}
|
||
|
||
type DefineBookStyle = {
|
||
chinese_style?: string
|
||
english_style?: string
|
||
id?: string
|
||
image?: string
|
||
}
|
||
|
||
/**
|
||
* 裁剪小视频的返回数据类型
|
||
*/
|
||
type BookFrameShortClip = {
|
||
startTime: number
|
||
endTime: number
|
||
videoPath: string
|
||
duration: number
|
||
}
|
||
|
||
type DeleteBookTaskDetailCondition = {
|
||
id?: string
|
||
bookId?: string
|
||
bookTaskId?: string
|
||
name?: string
|
||
}
|
||
|
||
//#region 替换小说数据
|
||
|
||
type BookReplaceData = {
|
||
before: string
|
||
after: string
|
||
replaceAll?: boolean
|
||
type: BookRepalceDataType
|
||
}
|
||
|
||
type ReplaceDataRes = {
|
||
bookTaskDetailId: string
|
||
newData: string | BookTaskDetail.CopywritingSubValue[]
|
||
type?: 'reversePrompt' | 'gptPrompt' | 'afterGpt' | 'prompt' | 'subValue'
|
||
reversePromptType?: 'prompt' | 'promptCN'
|
||
reversePromptId?: string
|
||
}
|
||
|
||
//#endregion
|
||
}
|