add exception interceptors

src/index.mts

@@ -39,6 +39,14 @@ const port = 7860
 
 let isRendering = false
 
+process.on('unhandledRejection', (reason: string, p: Promise<any>) => {
+  console.error('Unhandled Rejection at:', p, 'reason:', reason);
+})
+
+process.on('uncaughtException', (error: Error) => {
+  console.error(`Caught exception: ${error}\n` + `Exception origin: ${error.stack}`);
+})
+
 // fix this error: "PayloadTooLargeError: request entity too large"
 // there are multiple version because.. yeah well, it's Express!
 // app.use(bodyParser.json({limit: '50mb'}));

src/providers/image-generation/generateImageLCMGradio.mts

@@ -3,6 +3,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)
@@ -89,5 +90,9 @@ export async function generateImageLCMAsBase64(options: {
   if (!result?.length) {
     throw new Error(`the returned image was empty`)
   }
+
+  // const finalImage = await convertToWebp(result)
+  // return finalImage
+
   return result
 }

src/utils/image/convertToWebp.mts

@@ -0,0 +1,33 @@
+import sharp from "sharp"
+
+export async function convertToWebp(imgBase64: string): Promise<string> {
+  // Convert base64 to buffer
+  const tmpBuffer = Buffer.from(imgBase64, 'base64')
+
+  // Resize the buffer to the target size
+  const newBuffer = await sharp(tmpBuffer)
+    .webp({
+      // for options please see https://sharp.pixelplumbing.com/api-output#webp
+
+      // preset: "photo",
+
+      // effort: 3,
+
+      // for a PNG-like quality
+      // lossless: true,
+
+      // by default it is quality 80
+      quality: 90,
+
+      // nearLossless: true,
+
+      // use high quality chroma subsampling
+      smartSubsample: true,
+     })
+      .toBuffer()
+
+  // Convert the buffer back to base64
+  const newImageBase64 = newBuffer.toString('base64')
+
+  return `data:image/webp;base64,${newImageBase64}`
+}