apexify.js 3.0.1 → 3.1.0

package/lib/ai/functions/draw.js

@@ -1,514 +0,0 @@
-"use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __generator = (this && this.__generator) || function (thisArg, body) {
-    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
-    return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
-    function verb(n) { return function (v) { return step([n, v]); }; }
-    function step(op) {
-        if (f) throw new TypeError("Generator is already executing.");
-        while (g && (g = 0, op[0] && (_ = 0)), _) try {
-            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
-            if (y = 0, t) op = [op[0] & 2, t.value];
-            switch (op[0]) {
-                case 0: case 1: t = op; break;
-                case 4: _.label++; return { value: op[1], done: false };
-                case 5: _.label++; y = op[1]; op = [0]; continue;
-                case 7: op = _.ops.pop(); _.trys.pop(); continue;
-                default:
-                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
-                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
-                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
-                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
-                    if (t[2]) _.ops.pop();
-                    _.trys.pop(); continue;
-            }
-            op = body.call(thisArg, _);
-        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
-        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
-    }
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.drawImage = void 0;
-var translate = require("@iamtraction/google-translate");
-var sharp = require("sharp");
-var discord_js_1 = require("discord.js");
-var axios = require("axios");
-var api = require("api");
-var sdk = api("@prodia/v1.3.0#be019b2kls0gqss3");
-sdk.auth("43435e1c-cab1-493f-a224-f51e4b97ce8d");
-var validHercaiModals = [
-    "v1",
-    "v2",
-    "v2-beta",
-    "lexica",
-    "prodia",
-    "animefy",
-    "raava",
-    "shonin",
-    "v3",
-];
-var 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]",
-];
-function drawImage(message, numOfImages, textToDraw, hercai, imageModel, nsfw) {
-    return __awaiter(this, void 0, void 0, function () {
-        function retry(fn, retriesLeft) {
-            if (retriesLeft === void 0) { retriesLeft = maxRetryAttempts; }
-            return __awaiter(this, void 0, void 0, function () {
-                var error_1;
-                return __generator(this, function (_a) {
-                    switch (_a.label) {
-                        case 0:
-                            _a.trys.push([0, 2, , 4]);
-                            return [4 /*yield*/, fn()];
-                        case 1: return [2 /*return*/, _a.sent()];
-                        case 2:
-                            error_1 = _a.sent();
-                            if (retriesLeft === 0) {
-                                throw error_1;
-                            }
-                            console.error("Retry failed, ".concat(retriesLeft, " attempts left. Error: ").concat(error_1.message));
-                            return [4 /*yield*/, new Promise(function (resolve) { return setTimeout(resolve, retryInterval); })];
-                        case 3:
-                            _a.sent();
-                            return [2 /*return*/, retry(fn, retriesLeft - 1)];
-                        case 4: return [2 /*return*/];
-                    }
-                });
-            });
-        }
-        var maxRetryAttempts, retryInterval, response, translatedText_1, attachData, buttonsRow1, imageUrls, _loop_1, _0x4d7fb6, state_1, selectMenu_1, options, row1, row2, e_1;
-        var _a;
-        return __generator(this, function (_b) {
-            switch (_b.label) {
-                case 0:
-                    maxRetryAttempts = 3;
-                    retryInterval = 10000;
-                    _b.label = 1;
-                case 1:
-                    _b.trys.push([1, 8, , 9]);
-                    if (numOfImages > 4 || numOfImages <= 0) {
-                        throw new Error("Number of images can't be greater than 4 or smaller than 0.");
-                    }
-                    return [4 /*yield*/, retry(function () {
-                            return (0, translate)(textToDraw, {
-                                from: "auto",
-                                to: "en",
-                            });
-                        })];
-                case 2:
-                    translatedText_1 = _b.sent();
-                    attachData = [];
-                    buttonsRow1 = [];
-                    imageUrls = [];
-                    _loop_1 = function (_0x4d7fb6) {
-                        var generateResponse, generatedJobId, error_2, buffferedImage, res, resizedImage, attachment, urlButton, imageUrl, textToCheck_1, nsfwWords, attach, urlButton;
-                        return __generator(this, function (_c) {
-                            switch (_c.label) {
-                                case 0:
-                                    _c.trys.push([0, 7, , 12]);
-                                    if (!validHercaiModals.includes(imageModel)) return [3 /*break*/, 2];
-                                    return [4 /*yield*/, retry(function () {
-                                            return hercai.drawImage({
-                                                model: imageModel,
-                                                prompt: translatedText_1.text,
-                                            });
-                                        })];
-                                case 1:
-                                    response = _c.sent();
-                                    return [3 /*break*/, 6];
-                                case 2:
-                                    if (!validProdiaModals.includes(imageModel)) return [3 /*break*/, 5];
-                                    return [4 /*yield*/, sdk.generate({
-                                            model: imageModel,
-                                            prompt: translatedText_1.text,
-                                        })];
-                                case 3:
-                                    generateResponse = _c.sent();
-                                    generatedJobId = generateResponse.data.job;
-                                    return [4 /*yield*/, checkJobStatus(generatedJobId)];
-                                case 4:
-                                    response = _c.sent();
-                                    return [3 /*break*/, 6];
-                                case 5: throw new Error("Invalid modal name.");
-                                case 6: return [3 /*break*/, 12];
-                                case 7:
-                                    error_2 = _c.sent();
-                                    if (!(error_2.response && error_2.response.status === 429)) return [3 /*break*/, 8];
-                                    console.error("Too many requests. Please try again later.");
-                                    return [2 /*return*/, { value: message.reply("Please wait i am in a cool down for a minute") }];
-                                case 8:
-                                    if (!(error_2.response && error_2.response.status === 500)) return [3 /*break*/, 9];
-                                    console.error("Internal server error. Please try again later.");
-                                    return [2 /*return*/, { value: message.reply("Please wait i am in a cool down for a minute") }];
-                                case 9: return [4 /*yield*/, message.reply("Please wait i am in a cool down for a minute")];
-                                case 10:
-                                    _c.sent();
-                                    throw new Error("Error processing message in file");
-                                case 11: return [3 /*break*/, 12];
-                                case 12:
-                                    buffferedImage = void 0;
-                                    if (!(imageModel === "v3")) return [3 /*break*/, 15];
-                                    return [4 /*yield*/, retry(function () {
-                                            return axios.get(response.url, {
-                                                responseType: "arraybuffer",
-                                            });
-                                        })];
-                                case 13:
-                                    res = _c.sent();
-                                    buffferedImage = Buffer.from(res.data, "binary");
-                                    return [4 /*yield*/, sharp(buffferedImage)
-                                            .resize({
-                                            width: 0x320,
-                                        })
-                                            .toBuffer()];
-                                case 14:
-                                    resizedImage = _c.sent();
-                                    attachment = {
-                                        file: resizedImage,
-                                        name: "image_".concat(_0x4d7fb6 + 1, ".png"),
-                                    };
-                                    attachData.push(attachment);
-                                    urlButton = new discord_js_1.ButtonBuilder()
-                                        .setStyle(discord_js_1.ButtonStyle.Link)
-                                        .setLabel("Image ".concat(_0x4d7fb6 + 1))
-                                        .setURL(response.url);
-                                    buttonsRow1.push(urlButton);
-                                    imageUrls.push(response.url);
-                                    return [3 /*break*/, 18];
-                                case 15:
-                                    imageUrl = response.url || response;
-                                    if (!nsfw) return [3 /*break*/, 17];
-                                    return [4 /*yield*/, attemptImageCaptioning(imageUrl)];
-                                case 16:
-                                    textToCheck_1 = _c.sent();
-                                    nsfwWords = [
-                                        "anal",
-                                        "arousal",
-                                        "balls",
-                                        "blowjob",
-                                        "busty",
-                                        "butt",
-                                        "cameltoe",
-                                        "climax",
-                                        "clitoris",
-                                        "cock",
-                                        "crotch",
-                                        "cum",
-                                        "cumshot",
-                                        "cunnilingus",
-                                        "dirty",
-                                        "dirtytalk",
-                                        "dildo",
-                                        "erect",
-                                        "erogenous",
-                                        "escort",
-                                        "explicit",
-                                        "facial",
-                                        "fetish",
-                                        "flirt",
-                                        "foreplay",
-                                        "genitals",
-                                        "groin",
-                                        "hardcore",
-                                        "hardon",
-                                        "horniness",
-                                        "horny",
-                                        "kamasutra",
-                                        "kinky",
-                                        "lewd",
-                                        "lingerie",
-                                        "lust",
-                                        "lustful",
-                                        "masturbate",
-                                        "mature",
-                                        "milf",
-                                        "naughty",
-                                        "naked",
-                                        "nipples",
-                                        "nude",
-                                        "obscene",
-                                        "oral",
-                                        "orgasm",
-                                        "penetration",
-                                        "penis",
-                                        "pleasure",
-                                        "porn",
-                                        "prostitute",
-                                        "provocative",
-                                        "pubic",
-                                        "pussy",
-                                        "seduce",
-                                        "seductive",
-                                        "sensual",
-                                        "sex",
-                                        "sexual",
-                                        "sperm",
-                                        "strip",
-                                        "striptease",
-                                        "swinger",
-                                        "testicles",
-                                        "thong",
-                                        "threesome",
-                                        "undies",
-                                        "undress",
-                                        "vagina",
-                                        "vibrator",
-                                        "wank",
-                                        "wet",
-                                        "hentai",
-                                        "bdsm",
-                                        "gay",
-                                        "lesbian",
-                                        "femboy",
-                                        "ass",
-                                    ];
-                                    if (textToCheck_1 && nsfwWords.some(function (word) { return textToCheck_1 === null || textToCheck_1 === void 0 ? void 0 : textToCheck_1.includes(word); })) {
-                                        message.reply("Warning ⚠️. The generated image contatining nsfw content. Turn off nsfw to sedn nsfw images.");
-                                    }
-                                    _c.label = 17;
-                                case 17:
-                                    attach = new discord_js_1.AttachmentBuilder(imageUrl, { name: "image_".concat(_0x4d7fb6 + 1, ".png") });
-                                    attachData.push(attach);
-                                    urlButton = new discord_js_1.ButtonBuilder()
-                                        .setStyle(discord_js_1.ButtonStyle.Link)
-                                        .setLabel("Image ".concat(_0x4d7fb6 + 1))
-                                        .setURL(imageUrl);
-                                    buttonsRow1.push(urlButton);
-                                    imageUrls.push(imageUrl);
-                                    _c.label = 18;
-                                case 18: return [4 /*yield*/, new Promise(function (resolve) { return setTimeout(resolve, 0x7d0); })];
-                                case 19:
-                                    _c.sent();
-                                    return [2 /*return*/];
-                            }
-                        });
-                    };
-                    _0x4d7fb6 = 0x0;
-                    _b.label = 3;
-                case 3:
-                    if (!(_0x4d7fb6 < numOfImages)) return [3 /*break*/, 6];
-                    return [5 /*yield**/, _loop_1(_0x4d7fb6)];
-                case 4:
-                    state_1 = _b.sent();
-                    if (typeof state_1 === "object")
-                        return [2 /*return*/, state_1.value];
-                    _b.label = 5;
-                case 5:
-                    _0x4d7fb6++;
-                    return [3 /*break*/, 3];
-                case 6:
-                    selectMenu_1 = new discord_js_1.StringSelectMenuBuilder()
-                        .setCustomId("image_processing_select")
-                        .setPlaceholder("Select Image Process");
-                    options = [
-                        "Resizing",
-                        "Color Filters",
-                        "Convert Image Extension",
-                        "Brightness",
-                        "Sepia",
-                        "Pixelate",
-                        "Blur",
-                        "Fade",
-                        "Opaque",
-                        "Grayscale",
-                        "Contrast",
-                    ];
-                    options.forEach(function (option, index) {
-                        var selectOption = new discord_js_1.StringSelectMenuOptionBuilder()
-                            .setLabel(option)
-                            .setValue("process_".concat(index + 1));
-                        selectMenu_1.addOptions(selectOption);
-                    });
-                    row1 = (_a = new discord_js_1.ActionRowBuilder()).addComponents.apply(_a, buttonsRow1);
-                    row2 = new discord_js_1.ActionRowBuilder().addComponents(selectMenu_1);
-                    return [4 /*yield*/, message.reply({
-                            files: attachData,
-                            components: [row1, row2],
-                            allowedMentions: { repliedUser: false },
-                        })];
-                case 7:
-                    _b.sent();
-                    return [2 /*return*/, imageUrls];
-                case 8:
-                    e_1 = _b.sent();
-                    console.error("Error in drawImage:", e_1.message);
-                    if (e_1.response) {
-                        console.error("Status code:", e_1.response.status);
-                        console.error("Response data:", e_1.response.data);
-                    }
-                    message.reply({
-                        content: "An error occurred while processing the images.",
-                        allowedMentions: { repliedUser: false },
-                    });
-                    return [2 /*return*/, []];
-                case 9: return [2 /*return*/];
-            }
-        });
-    });
-}
-exports.drawImage = drawImage;
-function checkJobStatus(jobId) {
-    return __awaiter(this, void 0, void 0, function () {
-        var getJobResponse, jobData, error_3;
-        return __generator(this, function (_a) {
-            switch (_a.label) {
-                case 0:
-                    _a.trys.push([0, 5, , 6]);
-                    return [4 /*yield*/, sdk.getJob({ jobId: jobId })];
-                case 1:
-                    getJobResponse = _a.sent();
-                    jobData = getJobResponse.data;
-                    if (!(jobData.status === "generating" || jobData.status === "queued")) return [3 /*break*/, 3];
-                    return [4 /*yield*/, new Promise(function (resolve) { return setTimeout(resolve, 3000); })];
-                case 2:
-                    _a.sent();
-                    return [2 /*return*/, checkJobStatus(jobId)];
-                case 3:
-                    if (jobData.status === "succeeded") {
-                        return [2 /*return*/, jobData.imageUrl];
-                    }
-                    else {
-                        console.error("Job failed:", jobData);
-                        return [2 /*return*/, null];
-                    }
-                    _a.label = 4;
-                case 4: return [3 /*break*/, 6];
-                case 5:
-                    error_3 = _a.sent();
-                    console.error(error_3);
-                    return [2 /*return*/, null];
-                case 6: return [2 /*return*/];
-            }
-        });
-    });
-}
-function attemptImageCaptioning(imageUrl) {
-    return __awaiter(this, void 0, void 0, function () {
-        var retryCount, maxRetries, fetchData, error_4, error_5;
-        var _this = this;
-        return __generator(this, function (_a) {
-            switch (_a.label) {
-                case 0:
-                    _a.trys.push([0, 7, , 8]);
-                    retryCount = 0;
-                    maxRetries = 3;
-                    fetchData = function () { return __awaiter(_this, void 0, void 0, function () {
-                        var response, error_6;
-                        return __generator(this, function (_a) {
-                            switch (_a.label) {
-                                case 0:
-                                    _a.trys.push([0, 2, , 3]);
-                                    return [4 /*yield*/, 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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",
-                                            },
-                                        })];
-                                case 1:
-                                    response = _a.sent();
-                                    if (response.status === 200) {
-                                        return [2 /*return*/, response.data[0].generated_text];
-                                    }
-                                    else {
-                                        console.error("Failed to fetch image captioning API: ".concat(response.statusText));
-                                        return [2 /*return*/, null];
-                                    }
-                                    return [3 /*break*/, 3];
-                                case 2:
-                                    error_6 = _a.sent();
-                                    console.error("Error fetching data: ".concat(error_6.message));
-                                    throw error_6;
-                                case 3: return [2 /*return*/];
-                            }
-                        });
-                    }); };
-                    _a.label = 1;
-                case 1:
-                    if (!(retryCount < maxRetries)) return [3 /*break*/, 6];
-                    _a.label = 2;
-                case 2:
-                    _a.trys.push([2, 4, , 5]);
-                    return [4 /*yield*/, fetchData()];
-                case 3: return [2 /*return*/, _a.sent()];
-                case 4:
-                    error_4 = _a.sent();
-                    console.error("Error fetching data (Retry ".concat(retryCount + 1, "): ").concat(error_4.message));
-                    retryCount++;
-                    return [3 /*break*/, 5];
-                case 5: return [3 /*break*/, 1];
-                case 6: return [2 /*return*/, null];
-                case 7:
-                    error_5 = _a.sent();
-                    console.error("Error in attemptImageCaptioning: ".concat(error_5.message));
-                    return [2 /*return*/, null];
-                case 8: return [2 /*return*/];
-            }
-        });
-    });
-}