Update index.js

index.js

@@ -13,6 +13,7 @@ const nsfwjs = require('nsfwjs')
 const jpegjs = require('jpeg-js')
 const Jimp = require('jimp')
 const fileType = require("file-type")
+const Waifu2X = require('@ibaraki-douji/waifu2x')
 //const Stress = require('./lib/ddos.js');
 //const { BingChat } = (await import("bing-chat")).default
 const { acytoo, chatgpt_4 } = require("./lib/chatgpt.js")
@@ -224,6 +225,40 @@ app.post('/api/upscaler2', async (req, res) => {
 		res.json({ error: true, message: e === '[object Object]' ? 'Internal Server Error' : e })
 	}
 })
+app.post('/api/upscaler2', async (req, res) => {
+	try {
+		console.log(req.body)
+		const { images, status } = req.body
+        if (!images) return res.json({ success: false, message: 'Required an images!' })
+        if (!status) return res.json({ success: false, message: 'Required an status text!' })
+
+        if(status !== apikey) return res.json({ success: false, message: 'Invalid status!' })
+        if (/^(https?|http):\/\//i.test(images)) {
+          const data_img = await axios.request({
+            method: "GET",
+            url: images,
+            responseType: "arraybuffer"
+          })
+          const response = await upscaleImageV2(data_img.data)
+          const type_img = await fileType.fromBuffer(response)
+          res.setHeader('Content-Type', type_img.mime)
+          res.send(response)
+        } else if (images && typeof images == 'string' && isBase64(images)) {
+	   	  const response = await upscaleImageV2(Buffer.from(images, "base64"))
+          const type_img = await fileType.fromBuffer(response)
+          res.setHeader('Content-Type', type_img.mime)
+          res.send(response)
+        } else {
+          res.json({
+            success: false, message: 'No url or base64 detected!!' 
+          })
+        }
+	} catch (e) {
+		console.log(e)
+		e = String(e)
+		res.json({ error: true, message: e === '[object Object]' ? 'Internal Server Error' : e })
+	}
+})
 app.post('/api/toanime2', async (req, res) => {
 	try {
 		console.log(req.body)
@@ -516,6 +551,27 @@ async function upscaleImage(imageBuffer) {
     });
 }
 
+async function upscaleImageV2(imageBuffer) {
+  return new Promise(async (resolve, reject) => {
+    try {
+      const upscale = await Waifu2X.upscale(imageBuffer, null, {
+        noise: 0,
+        scale: 2,
+        gpu: -1,
+        ramLimit: 500,
+        outputAsBuffer: true
+      });
+
+      upscale.finishedPromise().then(() => {
+        const outputBuffer = upscale.endBuffer;
+        resolve(outputBuffer); // ArrayBuffer
+      });
+    } catch(e) {
+      reject(e.message);
+    }
+  });
+}
+
 async function processImageUpscaler(images, denoise, format, type) {
     return new Promise(async (resolve, reject) => {
       try {