From de0601205356cddfbe465f43dc9c46d9e1aed5d345a7e001cf170379becf8242 Mon Sep 17 00:00:00 2001
From: aslan <jan.trencansky@protonmail.com>
Date: Fri, 26 Dec 2025 13:52:46 +0100
Subject: [PATCH] Rework AI alts

---
 package.json              |   2 +-
 src/modules/ai/ai.ts      |  10 +--
 src/modules/ai/alts.ts    | 166 ++++++++++++++++++++++++++++++++++++++
 src/modules/ai/animals.ts | 102 -----------------------
 src/modules/ai/types.ts   |   8 +-
 5 files changed, 176 insertions(+), 112 deletions(-)
 create mode 100644 src/modules/ai/alts.ts
 delete mode 100644 src/modules/ai/animals.ts

diff --git a/package.json b/package.json
index 01587b7..146b882 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
     "name": "aslobot-matrix",
-    "version": "0.5.3",
+    "version": "0.6.0",
     "description": "",
     "license": "ISC",
     "author": "",
diff --git a/src/modules/ai/ai.ts b/src/modules/ai/ai.ts
index 12b9533..dae808d 100644
--- a/src/modules/ai/ai.ts
+++ b/src/modules/ai/ai.ts
@@ -2,7 +2,7 @@ import { MatrixClient, MsgType } from "matrix-js-sdk";
 import type { ICallbackStore } from "../types.js";
 import { config } from "../../config.js";
 import { getTextGemini, getImageGemini } from "../../services/ai/ai.js";
-import { animals } from "./animals.js";
+import { alts } from "./alts.js";
 
 let client: MatrixClient;
 
