CCF_100/NoMoreAcronyms
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Implement OCR

Browse Source
This commit is contained in:
Chloe Fontenot 🏳️‍⚧️ 2024-08-09 23:08:51 -05:00
parent 0667dd70d8
commit 1406378c9f
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.cjs
View File

@ -198,7 +198,7 @@ function getMatchingPhrase(inputString, targetCharacter) {
message.delete();
return true;
}
});
})();
return false;
}
if (checkLoop(message, checkString)) {
@ -209,9 +209,10 @@ function getMatchingPhrase(inputString, targetCharacter) {
(async () => {
// Does the message have an attachment? Check the attachment with OCR
console.log(message.attachments);
if (message.attachments.contentType.includes("image")) {
message.attachments.forEach(i => {
if (i.contentType.includes("image")) {
const worker = await createWorker('eng');
const ret = await worker.recognize(message.attachments.proxyURL);
const ret = await worker.recognize(i.proxyURL);
console.log("OCR Results: " + ret.data.text);
checkString = ret.data.text;
await worker.terminate();
@ -222,6 +223,7 @@ function getMatchingPhrase(inputString, targetCharacter) {
checkLoop(message, checkString);
}
})();
})();
}
/*
Main bot logic. Check for abbreviations by splitting the message at any non-word characters, then if we find a match, create a new array, replace the abbreviation with the phrase, then call replyMessage() and reply to the original message with the corrected one.