atridad/himbot
1
0
Fork 0
Code Projects Releases Packages 1 Activity Actions

upgraded the shit outta these

Browse source
This commit is contained in:
Atridad Lahiji 2023-07-07 21:33:53 -06:00
parent 099cc182d1
commit 3d0466152c
No known key found for this signature in database
2 changed files with 70 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
src/commands/fancypic.ts
View file

@ -6,7 +6,7 @@ import { AttachmentBuilder, Message } from 'discord.js';
@ApplyOptions<Command.Options>({
description: 'Make a picture... but high res!',
options: ['prompt'],
options: ['prompt', 'number of pictures'],
// 10mins
cooldownDelay: 300_000,
cooldownLimit: 1,
@ -21,21 +21,40 @@ export class UserCommand extends Command {
builder
.setName(this.name)
.setDescription(this.description)
.addStringOption((option) => option.setName('prompt').setDescription('Make a picture... but high res!').setRequired(true))
.addStringOption((option) =>
option.setName('prompt').setDescription('The prompt you will use to generate an image!').setRequired(true)
)
.addStringOption((option) =>
option
.setName('amount')
.setDescription('The number of images you would like to generate. Maximum 2.')
.setChoices(
...[
{ name: '1', value: '1' },
{ name: '2', value: '2' }
]
)
)
);
}
// Message command
public async messageRun(message: Message, args: Args) {
return this.picHr(message, args.getOption('prompt') || message.content.split('!fancypic ')[1]);
const amount = Math.abs(Number(args.getOption('amount')));
return this.picHr(message, args.getOption('prompt') || 'Scold me for not passing any prompt in.', amount <= 2 ? amount : 1);
}
// Chat Input (slash) command
public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
return this.picHr(interaction, interaction.options.getString('prompt') || 'NOTHING');
const amount = Number(interaction.options.getString('amount'));
return this.picHr(interaction, interaction.options.getString('prompt') || 'NOTHING', amount || 1);
}
private async picHr(interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction, prompt: string) {
private async picHr(
interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction,
prompt: string,
amount: number
) {
const askMessage =
interactionOrMessage instanceof Message
? await interactionOrMessage.channel.send({ content: '🤔 Thinking... 🤔' })
@ -69,8 +88,9 @@ export class UserCommand extends Command {
clip_guidance_preset: 'FAST_BLUE',
height: 1024,
width: 1024,
samples: 1,
steps: 32
samples: amount,
steps: 32,
seed: Number(String(interactionOrMessage.member?.user.id).substring(0, 5)) || 0
})
});
@ -94,7 +114,11 @@ export class UserCommand extends Command {
});
} else {
const responseJSON = (await imageGenResponse.json()) as GenerationResponse;
const imageAttachment = new AttachmentBuilder(Buffer.from(responseJSON.artifacts[0].base64, 'base64'));
const imageAttachment: AttachmentBuilder[] = [];
for (let i = 0; i < responseJSON.artifacts.length - 1; i++) {
imageAttachment.push(new AttachmentBuilder(Buffer.from(responseJSON.artifacts[i].base64, 'base64')));
}
const newCreditCountResponse = await fetch(`https://api.stability.ai/v1/user/balance`, {
method: 'GET',
@ -114,12 +138,12 @@ export class UserCommand extends Command {
}` || 'ERROR!';
if (interactionOrMessage instanceof Message) {
return askMessage.edit({ content, files: [imageAttachment] });
return askMessage.edit({ content, files: imageAttachment });
}
return interactionOrMessage.editReply({
content: content,
files: [imageAttachment]
files: imageAttachment
});
}
} else {

46
src/commands/pic.ts
View file

@ -6,7 +6,7 @@ import { AttachmentBuilder, Message } from 'discord.js';
@ApplyOptions<Command.Options>({
description: 'Make a picture!',
options: ['prompt'],
options: ['prompt', 'amount'],
// 10mins
cooldownDelay: 100_000,
cooldownLimit: 1,
@ -21,21 +21,42 @@ export class UserCommand extends Command {
builder
.setName(this.name)
.setDescription(this.description)
.addStringOption((option) => option.setName('prompt').setDescription('Make a picture!').setRequired(true))
.addStringOption((option) =>
option.setName('prompt').setDescription('The prompt you will use to generate an image!').setRequired(true)
)
.addStringOption((option) =>
option
.setName('amount')
.setDescription('The number of images you would like to generate. Maximum 4.')
.setChoices(
...[
{ name: '1', value: '1' },
{ name: '2', value: '2' },
{ name: '3', value: '3' },
{ name: '4', value: '4' }
]
)
)
);
}
// Message command
public async messageRun(message: Message, args: Args) {
return this.pic(message, args.getOption('prompt') || message.content.split('!pic ')[1]);
const amount = Math.abs(Number(args.getOption('amount')));
return this.pic(message, args.getOption('prompt') || 'Scold me for not passing any prompt in.', amount <= 4 ? amount : 1);
}
// Chat Input (slash) command
public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
return this.pic(interaction, interaction.options.getString('prompt') || 'NOTHING');
const amount = Number(interaction.options.getString('amount'));
return this.pic(interaction, interaction.options.getString('prompt') || 'NOTHING', amount || 1);
}
private async pic(interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction, prompt: string) {
private async pic(
interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction,
prompt: string,
amount: number
) {
const askMessage =
interactionOrMessage instanceof Message
? await interactionOrMessage.channel.send({ content: '🤔 Thinking... 🤔' })
@ -69,8 +90,9 @@ export class UserCommand extends Command {
clip_guidance_preset: 'FAST_BLUE',
height: 512,
width: 512,
samples: 1,
steps: 32
samples: amount,
steps: 32,
seed: Number(String(interactionOrMessage.member?.user.id).substring(0, 5)) || 0
})
});
@ -94,7 +116,11 @@ export class UserCommand extends Command {
});
} else {
const responseJSON = (await imageGenResponse.json()) as GenerationResponse;
const imageAttachment = new AttachmentBuilder(Buffer.from(responseJSON.artifacts[0].base64, 'base64'));
const imageAttachment: AttachmentBuilder[] = [];
for (let i = 0; i < responseJSON.artifacts.length; i++) {
imageAttachment.push(new AttachmentBuilder(Buffer.from(responseJSON.artifacts[i].base64, 'base64')));
}
const newCreditCountResponse = await fetch(`https://api.stability.ai/v1/user/balance`, {
method: 'GET',
@ -114,12 +140,12 @@ export class UserCommand extends Command {
}` || 'ERROR!';
if (interactionOrMessage instanceof Message) {
return askMessage.edit({ content, files: [imageAttachment] });
return askMessage.edit({ content, files: imageAttachment });
}
return interactionOrMessage.editReply({
content: content,
files: [imageAttachment]
files: imageAttachment
});
}
} else {