Update index.js

index.js

@@ -9,9 +9,11 @@ const FormData = require("form-data")
 const tfjs = require('@tensorflow/tfjs-node')
 const nsfwjs = require('nsfwjs')
 const jpegjs = require('jpeg-js')
+const fileType = require("file-type")
 //const { BingChat } = (await import("bing-chat")).default
 
 const { insta_iwaId, insta_iwaIdUrl, insta_iwa, insta_iwaTag } = require("./lib/instagram.js")
+const { webpToJpg } = require("./lib/convertFormat.js")
 const apikey = "@SadTeam77"
 
 const app = express()
@@ -180,11 +182,23 @@ app.post('/api/nsfw-check', async (req, res) => {
             result: response
           })
         } else if (images && typeof images == 'string' && isBase64(images)) {
-	   	  const response = await check_nsfw(Buffer.from(images, "base64"))
-          res.json({
-            status: "ok",
-            result: response
-          })
+	   	  const img = Buffer.from(images, "base64")
+          const type = fileType.fileTypeFromBuffer(img)
+          if (type.ext == "jpg") {
+            let response = await check_nsfw(img)
+            res.json({
+              status: "ok",
+              result: response
+            })
+          }
+          if (type.ext == "webp") {
+            let converting = await webToJpg(img)
+            let response = await check_nsfw(converting)
+            res.json({
+              status: "ok",
+              result: response
+            })
+          }
         } else {
           res.json({
             success: false, message: 'No url or base64 detected!!'