回退生成视频提示词
This commit is contained in:
parent
a78c7a2b43
commit
6e3a0c6d2d
@ -50,29 +50,6 @@ export default router.post(
|
|||||||
}
|
}
|
||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
const assetsIds = images.map((i) => {
|
|
||||||
if (i._type == "storyboard") {
|
|
||||||
return i.associateAssetsIds;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
const setIds = new Set(assetsIds.filter(Boolean).flat());
|
|
||||||
|
|
||||||
const assetsData = await u
|
|
||||||
.db("o_assets")
|
|
||||||
.leftJoin("o_image", "o_assets.imageId", "o_image.id")
|
|
||||||
.where("o_assets.id", "in", Array.from(setIds))
|
|
||||||
.select("o_assets.id", "o_image.filePath", "o_assets.name", "o_assets.type");
|
|
||||||
await Promise.all(
|
|
||||||
assetsData.map(async (i) => {
|
|
||||||
if (i.filePath) {
|
|
||||||
i.filePath = await u.oss.getFileUrl(i.filePath);
|
|
||||||
}
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
const assetsRecord: Record<number, any> = {};
|
|
||||||
assetsData.forEach((i) => {
|
|
||||||
assetsRecord[i.id] = i;
|
|
||||||
});
|
|
||||||
|
|
||||||
// 拆分 assets 和 storyboard
|
// 拆分 assets 和 storyboard
|
||||||
const assets: any[] = [];
|
const assets: any[] = [];
|
||||||
@ -85,7 +62,7 @@ export default router.post(
|
|||||||
type: item.type,
|
type: item.type,
|
||||||
name: item.name,
|
name: item.name,
|
||||||
});
|
});
|
||||||
if (item._type === "storyboard") {
|
if (item._type === "storyboard")
|
||||||
storyboard.push({
|
storyboard.push({
|
||||||
videoDesc: item.videoDesc,
|
videoDesc: item.videoDesc,
|
||||||
prompt: item.prompt,
|
prompt: item.prompt,
|
||||||
@ -94,30 +71,10 @@ export default router.post(
|
|||||||
associateAssetsIds: item.associateAssetsIds,
|
associateAssetsIds: item.associateAssetsIds,
|
||||||
shouldGenerateImage: item.shouldGenerateImage,
|
shouldGenerateImage: item.shouldGenerateImage,
|
||||||
});
|
});
|
||||||
if (item.associateAssetsIds && item.associateAssetsIds.length) {
|
|
||||||
item.associateAssetsIds.forEach((i: number) => {
|
|
||||||
const data = assetsRecord[i];
|
|
||||||
const extingIndex = assets.find((sub) => sub.id == data.id);
|
|
||||||
if (data && !extingIndex) {
|
|
||||||
assets.push({
|
|
||||||
id: data.id,
|
|
||||||
type: data.type,
|
|
||||||
name: data.name,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
const [id, modelData] = model.split(":");
|
const [id, modelData] = model.split(":");
|
||||||
const projectData = await u.db("o_project").select("*").where({ id: projectId }).first();
|
const projectData = await u.db("o_project").select("*").where({ id: projectId }).first();
|
||||||
const videoPrompt = await u.db("o_prompt").where("type", "videoPromptGeneration").first();
|
const videoPrompt = await u.db("o_prompt").where("type", "videoPromptGeneration").first();
|
||||||
let videoPromptGeneration = "" as string | undefined;
|
|
||||||
if (videoPrompt && videoPrompt.useData) {
|
|
||||||
videoPromptGeneration = videoPrompt.useData;
|
|
||||||
} else {
|
|
||||||
videoPromptGeneration = videoPrompt?.data ?? undefined;
|
|
||||||
}
|
|
||||||
const artStyle = projectData?.artStyle || "无";
|
const artStyle = projectData?.artStyle || "无";
|
||||||
const visualManual = u.getArtPrompt(artStyle, "art_skills", "art_storyboard_video");
|
const visualManual = u.getArtPrompt(artStyle, "art_skills", "art_storyboard_video");
|
||||||
const content = `
|
const content = `
|
||||||
@ -137,7 +94,7 @@ export default router.post(
|
|||||||
|
|
||||||
try {
|
try {
|
||||||
const { text } = await u.Ai.Text("universalAi").invoke({
|
const { text } = await u.Ai.Text("universalAi").invoke({
|
||||||
system: videoPromptGeneration,
|
system: videoPrompt?.data!,
|
||||||
messages: [
|
messages: [
|
||||||
{
|
{
|
||||||
role: "assistant",
|
role: "assistant",
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user