File size: 3,033 Bytes
660842c 4905b6b 660842c 4905b6b 660842c 4905b6b 660842c 4905b6b 660842c |
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 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 |
"use server"
import { generateSeed } from "@/lib/generateSeed"
import { UpscalingParams } from "@/types"
import { addBase64HeaderToPng } from "./addBase64HeaderToPng"
const gradioApi = `https://jbilcke-hf-image-upscaling-api.hf.space`
const microserviceApiKey = `${process.env.MICROSERVICE_API_SECRET_TOKEN || ""}`
export async function upscale({
imageAsBase64,
prompt,
negativePrompt,
scaleFactor,
nbSteps,
seed,
}: UpscalingParams): Promise<string> {
const addedPrompt = [
"clean",
"high-resolution",
"8k",
"best quality",
"masterpiece",
"crisp",
"sharp",
"intricate details"
].join(", ")
const negPrompt = [
negativePrompt,
"pixelated",
"pixels",
"noise",
"blur",
"motion blur",
"lowres",
"oversmooth",
"longbody",
"bad anatomy",
"bad hands",
"missing fingers",
"extra digit",
"fewer digits",
"cropped",
"worst quality",
"low quality",
"artificial",
"unrealistic",
"watermark",
"trademark",
"error",
"mistake"
].join(", ")
const conditioningScale = 1.4
const classifierFreeGuidance = 9.5
// remember: a space needs to be public for the classic fetch() to work
const res = await fetch(gradioApi + (gradioApi.endsWith("/") ? "" : "/") + "api/predict", {
method: "POST",
headers: {
"Content-Type": "application/json",
// Authorization: `Bearer ${hfApiToken}`,
},
body: JSON.stringify({
fn_index: 0, // <- is it 0 or 1?
data: [
microserviceApiKey,
imageAsBase64, // blob in 'parameter_5' Image component
prompt, // string in 'Prompt' Textbox component
addedPrompt, // string in 'Added Prompt' Textbox component
negPrompt, // string in 'Negative Prompt' Textbox component
nbSteps, // number (numeric value between 10 and 50) in 'Denoise Steps' Slider component
scaleFactor, // number (numeric value between 1 and 4) in 'Upsample Scale' Slider component
conditioningScale, // number (numeric value between 0.5 and 1.5) in 'Conditioning Scale' Slider component
classifierFreeGuidance, // number (numeric value between 0.1 and 10.0) in 'Classier-free Guidance' Slider component
seed || generateSeed(), // number (numeric value between -1 and 2147483647) in 'Seed' Slider component
],
}),
cache: "no-store",
// we can also use this (see https://vercel.com/blog/vercel-cache-api-nextjs-cache)
// next: { revalidate: 1 }
})
const { data } = await res.json()
// console.log("data:", data)
// Recommendation: handle errors
if (res.status !== 200 || !Array.isArray(data)) {
// This will activate the closest `error.js` Error Boundary
throw new Error(`Failed to fetch data (status: ${res.status})`)
}
// console.log("data:", data.slice(0, 50))
const base64Content = (data?.[0] || "") as string
if (!base64Content) {
throw new Error(`invalid response (no content)`)
}
return addBase64HeaderToPng(base64Content)
} |