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

Cleanup!

Browse source
This commit is contained in:
Atridad Lahiji 2023-11-10 18:47:06 -07:00
parent 1ca5dbf987
commit e9d9b202b0
No known key found for this signature in database
5 changed files with 33 additions and 199 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
src/commands/credits.ts
View file

@ -1,51 +0,0 @@
import { ApplyOptions } from '@sapphire/decorators';
import { Command } from '@sapphire/framework';
import { Message } from 'discord.js';
// @ts-ignore
@ApplyOptions<Command.Options>({
description: 'Check how many credits I have left!'
})
export class UserCommand extends Command {
// Register Chat Input and Context Menu command
public override registerApplicationCommands(registry: Command.Registry) {
registry.registerChatInputCommand((builder) => builder.setName(this.name).setDescription(this.description));
}
// Message command
public async messageRun(message: Message) {
return this.credits(message);
}
// Chat Input (slash) command
public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
return this.credits(interaction);
}
private async credits(interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction) {
const askMessage =
interactionOrMessage instanceof Message
? await interactionOrMessage.channel.send({ content: '🤔 Thinking... 🤔' })
: await interactionOrMessage.reply({ content: '🤔 Thinking... 🤔', fetchReply: true });
const creditCountResponse = await fetch(`https://api.stability.ai/v1/user/balance`, {
method: 'GET',
headers: {
'Content-Type': 'application/json',
Authorization: `Bearer ${process.env.STABILITY_API_KEY}`
}
});
const balance = (await creditCountResponse.json()).credits || 0;
const content = `I have ${balance} credits remaining for image generation!`;
if (interactionOrMessage instanceof Message) {
return askMessage.edit({ content });
}
return interactionOrMessage.editReply({
content: content
});
}
}

35
src/commands/disclosure.ts
View file

@ -1,35 +0,0 @@
import { ApplyOptions } from '@sapphire/decorators';
import { Command } from '@sapphire/framework';
import { Message } from 'discord.js';
// @ts-ignore
@ApplyOptions<Command.Options>({
description: 'Disclosures for privacy.'
})
export class UserCommand extends Command {
// Register Chat Input and Context Menu command
public override registerApplicationCommands(registry: Command.Registry) {
registry.registerChatInputCommand((builder) => builder.setName(this.name).setDescription(this.description));
}
// Message command
public async messageRun(message: Message) {
return this.sendDisclosure(message);
}
// Chat Input (slash) command
public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
return this.sendDisclosure(interaction);
}
private async sendDisclosure(interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction) {
interactionOrMessage instanceof Message
? await interactionOrMessage.channel.send({
content: `Commands such as /ask and /pic do not send your discord data to the AI providers. They will, however, send anything you include in the prompts. /ask uses OpenAI's gpt3.5 model and /pic uses Stable Diffusion. All other commands and the data associated stays with HimBot's server and Discord's Servers.`
})
: await interactionOrMessage.reply({
content: `Commands such as /ask and /pic do not send your discord data to the AI providers. They will, however, send anything you include in the prompts. /ask uses OpenAI's gpt3.5 model and /pic uses Stable Diffusion. All other commands and the data associated stays with HimBot's server and Discord's Servers.`,
fetchReply: true
});
}
}

5
src/commands/fancypic.ts
View file

@ -12,7 +12,7 @@ const openai = new OpenAI({
description: 'Pic... but better!', description: 'Pic... but better!',
options: ['prompt'], options: ['prompt'],
// 10mins // 10mins
cooldownDelay: 400_000, cooldownDelay: 500_000,
cooldownLimit: 1, cooldownLimit: 1,
// Yes... I did hardcode myself. // Yes... I did hardcode myself.
cooldownFilteredUsers: ['83679718401904640'], cooldownFilteredUsers: ['83679718401904640'],
@ -52,7 +52,8 @@ export class UserCommand extends Command {
model: 'dall-e-3', model: 'dall-e-3',
prompt, prompt,
n: 1, n: 1,
size: '1024x1024' size: '1024x1024',
quality: 'hd'
}); });
const imageUrl = response.data[0].url || ''; const imageUrl = response.data[0].url || '';

41
src/commands/meow.ts
View file

@ -1,41 +0,0 @@
import { ApplyOptions } from '@sapphire/decorators';
import { Command } from '@sapphire/framework';
import { Message } from 'discord.js';
// @ts-ignore
@ApplyOptions<Command.Options>({
description: 'Meow!'
})
export class UserCommand extends Command {
// Register Chat Input and Context Menu command
public override registerApplicationCommands(registry: Command.Registry) {
// Register Chat Input command
registry.registerChatInputCommand({
name: this.name,
description: this.description
});
}
// Message command
public async messageRun(message: Message) {
return this.sendMeow(message);
}
// Chat Input (slash) command
public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
return this.sendMeow(interaction);
}
private async sendMeow(interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction) {
const catResponse = await fetch('https://cataas.com/cat?json=true');
const catData = await catResponse.json();
interactionOrMessage instanceof Message
? await interactionOrMessage.channel.send({
content: `https://cataas.com/${catData.url}`
})
: await interactionOrMessage.reply({
content: `https://cataas.com/${catData.url}`,
fetchReply: true
});
}
}

100
src/commands/pic.ts
View file

