File size: 2,676 Bytes
4f64da5 637dd5c f4af987 a438bb5 f4af987 4f64da5 637dd5c a438bb5 637dd5c a438bb5 637dd5c a438bb5 637dd5c 4f64da5 637dd5c |
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 |
"use server"
import Gorgon from "@gorgonjs/gorgon"
import { RenderedScene } from "./types"
import { Engine, EngineType } from "./engines"
// note: there is no / at the end in the variable
// so we have to add it ourselves if needed
const apiUrl = process.env.RENDERING_ENGINE_API
const cacheDurationInSec = 30 * 60 // 30 minutes
export async function render({
prompt,
actionnables = [],
engine,
}: {
prompt: string
actionnables: string[]
engine: Engine
}) {
if (!prompt) {
console.error(`cannot call the rendering API without a prompt, aborting..`)
throw new Error(`cannot call the rendering API without a prompt, aborting..`)
}
if (!Array.isArray(actionnables) || !actionnables.length) {
console.error(`cannot call the rendering API without actionnables, aborting..`)
throw new Error(`cannot call the rendering API without actionnables, aborting..`)
}
const nbFrames = engine.type === "video" ? 8 : 1
const cacheKey = `render/${JSON.stringify({ prompt, actionnables, nbFrames, type: engine.type })}`
return await Gorgon.get(cacheKey, async () => {
let defaulResult: RenderedScene = {
assetUrl: "",
maskBase64: "",
error: "",
segments: []
}
try {
console.log(`calling ${apiUrl}/render with prompt: ${prompt}`)
const res = await fetch(`${apiUrl}/render`, {
method: "POST",
headers: {
Accept: "application/json",
"Content-Type": "application/json",
// Authorization: `Bearer ${process.env.VC_SECRET_ACCESS_TOKEN}`,
},
body: JSON.stringify({
prompt,
// nbFrames: 8 and nbSteps: 15 --> ~10 sec generation
nbFrames, // when nbFrames is 1, we will only generate static images
nbSteps: 20,
actionnables,
segmentation: "firstframe", // one day we will remove this param, to make it automatic
}),
cache: 'no-store',
// we can also use this (see https://vercel.com/blog/vercel-cache-api-nextjs-cache)
// next: { revalidate: 1 }
})
// console.log("res:", res)
// The return value is *not* serialized
// You can return Date, Map, Set, etc.
// Recommendation: handle errors
if (res.status !== 200) {
// This will activate the closest `error.js` Error Boundary
throw new Error('Failed to fetch data')
}
const response = (await res.json()) as RenderedScene
// console.log("response:", response)
return response
} catch (err) {
console.error(err)
Gorgon.clear(cacheKey)
return defaulResult
}
}, cacheDurationInSec * 1000)
} |