joan/Haha-Yes
1
0
Fork 0
forked from Supositware/Haha-Yes
Code Pull requests Activity

Avoid stupid leak?

Browse source
This commit is contained in:
Supositware 2022-10-16 22:37:34 +02:00
parent 332f1730d6
commit 5a8ec1dbe7
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
commands/fun/stablediffusion.js
View file

@ -16,20 +16,11 @@ export default {
category: 'fun',
async execute(interaction, args, client) {
await interaction.deferReply();
generate(interaction, args.prompt);
client.on('interactionCreate', async (interactionMenu) => {
if (interaction.user !== interactionMenu.user) return;
if (!interactionMenu.isButton) return;
if (interactionMenu.customId === `regenerate${interactionMenu.user.id}`) {
await interactionMenu.deferReply();
await generate(interactionMenu, args.prompt);
}
});
generate(interaction, args.prompt, client);
},
};
async function generate(i, prompt) {
async function generate(i, prompt, client) {
const body = {
prompt: prompt,
params: {
@ -60,11 +51,20 @@ async function generate(i, prompt) {
.addComponents(
new ButtonBuilder()
.setCustomId(`regenerate${i.user.id}`)
.setLabel('🔄')
.setLabel('🔄 Regenerate')
.setStyle(ButtonStyle.Primary),
);
fs.writeFileSync(`${os.tmpdir()}/${i.id}.png`, response.generations[0].img, 'base64');
await i.editReply({ embeds: [stableEmbed], components: [row], files: [`${os.tmpdir()}/${i.id}.png`] });
client.once('interactionCreate', async (interactionMenu) => {
if (i.user !== interactionMenu.user) return;
if (!interactionMenu.isButton) return;
if (interactionMenu.customId === `regenerate${interactionMenu.user.id}`) {
await interactionMenu.deferReply();
await generate(interactionMenu, prompt, client);
}
});
}