mirror of
https://gitlab.com/bignutty/labscore.git
synced 2025-06-09 14:43:05 -04:00
cmd
This commit is contained in:
parent
d6665bdf47
commit
a5e1dd506c
1 changed files with 59 additions and 0 deletions
59
commands/message/image/dalle.js
Normal file
59
commands/message/image/dalle.js
Normal file
|
@ -0,0 +1,59 @@
|
||||||
|
const { getRecentImage } = require("../../../labscore/utils/attachment");
|
||||||
|
const { createEmbed } = require("../../../labscore/utils/embed");
|
||||||
|
const { editOrReply } = require("../../../labscore/utils/message");
|
||||||
|
|
||||||
|
const superagent = require('superagent');
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
name: 'dalle',
|
||||||
|
label: 'query',
|
||||||
|
metadata: {
|
||||||
|
description: 'dalle',
|
||||||
|
examples: ['dalle'],
|
||||||
|
category: 'image',
|
||||||
|
usage: 'dalle <text>'
|
||||||
|
},
|
||||||
|
ratelimit: {
|
||||||
|
type: 'guild',
|
||||||
|
limit: 1,
|
||||||
|
duration: 15000
|
||||||
|
},
|
||||||
|
run: async (context, args) => {
|
||||||
|
context.triggerTyping();
|
||||||
|
let response = await editOrReply(context, { embeds: [createEmbed("loading", context, `Synthesizing images...`)] })
|
||||||
|
|
||||||
|
// Display a message if server might be busy.
|
||||||
|
let notice = setTimeout(() => {
|
||||||
|
response.edit({ embeds: [createEmbed("loading", context, "DALL-E Server might be under heavy load, generation takes a bit...")] })
|
||||||
|
}, 15000)
|
||||||
|
|
||||||
|
try{
|
||||||
|
let t = Date.now();
|
||||||
|
|
||||||
|
let img = await superagent.post(`https://bf.dallemini.ai/generate`)
|
||||||
|
.send({
|
||||||
|
prompt: args.query
|
||||||
|
})
|
||||||
|
|
||||||
|
clearTimeout(notice)
|
||||||
|
|
||||||
|
let embeds = [];
|
||||||
|
let files = [];
|
||||||
|
|
||||||
|
for(let i = 0; i < 4; i++){
|
||||||
|
embeds.push(createEmbed("default", context, {image: {url:`attachment://dalle${i}.jpeg`}, 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: `dalle${i}.jpeg`,
|
||||||
|
value: Buffer.from(img.body.images[i], 'base64')
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
await response.edit({ embeds, files })
|
||||||
|
}catch(e){
|
||||||
|
console.log(e)
|
||||||
|
clearTimeout(notice)
|
||||||
|
if(e.response.status == 503) return await response.edit({embeds:[createEmbed("error", context, `DALL-E server is busy, try again later.`)]})
|
||||||
|
await response.edit({embeds:[createEmbed("error", context, `Image generation failed.`)]})
|
||||||
|
}
|
||||||
|
},
|
||||||
|
};
|
Loading…
Add table
Add a link
Reference in a new issue