Spaces:
Sleeping
Sleeping
File size: 2,334 Bytes
3d4392e 8101ed0 3d4392e 8101ed0 3d4392e 8101ed0 3d4392e 8101ed0 3d4392e 8101ed0 3d4392e 8101ed0 3d4392e 8101ed0 3d4392e 8101ed0 3d4392e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 |
import { NextResponse, NextRequest } from "next/server"
import queryString from "query-string"
import { newRender, getRender } from "../../providers/videochain/renderWithVideoChain"
import { generateSeed } from "@/lib/utils/generateSeed"
import { sleep } from "@/lib/utils/sleep"
import { getNegativePrompt, getPositivePrompt } from "../../utils/imagePrompts"
import { getContentType } from "@/lib/data/getContentType"
export async function GET(req: NextRequest) {
const qs = queryString.parseUrl(req.url || "")
const query = (qs || {}).query
let prompt = ""
try {
prompt = decodeURIComponent(query?.p?.toString() || "").trim()
} catch (err) {}
if (!prompt) {
return NextResponse.json({ error: 'no prompt provided' }, { status: 400 });
}
let format = "binary"
try {
const f = decodeURIComponent(query?.f?.toString() || "").trim()
if (f === "json" || f === "binary") { format = f }
} catch (err) {}
// console.log("calling await newRender")
prompt = getPositivePrompt(prompt)
const negativePrompt = getNegativePrompt()
let render = await newRender({
prompt,
negativePrompt,
nbFrames: 1,
nbFPS: 1,
nbSteps: 8,
width: 1024,
height: 576,
turbo: true,
shouldRenewCache: true,
seed: generateSeed()
})
let attempts = 10
while (attempts-- > 0) {
if (render.status === "completed") {
if (format === "json") {
return NextResponse.json(render, {
status: 200,
statusText: "OK",
})
} else {
const contentType = getContentType(render.assetUrl)
const base64String = render.assetUrl.split(";base64,").pop() || ""
const data = Buffer.from(base64String, "base64")
const headers = new Headers()
headers.set('Content-Type', contentType)
return new NextResponse(data, {
status: 200,
statusText: "OK",
headers
})
}
}
if (render.status === "error") {
return NextResponse.json(render, {
status: 200,
statusText: "OK",
})
}
await sleep(2000) // minimum wait time
// console.log("asking getRender")
render = await getRender(render.renderId)
}
return NextResponse.json({
"error": "failed to call VideoChain (timeout expired)"
}, {
status: 500,
})
}
|