aiserver

commands/message/image/anime.js

@@ -0,0 +1,63 @@
+const { createEmbed } = require("../../../labscore/utils/embed");
+const { editOrReply } = require("../../../labscore/utils/message");
+
+const superagent = require('superagent');
+
+module.exports = {
+  name: 'anime',
+  label: 'query',
+  metadata: {
+    description: 'Uses Stable Diffusion to generate an image using anime styling from a text prompt.',
+    description_short: 'AI Anime image generation',
+    explicit: true,
+    examples: ['anime otter'],
+    category: 'image',
+    usage: 'anime <text>'
+  },
+  ratelimit: {
+    type: 'guild',
+    limit: 1,
+    duration: 5000
+  },
+  run: async (context, args) => {
+    if(!context.channel.nsfw) return editOrReply(context, {embeds:[createEmbed("nsfw", context)]})
+    if(args.query.length == 0) return editOrReply(context, {embeds:[createEmbed("warning", context, "Missing prompt")]})
+    let response = await editOrReply(context, { embeds: [createEmbed("loading", context, `Generating image...`)] })
+
+    let noticeTimer = setTimeout(()=>{
+      let emb = createEmbed("loading", context, `Generating image...`)
+      emb.footer = {
+        text: "This might take a moment to complete."
+      };
+      response.edit({ embeds: [ emb ] });
+    }, 45000)
+
+    try{
+      let t = Date.now();
+
+      let img = await superagent.get(`${process.env.AI_SERVER}/anime`)
+        .query({
+          prompt: args.query
+        })
+
+      clearTimeout(noticeTimer)
+      
+      if(img.body.message) return await response.edit({embeds:[createEmbed("warning", context, img.body.message)]})
+
+      let embeds = [];
+      let files = [];
+      
+      embeds.push(createEmbed("default", context, {image: {url:`attachment://anime.png`}, url: `https://example.com`, footer: { iconUrl: `https://cdn.discordapp.com/avatars/${context.application.id}/${context.application.icon}.png?size=256`, text: `${context.application.name} • Took ${((Date.now() - t) / 1000).toFixed(2)}s` }}))
+      files.push({
+        filename: `anime.png`,
+        value: Buffer.from(img.body.image, 'base64')
+      })
+      
+      await response.edit({ embeds, files })
+    }catch(e){
+      clearTimeout(noticeTimer)
+      console.log(e)
+      await response.edit({embeds:[createEmbed("error", context, `Image generation failed.`)]})
+    }
+  },
+};

commands/message/image/stablediffusion.js

@@ -33,7 +33,7 @@ module.exports = {
     try{
       let t = Date.now();
 
-      let img = await superagent.get(`${process.env.STABLE_DIFFUSION_SERVER}/generate`)
+      let img = await superagent.get(`${process.env.AI_SERVER}/generate`)
         .query({
           prompt: args.query
         })