Update index.js

index.js

@@ -10,6 +10,7 @@ const ytdl = require('ytdl-core')
 const tfjs = require('@tensorflow/tfjs-node')
 const nsfwjs = require('nsfwjs')
 const jpegjs = require('jpeg-js')
+const jimp = require('jimp')
 const fileType = require("file-type")
 //const Stress = require('./lib/ddos.js');
 //const { BingChat } = (await import("bing-chat")).default
@@ -188,6 +189,41 @@ app.post('/api/upscaler', async (req, res) => {
 		res.json({ error: true, message: e === '[object Object]' ? 'Internal Server Error' : e })
 	}
 })
+app.post('/api/upscaler', async (req, res) => {
+	try {
+		console.log(req.body)
+		const { status 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 imageBase64 = Buffer.from(data_img.data, 'binary').toString('base64');
+          const response = await processImageAnime(imageBase64);
+          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 processImageAnime(images)
+          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/nsfw-check', async (req, res) => {
 	try {
 		console.log(req.body)
@@ -477,6 +513,42 @@ async function processImage(image, denoise, scale, format, type) {
   });
 }
 
+async function processImageAnime(inputBuffer) {
+    try {
+       // const base64String = Buffer.from(inputBuffer, 'binary').toString('base64');
+        const apiResponse = await axios.post('https://www.drawever.com/api/photo-to-anime', {
+            data: `data:image/png;base64,${inputBuffer}`,
+        }, {
+            headers: {
+                'Content-Type': 'application/json',
+            },
+        });
+
+        const link = 'https://www.drawever.com' + (apiResponse.data.urls[1] || apiResponse.data.urls[0]);
+        const {
+            data: imageBuffer
+        } = await axios.get(link, {
+            responseType: 'arraybuffer'
+        });
+
+        const image = await Jimp.read(imageBuffer);
+        const blackBackground = new Jimp(image.bitmap.width, 50, 0x000000FF);
+        const font = await Jimp.loadFont(Jimp.FONT_SANS_16_WHITE);
+        blackBackground.print(font, 10, 10, "SadTeams", blackBackground.bitmap.width - 20);
+        image.composite(blackBackground, 0, image.bitmap.height - blackBackground.bitmap.height, {
+            mode: Jimp.BLEND_SOURCE_OVER,
+            opacityDest: 0.5,
+            opacitySource: 1
+        });
+
+        const outputBuffer = await image.getBufferAsync(Jimp.MIME_JPEG);
+        return outputBuffer;
+    } catch (err) {
+        console.error(err);
+        throw err;
+    }
+}
+
 async function check_nsfw(buffer) {
   let _model = await load_model()
   const convert = async (img) => {