@ -1,15 +1,18 @@
import { ApplyOptions } from '@sapphire/decorators'; import { ApplyOptions } from '@sapphire/decorators';
import { Args, BucketScope, Command } from '@sapphire/framework'; import { Args, BucketScope, Command } from '@sapphire/framework';
import { AttachmentBuilder, Message } from 'discord.js'; import { AttachmentBuilder, Message } from 'discord.js';
import OpenAI from 'openai';
// This is literally the world's messiest Typescript code. Please don't judge me... const openai = new OpenAI({
apiKey: process.env.OPENAI_API_KEY
});
// @ts-ignore // @ts-ignore
@ApplyOptions<Command.Options>({ @ApplyOptions<Command.Options>({
description: 'Make a picture... but high res!', description: 'Pic... but better!',
options: ['prompt'], options: ['prompt'],
// 10mins // 10mins
cooldownDelay: 300_000, cooldownDelay: 400_000,
cooldownLimit: 1, cooldownLimit: 1,
// Yes... I did hardcode myself. // Yes... I did hardcode myself.
cooldownFilteredUsers: ['83679718401904640'], cooldownFilteredUsers: ['83679718401904640'],
@ -30,62 +33,34 @@ export class UserCommand extends Command {
// Message command // Message command
public async messageRun(message: Message, args: Args) { public async messageRun(message: Message, args: Args) {
return this.picHr(message, args.getOption('prompt') || 'Scold me for not passing any prompt in.'); return this.pic(message, args.getOption('prompt') || 'NOTHING');
} }
// Chat Input (slash) command // Chat Input (slash) command
public async chatInputRun(interaction: Command.ChatInputCommandInteraction) { public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
return this.picHr(interaction, interaction.options.getString('prompt') || 'NOTHING'); return this.pic(interaction, interaction.options.getString('prompt') || 'NOTHING');
} }
private async picHr(interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction, prompt: string) { private async pic(interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction, prompt: string) {
const askMessage = const askMessage =
interactionOrMessage instanceof Message interactionOrMessage instanceof Message
? await interactionOrMessage.channel.send({ content: '🤔 Thinking... 🤔' }) ? await interactionOrMessage.channel.send({ content: '🤔 Thinking... 🤔' })
: await interactionOrMessage.reply({ content: '🤔 Thinking... 🤔', fetchReply: true }); : await interactionOrMessage.reply({ content: '🤔 Thinking... 🤔', fetchReply: true });
const creditCountResponse = await fetch(`https://api.stability.ai/v1/user/balance`, { try {
method: 'GET', const response = await openai.images.generate({
headers: { model: 'dall-e-3',
'Content-Type': 'application/json', prompt,
Authorization: `Bearer ${process.env.STABILITY_API_KEY}` n: 1,
} size: '1024x1024',
}); quality: 'standard'
const balance = (await creditCountResponse.json()).credits || 0;
if (balance > 5) {
const imageGenResponse = await fetch(`https://api.stability.ai/v1/generation/stable-diffusion-xl-1024-v1-0/text-to-image`, {
method: 'POST',
headers: {
'Content-Type': 'application/json',
Accept: 'application/json',
Authorization: `Bearer ${process.env.STABILITY_API_KEY}`
},
body: JSON.stringify({
text_prompts: [
{
text: prompt
}
],
cfg_scale: 6,
clip_guidance_preset: 'FAST_BLUE',
height: 1024,
width: 1024,
steps: 32,
seed: Number(String(interactionOrMessage.member?.user.id).substring(0, 5)) || 0
})
}); });
interface GenerationResponse { const imageUrl = response.data[0].url || '';
artifacts: Array<{ // get an array buffer
base64: string; const imageBuffer = await fetch(imageUrl).then((r) => r.arrayBuffer());
seed: number;
finishReason: string;
}>;
}
if (!imageGenResponse.ok) { if (response.data.length === 0) {
const content = `Sorry, I can't complete the prompt for: ${prompt}`; const content = `Sorry, I can't complete the prompt for: ${prompt}`;
if (interactionOrMessage instanceof Message) { if (interactionOrMessage instanceof Message) {
@ -96,29 +71,16 @@ export class UserCommand extends Command {
content: content content: content
}); });
} else { } else {
const responseJSON = (await imageGenResponse.json()) as GenerationResponse;
const imageAttachment: AttachmentBuilder[] = []; const imageAttachment: AttachmentBuilder[] = [];
for (let i = 0; i < responseJSON.artifacts.length; i++) { imageAttachment.push(
imageAttachment.push( new AttachmentBuilder(Buffer.from(new Uint8Array(imageBuffer)), {
new AttachmentBuilder(Buffer.from(responseJSON.artifacts[i].base64, 'base64'), { name: 'himbot_response.jpg',
name: 'himbot_response.jpg', description: `An image generated by Himbot using the prompt: ${prompt}`
description: `An image generated by Himbot using the prompt: ${prompt}` })
}) );
);
}
const newCreditCountResponse = await fetch(`https://api.stability.ai/v1/user/balance`, { const content = `Prompt: ${prompt}:`;
method: 'GET',
headers: {
'Content-Type': 'application/json',
Authorization: `Bearer ${process.env.STABILITY_API_KEY}`
}
});
const newBalance = (await newCreditCountResponse.json()).credits || 0;
const content = `Credits Used: ${balance - newBalance}\nPrompt: ${prompt}` || 'ERROR!';
if (interactionOrMessage instanceof Message) { if (interactionOrMessage instanceof Message) {
return askMessage.edit({ content, files: imageAttachment }); return askMessage.edit({ content, files: imageAttachment });
@ -129,16 +91,14 @@ export class UserCommand extends Command {
files: imageAttachment files: imageAttachment
}); });
} }
} else { } catch (error) {
const content = `Oops! We're out of credits for this. If you'd like to help fund this command, please type "/support" for details!`; const content = "Sorry, I can't complete the prompt for: " + prompt + '\n' + 'Error: ' + error;
if (interactionOrMessage instanceof Message) { if (interactionOrMessage instanceof Message) {
return askMessage.edit({ content }); return askMessage.edit({ content });
} }
return interactionOrMessage.editReply({ return interactionOrMessage.editReply({ content });
content: content
});
} }
} }
} }