Update index.js

index.js

@@ -31,6 +31,8 @@ import { ApiKey } from './models.js';
 import { Readable } from "stream";
 import { randomBytes } from "crypto";
 import { AkenoaiJs } from "akenoaijs";
+import tf from '@tensorflow/tfjs-node';
+import nsfwjs from 'nsfwjs';
 
 import { 
   CheckMilWare,
@@ -131,11 +133,66 @@ const AllJsonReques = {
   },
 }
 
+let model;
+
+nsfwjs.load("https://raw.githubusercontent.com/infinitered/nsfwjs/master/models/inception_v3/model.json", { size: 299 })
+  .then((loadedModel) => {
+    model = loadedModel;
+    console.log('Model loaded');
+  })
+  .catch((error) => {
+    console.error('Error loading model:', error);
+  });
 
 app.get("/api/v1/json/all", async (req, res) => {
   res.json(AllJsonReques);
 });
 
+app.get('api/v1/nsfw', async (req, res) => {
+  try {
+    const { url } = req.query;
+    if (!url) {
+      return res.status(400).json({ message: 'Invalid URL.' });
+    }
+
+    const response = await axios.get(url, { responseType: 'arraybuffer' });
+    const imageBuffer = Buffer.from(response.data);
+
+    let imageTensor;
+
+    if (url.endsWith('.gif')) {
+      const jpgBuffer = await sharp(imageBuffer)
+        .resize({ width: 299, height: 299 })
+        .toFormat('jpeg')
+        .toBuffer();
+
+      imageTensor = tf.node.decodeImage(jpgBuffer, 3);
+    } else {
+      imageTensor = tf.node.decodeImage(imageBuffer, 3);
+    }
+
+    const predictions = await model.classify(imageTensor);
+    imageTensor.dispose();
+
+    const formattedPredictions = predictions.reduce((acc, { className, probability }) => {
+      acc[className] = probability;
+      return acc;
+    }, {});
+
+    res.json(formattedPredictions);
+  } catch (error) {
+    console.error('Error processing image:', error);
+    
+    if (error.response) {
+      return res.status(error.response.status).json({ message: 'Error fetching image from URL.', details: error.message });
+    } else if (error.code === 'ERR_INVALID_URL') {
+      return res.status(400).json({ message: 'Invalid image URL.', details: error.message });
+    } else {
+      return res.status(500).json({ message: 'Internal server error.', details: error.message });
+    }
+  }
+});
+
 app.get('/broadcast-user', async (req, res) => {
     const dbClient = new Database("AkenoXJs");
     const collection = dbClient.collection("api_keys");