apexify.js 3.3.11 → 4.0.1

package/lib/ai/functions/draw.ts

@@ -2,91 +2,21 @@ 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 = 4;
   const retryInterval = 5000;
@@ -126,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();
@@ -201,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",
@@ -284,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.");
           }
         }
 
@@ -345,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 },
     });
 
@@ -435,4 +831,4 @@ async function attemptImageCaptioning(imageUrl: string) {
   }
 }
 
-export { aiImagine };
+export { aiImagine };

package/lib/ai/functions/generateVoiceResponse.ts

@@ -4,9 +4,10 @@ import axios from "axios";
 import { createWorker, recognize } from "tesseract.js";
 import { chunkString } from "./chunkString";
 import { aiImagine } from "./draw";
+import { ImageModals } from './validOptions';
 import fs from "fs";
 
-const herc = new Hercai();
+const hercai = new Hercai('6eZZOdDwm6Epdzn8mnhcX9SBDkxvoNYcNj9ILS0P44=');
 let isProcessing = false;
 type ChatModelOption = any | "v3" | "v3-32k" | "turbo" | "turbo-16k" | "gemini" ;
 
@@ -16,14 +17,17 @@ async function aiVoice(
   finalText: string,
   hercai: any,
   drawValid: any,
-  imageModal: string,
-  chatModal: string | "v3" | "v3-32k" | "turbo" | "turbo-16k" | "gemini" | "starChat" | "apexChat" | 'gemma-v3' | 'gemma-v4' | 'zephyr-beta' | undefined,
+  imageModal: ImageModals,
+  chatModal: ChatModelOption,
   voiceModal: string,
   voice_code: string,
   type: string,
   apiKey: string,
   nsfw: boolean,
-  nsfwKeyWords: string[]
+  nsfwKeyWords: string[],
+  deepCheck: boolean,
+  enhancer: any,
+  buttons: any[]
 ) {
   if (message.author.bot || isProcessing || !message.guild) {
     return;
@@ -35,7 +39,7 @@ async function aiVoice(
     let msg = message.content;
 
     if (drawValid) {
-      return await aiImagine(message, numOfImages, finalText, hercai, imageModal, nsfw, nsfwKeyWords);
+      return await aiImagine(message, numOfImages, finalText, hercai, imageModal, nsfw, nsfwKeyWords, deepCheck, enhancer, buttons);
     }
 
     if (message.attachments.size > 0) {
@@ -47,7 +51,7 @@ async function aiVoice(
       msg += "\n" + text;
     }
 
-    msg = (await herc.question({
+    msg = (await hercai.betaQuestion({
       model: chatModal as ChatModelOption,
       content: msg,
     })).reply.replace(/\n/g, " ");
@@ -76,6 +80,7 @@ async function aiVoice(
             },
           });
           await message.reply({
+            components: buttons,
             files: [
               {
                 attachment: mp3Stream.data,
@@ -116,6 +121,7 @@ async function aiVoice(
         const audioData = response.data.audioStream;
         fs.writeFileSync("output.ogg", Buffer.from(audioData, "base64"));
         await message.reply({
+          components: buttons,
           files: [
             {
               attachment: "output.ogg",
@@ -128,7 +134,7 @@ async function aiVoice(
         console.error(error);
         throw new Error('Error generating audio file.');
       }
-    } else if (imageModal === "zenithAi") {
+    } else if (voiceModal === "zenithAI") {
       if (!apiKey) {
         throw new Error('Error: The zenithAi model requires a premium account for generation. Please visit our support server at https://discord.gg/4uhpr7w8wW for assistance.');
       }
@@ -154,7 +160,9 @@ async function aiVoice(
       try {
         const response = await axios.request(options);
         const audioUrl = response.data.result.audio_url;
-        await message.reply(audioUrl, {
+        await message.reply({
+          content: audioUrl,
+          components: buttons,
           allowedMentions: { repliedUser: false },
         });
       } catch (error) {