apexify.js 4.4.36 → 4.5.1

package/lib/ai/ApexModules.ts

@@ -3,22 +3,18 @@ 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";
-import { llamaChat, mixtral, openChat, v4 } from './utils';
-import { codellama } from './modals-chat/codellama';
-import { bard } from './modals-chat/bard';
-import { bing } from './modals-chat/bing';
-import { gemini_v2 } from './modals-chat/geminiV2';
-import { whisper } from './modals-chat/whisper';
-import { llama2 } from './modals-chat/llamav2';
-import { llama3 } from './modals-chat/llamav3';
-import { gemma } from './modals-chat/gemma';
-import { mixtral2 } from './modals-chat/mixtralv2';
+import { groqChatModels, hercChatModels, otherChatModel, rsnChatModels, validateModels } from "./functions/validOptions";
+import { whisper } from './modals-chat/groq/whisper';
 import { connect } from 'verse.db';
-import { geminiFast } from './modals-chat/geminiFast';
-import { flux } from './modals-images/flux';
-import { cartoon } from './modals-images/cartoon';
+import { otherChatModels } from './modals-chat/others/otherModels';
+import { rsnAPI } from './modals-chat/rsn/rsnChat';
+import { chatGroq } from './modals-chat/groq/chatgroq';
+import { geminiFast } from './modals-chat/gemini/geminiFast';
+import { cartoon } from './modals-chat/freesedgpt/cartoon';
+import { flux } from './modals-chat/freesedgpt/flux';
+import { gpt4o } from './modals-chat/freesedgpt/chat';
+import { groqAnalyzer } from './utils';
+
 
 const dbConfig = {
   adapter: 'json',
@@ -50,21 +46,19 @@ type hercmodals = "v1" | "v2" | "v2-beta" | "v3" | "lexica" | "prodia" | "animef
 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, nsfwWords = [], count = 2, negative_prompt = "", sampler = "DPM++ 2M Karras", height = 512, width = 512, cfg_scale = 9, steps = 20, seed = -1, image_style = "cinematic" } = options;
-
-  sdk.auth(options.Api_Key || 'eaebff6e-c7b2-477c-8edd-9aa91becf1e3');
+  const imageType = await validateModels;
+  const { Api_Key = 'eaebff6e-c7b2-477c-8edd-9aa91becf1e3', nsfw = false, deepCheck = false, nsfwWords = [], count = 2, negative_prompt = "", sampler = "DPM++ 2M Karras", height = 512, width = 512, cfg_scale = 9, steps = 20, seed = -1, image_style = "cinematic" } = options;
 
   const translatedText = await translate(prompt, {
      from: "auto",
-     to: "en",
-  });
+     to: "en"
+    });
 
   if (count > 4 || count <= 0) throw new Error("Inavlid usage. Count can't be less than 0 or more than 4.");