@@ -18,8 +18,8 @@ const registerModuleAI = (
     });
     callbackStore.messageCallbacks.push({
         startConditions: [
-            ...animals.map((animal) => `${animal.name} when`),
-            ...animals.map((animal) => `${animal.name} ked`),
+            ...alts.map((alt) => `${alt.key} when`),
+            ...alts.map((alt) => `${alt.key} ked`),
             `!img`,
         ],
         callbackFunc: onImageGen,
@@ -37,8 +37,8 @@ const onAI = async (text: string, roomId: string) => {
 
 const onImageGen = async (text: string, roomId: string) => {
     let textMod = text.replace("!img", "").trim().toLowerCase();
-    animals.forEach((animal) => {
-        textMod = textMod.replaceAll(animal.name, animal.animal);
+    alts.forEach((alt) => {
+        textMod = textMod.replaceAll(alt.key, alt.alt);
     });
 
     const buffer = await getImageGemini(textMod);
diff --git a/src/modules/ai/alts.ts b/src/modules/ai/alts.ts
new file mode 100644
index 0000000..919c03b
--- /dev/null
+++ b/src/modules/ai/alts.ts
@@ -0,0 +1,166 @@
+import type { IAIAlt } from "./types.js";
+
+const alts: IAIAlt[] = [
+    {
+        key: "satek",
+        alt: "black cat",
+    },
+    {
+        key: "sateg",
+        alt: "black cat",
+    },
+    {
+        key: "sadeg",
+        alt: "black cat",
+    },
+    {
+        key: "satko",
+        alt: "black cat",
+    },
+    {
+        key: "sadko",
+        alt: "black cat",
+    },
+    {
+        key: "vlado",
+        alt: "black cat",
+    },
+    {
+        key: "vladko",
+        alt: "black cat",
+    },
+    {
+        key: "vladimir",
+        alt: "black cat",
+    },
+    {
+        key: "macag",
+        alt: "black cat",
+    },
+    {
+        key: "vladko",
+        alt: "black cat",
+    },
+    {
+        key: "macica",
+        alt: "white cat",
+    },
+    {
+        key: "macico",
+        alt: "white cat",
+    },
+    {
+        key: "gabor",
+        alt: "hedgehog",
+    },
+    {
+        key: "martin",
+        alt: "hedgehog",
+    },
+    {
+        key: "madys",
+        alt: "beaver",
+    },
+    {
+        key: "mandak",
+        alt: "beaver",
+    },
+    {
+        key: "mando",
+        alt: "beaver",
+    },
+    {
+        key: "mandik",
+        alt: "beaver",
+    },
+    {
+        key: "madik",
+        alt: "beaver",
+    },
+    {
+        key: "madeg",
+        alt: "beaver",
+    },
+    {
+        key: "janys",
+        alt: "lion",
+    },
+    {
+        key: "jano",
+        alt: "lion",
+    },
+    {
+        key: "janeg",
+        alt: "lion",
+    },
+    {
+        key: "aslan",
+        alt: "lion",
+    },
+    {
+        key: "aslo",
+        alt: "lion",
+    },
+    {
+        key: "marek",
+        alt: "purple snake",
+    },
+    {
+        key: "mareg",
+        alt: "purple snake",
+    },
+    {
+        key: "maceg",
+        alt: "purple snake",
+    },
+    {
+        key: "rod",
+        alt: "purple snake",
+    },
+    {
+        key: "lara",
+        alt: "ginger dog",
+    },
+    {
+        key: "kostur",
+        alt: "ginger dog",
+    },
+    {
+        key: "kosturik",
+        alt: "ginger dog",
+    },
+    {
+        key: "tato",
+        alt: "troll",
+    },
+    {
+        key: "mama",
+        alt: "angel",
+    },
+    {
+        key: "monika",
+        alt: "angel",
+    },
+    {
+        key: "puk",
+        alt: "rat",
+    },
+    {
+        key: "becky",
+        alt: "giantess",
+    },
+    {
+        key: "beky",
+        alt: "giantess",
+    },
+    {
+        key: "beki",
+        alt: "giantess",
+    },
+    {
+        key: "veronika",
+        alt: "lgbt princess",
+    },
+];
+
+export { alts };
diff --git a/src/modules/ai/animals.ts b/src/modules/ai/animals.ts
deleted file mode 100644
index f45bf4a..0000000
--- a/src/modules/ai/animals.ts
+++ /dev/null
@@ -1,102 +0,0 @@
-import type { IAnimal } from "./types.js";
-
-const animals: IAnimal[] = [
-    {
-        name: "satek",
-        animal: "black cat",
-    },
-    {
-        name: "satko",
-        animal: "black cat",
-    },
-    {
-        name: "vlado",
-        animal: "black cat",
-    },
-    {
-        name: "macag",
-        animal: "black cat",
-    },
-    {
-        name: "vladko",
-        animal: "black cat",
-    },
-    {
-        name: "macica",
-        animal: "white cat",
-    },
-    {
-        name: "macico",
-        animal: "white 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: "madeg",
-        animal: "beaver",
-    },
-    {
-        name: "janys",
-        animal: "lion",
-    },
-    {
-        name: "jano",
-        animal: "lion",
-    },
-    {
-        name: "janeg",
-        animal: "lion",
-    },
-    {
-        name: "aslan",
-        animal: "lion",
-    },
-    {
-        name: "aslo",
-        animal: "lion",
-    },
-    {
-        name: "marek",
-        animal: "purple snake",
-    },
-    {
-        name: "mareg",
-        animal: "purple snake",
-    },
-    {
-        name: "maceg",
-        animal: "purple snake",
-    },
-    {
-        name: "rod",
-        animal: "purple snake",
-    },
-];
-
-export { animals };
diff --git a/src/modules/ai/types.ts b/src/modules/ai/types.ts
index e4e1a54..c7b0c35 100644
--- a/src/modules/ai/types.ts
+++ b/src/modules/ai/types.ts
@@ -1,6 +1,6 @@
-interface IAnimal {
-    name: string;
-    animal: string;
+interface IAIAlt {
+    key: string;
+    alt: string;
 }
 
-export { type IAnimal };
+export { type IAIAlt };