xiao/commands/analyze/nsfw.js

const Command = require('../../structures/Command');
const request = require('node-superfetch');
const { stripIndents } = require('common-tags');
const { isImageNSFW } = require('../../util/Util');

module.exports = class NsfwCommand extends Command {
	constructor(client) {
		super(client, {
			name: 'nsfw',
			aliases: ['nsfw-image', 'nsfw-img', 'img-nsfw', 'image-nsfw'],
			group: 'analyze',
			memberName: 'nsfw',
			description: 'Determines if an image is NSFW.',
			throttling: {
				usages: 2,
				duration: 30
			},
			args: [
				{
					key: 'image',
					prompt: 'What image would you like to test?',
					type: 'image-or-avatar'
				}
			]
		});
	}

	async run(msg, { image }) {
		try {
			const { body } = await request.get(image);
			const predictions = await isImageNSFW(this.client.nsfwModel, body, false);
			const formatted = predictions.map(result => `${Math.round(result.probability * 100)}% ${result.className}`);
			return msg.reply(stripIndents`
				**This image gives the following results:**
				${formatted.join('\n')}
			`);
		} catch (err) {
			return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
		}
	}
};