pissbot-9000/commands/message/genai/emogen.js

const { emogen } = require("#api");

const { createEmbed } = require("#utils/embed");
const { iconPill } = require("#utils/markdown");
const { editOrReply } = require("#utils/message");
const { hasFeature } = require("#utils/testing");

// TODO: Turn this into a general purpose permissions constant
const { Permissions } = require("detritus-client/lib/constants");

module.exports = {
  name: 'emogen',
  label: 'text',
  metadata: {
    description: `${iconPill("generative_ai", "LIMITED TESTING")}\n\nEmogen`,
    description_short: 'Emogen',
    examples: ['emogen otter eating a slice of pizza'],
    category: 'limited',
    usage: 'emogen <prompt>'
  },
  args: [
    { name: 'style', default: 'microsoft', required: false, help: "Emogen Style" }
  ],
  permissionsClient: [Permissions.EMBED_LINKS, Permissions.SEND_MESSAGES, Permissions.USE_EXTERNAL_EMOJIS, Permissions.ATTACH_FILES, Permissions.READ_MESSAGE_HISTORY],
  run: async (context, args) => {
    if(!await hasFeature(context, "ai/imagen")) return;
    context.triggerTyping();

    if(!args.text) return editOrReply(context, createEmbed("warning", context, `Missing Parameter (prompt).`))

    try{
      await editOrReply(context, createEmbed("loading", context, "Generating emoji..."));

      let res = await emogen(context, args.text, args.style)
      
      let file = `emogen.${Date.now()}.png`;

      return editOrReply(context, {
        embeds: [
          createEmbed("image", context, {
            url: file
          })
        ],
        files: [{
          filename: file,
          value: res.response.body
        }]});
    } catch(e){
      console.log(e)
      if(e.response?.body?.message) return editOrReply(context, createEmbed("error", context, e.response.body.message))
      return editOrReply(context, createEmbed("error", context, `Unable to generate emoji.`))
    }
  }
};