Split prompts into pre/prompt/post and tidy the AIGen video form.
Keep wrappers off the library mid-prompt so Use as input and Extend restore the same split, send stills to xAIGen as hidden inputs, and stop the two-column library from growing with the list. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -0,0 +1,48 @@
|
||||
import { passwordAuthEnabled, privateInstanceId } from '~/server/utils/localAuth'
|
||||
import { saveStill } from '~/server/utils/library'
|
||||
|
||||
function ingestOwner() {
|
||||
if (passwordAuthEnabled()) return privateInstanceId()
|
||||
return 'local'
|
||||
}
|
||||
|
||||
function assertIngestToken(event: Parameters<typeof getHeader>[0]) {
|
||||
const expected = String(useRuntimeConfig().backupIngestToken || process.env.BACKUP_INGEST_TOKEN || '').trim()
|
||||
if (!expected) {
|
||||
throw createError({ statusCode: 503, statusMessage: 'Backup ingest is not configured' })
|
||||
}
|
||||
const header = String(getHeader(event, 'authorization') || '')
|
||||
if (header !== `Bearer ${expected}`) {
|
||||
throw createError({ statusCode: 401, statusMessage: 'Unauthorized' })
|
||||
}
|
||||
}
|
||||
|
||||
export default defineEventHandler(async (event) => {
|
||||
assertIngestToken(event)
|
||||
const form = await readMultipartFormData(event)
|
||||
if (!form?.length) throw createError({ statusCode: 400, statusMessage: 'Expected an image upload' })
|
||||
const fields: Record<string, string> = {}
|
||||
let image: { filename: string; data: Buffer } | null = null
|
||||
for (const part of form) {
|
||||
if (part.name === 'image' && part.filename && part.data?.length) {
|
||||
image = { filename: part.filename, data: part.data }
|
||||
} else if (part.name && part.data) {
|
||||
fields[part.name] = part.data.toString('utf8')
|
||||
}
|
||||
}
|
||||
if (!image) throw createError({ statusCode: 400, statusMessage: 'An image is required' })
|
||||
const owner = ingestOwner()
|
||||
const still = await saveStill({
|
||||
ownerKey: owner,
|
||||
folderId: 'inbox',
|
||||
filename: image.filename,
|
||||
data: image.data,
|
||||
width: Number(fields.width || 0),
|
||||
height: Number(fields.height || 0),
|
||||
hideInput: true,
|
||||
role: 'input',
|
||||
name: (fields.name || '').trim().slice(0, 80) || undefined
|
||||
})
|
||||
if (!still) throw createError({ statusCode: 400, statusMessage: 'Backup studio already has that still' })
|
||||
return { ok: true, id: still.id }
|
||||
})
|
||||
@@ -0,0 +1,38 @@
|
||||
import { readFileSync } from 'node:fs'
|
||||
import { isXaigenStudio } from '~/utils/videoModels'
|
||||
import { assertLibraryOwner, getStill, stillPath } from '~/server/utils/library'
|
||||
|
||||
export default defineEventHandler(async (event) => {
|
||||
if (isXaigenStudio()) {
|
||||
throw createError({ statusCode: 404, statusMessage: 'Not found' })
|
||||
}
|
||||
const config = useRuntimeConfig()
|
||||
const url = String(config.backupStudioUrl || process.env.BACKUP_STUDIO_URL || '').replace(/\/$/, '')
|
||||
const token = String(config.backupStudioToken || process.env.BACKUP_STUDIO_TOKEN || '').trim()
|
||||
if (!url || !token) {
|
||||
throw createError({ statusCode: 503, statusMessage: 'Backup studio is not configured' })
|
||||
}
|
||||
const { owner } = assertLibraryOwner(event)
|
||||
const id = String(getRouterParam(event, 'id') || '')
|
||||
const still = getStill(owner, id)
|
||||
const path = stillPath(owner, still.id)
|
||||
const data = readFileSync(path)
|
||||
const body = new FormData()
|
||||
body.append('image', new Blob([Uint8Array.from(data)]), still.filename || 'still.png')
|
||||
body.append('name', still.name || still.filename || 'still')
|
||||
body.append('width', String(still.width || 0))
|
||||
body.append('height', String(still.height || 0))
|
||||
const res = await fetch(`${url}/api/library/backup-ingest`, {
|
||||
method: 'POST',
|
||||
headers: { Authorization: `Bearer ${token}` },
|
||||
body
|
||||
})
|
||||
if (!res.ok) {
|
||||
const message = await res.text().catch(() => '')
|
||||
throw createError({
|
||||
statusCode: res.status === 401 ? 502 : res.status,
|
||||
statusMessage: message || 'Backup studio refused the still'
|
||||
})
|
||||
}
|
||||
return { ok: true }
|
||||
})
|
||||
Reference in New Issue
Block a user