mirrors/pissbot-9000
1
0
Fork 0
mirror of https://gitlab.com/bignutty/labscore.git synced 2025-06-09 06:33:05 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

support translate enhancements in ocrtr

Browse source
This commit is contained in:
derpystuff 2023-03-18 14:26:32 +01:00
parent a63cea73a6
commit 37a6c40d30
1 changed files with 12 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
commands/message/utils/ocrtr.js
View file

@ -1,10 +1,11 @@
const { googleVisionOcr, googleTranslate } = require("../../../labscore/api"); const { googleVisionOcr, googleTranslate } = require("../../../labscore/api");
const { TRANSLATE_LANGUAGES, TRANSLATE_LANGUAGE_MAPPINGS } = require("../../../labscore/constants");
const { getRecentImage } = require("../../../labscore/utils/attachment"); const { getRecentImage } = require("../../../labscore/utils/attachment");
const { createEmbed } = require("../../../labscore/utils/embed"); const { createEmbed } = require("../../../labscore/utils/embed");
const { codeblock, icon, highlight } = require("../../../labscore/utils/markdown"); const { codeblock, icon, highlight, pill } = require("../../../labscore/utils/markdown");
const { editOrReply } = require("../../../labscore/utils/message"); const { editOrReply } = require("../../../labscore/utils/message");
const { STATICS } = require("../../../labscore/utils/statics"); const { STATICS } = require("../../../labscore/utils/statics");
const { isSupported } = require("../../../labscore/utils/translate"); const { isSupported, getCodeFromAny } = require("../../../labscore/utils/translate");
module.exports = { module.exports = {
name: 'ocrtr', name: 'ocrtr',
@ -29,6 +30,9 @@ module.exports = {
if(!isSupported(args.to)) return editOrReply(context, createEmbed("warning", context, "Invalid language (to).")) if(!isSupported(args.to)) return editOrReply(context, createEmbed("warning", context, "Invalid language (to)."))
if(!isSupported(args.from)) return editOrReply(context, createEmbed("warning", context, "Invalid language (from).")) if(!isSupported(args.from)) return editOrReply(context, createEmbed("warning", context, "Invalid language (from)."))
args.to = getCodeFromAny(args.to)
args.from = getCodeFromAny(args.from)
let image = await getRecentImage(context, 50) let image = await getRecentImage(context, 50)
if (!image) return editOrReply(context, { embeds: [createEmbed("warning", context, "No images found.")] }) if (!image) return editOrReply(context, { embeds: [createEmbed("warning", context, "No images found.")] })
@ -43,14 +47,18 @@ module.exports = {
try{ try{
let translate = await googleTranslate(context, ocr.response.body.text, args.to, args.from) let translate = await googleTranslate(context, ocr.response.body.text, args.to, args.from)
let fromFlag = TRANSLATE_LANGUAGE_MAPPINGS[translate.response.body.language.from || args.from] || ''
let toFlag = TRANSLATE_LANGUAGE_MAPPINGS[translate.response.body.language.to] || ''
return editOrReply(context, createEmbed("default", context, { return editOrReply(context, createEmbed("default", context, {
description: `${icon("locale")} ${highlight(`${translate.response.body.language.from} -> ${translate.response.body.language.to}`)}\n${codeblock("ansi", [translate.response.body.translation])}`, description: `${icon("locale")} ​ ${fromFlag} ${pill(TRANSLATE_LANGUAGES[translate.response.body.language.from || args.from])} ​ ​ ​​${icon("arrow_right")} ​ ​ ​ ​${toFlag} ${pill(TRANSLATE_LANGUAGES[translate.response.body.language.to])}\n${codeblock("ansi", [translate.response.body.translation])}`,
thumbnail: { thumbnail: {
url: image url: image
}, },
footer: { footer: {
iconUrl: STATICS.google, iconUrl: STATICS.google,
text: `Google Cloud Vision • ${context.application.name} • Took ${parseFloat(ocr.timings) + parseFloat(translate.timings)}s` text: `Google Cloud Vision • ${context.application.name}`
} }
})) }))
}catch(e){ }catch(e){