Spaces:
Paused
Paused
File size: 1,473 Bytes
8de10f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import Tesseract from 'tesseract.js';
import { writeFile, unlink } from 'fs/promises';
import config from '../../config.cjs';
const givetextCommand = async (m, Matrix) => {
const prefix = config.PREFIX;
const cmd = m.body.startsWith(prefix) ? m.body.slice(prefix.length).split(' ')[0].toLowerCase() : '';
const arg = m.body.slice(prefix.length + cmd.length).trim();
const validCommands = ['givetext', 'extract'];
if (validCommands.includes(cmd)) {
if (!m.quoted || m.quoted.mtype !== 'imageMessage') {
return m.reply(`Send/Reply with an image to extract text ${prefix + cmd}`);
}
let lang = 'eng';
if (args.length > 0) {
lang = args[0];
}
try {
const media = await m.quoted.download();
if (!media) throw new Error('Failed to download media.');
const filePath = `./${Date.now()}.png`;
await writeFile(filePath, media);
const { data: { text } } = await Tesseract.recognize(filePath, lang, {
logger: m => console.log(m)
});
const responseMessage = `Extracted Text:\n\n${text}`;
await Matrix.sendMessage(m.from, { text: responseMessage }, { quoted: m });
await unlink(filePath);
} catch (error) {
console.error("Error extracting text from image:", error);
await Matrix.sendMessage(m.from, { text: 'Error extracting text from image.' }, { quoted: m });
}
}
};
export default givetextCommand;
|