80 lines
2.5 KiB
JavaScript
80 lines
2.5 KiB
JavaScript
const fs = require("fs");
|
|
const Discord = require("discord.js");
|
|
const yaml = require("js-yaml");
|
|
const generateImage = require('../Structures/generateImage.js');
|
|
|
|
const supportbot = yaml.load(fs.readFileSync("./Configs/supportbot.yml", "utf8"));
|
|
const Command = require("../Structures/Command.js");
|
|
|
|
module.exports = new Command({
|
|
name: "image",
|
|
description: "Create an image through SupportBot AI",
|
|
type: Discord.ApplicationCommandType.ChatInput,
|
|
options: [
|
|
{
|
|
name: "prompt",
|
|
description: "The prompt to generate an image from",
|
|
type: Discord.ApplicationCommandOptionType.String,
|
|
required: true,
|
|
},
|
|
{
|
|
name: "negativeprompt",
|
|
description: "The negative prompt to exclude from the image",
|
|
type: Discord.ApplicationCommandOptionType.String,
|
|
required: false,
|
|
},
|
|
{
|
|
name: "cfgscale",
|
|
description: "The cfg scale for the image generation",
|
|
type: Discord.ApplicationCommandOptionType.Number,
|
|
required: false,
|
|
}
|
|
],
|
|
permissions: "SendMessages",
|
|
|
|
async run(interaction) {
|
|
const prompt = interaction.options.getString("prompt");
|
|
const negativePrompt = interaction.options.getString("negativeprompt") || "";
|
|
const cfgScale = interaction.options.getNumber("cfgscale") || 7.5; // Default value
|
|
|
|
try {
|
|
const imageUrl = await generateImage(prompt, negativePrompt, cfgScale);
|
|
|
|
const ImageEmbed = new Discord.EmbedBuilder()
|
|
.setTitle("Generated Image")
|
|
.setDescription(`Here is the image generated for your prompt: "${prompt}"`)
|
|
.setImage(imageUrl)
|
|
.setColor(supportbot.Embed.Colours.General)
|
|
.setFooter({
|
|
text: supportbot.Embed.Footer,
|
|
iconURL: interaction.user.displayAvatarURL(),
|
|
});
|
|
|
|
if (!interaction.replied) {
|
|
await interaction.reply({
|
|
ephemeral: true,
|
|
embeds: [ImageEmbed],
|
|
});
|
|
} else {
|
|
await interaction.followUp({
|
|
ephemeral: true,
|
|
embeds: [ImageEmbed],
|
|
});
|
|
}
|
|
} catch (error) {
|
|
console.error('Error generating image:', error);
|
|
if (!interaction.replied) {
|
|
await interaction.reply({
|
|
ephemeral: true,
|
|
content: `An error occurred while generating the image: ${error.message}. Please try again later.`,
|
|
});
|
|
} else {
|
|
await interaction.followUp({
|
|
ephemeral: true,
|
|
content: `An error occurred while generating the image: ${error.message}. Please try again later.`,
|
|
});
|
|
}
|
|
}
|
|
},
|
|
});
|