src/services/ImageRecognitionService.ts

/**
 * This file is part of SudoBot.
 *
 * Copyright (C) 2021-2024 OSN Developers.
 *
 * SudoBot is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * SudoBot is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
 */

import Tesseract, { createWorker } from "tesseract.js";
import Service from "../core/Service";
import { log } from "../utils/logger";

export const name = "imageRecognitionService";

export default class ImageRecognitionService extends Service {
    protected worker: Tesseract.Worker | null = null;
    protected timeout: NodeJS.Timeout | null = null;

    protected async createWorkerIfNeeded() {
        if (!this.worker && !this.timeout) {
            log("Spawning new tesseract worker for image recognition");
            this.worker = await createWorker("eng");
            this.setTimeout();
        } else if (this.worker && this.timeout) {
            log("Using existing tesseract worker for image recognition");
            clearTimeout(this.timeout);
            this.setTimeout();
        }
    }

    protected setTimeout() {
        this.timeout = setTimeout(() => {
            log("Terminating existing tesseract worker due to inactivity");
            this.worker?.terminate();
            this.timeout = null;
        }, 60_000);
    }

    async recognize(image: Tesseract.ImageLike) {
        await this.createWorkerIfNeeded();
        return this.worker!.recognize(image);
    }
}
1	rakinar2	577	/**
2			* This file is part of SudoBot.
3			*
4			* Copyright (C) 2021-2024 OSN Developers.
5			*
6			* SudoBot is free software; you can redistribute it and/or modify it
7			* under the terms of the GNU Affero General Public License as published by
8			* the Free Software Foundation, either version 3 of the License, or
9			* (at your option) any later version.
10			*
11			* SudoBot is distributed in the hope that it will be useful, but
12			* WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU Affero General Public License for more details.
15			*
16			* You should have received a copy of the GNU Affero General Public License
17			* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18			*/
19
20			import Tesseract, { createWorker } from "tesseract.js";
21			import Service from "../core/Service";
22			import { log } from "../utils/logger";
23
24			export const name = "imageRecognitionService";
25
26			export default class ImageRecognitionService extends Service {
27			protected worker: Tesseract.Worker \| null = null;
28			protected timeout: NodeJS.Timeout \| null = null;
29
30			protected async createWorkerIfNeeded() {
31			if (!this.worker && !this.timeout) {
32			log("Spawning new tesseract worker for image recognition");
33			this.worker = await createWorker("eng");
34			this.setTimeout();
35			} else if (this.worker && this.timeout) {
36			log("Using existing tesseract worker for image recognition");
37			clearTimeout(this.timeout);
38			this.setTimeout();
39			}
40			}
41
42			protected setTimeout() {
43			this.timeout = setTimeout(() => {
44			log("Terminating existing tesseract worker due to inactivity");
45			this.worker?.terminate();
46			this.timeout = null;
47			}, 60_000);
48			}
49
50			async recognize(image: Tesseract.ImageLike) {
51			await this.createWorkerIfNeeded();
52			return this.worker!.recognize(image);
53			}
54			}