mirror of
https://gitlab.com/bignutty/labscore.git
synced 2025-06-09 06:33:05 -04:00
move editimage to aiserver
This commit is contained in:
parent
f35586fe89
commit
10c15fb179
2 changed files with 58 additions and 39 deletions
|
@ -1,39 +0,0 @@
|
|||
const { getRecentImage } = require("../../../../labscore/utils/attachment");
|
||||
const { createEmbed } = require("../../../../labscore/utils/embed");
|
||||
const { editOrReply } = require("../../../../labscore/utils/message");
|
||||
|
||||
const { deepdream, imageedit } = require('../../../../labscore/api')
|
||||
|
||||
module.exports = {
|
||||
name: 'editimage',
|
||||
aliases: ['ei'],
|
||||
label: 'prompt',
|
||||
metadata: {
|
||||
description: 'Edits an image using AI.',
|
||||
description_short: 'AI image editing',
|
||||
examples: ['editimage With a crown'],
|
||||
category: 'image',
|
||||
usage: 'editimage <prompt>'
|
||||
},
|
||||
run: async (context, args) => {
|
||||
context.triggerTyping();
|
||||
try{
|
||||
if(!args.prompt) return editOrReply(context, { embeds: [createEmbed("warning", context, "Missing prompt.")] })
|
||||
|
||||
let image = await getRecentImage(context, 50)
|
||||
if(!image) return editOrReply(context, { embeds: [createEmbed("warning", context, "No images found.")] })
|
||||
|
||||
let res = await imageedit(context, image, args.prompt)
|
||||
|
||||
if(res.response.body.status == 1) return editOrReply(context, {embeds:[createEmbed("warning", context, res.response.body.errors[0])]})
|
||||
|
||||
return editOrReply(context, createEmbed("image", context, {
|
||||
url: res.response.body.image,
|
||||
time: res.timings
|
||||
}))
|
||||
}catch(e){
|
||||
console.log(e)
|
||||
return editOrReply(context, {embeds:[createEmbed("error", context, `Image editing timed out.`)]})
|
||||
}
|
||||
},
|
||||
};
|
58
commands/message/image/imageedit.js
Normal file
58
commands/message/image/imageedit.js
Normal file
|
@ -0,0 +1,58 @@
|
|||
const { createEmbed } = require("../../../labscore/utils/embed");
|
||||
const { editOrReply } = require("../../../labscore/utils/message");
|
||||
const { getRecentImage } = require("../../../labscore/utils/attachment");
|
||||
|
||||
const superagent = require('superagent');
|
||||
|
||||
module.exports = {
|
||||
name: 'editimage',
|
||||
label: 'prompt',
|
||||
aliases: ['ei'],
|
||||
metadata: {
|
||||
description: 'Edits an image using AI.',
|
||||
description_short: 'AI image editing',
|
||||
examples: ['editimage Wearing a crown'],
|
||||
category: 'image',
|
||||
usage: 'editimage <prompt>'
|
||||
},
|
||||
ratelimit: {
|
||||
type: 'guild',
|
||||
limit: 1,
|
||||
duration: 5000
|
||||
},
|
||||
run: async (context, args) => {
|
||||
if(!args.prompt) return editOrReply(context, { embeds: [createEmbed("warning", context, "Missing prompt.")] })
|
||||
|
||||
let image = await getRecentImage(context, 50)
|
||||
if(!image) return editOrReply(context, { embeds: [createEmbed("warning", context, "No images found.")] })
|
||||
|
||||
let response = await editOrReply(context, { embeds: [createEmbed("loading", context, `Editing image...`)] })
|
||||
|
||||
let noticeTimer = setTimeout(()=>{
|
||||
let emb = createEmbed("loading", context, `Editing image...`)
|
||||
emb.footer = {
|
||||
text: "This might take several minutes to complete."
|
||||
};
|
||||
response.edit({ embeds: [ emb ] });
|
||||
}, 45000)
|
||||
|
||||
try{
|
||||
let img = await superagent.get(`${process.env.AI_SERVER}/deepai/imageeditor`)
|
||||
.query({
|
||||
prompt: args.prompt,
|
||||
url: image
|
||||
})
|
||||
|
||||
clearTimeout(noticeTimer)
|
||||
|
||||
await response.edit({ embeds: [
|
||||
createEmbed("image", context, {
|
||||
url: img.body.image
|
||||
})
|
||||
] })
|
||||
}catch(e){
|
||||
console.log(e)
|
||||
await response.edit({embeds:[createEmbed("error", context, `Image generation failed.`)]})
|
||||
}
|
||||
},
|
||||
};
|
Loading…
Add table
Add a link
Reference in a new issue