Update index.js

index.js

@@ -8,7 +8,7 @@ import { fileTypeFromBuffer } from "file-type";
 import { Client } from "@gradio/client";
 import { stablediff } from "./lib/diffusion.js";
 import { askOpenGPT4o } from "./lib/chatgpt.js";
-import { upscaler } from "./lib/enchance.js";
+import { processImageUpscaler } from "./lib/enchance.js";
 
 const app = express();
 app.set('json spaces', 4);
@@ -80,8 +80,10 @@ app.post('/api/img2img', async (req, res) => {
 app.post('/api/upscaler', async (req, res) => {
 	try {
 		console.log(req.body)
-		const { images, status } = req.body
+		const { images, filenames, mimetype, status } = req.body
         if (!images) return res.json({ success: false, message: 'Required an images!' })
+        if (!filenames) return res.json({ success: false, message: 'Required an filenames!' })
+        if (!mimetype) return res.json({ success: false, message: 'Required an mimetype!' })
         if (!status) return res.json({ success: false, message: 'Required an status text!' })
 
         if(status !== apikey) return res.json({ success: false, message: 'Invalid status!' })
@@ -91,14 +93,14 @@ app.post('/api/upscaler', async (req, res) => {
             url: images,
             responseType: "arraybuffer"
           })
-          const response = await upscaler(data_img.data.toString("base64"))
-          //const type_img = await fileTypeFromBuffer(response)
-          //res.setHeader('Content-Type', type_img.mime)
+          const response = await processImageUpscaler({buffer: data_img.data, method: "enchance", filenames, mimetype})
+          const type_img = await fileTypeFromBuffer(response)
+          res.setHeader('Content-Type', type_img.mime)
           res.json(response)
         } else if (images && typeof images == 'string' && isBase64(images)) {
-	   	  const response = await upscaler(images)
-          //const type_img = await fileTypeFromBuffer(response)
-          //res.setHeader('Content-Type', type_img.mime)
+	   	  const response = await processImageUpscaler({buffer: Buffer.from(data_img.data, "base64"), method: "enchance", filenames, mimetype})
+          const type_img = await fileTypeFromBuffer(response)
+          res.setHeader('Content-Type', type_img.mime)
           res.json(response)
         } else {
           res.json({