Update index.js

index.js

@@ -8,6 +8,7 @@ const axios = require("axios")
 const FormData = require("form-data")
 const ytdl = require('ytdl-core')
 const tfjs = require('@tensorflow/tfjs-node')
+const Upscaler = require('upscaler')
 const nsfwjs = require('nsfwjs')
 const jpegjs = require('jpeg-js')
 const Jimp = require('jimp')
@@ -169,7 +170,7 @@ app.post('/api/upscaler', async (req, res) => {
             url: images,
             responseType: "arraybuffer"
           })
-          const response = await processImage(data_img.data, denoise, scale, format, type)
+          const response = await processImageUpscaler(data_img.data, denoise, scale, format, type)
           const type_img = await fileType.fromBuffer(response)
           res.setHeader('Content-Type', type_img.mime)
           res.send(response)
@@ -189,6 +190,40 @@ app.post('/api/upscaler', 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 } = 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 upscaleImage(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 upscaleImage(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)
@@ -455,62 +490,84 @@ function toPDF(urls) {
 	})
 }
 
-async function processImage(image, denoise, scale, format, type) {
-  return new Promise(async (resolve, reject) => {
-    try {
-      let type_img = await fileType.fromBuffer(image)
-      let random_numbers = Math.floor(Math.random() * 1000);
-
-      const formData = new FormData();
-      formData.append("denoise", denoise);
-      formData.append("scale", scale);
-      formData.append("format", format);
-      formData.append("type", type);
-      formData.append("file", image, {
-        filename:
-          "images_" + random_numbers.toString().padStart(3, "0") + "." + type_img.ext,
-        contentType: type_img.mime,
-      });
-
-      const response = await axios.request({
-        method: "POST",
-        url: "https://api.alcaamado.es/ns-api-waifu2x/v1/convert",
-        data: formData,
-        debug: true,
-        headers: {
-          Authority: "api.alcaamado.es",
-          Accept: "application/json",
-          Referer: "https://waifu2x.pro/",
-          Origin: "https://waifu2x.pro",
-          "User-Agent":
-            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
-        },
-      });
-      
-      const images = await axios.request({
-        method: "GET",
-        url:
-          "https://api.alcaamado.es/api/v2/waifu2x/get?hash=" +
-          response.data.hash +
-          "&type=" +
-          format,
-        headers: {
-          Accept: "image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8",
-          "Content-Type": "image/jpg",
-          Referer: "https://waifu2x.pro/",
-          "User-Agent":
-            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
-        },
-        responseType: "arraybuffer",
-      });
-
-      // Mengonversi arraybuffer ke Buffer
-      //const buffer = Buffer.from(images.data);
-      resolve(images.data);
-    } catch (error) {
-      reject(error);
-    }
+async function upscaleImage(images) {
+  const upscaler = new Upscaler();
+  const image = tfjs.node.decodeImage(images, 3);
+
+  const tensor = await upscaler.upscale(image, {
+    scale: 4, // upscale by a factor of 4
+    sharpen: true, // apply sharpening
+    denoise: true, // apply denoising
+    contrast: 1.5, // increase contrast by 50%
+    colorEnhance: true, // enhance colors
   });
+
+  const upscaledTensor = await tf.node.encodePng(tensor);
+  const arrayBuffer = await upscaledTensor.arrayBuffer();
+
+  // dispose the tensors!
+  image.dispose();
+  tensor.dispose();
+  upscaledTensor.dispose();
+
+  return arrayBuffer;
+}
+
+async function processImageUpscaler(images, denoise, format, type) {
+    return new Promise(async (resolve, reject) => {
+      try {
+        const formData = new FormData();
+        formData.append("denoise", denoise);
+        formData.append("scale", "true");
+        formData.append("format", format);
+        formData.append("type", type);
+        formData.append("file", images, {
+          filename:
+            "images_" + "downlaod.jpg",
+          contentType: "image/jpeg",
+        });
+
+        // Convert FormData to Buffer
+       // const bufferFormData = await formData.getBuffer();
+
+        const response = await axios.request({
+          method: "POST",
+          url: "https://api.alcaamado.es/ns-api-waifu2x/v1/convert",
+          data: formData,
+          debug: true,
+          headers: {
+            Authority: "api.alcaamado.es",
+            Accept: "application/json",
+            Referer: "https://waifu2x.pro/",
+            Origin: "https://waifu2x.pro",
+            "User-Agent":
+              "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+            "Content-Type": `multipart/form-data; boundary=${formData._boundary}`,
+          },
+        });
+
+        const images = await axios.request({
+          method: "GET",
+          url:
+            "https://api.alcaamado.es/api/v2/waifu2x/get?hash=" +
+            response.data.hash +
+            "&type=" +
+            format,
+          headers: {
+            Accept: "image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8",
+            "Content-Type": "image/jpg",
+            Referer: "https://waifu2x.pro/",
+            "User-Agent":
+              "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+          },
+          responseType: "arraybuffer",
+        });
+
+        resolve(images.data)
+      } catch (error) {
+        reject(error);
+      }
+    });
 }
 
 async function processImageAnime(inputBuffer) {