Compare commits

..

2 commits

Author SHA1 Message Date
c5cacbb78b Maybe fix that thing 2022-10-16 20:09:59 +02:00
5b426b55ff Stable diffusion! 2022-10-16 20:09:48 +02:00
2 changed files with 69 additions and 1 deletions

View file

@ -0,0 +1,68 @@
import { SlashCommandBuilder, EmbedBuilder, ButtonBuilder, ActionRowBuilder, ButtonStyle } from 'discord.js';
import fetch from 'node-fetch';
import fs from 'node:fs';
import os from 'node:os';
export default {
data: new SlashCommandBuilder()
.setName('stablediffusion')
.setDescription('AI generated image with stable diffusion (This can take very long)')
.addStringOption(option =>
option.setName('prompt')
.setDescription('What do you want the AI to generate?')
.setRequired(true)),
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);
}
});
},
};
async function generate(i, prompt) {
const body = {
prompt: prompt,
params: {
n: 1,
width: 512,
height: 512,
},
nsfw: true,
censor_nsfw: i.channel.nsfw ? true : false,
};
const fetchParameters = {
method: 'post',
body: JSON.stringify(body),
headers: { 'Content-Type': 'application/json', 'apikey': '0000000000' },
};
let response = await fetch('https://stablehorde.net/api/v2/generate/sync', fetchParameters);
response = await response.json();
const stableEmbed = new EmbedBuilder()
.setColor(i.member ? i.member.displayHexColor : 'Navy')
.setTitle(prompt)
.setURL('https://aqualxx.github.io/stable-ui/')
.setFooter({ text: `Seed: ${response.generations[0].seed} worker ID: ${response.generations[0].worker_id} worker name: ${response.generations[0].worker_name}` });
const row = new ActionRowBuilder()
.addComponents(
new ButtonBuilder()
.setCustomId(`regenerate${i.user.id}`)
.setLabel('🔄')
.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`] });
}

View file

@ -26,7 +26,7 @@ export function rand(text, interaction) {
const variables = [ const variables = [
{ {
name: /\[author\]/, name: /\[author\]/,
value: interaction ? interaction.author.username : '', value: interaction ? interaction.author ? interaction.author.username : '' : '',
}, },
{ {
name: /\[member\]/, name: /\[member\]/,