wangxuming/NianAIGC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add admin accounts and image templates

Browse Source
This commit is contained in:
inman
2026-07-03 11:25:25 +08:00
parent d98e58adfa
commit c3ea9f0eb1
61 changed files with 7016 additions and 199 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/prompt/assembler.ts
View File

@@ -1,4 +1,5 @@
import { getTemplateById, type VideoTemplate } from "@/lib/content/video-templates";
import { extractMaterialPlaceholders } from "@/lib/prompt/material-placeholders";
export type StoryboardScene = {
id: string;
@@ -112,12 +113,8 @@ export function assemblePrompt(input: PromptAssemblyInput): PromptAssemblyResult
export function extractMaterialRequirements(prompt: string) {
const requirements = { image: 0, video: 0, audio: 0 };
for (const match of prompt.matchAll(/@(参考视频|图片|图|视频|音频)(\d*)/g)) {
const kind = match[1];
const index = match[2] ? Number(match[2]) : 1;
if (kind === "图片" || kind === "图") requirements.image = Math.max(requirements.image, index);
if (kind === "视频" || kind === "参考视频") requirements.video = Math.max(requirements.video, index);
if (kind === "音频") requirements.audio = Math.max(requirements.audio, index);
for (const placeholder of extractMaterialPlaceholders(prompt)) {
requirements[placeholder.type] = Math.max(requirements[placeholder.type], placeholder.index);
}
return requirements;
}