Update index.js

index.js

@@ -120,15 +120,25 @@ app.post('/api/toanime', async (req, res) => {
 app.post('/api/waifu2x', async (req, res) => {
 	try {
 		console.log(req.body)
-		const { url, format, status } = req.body
-        if (!url) return res.json({ success: false, message: 'Required an url!' })
+		const { images, format, status } = req.body
+        if (!images) return res.json({ success: false, message: 'Required an images!' })
         if (!format) return res.json({ success: false, message: 'Required an format size!' })
         if (!status) return res.json({ success: false, message: 'Required an status text!' })
 
         if(status !== apikey) return res.json({ success: false, message: 'Invalid status!' })
-		const response = await waifu2x(url, format)
+        if (!images && typeof images !== 'string' && !isBase64(images)) {
+          const data_img = await axios.request({
+            method: "GET",
+            url: images
+          })
+          const response = await waifu2x(data_img, format)
+          res.setHeader('Content-Type', 'image/jpg')
+          res.send(response)
+        } else {
+		const response = await waifu2x(Buffer.from(images, "base64", format)
         res.setHeader('Content-Type', 'image/jpg')
         res.send(response)
+        }
 	} catch (e) {
 		console.log(e)
 		e = String(e)
@@ -218,9 +228,9 @@ async function acytoo(text, model) {
     })
   return res.data
 }
-async function waifu2x(urls, formats) {
+async function waifu2x(image, formats) {
     // data
-  let img = await axios.get(urls, { responseType: "arraybuffer"})
+  //let img = await axios.get(urls, { responseType: "arraybuffer"})
   let random_numbers = Math.floor(Math.random() * 1000)
   let format
     if(formats == "Medium") {
@@ -234,7 +244,7 @@ async function waifu2x(urls, formats) {
     const formData = new FormData()
     formData.append("denoise", format)
     formData.append("scale", "true")
-    formData.append("file", img.data, {
+    formData.append("file", image, {
         filename: "images_" + random_numbers.toString().padStart(3, '0') + ".jpg",
         contentType: "image/jpeg"
     })