arthur/xiao
1
0
Fork 0
mirror of https://github.com/arthur-pbty/xiao.git synced 2026-06-03 23:36:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

Test screenshot images in tensorflow

Browse Source
This commit is contained in:
Dragon Fire
2021-03-22 18:39:34 -04:00
parent bbe2702397
commit 17c25fe040
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
commands/other/screenshot.js
+5
View File
@@ -1,6 +1,7 @@
const Command = require('../../structures/Command');
const request = require('node-superfetch');
const { URL } = require('url');
const { isImageNSFW } = require('../../util/Util');
module.exports = class ScreenshotCommand extends Command {
constructor(client) {
@@ -46,6 +47,10 @@ module.exports = class ScreenshotCommand extends Command {
if (nsfw) return msg.reply('This site is NSFW.');
}
const { body } = await request.get(`https://image.thum.io/get/width/1920/crop/675/noanimate/${url.href}`);
if (!msg.channel.nsfw) {
const aiDetect = await isImageNSFW(this.client.nsfwModel, body);
if (aiDetect) return msg.reply('This site isn\'t NSFW, but the resulting image was.');
}
return msg.say({ files: [{ attachment: body, name: 'screenshot.png' }] });
} catch (err) {
if (err.status === 404) return msg.say('Could not find any results. Invalid URL?');