-  if ((width || height) > 1024 || (width || height) <= 0) throw new Error("Inavlid usage. Image width/height can't be less than 0 or more than 1024.");
+  if ((width || height) <= 0) throw new Error("Inavlid usage. Image width/height can't be less than 0 or more than 1024.");
 
   for (let i = 0; i < count; i++) {
-     if (imageType.validHercaiModals.includes(model)) {
+     if (imageType.validHercaiModels.includes(model)) {
         if (model === 'prodia-v2') {
            imageURL = (await hercai.betaDrawImage({ 
               prompt: translatedText.text, 
@@ -83,7 +77,8 @@ async function ApexImagine(model: string, prompt: string, options: ApexImagineOp
               negative_prompt: negative_prompt 
            })).url;
         }
-     } else if (imageType.validProdiaModals.includes(model)) {
+     } else if (imageType.validProdiaModels.includes(model)) {
+      sdk.auth(Api_Key);
         const generating = await sdk.generate({
            model: model,
            prompt: translatedText.text,
@@ -100,6 +95,7 @@ async function ApexImagine(model: string, prompt: string, options: ApexImagineOp
         const generatedJobId = generating.data.job;
         imageURL = await checkJobStatus(generatedJobId);
      } else if (imageType.validSXDL.includes(model)) {
+      sdk.auth(Api_Key);
         const generating = await sdk.sdxlGenerate({
            model: model,
            prompt: translatedText.text,
@@ -115,15 +111,18 @@ async function ApexImagine(model: string, prompt: string, options: ApexImagineOp
 
         const generatedJobId = generating.data.job;
         imageURL = await checkJobStatus(generatedJobId);
-     } else if (model === 'flux') {
-      const response = await flux(prompt);
-      if (response !== null) imageURL = response;
-      if (response === null) throw new Error('This modal got broke please inform the owner directly!')
-     } else if (model === 'cartoon') {
-      const response = await cartoon(prompt);
-      if (response !== null) imageURL = response;
-      if (response === null) throw new Error('This modal got broke please inform the owner directly!')
-     } else {
+     } else if (model === 'flux-schnell') {
+      const gen = await flux(translatedText.text,  options.Api_Key as string);
+      if (gen === null) throw new Error('this model reached rate limit. Provide your own api key from: "https://discord.gg/94qUZWhwFE"');
+      imageURL = gen
+
+    } else if (model === 'real-cartoon-xl-v6') {
+      const gen = await cartoon(translatedText.text, options.Api_Key as string);
+      if (gen === null) throw new Error('this model reached rate limit. Provide your own api key from: "https://discord.gg/94qUZWhwFE"');
+      imageURL = gen
+    } else if (model === 'dalle') { 
+      imageURL = await rsnAPI({ API_KEY: Api_Key, prompt: translatedText.text, apiName: 'dalle' });
+    } else {
       throw new Error("Invalid model provided. Please check docs/npm page for valid models.");
      }
 
@@ -618,7 +617,7 @@ async function ApexImagine(model: string, prompt: string, options: ApexImagineOp
           let shouldExclude = false;
 
           if (nsfw === true) {
-             const caption = await apexChecker(imageURL);
+             const caption = await groqAnalyzer({ imgURL: imageURL, prompt });
              if (!caption) return;
              shouldExclude = NSFWWORDS.some(word => caption.includes(word));
              shouldExclude = nsfwWords.some(word => caption.includes(word));
@@ -678,7 +677,7 @@ async function ApexChat(model: string, prompt: string, { userId, memory, limit,
       }
 
       const responses = await Promise.all(chunks.map(async (chunk) => {
-        return await processChunk(model, chunk, {});
+        return await processChunk(model, chunk, { ApiKey: Api_key, personality: instruction });
       }));
 
       response = responses.join('');
@@ -699,116 +698,27 @@ async function ApexChat(model: string, prompt: string, { userId, memory, limit,
 
 async function processChunk(model: string, prompt: string, { ApiKey, personality }: { ApiKey?: string, personality?: string }): Promise<string> {
   let response: string;
-  switch (model) {
-    case 'v3':
-    case 'v3-32k':
-    case 'turbo':
-    case 'turbo-16k':
-    case 'gemini':
+  switch (true) {
+    case hercChatModels.includes(model):
       response = (await hercai.question({ model: model as ChatModelOption, content: prompt, personality })).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;
-    case 'v4':
-      response = await v4({ API_KEY: ApiKey, prompt });
-      break;
-    case 'openChat':
-      response = await openChat({ API_KEY: ApiKey, prompt });
-      break;
-    case 'mixtral':
-      response = await mixtral({ API_KEY: ApiKey, prompt });
-      break;
-    case 'llama':
-      response = await llamaChat({ API_KEY: ApiKey, prompt });
-      break;
-    case 'llama-v2':
-      response = await llama2(prompt, ApiKey);
-      break;
-    case 'llama-v3':
-      response = await llama3(prompt, ApiKey);
-      break;
-    case 'gemma':
-      response = await gemma(prompt, ApiKey);
-      break;
-    case 'mixtral-v2':
-      response = await mixtral2(prompt, ApiKey);
+    case otherChatModel.includes(model):
+      response = await otherChatModels({ modelName: model as any, prompt });
       break;
-    case 'codellama':
-      response = await codellama({ API_KEY: ApiKey, prompt });
+    case rsnChatModels.includes(model):
+      response = await rsnAPI({ API_KEY: ApiKey, apiName: model, prompt });
       break;
-    case 'bard':
-      response = await bard({ API_KEY: ApiKey, prompt });
-      break;
-    case 'bing':
-      response = await bing({ API_KEY: ApiKey, prompt });
-      break;
-    case 'gemini-v2':
-      response = await gemini_v2({ API_KEY: ApiKey, prompt });
+    case groqChatModels.includes(model):
+      response = await chatGroq({ API_KEY: ApiKey, apiName: model, prompt, instruction: personality });
       break;
+    case model === 'gpt-4o':
+        response = await gpt4o({ ApiKey, 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): Promise<string | null> {
   try {