mirror of
https://gitlab.com/bignutty/labscore.git
synced 2025-06-08 14:13:02 -04:00
cleanup!
This commit is contained in:
parent
c598da21f1
commit
2380ac41d8
8 changed files with 5 additions and 88 deletions
|
@ -1,82 +0,0 @@
|
|||
const { createEmbed } = require("../../../labscore/utils/embed");
|
||||
const { editOrReply } = require("../../../labscore/utils/message");
|
||||
|
||||
const superagent = require('superagent');
|
||||
const { getRecentImage } = require("../../../labscore/utils/attachment");
|
||||
|
||||
module.exports = {
|
||||
name: 'anime',
|
||||
label: 'query',
|
||||
metadata: {
|
||||
description: 'Uses Stable Diffusion to generate an image using anime styling from a text prompt.\n\n**Replying** to a message with an image will use said image as a base for the AI (img2img), this does __not__ support inpainting/adding elements to the image.',
|
||||
description_short: 'AI Anime image generation',
|
||||
explicit: true,
|
||||
examples: ['anime otter'],
|
||||
category: 'disabled',
|
||||
usage: 'anime <text>'
|
||||
},
|
||||
ratelimit: {
|
||||
type: 'guild',
|
||||
limit: 1,
|
||||
duration: 5000
|
||||
},
|
||||
run: async (context, args) => {
|
||||
return;
|
||||
if(!context.channel.nsfw) return editOrReply(context, {embeds:[createEmbed("nsfw", context)]})
|
||||
if(args.query.length == 0) return editOrReply(context, {embeds:[createEmbed("warning", context, "Missing prompt")]})
|
||||
|
||||
let hasUrl = false;
|
||||
let image;
|
||||
if(context.message.messageReference){
|
||||
image = await getRecentImage(context, 50)
|
||||
hasUrl = true;
|
||||
if(!image) return editOrReply(context, createEmbed("warning", context, "No images found."))
|
||||
}
|
||||
|
||||
let response = await editOrReply(context, { embeds: [createEmbed("loading", context, `Generating image...`)] })
|
||||
|
||||
let noticeTimer = setTimeout(()=>{
|
||||
let emb = createEmbed("loading", context, `Generating image...`)
|
||||
emb.footer = {
|
||||
text: "This might take a moment to complete."
|
||||
};
|
||||
response.edit({ embeds: [ emb ] });
|
||||
}, 45000)
|
||||
|
||||
try{
|
||||
let t = Date.now();
|
||||
|
||||
let img;
|
||||
if(hasUrl){
|
||||
img = await superagent.get(`${process.env.AI_SERVER}/anime/img2img`)
|
||||
.query({
|
||||
prompt: args.query,
|
||||
url: image
|
||||
})
|
||||
} else {
|
||||
img = await superagent.get(`${process.env.AI_SERVER}/anime`)
|
||||
.query({
|
||||
prompt: args.query
|
||||
})
|
||||
}
|
||||
clearTimeout(noticeTimer)
|
||||
|
||||
if(img.body.message) return await response.edit({embeds:[createEmbed("warning", context, img.body.message)]})
|
||||
|
||||
let embeds = [];
|
||||
let files = [];
|
||||
|
||||
embeds.push(createEmbed("default", context, {image: {url:`attachment://anime.png`}, url: `https://example.com`, footer: { iconUrl: `https://cdn.discordapp.com/avatars/${context.application.id}/${context.application.icon}.png?size=256`, text: `${context.application.name} • Took ${((Date.now() - t) / 1000).toFixed(2)}s` }}))
|
||||
files.push({
|
||||
filename: `anime.png`,
|
||||
value: Buffer.from(img.body.image, 'base64')
|
||||
})
|
||||
|
||||
await response.edit({ embeds, files })
|
||||
}catch(e){
|
||||
clearTimeout(noticeTimer)
|
||||
console.log(e)
|
||||
await response.edit({embeds:[createEmbed("error", context, `Image generation failed.`)]})
|
||||
}
|
||||
},
|
||||
};
|
|
@ -1,7 +1,8 @@
|
|||
const { createEmbed } = require("../../../labscore/utils/embed");
|
||||
const { editOrReply } = require("../../../labscore/utils/message");
|
||||
const { icon, highlight, link } = require('../../../labscore/utils/markdown');
|
||||
|
||||
const superagent = require('superagent');
|
||||
const { icon, highlight, link } = require('../../../labscore/utils/markdown')
|
||||
|
||||
const applicationFlags = {
|
||||
EMBEDDED_RELEASED: 1,
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
const { Constants } = require("detritus-client");
|
||||
const Permissions = Constants.Permissions;
|
||||
|
||||
const { icon, pill } = require("../../../labscore/utils/markdown");
|
||||
const { icon } = require("../../../labscore/utils/markdown");
|
||||
|
||||
// TODO: copy pasted from v1, rework this eventually
|
||||
|
||||
|
|
|
@ -1,5 +1,4 @@
|
|||
const { createEmbed, formatPaginationEmbeds } = require('../../../labscore/utils/embed')
|
||||
const { link } = require('../../../labscore/utils/markdown')
|
||||
const { editOrReply } = require('../../../labscore/utils/message')
|
||||
const { STATICS } = require('../../../labscore/utils/statics')
|
||||
|
||||
|
|
|
@ -1,5 +1,4 @@
|
|||
const { createEmbed, formatPaginationEmbeds } = require('../../../labscore/utils/embed')
|
||||
const { link } = require('../../../labscore/utils/markdown')
|
||||
const { editOrReply } = require('../../../labscore/utils/message')
|
||||
const { STATICS } = require('../../../labscore/utils/statics')
|
||||
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
const { createEmbed, formatPaginationEmbeds, page } = require('../../../labscore/utils/embed')
|
||||
const { link, pill, citation } = require('../../../labscore/utils/markdown')
|
||||
const { link, citation } = require('../../../labscore/utils/markdown')
|
||||
const { editOrReply } = require('../../../labscore/utils/message')
|
||||
const { STATICS } = require('../../../labscore/utils/statics')
|
||||
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
const { createEmbed, formatPaginationEmbeds } = require('../../../labscore/utils/embed')
|
||||
const { link, icon, highlight, iconPill } = require('../../../labscore/utils/markdown')
|
||||
const { link, iconPill } = require('../../../labscore/utils/markdown')
|
||||
const { editOrReply } = require('../../../labscore/utils/message')
|
||||
const { STATICS } = require('../../../labscore/utils/statics')
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue