diff --git a/src/commands/fun/gelbooru.js b/src/commands/fun/gelbooru.js index 8d1f3b1..de6911d 100644 --- a/src/commands/fun/gelbooru.js +++ b/src/commands/fun/gelbooru.js @@ -94,6 +94,15 @@ function formatTime(time) { return `${(Number(time) / 1e6).toFixed(2)}ms` } +const aiTags = [ + "ai_art", + "ai_generated" +]; + +function isAIPost(post) { + return aiTags.map(i=>post.tags.includes(i)).includes(true); +} + module.exports = { data, async execute(interaction) { @@ -113,7 +122,7 @@ module.exports = { await interaction.followUp("Could not find any post matching tags."); return; } - if (!nsfw && isNSFWPost(newPost)) continue; + if (!nsfw && isNSFWPost(newPost) || isAIPost(newPost)) continue; post = newPost; }