apexify.js 4.0.12 → 4.1.0

package/lib/ai/ApexModules.ts

@@ -0,0 +1,733 @@
+import { Hercai } from 'hercai';
+import axios from 'axios';
+import api  from "api";
+import translate from "@iamtraction/google-translate";
+const sdk = api("@prodia/v1.3.0#be019b2kls0gqss3");
+import { apexai, starChat, facebook_ai, yi_34b } from './modals-chat/modals';
+import { validOptions } from "./functions/validOptions";
+
+sdk.auth("43435e1c-cab1-493f-a224-f51e4b97ce8d");
+const hercai = new Hercai('6eZZOdDwm6Epdzn8mnhcX9SBDkxvoNYcNj9ILS0P44=');
+
+export interface ApexImagineOptions {
+   nsfw?: boolean;
+   deepCheck?: boolean;
+   count?: number;
+   negative_prompt?: string;
+   sampler?: any;
+   image_style?: any;
+   width?: number;
+   height?: number;
+   steps?: number;
+   seed?: number;
+   cfg_scale?: number;
+}
+
+type ChatModelOption = "v3" | "v3-32k" | "turbo" | "turbo-16k" | "gemini" ;
+type hercmodals = "v1" | "v2" | "v2-beta" | "v3" | "lexica" | "prodia" | "animefy" | "raava" | "shonin" | "simurg";
+
+async function ApexImagine(model: string, prompt: string, options: ApexImagineOptions): Promise<string[] | undefined> {
+  let imageURL: string | null;
+  let response: string[] = [];
+  const imageType = await validOptions;
+  
+  const { nsfw = false, deepCheck = false, count = 2, negative_prompt = "", sampler = "DPM-Solver", height = 512, width = 512, cfg_scale = 9, steps = 20, seed = -1, image_style = "Cinematic" } = options;
+
+  const translatedText = await translate(prompt, {
+     from: "auto",
+     to: "en",
+  });
+
+  for (let i = 0; i < count; i++) {
+     if (imageType.validHercaiModals.includes(model)) {
+        if (model === 'prodia') {
+           imageURL = (await hercai.betaDrawImage({ 
+              prompt: translatedText.text, 
+              negative_prompt: negative_prompt, 
+              sampler: sampler, 
+              image_style: image_style, 
+              width: width, 
+              height: height, 
+              scale: cfg_scale, 
+              steps: steps 
+           })).url;
+        } else {
+           imageURL = (await hercai.drawImage({ 
+              model: model as hercmodals, 
+              prompt: translatedText.text, 
+              negative_prompt: negative_prompt 
+           })).url;
+        }
+     } else if (imageType.validProdiaModals.includes(model)) {
+        const generating = await sdk.generate({
+           model: model,
+           prompt: translatedText.text,
+           negative_prompt: negative_prompt,
+           style_preset: image_style,
+           cfg_scale: cfg_scale,
+           sampler: sampler,
+           seed: seed,
+           steps: steps,
+           width: width,
+           height: height
+        });
+
+        const generatedJobId = generating.data.job;
+        imageURL = await checkJobStatus(generatedJobId);
+     } else if (imageType.validSXDL.includes(model)) {
+        const generating = await sdk.sdxlGenerate({
+           model: model,
+           prompt: translatedText.text,
+           negative_prompt: negative_prompt,
+           style_preset: image_style,
+           cfg_scale: cfg_scale,
+           sampler: sampler,
+           seed: seed,
+           steps: steps,
+           width: width,
+           height: height
+        });
+
+        const generatedJobId = generating.data.job;
+        imageURL = await checkJobStatus(generatedJobId);
+     } else {
+        throw new Error("Invalid model provided. Please check docs/mpn page for valid models.");
+     }
+
+     if (imageURL) {
+        if (nsfw === false && deepCheck === false) {
+          response.push(imageURL);
+        } else {
+
+          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",
+            "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",
+          ];
+
+          let shouldExclude = false;
+          const caption = await apexChecker(imageURL);
+
+          if (nsfw === true && caption) {
+             shouldExclude = nsfwWords.some(word => caption.includes(word));
+          }
+
+          if (deepCheck === true) {
+             shouldExclude = shouldExclude || nsfwWords.some(word => translatedText.text.includes(word));
+          }
+
+          if (!shouldExclude) {
+            response.push(imageURL);
+          }
+        }
+     }
+  }
+  return response;
+}
+
+async function ApexChat(model: string, prompt: string): Promise<string> {
+  try {
+    if (prompt.length >= 2000) {
+      const chunks = [];
+      while (prompt.length > 0) {
+        chunks.push(prompt.substring(0, 2000));
+        prompt = prompt.substring(2000);
+      }
+
+      const responses = await Promise.all(chunks.map(async (chunk) => {
+        return await processChunk(model, chunk);
+      }));
+
+      return responses.join('');
+    } else {
+      return await processChunk(model, prompt);
+    }
+  } catch (error: any) {
+    console.error(error.message); 
+    return '';
+  }
+}
+
+async function processChunk(model: string, prompt: string): Promise<string> {
+  let response: string;
+
+  switch (model) {
+    case 'v3':
+    case 'v3-32k':
+    case 'turbo':
+    case 'turbo-16k':
+    case 'gemini':
+      response = (await hercai.question({ model: model as ChatModelOption, content: prompt })).reply;
+      break;
+    case 'apexChat':
+      response = await apexai(prompt);
+      break;
+    case 'starChat':
+      response = await starChat(prompt);
+      break;
+      case 'facebook-ai':
+        response = await facebook_ai(prompt);
+      break;
+      case 'yi-ai':
+        response = await yi_34b(prompt);
+      break;
+    default:
+      throw new Error('Invalid model.');
+  }
+  return response;
+}
+
+async function apexChecker(urls: any) {
+  try {
+    let retryCount = 0;
+    const maxRetries = 3;
+
+    const fetchData = async () => {
+      try {
+        const response = await axios.post(
+          `https://api-inference.huggingface.co/models/Salesforce/blip-image-captioning-base`,
+          { image: urls },
+          {
+            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 (e: any) {
+        console.error(`Error fetching data: ${e.message}`);
+        throw e;
+      }
+    };
+
+    while (retryCount < maxRetries) {
+      try {
+        return await fetchData();
+      } catch (e: any) {
+        console.error(
+          `Error fetching data (Retry ${retryCount + 1}): ${e.message}`,
+        );
+        retryCount++;
+      }
+    }
+
+    return null;
+  } catch (e: any) {
+    console.error(`Error in attemptImageCaptioning: ${e.message}`);
+    return null;
+  }
+}
+
+async function checkJobStatus(jobId: string, retryCount = 3): Promise<string | null> {
+  try {
+    const getJobResponse = await sdk.getJob({ jobId: jobId });
+    const jobData = getJobResponse.data;
+
+    if (jobData.status === "generating" || jobData.status === "queued") {
+        if (retryCount > 0) {
+            await new Promise(resolve => setTimeout(resolve, 3000));
+            return checkJobStatus(jobId, retryCount - 1);
+        } else {
+            console.error("Job failed after multiple retries:", jobData);
+            return null;
+        }
+    } else if (jobData.status === "succeeded") {
+        return jobData.imageUrl;
+    } else {
+        console.error("Job failed:", jobData);
+        return null;
+    }
+  } catch (error) {
+    console.error(error);
+    return null;
+  }
+}
+
+
+export { ApexImagine, ApexChat };