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

Error message for ocr, fix attachment lookup

Browse source
This commit is contained in:
derpystuff 2022-06-22 22:58:44 +02:00
parent c384cb8022
commit 2ea818f30f
3 changed files with 15 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -18,7 +18,7 @@ module.exports = {
const { message } = args;
let attachment = getMessageAttachment(message)
if(attachment && validateAttachment(attachment)){
if(attachment && validateAttachment(attachment, "image")){
attachment = attachment.url
} else {
delete attachment;
@ -27,12 +27,14 @@ module.exports = {
let ocr = await googleVisionOcr(context, attachment)
if(ocr.response.body.text.length == 0) return context.editOrRespond({ embeds: [createEmbed("warning", context, "No text detected.")], flags: MessageFlags.EPHEMERAL })
await context.editOrRespond({
embeds: [createEmbed("default", context, {
thumbnail: {
url: attachment
},
description: codeblock("ansi", [ocr.response.body.text]),
description: codeblock("ansi", ["​" + ocr.response.body.text]),
footer: {
iconUrl: STATICS.google,
text: `Google Cloud Vision • ${context.application.name} • Took ${ocr.timings}s`

5
commands/message/utils/ocr.js
View file

@ -18,7 +18,12 @@ module.exports = {
let image = await getRecentImage(context, 50)
if (!image) return editOrReply(context, { embeds: [createEmbed("warning", context, "No images found.")] })
console.log(image)
let ocr = await googleVisionOcr(context, image)
if(ocr.response.body.text.length == 0) return editOrReply(context, { embeds: [createEmbed("warning", context, "No text detected.")] })
return editOrReply(context, createEmbed("default", context, {
thumbnail: {
url: image

10
labscore/utils/attachment.js
View file

@ -85,10 +85,12 @@ module.exports.getRecentVideo = async function(context, limit) {
return attachments;
}
function validateAttachment(attachment){
if (attachment.contentType && attachmentTypes.image.includes(attachment.contentType)) { // discord attachment
function validateAttachment(attachment, type){
console.log(attachment)
let allowedTypes = attachmentTypes[type]
if (attachment.contentType && allowedTypes.includes(attachment.contentType)) { // discord attachment
return true
} else if (!attachment.content_type) { // other form of media
} else if (!attachment.contentType) { // other form of media
return true
} else {
return false
@ -102,7 +104,7 @@ module.exports.getRecentImage = async function(context, limit){
let at;
for (const a of attachments) {
if(validateAttachment(a) && at === undefined) at = a.url
if(validateAttachment(a, "image") && at === undefined) at = a.url
}
return at;
}