add conversion to webp

src/providers/image-generation/generateImageLCMGradio.mts

@@ -81,7 +81,7 @@ export async function generateImageLCMAsBase64(options: {
     seed, // number (numeric value between 0 and 2147483647) in 'Seed' Slider component		
     width, // number (numeric value between 256 and 1024) in 'Width' Slider component		
     height, // number (numeric value between 256 and 1024) in 'Height' Slider component		
-    0.0, // can be disabled for LCM-LORA-SSD-1B
+    0.0, // can be disabled for LCM SDXL
     nbSteps, // number (numeric value between 2 and 8) in 'Number of inference steps for base' Slider component			
     secretToken
   ])) as any
@@ -91,8 +91,11 @@ export async function generateImageLCMAsBase64(options: {
     throw new Error(`the returned image was empty`)
   }
 
-  // const finalImage = await convertToWebp(result)
-  // return finalImage
-
-  return result
+  try {
+    const finalImage = await convertToWebp(result)
+    return finalImage
+  } catch (err) {
+    // console.log("err:", err)
+    throw new Error(err)
+  }
 }

src/providers/image-generation/generateImageSDXLGradio.mts

@@ -2,6 +2,7 @@ import { client } from "@gradio/client"
 
 import { generateSeed } from "../../utils/misc/generateSeed.mts"
 import { getValidNumber } from "../../utils/validators/getValidNumber.mts"
+import { convertToWebp } from "../../utils/image/convertToWebp.mts"
 
 // TODO add a system to mark failed instances as "unavailable" for a couple of minutes
 // console.log("process.env:", process.env)
@@ -93,5 +94,12 @@ export async function generateImageSDXLAsBase64(options: {
   if (!result?.length) {
     throw new Error(`the returned image was empty`)
   }
-  return result
+
+  try {
+    const finalImage = await convertToWebp(result)
+    return finalImage
+  } catch (err) {
+    // console.log("err:", err)
+    throw new Error(err)
+  }
 }

src/utils/image/convertToWebp.mts

@@ -1,8 +1,16 @@
 import sharp from "sharp"
 
-export async function convertToWebp(imgBase64: string): Promise<string> {
+export async function convertToWebp(imgBase64: string = ""): Promise<string> {
+
+  const base64WithoutHeader = imgBase64.split(";base64,")[1] || ""
+
+  if (!base64WithoutHeader) {
+    const slice = `${imgBase64 || ""}`.slice(0, 50)
+    throw new Error(`couldn't process input image "${slice}..."`)
+  }
+
   // Convert base64 to buffer
-  const tmpBuffer = Buffer.from(imgBase64, 'base64')
+  const tmpBuffer = Buffer.from(base64WithoutHeader, 'base64')
 
   // Resize the buffer to the target size
   const newBuffer = await sharp(tmpBuffer)