Rework ai image gen animals

src/modules/ai/ai.ts

@@ -3,6 +3,7 @@ import type { ICallbackStore } from "../types.js";
 import { config } from "../../config.js";
 import { getImageNanoBanana, getTextGemini } from "../../services/ai/ai.js";
 import type { IAnimal } from "./types.js";
+import { animals } from "./animals.js";
 
 let client: MatrixClient;
 
@@ -17,76 +18,14 @@ const registerModuleAI = (
         callbackFunc: onAI,
     });
     callbackStore.messageCallbacks.push({
-        startConditions: [
+        startConditions: animals
-            `satek when `,
+            .map((animal) => `${animal.name} when`)
-            `satek ked `,
+            .concat(animals.map((animal) => `${animal.name} ked`)),
-            `gabor when `,
-            `gabor ked `,
-            `martin when `,
-            `martin ked `,
-            `madys when `,
-            `madys ked `,
-            `mandak when `,
-            `mandak ked `,
-            `mando when `,
-            `mando ked `,
-            `mandik when `,
-            `mandik ked `,
-            `madik when `,
-            `madik ked `,
-            `janys when `,
-            `janys ked `,
-            `jano when `,
-            `jano ked `,
-        ],
         callbackFunc: onImageGen,
     });
 };
 
 const getAnimal = (name: string): IAnimal | undefined => {
-    const animals: IAnimal[] = [
-        {
-            name: "satek",
-            animal: "black cat",
-        },
-        {
-            name: "gabor",
-            animal: "hedgehog",
-        },
-        {
-            name: "martin",
-            animal: "hedgehog",
-        },
-        {
-            name: "madys",
-            animal: "beaver",
-        },
-        {
-            name: "mandak",
-            animal: "beaver",
-        },
-        {
-            name: "mando",
-            animal: "beaver",
-        },
-        {
-            name: "mandik",
-            animal: "beaver",
-        },
-        {
-            name: "madik",
-            animal: "beaver",
-        },
-        {
-            name: "janys",
-            animal: "lion",
-        },
-        {
-            name: "jano",
-            animal: "lion",
-        },
-    ];
-
     const foundAnimals = animals
         .map((animal) => {
             if (name.includes(animal.name)) {
@@ -108,18 +47,22 @@ const onAI = async (text: string, roomId: string) => {
 };
 
 const onImageGen = async (text: string, roomId: string) => {
-    const animal = getAnimal(text.trim().split(/\s+/)[0] ?? "");
+    const firstAnimal = getAnimal(text.trim().split(/\s+/)[0] ?? "");
-    if (!animal) {
+    if (!firstAnimal) {
         return;
     }
 
-    const textMod = text.replace(animal.name, animal.animal);
+    let textMod = text;
+    animals.forEach((animal) => {
+        textMod = textMod.replaceAll(animal.name, animal.animal);
+    });
+
     const buffer = await getImageNanoBanana(textMod.trim());
     if (!buffer || buffer.length < 10) {
         return;
     }
 
-    const imageName = `photo-${animal.name}.png`;
+    const imageName = `photo-${firstAnimal.name}.png`;
 
     const uploadResult = await client.uploadContent(buffer, {
         type: "image/png",

src/modules/ai/animals.ts

@@ -0,0 +1,46 @@
+import type { IAnimal } from "./types.js";
+
+const animals: IAnimal[] = [
+    {
+        name: "satek",
+        animal: "black cat",
+    },
+    {
+        name: "gabor",
+        animal: "hedgehog",
+    },
+    {
+        name: "martin",
+        animal: "hedgehog",
+    },
+    {
+        name: "madys",
+        animal: "beaver",
+    },
+    {
+        name: "mandak",
+        animal: "beaver",
+    },
+    {
+        name: "mando",
+        animal: "beaver",
+    },
+    {
+        name: "mandik",
+        animal: "beaver",
+    },
+    {
+        name: "madik",
+        animal: "beaver",
+    },
+    {
+        name: "janys",
+        animal: "lion",
+    },
+    {
+        name: "jano",
+        animal: "lion",
+    },
+];
+
+export { animals };