apexify.js 3.3.11 → 4.0.1

package/dist/ai/functions/draw.js

@@ -7,84 +7,14 @@ exports.aiImagine = void 0;
 const google_translate_1 = __importDefault(require("@iamtraction/google-translate"));
 const sharp_1 = __importDefault(require("sharp"));
 const discord_js_1 = require("discord.js");
+const axios_1 = __importDefault(require("axios"));
+const validOptions_1 = require("./validOptions");
 const api_1 = __importDefault(require("api"));
 const sdk = (0, api_1.default)("@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]",
-];
-async function aiImagine(message, numOfImages, textToDraw, hercai, imageModel, nsfw, nsfwKeyWords) {
-    var _a, _b, _c, _d, _e;
-    const maxRetryAttempts = 3;
-    const retryInterval = 10000;
+async function aiImagine(message, numOfImages, textToDraw, hercai, imageModel, nsfw, nsfwKeyWords, deepCheck, enhancer, buttons) {
+    var _a, _b, _c, _d, _e, _f, _g, _h;
+    const maxRetryAttempts = 4;
+    const retryInterval = 5000;
     let response;
     async function retry(fn, retriesLeft = maxRetryAttempts) {
         try {
@@ -112,21 +42,42 @@ async function aiImagine(message, numOfImages, textToDraw, hercai, imageModel, n
         const imageUrls = [];
         for (let _0x4d7fb6 = 0x0; _0x4d7fb6 < numOfImages; _0x4d7fb6++) {
             try {
-                if (validHercaiModals.includes(imageModel)) {
+                if (validOptions_1.validOptions.validHercaiModals.includes(imageModel)) {
                     response = await retry(() => hercai.drawImage({
                         model: imageModel,
                         prompt: translatedText.text,
                     }));
                 }
-                else if (validProdiaModals.includes(imageModel)) {
+                else if (validOptions_1.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 ((_b = message.channel) === null || _b === void 0 ? void 0 : _b.sendTyping());
                     const generatedJobId = generateResponse.data.job;
                     response = await checkJobStatus(generatedJobId);
                 }
+                else if (validOptions_1.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 ((_c = message.channel) === null || _c === void 0 ? void 0 : _c.sendTyping());
+                    const generatedJobId = generateResponse.data.job;
+                    response = await checkJobStatus(generatedJobId);
+                }
                 else {
                     throw new Error("Invalid modal name.");
                 }
@@ -147,20 +98,11 @@ async function aiImagine(message, numOfImages, textToDraw, hercai, imageModel, n
             }
             let buffferedImage;
             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();
-                });
-                await ((_c = message.channel) === null || _c === void 0 ? void 0 : _c.sendTyping());
-                buffferedImage = Buffer.from(res, "binary");
+                const res = await retry(() => axios_1.default.get(response.url, {
+                    responseType: "arraybuffer",
+                }));
+                await ((_d = message.channel) === null || _d === void 0 ? void 0 : _d.sendTyping());
+                buffferedImage = Buffer.from(res.data, "binary");
                 const resizedImage = await (0, sharp_1.default)(buffferedImage)
                     .resize({
                     width: 0x320,
@@ -180,10 +122,411 @@ async function aiImagine(message, numOfImages, textToDraw, hercai, imageModel, n
             }
             else {
                 const imageUrl = response.url || response;
-                await ((_d = message.channel) === null || _d === void 0 ? void 0 : _d.sendTyping());
+                await ((_e = message.channel) === null || _e === void 0 ? void 0 : _e.sendTyping());
                 if (nsfw) {
                     const textToCheck = await attemptImageCaptioning(imageUrl);
                     const nsfwWords = [
+                        "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",
@@ -266,11 +609,19 @@ async function aiImagine(message, numOfImages, textToDraw, hercai, imageModel, n
                     ];
                     if (nsfwKeyWords.length > 0) {
                         if (textToCheck && nsfwKeyWords.some(word => textToCheck === null || textToCheck === void 0 ? void 0 : 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 (deepCheck) {
+                        if (translatedText.text && nsfwWords.some(word => { var _a; return (_a = translatedText.text) === null || _a === void 0 ? void 0 : _a.includes(word); })) {
+                            return message.reply("Warning ⚠️. Your prompt contains **`NSFW/Illegal/Prohibited`** words. Please refrain from doing this.");
+                        }
+                        if (translatedText.text && nsfwKeyWords.some(word => { var _a; return (_a = translatedText.text) === null || _a === void 0 ? void 0 : _a.includes(word); })) {
+                            return message.reply("Warning ⚠️. Your prompt contains **`NSFW/Illegal/Prohibited`** words. Please refrain from doing this.");
                         }
                     }
                     if (textToCheck && nsfwWords.some(word => textToCheck === null || textToCheck === void 0 ? void 0 : 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.");
                     }
                 }
                 const attach = new discord_js_1.AttachmentBuilder(imageUrl, { name: `image_${_0x4d7fb6 + 1}.png` });
@@ -306,12 +657,36 @@ async function aiImagine(message, numOfImages, textToDraw, hercai, imageModel, n
                 .setValue(`process_${index + 1}`);
             selectMenu.addOptions(selectOption);
         });
-        await ((_e = message.channel) === null || _e === void 0 ? void 0 : _e.sendTyping());
+        await ((_f = message.channel) === null || _f === void 0 ? void 0 : _f.sendTyping());
         const row1 = new discord_js_1.ActionRowBuilder().addComponents(...buttonsRow1);
         const row2 = new discord_js_1.ActionRowBuilder().addComponents(selectMenu);
+        if (enhancer && enhancer.enhanceModal) {
+            for (const imageUrl of imageUrls) {
+                try {
+                    if (!((_g = validOptions_1.validOptions.validEnhancers) === null || _g === void 0 ? void 0 : _g.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 ((_h = message.channel) === null || _h === void 0 ? void 0 : _h.sendTyping());
+                    const generatedJobId = generateResponse.data.job;
+                    const enhancedImageUrl = await checkJobStatus(generatedJobId);
+                    if (enhancedImageUrl) {
+                        attachData.push(new discord_js_1.AttachmentBuilder(enhancedImageUrl, { name: `enhanced_image_${imageUrls.indexOf(imageUrl) + 1}.png` }));
+                    }
+                }
+                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 },
         });
         return imageUrls;
@@ -357,16 +732,14 @@ async function attemptImageCaptioning(imageUrl) {
         const maxRetries = 3;
         const fetchData = async () => {
             try {
-                const response = await fetch(`https://api-inference.huggingface.co/models/Salesforce/blip-image-captioning-base`, {
-                    method: 'POST',
+                const response = await axios_1.default.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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`,
                     },
-                    body: JSON.stringify({ image: imageUrl })
                 });
-                if (response.ok) {
-                    return await response.json();
+                if (response.status === 200) {
+                    return response.data[0].generated_text;
                 }
                 else {
                     console.error(`Failed to fetch image captioning API: ${response.statusText}`);
@@ -383,6 +756,7 @@ async function attemptImageCaptioning(imageUrl) {
                 return await fetchData();
             }
             catch (error) {
+                console.error(`Error fetching data (Retry ${retryCount + 1}): ${error.message}`);
                 retryCount++;
             }
         }