Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
"use server" | |
import { RenderRequest, RenderedScene } from "@/types/general" | |
// note: there is no / at the end in the variable | |
// so we have to add it ourselves if needed | |
const apiUrl = `${process.env.VIDEOCHAIN_API_URL || ""}` | |
const apiKey = `${process.env.VIDEOCHAIN_API_KEY || ""}` | |
export async function newRender({ | |
prompt, | |
negativePrompt, | |
nbFrames, | |
nbFPS, | |
nbSteps, | |
width, | |
height, | |
turbo, | |
shouldRenewCache, | |
seed, | |
}: { | |
prompt: string | |
negativePrompt: string | |
nbFrames: number | |
nbFPS: number | |
nbSteps: number | |
width: number | |
height: number | |
turbo: boolean | |
shouldRenewCache: boolean | |
seed?: number | |
}) { | |
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..`) | |
} | |
const cacheKey = `render/${JSON.stringify({ prompt })}` | |
// return await Gorgon.get(cacheKey, async () => { | |
let defaulResult: RenderedScene = { | |
renderId: "", | |
status: "error", | |
assetUrl: "", | |
durationInMs: 0, | |
maskUrl: "", | |
error: "failed to fetch the data", | |
alt: "", | |
segments: [] | |
} | |
// console.log("fetch api:", `${apiUrl}/render`) | |
try { | |
// console.log(`calling POST ${apiUrl}/render with seed ${seed} and prompt: ${prompt}`) | |
const res = await fetch(`${apiUrl}/render`, { | |
method: "POST", | |
headers: { | |
Accept: "application/json", | |
"Content-Type": "application/json", | |
Authorization: `Bearer ${apiKey}`, | |
}, | |
body: JSON.stringify({ | |
prompt, | |
negativePrompt, | |
// nbFrames: 8 and nbSteps: 15 --> ~10 sec generation | |
nbFrames, // when nbFrames is 1, we will only generate static images | |
nbFPS, | |
nbSteps, // 20 = fast, 30 = better, 50 = best | |
width, | |
height, | |
seed, | |
actionnables: [], | |
segmentation: "disabled", // one day we will remove this param, to make it automatic | |
upscalingFactor: 1, // let's disable upscaling right now | |
turbo, // always use turbo mode (it's for images only anyway) | |
// also what could be done iw that we could use the width and height to control this | |
cache: "ignore", // shouldRenewCache ? "renew" : "use", | |
// this is the "blocking mode", which help us having to poll the server | |
wait: true, | |
} as Partial<RenderRequest>), | |
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 | |
} | |
} | |
export async function getRender(renderId: string) { | |
if (!renderId) { | |
console.error(`cannot call the rendering API without a renderId, aborting..`) | |
throw new Error(`cannot call the rendering API without a renderId, aborting..`) | |
} | |
let defaulResult: RenderedScene = { | |
renderId: "", | |
status: "error", | |
assetUrl: "", | |
durationInMs: 0, | |
maskUrl: "", | |
error: "failed to fetch the data", | |
alt: "", | |
segments: [] | |
} | |
try { | |
// console.log(`calling GET ${apiUrl}/render with renderId: ${renderId}`) | |
const res = await fetch(`${apiUrl}/render/${renderId}`, { | |
method: "GET", | |
headers: { | |
Accept: "application/json", | |
"Content-Type": "application/json", | |
Authorization: `Bearer ${apiKey}`, | |
}, | |
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 | |
} | |
} |