apexify.js 3.3.10 → 4.0.0

package/lib/ai/functions/draw.ts

@@ -1,94 +1,25 @@
 import translate from "@iamtraction/google-translate";
 import sharp from "sharp";
 import { ButtonBuilder, StringSelectMenuBuilder, StringSelectMenuOptionBuilder, ButtonStyle, ActionRowBuilder, AttachmentBuilder, AttachmentData } from "discord.js";
+import axios from "axios";
+import { validOptions, ImageModals } from './validOptions';
 import api from "api";
 
-const sdk: any = api("@prodia/v1.3.0#be019b2kls0gqss3"); 
-sdk.auth("43435e1c-cab1-493f-a224-f51e4b97ce8d");
-
-const validHercaiModals = [
-  "v1",
-  "v2",
-  "v2-beta",
-  "lexica",
-  "prodia",
-  "animefy",
-  "raava",
-  "shonin",
-  "v3",
-];
-
-const validProdiaModals = [
-  "3Guofeng3_v34.safetensors [50f420de]",
-  "absolutereality_V16.safetensors [37db0fc3]",
-  "absolutereality_v181.safetensors [3d9d4d2b]",
-  "amIReal_V41.safetensors [0a8a2e61]",
-  "analog-diffusion-1.0.ckpt [9ca13f02]",
-  "anythingv3_0-pruned.ckpt [2700c435]",
-  "anything-v4.5-pruned.ckpt [65745d25]",
-  "anythingV5_PrtRE.safetensors [893e49b9]",
-  "AOM3A3_orangemixs.safetensors [9600da17]",
-  "blazing_drive_v10g.safetensors [ca1c1eab]",
-  "cetusMix_Version35.safetensors [de2f2560]",
-  "childrensStories_v13D.safetensors [9dfaabcb]",
-  "childrensStories_v1SemiReal.safetensors [a1c56dbb]",
-  "childrensStories_v1ToonAnime.safetensors [2ec7b88b]",
-  "Counterfeit_v30.safetensors [9e2a8f19]",
-  "cuteyukimixAdorable_midchapter3.safetensors [04bdffe6]",
-  "cyberrealistic_v33.safetensors [82b0d085]",
-  "dalcefo_v4.safetensors [425952fe]",
-  "deliberate_v2.safetensors [10ec4b29]",
-  "deliberate_v3.safetensors [afd9d2d4]",
-  "dreamlike-anime-1.0.safetensors [4520e090]",
-  "dreamlike-diffusion-1.0.safetensors [5c9fd6e0]",
-  "dreamlike-photoreal-2.0.safetensors [fdcf65e7]",
-  "dreamshaper_6BakedVae.safetensors [114c8abb]",
-  "dreamshaper_7.safetensors [5cf5ae06]",
-  "dreamshaper_8.safetensors [9d40847d]",
-  "edgeOfRealism_eorV20.safetensors [3ed5de15]",
-  "EimisAnimeDiffusion_V1.ckpt [4f828a15]",
-  "elldreths-vivid-mix.safetensors [342d9d26]",
-  "epicrealism_naturalSinRC1VAE.safetensors [90a4c676]",
-  "ICantBelieveItsNotPhotography_seco.safetensors [4e7a3dfd]",
-  "juggernaut_aftermath.safetensors [5e20c455]",
-  "lofi_v4.safetensors [ccc204d6]",
-  "lyriel_v16.safetensors [68fceea2]",
-  "majicmixRealistic_v4.safetensors [29d0de58]",
-  "mechamix_v10.safetensors [ee685731]",
-  "meinamix_meinaV9.safetensors [2ec66ab0]",
-  "meinamix_meinaV11.safetensors [b56ce717]",
-  "neverendingDream_v122.safetensors [f964ceeb]",
-  "openjourney_V4.ckpt [ca2f377f]",
-  "pastelMixStylizedAnime_pruned_fp16.safetensors [793a26e8]",
-  "portraitplus_V1.0.safetensors [1400e684]",
-  "protogenx34.safetensors [5896f8d5]",
-  "Realistic_Vision_V1.4-pruned-fp16.safetensors [8d21810b]",
-  "Realistic_Vision_V2.0.safetensors [79587710]",
-  "Realistic_Vision_V4.0.safetensors [29a7afaa]",
-  "Realistic_Vision_V5.0.safetensors [614d1063]",
-  "redshift_diffusion-V10.safetensors [1400e684]",
-  "revAnimated_v122.safetensors [3f4fefd9]",
-  "rundiffusionFX25D_v10.safetensors [cd12b0ee]",
-  "rundiffusionFX_v10.safetensors [cd4e694d]",
-  "sdv1_4.ckpt [7460a6fa]",
-  "v1-5-pruned-emaonly.safetensors [d7049739]",
-  "v1-5-inpainting.safetensors [21c7ab71]",
-  "shoninsBeautiful_v10.safetensors [25d8c546]",
-  "theallys-mix-ii-churned.safetensors [5d9225a4]",
-  "timeless-1.0.ckpt [7c4971d4]",
-  "toonyou_beta6.safetensors [980f6b15]",
-];
+const sdk: any = api("@prodia/v1.3.0#be019b2kls0gqss3");
 async function aiImagine(
   message: any,
   numOfImages: number,
   textToDraw: string,
   hercai: any,
-  imageModel: string,
+  imageModel: ImageModals,
   nsfw: boolean,
   nsfwKeyWords: string[],
+  deepCheck: boolean,
+  enhancer: any,
+  buttons: any[]
 ) {
-  const maxRetryAttempts = 3;
-  const retryInterval = 10000;
+  const maxRetryAttempts = 4;
+  const retryInterval = 5000;
   let response: any;
   async function retry(fn: any, retriesLeft = maxRetryAttempts) {
     try {
@@ -97,6 +28,7 @@ async function aiImagine(
       if (retriesLeft === 0) {
         throw error;
       }
+
       await new Promise((resolve) => setTimeout(resolve, retryInterval));
       return retry(fn, retriesLeft - 1);
     }
@@ -124,17 +56,40 @@ async function aiImagine(
 
     for (let _0x4d7fb6 = 0x0; _0x4d7fb6 < numOfImages; _0x4d7fb6++) {
       try {
-        if (validHercaiModals.includes(imageModel)) {
+        if (validOptions.validHercaiModals.includes(imageModel)) {
           response = await retry(() =>
             hercai.drawImage({
               model: imageModel,
               prompt: translatedText.text,
             }),
           );
-        } else if (validProdiaModals.includes(imageModel)) {
+        } else if (validOptions.validProdiaModals.includes(imageModel)) {
           const generateResponse = await sdk.generate({
             model: imageModel,
             prompt: translatedText.text,
+            negative_prompt: enhancer.negative_prompt,
+            style_preset: enhancer.imgStyle,
+            cfg_scale: enhancer.cfgScale,
+            sampler: enhancer.sampler,
+            seed: enhancer.seed,
+            steps: enhancer.steps,
+            });
+
+          await message.channel?.sendTyping();
+
+          const generatedJobId = generateResponse.data.job;
+          response = await checkJobStatus(generatedJobId);
+        } else if (validOptions.validSXDL.includes(imageModel)) {
+
+          const generateResponse = await sdk.sdxlGenerate({
+            model: imageModel,
+            prompt: translatedText.text,
+            negative_prompt: enhancer.negative_prompt,
+            style_preset: enhancer.imgStyle,
+            cfg_scale: enhancer.cfgScale,
+            sampler: enhancer.sampler,
+            seed: enhancer.seed,
+            steps: enhancer.steps,
           });
 
           await message.channel?.sendTyping();
@@ -159,45 +114,38 @@ async function aiImagine(
       let buffferedImage: any;
 
       if (imageModel === "v3") {
-        const res = await retry(async () => {
-            const resp = await fetch(response.url, {
-                method: 'GET',
-                headers: {
-                    'Content-Type': 'application/octet-stream'
-                }
-            });
-            if (!resp.ok) {
-                throw new Error("Failed to fetch image data.");
-            }
-            return await resp.arrayBuffer();
-        });
-    
+        const res = await retry(() =>
+          axios.get(response.url, {
+            responseType: "arraybuffer",
+          }),
+        );
+
         await message.channel?.sendTyping();
-    
-        buffferedImage = Buffer.from(res, "binary");
-    
+
+        buffferedImage = Buffer.from(res.data, "binary");
+
         const resizedImage = await sharp(buffferedImage)
-            .resize({
-                width: 0x320,
-            })
-            .toBuffer();
-    
+          .resize({
+            width: 0x320,
+          })
+          .toBuffer();
+
         const attachment = {
-            file: resizedImage,
-            name: `image_${_0x4d7fb6 + 1}.png`,
+          file: resizedImage,
+          name: `image_${_0x4d7fb6 + 1}.png`,
         } as AttachmentData;
-    
+
         attachData.push(attachment);
-    
+
         const urlButton = new ButtonBuilder()
-            .setStyle(ButtonStyle.Link)
-            .setLabel(`Image ${_0x4d7fb6 + 1}`)
-            .setURL(response.url);
-    
+          .setStyle(ButtonStyle.Link)
+          .setLabel(`Image ${_0x4d7fb6 + 1}`)
+          .setURL(response.url);
+
         buttonsRow1.push(urlButton);
-    
+
         imageUrls.push(response.url);
-    } else {
+      } else {
         const imageUrl = response.url || response;
 
         await message.channel?.sendTyping();
@@ -206,6 +154,407 @@ async function aiImagine(
           const textToCheck = await attemptImageCaptioning(imageUrl);
           
           const nsfwWords: string[] = [
+            "2g1c",
+            "2 girls 1 cup",
+            "acrotomophilia",
+            "alabama hot pocket",
+            "alaskan pipeline",
+            "anal",
+            "anilingus",
+            "anus",
+            "apeshit",
+            "arsehole",
+            "ass",
+            "asshole",
+            "assmunch",
+            "auto erotic",
+            "autoerotic",
+            "babeland",
+            "baby batter",
+            "baby juice",
+            "ball gag",
+            "ball gravy",
+            "ball kicking",
+            "ball licking",
+            "ball sack",
+            "ball sucking",
+            "bangbros",
+            "bangbus",
+            "bareback",
+            "barely legal",
+            "barenaked",
+            "bastard",
+            "bastardo",
+            "bastinado",
+            "bbw",
+            "bdsm",
+            "beaner",
+            "beaners",
+            "beaver cleaver",
+            "beaver lips",
+            "beastiality",
+            "bestiality",
+            "big black",
+            "big breasts",
+            "big knockers",
+            "big tits",
+            "bimbos",
+            "birdlock",
+            "bitch",
+            "bitches",
+            "black cock",
+            "blonde action",
+            "blonde on blonde action",
+            "blowjob",
+            "blow job",
+            "blow your load",
+            "blue waffle",
+            "blumpkin",
+            "bollocks",
+            "bondage",
+            "boner",
+            "boob",
+            "boobs",
+            "booty call",
+            "brown showers",
+            "brunette action",
+            "bukkake",
+            "bulldyke",
+            "bullet vibe",
+            "bullshit",
+            "bung hole",
+            "bunghole",
+            "busty",
+            "butt",
+            "buttcheeks",
+            "butthole",
+            "camel toe",
+            "camgirl",
+            "camslut",
+            "camwhore",
+            "carpet muncher",
+            "carpetmuncher",
+            "chocolate rosebuds",
+            "cialis",
+            "circlejerk",
+            "cleveland steamer",
+            "clit",
+            "clitoris",
+            "clover clamps",
+            "clusterfuck",
+            "cock",
+            "cocks",
+            "coprolagnia",
+            "coprophilia",
+            "cornhole",
+            "coon",
+            "coons",
+            "creampie",
+            "cum",
+            "cumming",
+            "cumshot",
+            "cumshots",
+            "cunnilingus",
+            "cunt",
+            "darkie",
+            "date rape",
+            "daterape",
+            "deep throat",
+            "deepthroat",
+            "dendrophilia",
+            "dick",
+            "dildo",
+            "dingleberry",
+            "dingleberries",
+            "dirty pillows",
+            "dirty sanchez",
+            "doggie style",
+            "doggiestyle",
+            "doggy style",
+            "doggystyle",
+            "dog style",
+            "dolcett",
+            "domination",
+            "dominatrix",
+            "dommes",
+            "donkey punch",
+            "double dong",
+            "double penetration",
+            "dp action",
+            "dry hump",
+            "dvda",
+            "eat my ass",
+            "ecchi",
+            "ejaculation",
+            "erotic",
+            "erotism",
+            "escort",
+            "eunuch",
+            "fag",
+            "faggot",
+            "fecal",
+            "felch",
+            "fellatio",
+            "feltch",
+            "female squirting",
+            "femdom",
+            "figging",
+            "fingerbang",
+            "fingering",
+            "fisting",
+            "foot fetish",
+            "footjob",
+            "frotting",
+            "fuck",
+            "fuck buttons",
+            "fuckin",
+            "fucking",
+            "fucktards",
+            "fudge packer",
+            "fudgepacker",
+            "futanari",
+            "gangbang",
+            "gang bang",
+            "gay sex",
+            "genitals",
+            "giant cock",
+            "girl on",
+            "girl on top",
+            "girls gone wild",
+            "goatcx",
+            "goatse",
+            "god damn",
+            "gokkun",
+            "golden shower",
+            "goodpoop",
+            "goo girl",
+            "goregasm",
+            "grope",
+            "group sex",
+            "g-spot",
+            "guro",
+            "hand job",
+            "handjob",
+            "hard core",
+            "hardcore",
+            "hentai",
+            "homoerotic",
+            "honkey",
+            "hooker",
+            "horny",
+            "hot carl",
+            "hot chick",
+            "how to kill",
+            "how to murder",
+            "huge fat",
+            "humping",
+            "incest",
+            "intercourse",
+            "jack off",
+            "jail bait",
+            "jailbait",
+            "jelly donut",
+            "jerk off",
+            "jigaboo",
+            "jiggaboo",
+            "jiggerboo",
+            "jizz",
+            "juggs",
+            "kike",
+            "kinbaku",
+            "kinkster",
+            "kinky",
+            "knobbing",
+            "leather restraint",
+            "leather straight jacket",
+            "lemon party",
+            "livesex",
+            "lolita",
+            "lovemaking",
+            "make me come",
+            "male squirting",
+            "masturbate",
+            "masturbating",
+            "masturbation",
+            "menage a trois",
+            "milf",
+            "missionary position",
+            "mong",
+            "motherfucker",
+            "mound of venus",
+            "mr hands",
+            "muff diver",
+            "muffdiving",
+            "nambla",
+            "nawashi",
+            "negro",
+            "neonazi",
+            "nigga",
+            "nigger",
+            "nig nog",
+            "nimphomania",
+            "nipple",
+            "nipples",
+            "nsfw",
+            'nsfw images',
+            "nude",
+            "nudity",
+            "nutten",
+            "nympho",
+            "nymphomania",
+            "octopussy",
+            "omorashi",
+            "one cup two girls",
+            'one guy one jar',
+            "orgasm",
+            "orgy",
+            'paedophile',
+            "paki",
+            "panties",
+            "panty",
+            "pedobear",
+            "pedophile",
+            'pegging',
+            'penis',
+            "phone sex",
+            'piece of shit',
+            "pikey",
+            "pissing",
+            "piss pig",
+            "pisspig",
+            "playboy",
+            "pleasure chest",
+            "pole smoker",
+            "ponyplay",
+            "poof",
+            "poon",
+            "poontang",
+            "punany",
+            "poop chute",
+            "poopchute",
+            "porn",
+            "porno",
+            "pornography",
+            "prince albert piercing",
+            "pthc",
+            "pubes",
+            "pussy",
+            "queaf",
+            "queef",
+            "quim",
+            "raghead",
+            "raging boner",
+            "rape",
+            "raping",
+            "rapist",
+            "rectum",
+            "reverse cowgirl",
+            "rimjob",
+            "rimming",
+            "rosy palm",
+            "rosy palm and her 5 sisters",
+            "rusty trombone",
+            "sadism",
+            "santorum",
+            "scat",
+            "schlong",
+            "scissoring",
+            "semen",
+            "sex",
+            "sexcam",
+            "sexo",
+            "sexy",
+            "sexual",
+            "sexually",
+            "sexuality",
+            "shaved beaver",
+            "shaved pussy",
+            "shemale",
+            "shibari",
+            "shit",
+            "shitblimp",
+            "shitty",
+            "shota",
+            "shrimping",
+            "skeet",
+            "slanteye",
+            "slut",
+            "s&m",
+            "smut",
+            "snatch",
+            "snowballing",
+            "sodomize",
+            "sodomy",
+            "spastic",
+            "spic",
+            "splooge",
+            "splooge moose",
+            "spooge",
+            "spread legs",
+            "spunk",
+            "strap on",
+            "strapon",
+            "strappado",
+            "strip club",
+            "style doggy",
+            "suck",
+            "sucks",
+            "suicide girls",
+            "sultry women",
+            "swastika",
+            "swinger",
+            "tainted love",
+            "taste my",
+            "tea bagging",
+            "threesome",
+            "throating",
+            "thumbzilla",
+            "tied up",
+            "tight white",
+            "tit",
+            "tits",
+            "titties",
+            "titty",
+            "topless",
+            "tosser",
+            "towelhead",
+            "tranny",
+            "tribadism",
+            "tub girl",
+            'tubgirl',
+            "tushy",
+            "twat",
+            "twink",
+            "twinkie",
+            "two girls one cup",
+            "undressing",
+            'upskirt',
+            "urethra play",
+            "urophilia",
+            "vagina",
+            "venus mound",
+            "viagra",
+            "vibrator",
+            "violet wand",
+            "vorarephilia",
+            "voyeur",
+            "voyeurweb",
+            "voyuer",
+            "vulva",
+            "wank",
+            "wetback",
+            "wet dream",
+            "white power",
+            "whore",
+            "worldsex",
+            "wrapping men",
+            "wrinkled starfish",
+            "xx",
+            "xxx",
+            "yaoi",
+            "yellow showers",
+            "yiffy",
+            "zoophilia",            
             "anal",
             "arousal",
             "balls",
@@ -289,11 +638,22 @@ async function aiImagine(
 
         if (nsfwKeyWords.length > 0) {
          if (textToCheck && nsfwKeyWords.some(word => textToCheck?.includes(word))) {
-          return message.reply("Warning ⚠️. The generated image contatining nsfw content. Turn off nsfw to sedn nsfw images.");
+          return message.reply("Warning ⚠️. The generated image contatining nsfw content. Turn off nsfw to send nsfw images.");
          } 
         }
-          if (textToCheck && nsfwWords.some(word => textToCheck?.includes(word))) {
-              return message.reply("Warning ⚠️. The generated image contatining nsfw content. Turn off nsfw to sedn nsfw images.");
+
+      if (deepCheck) {
+        if (translatedText.text && nsfwWords.some(word => translatedText.text?.includes(word))) {
+          return message.reply("Warning ⚠️. Your prompt contains **`NSFW/Illegal/Prohibited`** words. Please refrain from doing this.");
+        }
+        
+        if (translatedText.text && nsfwKeyWords.some(word => translatedText.text?.includes(word))) {
+          return message.reply("Warning ⚠️. Your prompt contains **`NSFW/Illegal/Prohibited`** words. Please refrain from doing this.");
+        }
+      }
+      
+        if (textToCheck && nsfwWords.some(word => textToCheck?.includes(word))) {
+              return message.reply("Warning ⚠️. The generated image contatining nsfw content. Turn off nsfw to send nsfw images.");
           }
         }
 
@@ -350,9 +710,40 @@ async function aiImagine(
     );
     const row2 = new ActionRowBuilder().addComponents(selectMenu);
 
+    if (enhancer && enhancer.enhanceModal) {
+      for (const imageUrl of imageUrls) {
+        try {
+
+        if (!validOptions.validEnhancers?.includes(enhancer.enhanceModal)) {
+          return await message.reply({ content: `Invalid enhancer modal (~~${enhancer.enhanceModal}~~). Please check documentation online at apexifyjs.jedi-studio.com.`})
+        }
+
+          const generateResponse = await sdk.upscale({
+            resize: 4,
+            model: enhancer.enhanceModal,
+            imageUrl,
+          });
+    
+          await message.channel?.sendTyping();
+    
+          const generatedJobId = generateResponse.data.job;
+          const enhancedImageUrl = await checkJobStatus(generatedJobId);
+    
+          if (enhancedImageUrl) {
+            attachData.push(new AttachmentBuilder(
+              enhancedImageUrl,
+              { name: `enhanced_image_${imageUrls.indexOf(imageUrl) + 1}.png` },
+            ) as AttachmentData);
+          }
+        } catch (error) {
+          console.error(`Error enhancing image at URL ${imageUrl}:`, error);
+        }
+      }
+    }
+    const allRows = [...buttons,  row1, row2]
     await message.reply({
       files: attachData,
-      components: [row1, row2],
+      components: allRows,
       allowedMentions: { repliedUser: false },
     });
 
@@ -392,44 +783,51 @@ async function checkJobStatus(jobId: number | string | any) {
 
 async function attemptImageCaptioning(imageUrl: string) {
   try {
-      let retryCount = 0;
-      const maxRetries = 3;
-
-      const fetchData = async () => {
-          try {
-              const response = await fetch(`https://api-inference.huggingface.co/models/Salesforce/blip-image-captioning-base`, {
-                  method: 'POST',
-                  headers: {
-                      "Content-Type": "application/json",
-                      Authorization: `Bearer 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`,
-                  },
-                  body: JSON.stringify({ image: imageUrl })
-              });
-
-              if (response.ok) {
-                  return await response.json();
-              } else {
-                  console.error(`Failed to fetch image captioning API: ${response.statusText}`);
-                  return null;
-              }
-          } catch (error: any) {
-              console.error(`Error fetching data: ${error.message}`);
-              throw error;
-          }
-      };
+    let retryCount = 0;
+    const maxRetries = 3;
 
-      while (retryCount < maxRetries) {
-          try {
-              return await fetchData();
-          } catch (error: any) {
-              retryCount++;
-          }
+    const fetchData = async () => {
+      try {
+        const response = await axios.post(
+          `https://api-inference.huggingface.co/models/Salesforce/blip-image-captioning-base`,
+          { image: imageUrl },
+          {
+            headers: {
+              "Content-Type": "application/json",
+              Authorization: `Bearer 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`,
+            },
+          },
+        );
+
+        if (response.status === 200) {
+          return response.data[0].generated_text;
+        } else {
+          console.error(
+            `Failed to fetch image captioning API: ${response.statusText}`,
+          );
+          return null;
+        }
+      } catch (error: any) {
+        console.error(`Error fetching data: ${error.message}`);
+        throw error;
       }
+    };
 
-      return null;
+    while (retryCount < maxRetries) {
+      try {
+        return await fetchData();
+      } catch (error: any) {
+        console.error(
+          `Error fetching data (Retry ${retryCount + 1}): ${error.message}`,
+        );
+        retryCount++;
+      }
+    }
+
+    return null;
   } catch (error: any) {
-      console.error(`Error in attemptImageCaptioning: ${error.message}`);
-      return null;
+    console.error(`Error in attemptImageCaptioning: ${error.message}`);
+    return null;
   }
 }