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

add translate action to ctx ocr

Browse source
This commit is contained in:
bignutty 2024-11-07 20:27:25 +01:00
parent 13e2437b69
commit 2d25e87249
3 changed files with 61 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

61
commands/interaction/context/ocr.js
View file

@ -1,13 +1,15 @@
const { googleVisionOcr } = require('#api');
const { googleVisionOcr, googleTranslate } = require('#api');
const { TRANSLATE_DISPLAY_MAPPINGS, TRANSLATE_LANGUAGES } = require('#constants');
const { getMessageAttachment, validateAttachment } = require('#utils/attachment');
const { createEmbed } = require('#utils/embed');
const { acknowledge } = require('#utils/interactions');
const { codeblock } = require('#utils/markdown');
const { codeblock, pill, icon } = require('#utils/markdown');
const { editOrReply } = require('#utils/message');
const { STATICS } = require('#utils/statics');
const { STATICS, STATIC_ICONS, STATIC_ACTIONS } = require('#utils/statics');
const { ApplicationCommandTypes, MessageFlags } = require("detritus-client/lib/constants");
const { Components } = require('detritus-client/lib/utils');
module.exports = {
name: 'OCR',
@ -38,15 +40,52 @@ module.exports = {
if(ocr.response.body.status == 1) return editOrReply(context, createEmbed("warning", context, ocr.response.body.text))
await editOrReply(context, createEmbed("default", context, {
thumbnail: {
url: attachment
},
description: codeblock("ansi", ["​" + ocr.response.body.text.substr(0,3900)]),
footer: {
iconUrl: STATICS.google,
text: `Google Cloud Vision • ${context.application.name} • Took ${ocr.timings}s`
const components = new Components({
timeout: 100000,
run: async (ctx) => {
if (ctx.userId !== context.userId) return await ctx.respond(InteractionCallbackTypes.DEFERRED_UPDATE_MESSAGE);
await ctx.editOrRespond({
embeds: [createEmbed("ai_custom", context, STATIC_ACTIONS.translate)]
})
let translate = await googleTranslate(context, ocr.response.body.text, "en", "auto")
let fromFlag = TRANSLATE_DISPLAY_MAPPINGS[translate.response.body.language.from || sourceLanguage] || ''
let toFlag = TRANSLATE_DISPLAY_MAPPINGS[translate.response.body.language.to] || ''
return editOrReply(context, createEmbed("default", context, {
description: `${icon("locale")} ​ ${fromFlag} ${pill(TRANSLATE_LANGUAGES[translate.response.body.language.from || sourceLanguage] || translate.response.body.language.from || "Detected Language")} ​ ​ ​​${icon("arrow_right")} ​ ​ ​ ​${toFlag} ${pill(TRANSLATE_LANGUAGES[translate.response.body.language.to] || translate.response.body.language.to)}\n${codeblock("ansi", [translate.response.body.translation])}`,
thumbnail: {
url: attachment
},
footer: {
iconUrl: STATICS.googletranslate,
text: `Google Translate • ${context.application.name}`
}
}))
}
})
components.addButton({
label: "Translate",
emoji: icon("button_translate"),
custom_id: "actions-translate",
style: 2
})
await editOrReply(context, createEmbed("default", context, {
embeds: [{
thumbnail: {
url: attachment
},
description: codeblock("ansi", ["​" + ocr.response.body.text.substr(0,3900)]),
footer: {
iconUrl: STATICS.google,
text: `Google Cloud Vision • ${context.application.name} • Took ${ocr.timings}s`
}
}],
components
}))
}catch(e){
console.log(e)