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

update transcription to use obelisk service

Browse source
This commit is contained in:
derpystuff 2024-04-08 20:29:59 +02:00
parent 8f2727456b
commit 0cac2568d8
4 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
commands/interaction/context/transcribe.js
View file

@ -1,7 +1,7 @@
const { Constants } = require('detritus-client'); const { Constants } = require('detritus-client');
const { InteractionCallbackTypes, ApplicationCommandTypes, MessageFlags } = Constants; const { InteractionCallbackTypes, ApplicationCommandTypes, MessageFlags } = Constants;
const { googleSpeechRecognitionWithLabels } = require('../../../labscore/api'); const { transcribeWithSpeakerLabelsObelisk } = require('../../../labscore/api/obelisk');
const { createEmbed } = require('../../../labscore/utils/embed'); const { createEmbed } = require('../../../labscore/utils/embed');
const { codeblock } = require('../../../labscore/utils/markdown'); const { codeblock } = require('../../../labscore/utils/markdown');
@ -34,7 +34,7 @@ module.exports = {
flags: MessageFlags.EPHEMERAL flags: MessageFlags.EPHEMERAL
}) })
const recog = await googleSpeechRecognitionWithLabels(context, message.attachments.first().url) const recog = await transcribeWithSpeakerLabelsObelisk(context, message.attachments.first().url)
return editOrReply(context, { return editOrReply(context, {
embeds: [createEmbed("default", context, { embeds: [createEmbed("default", context, {

4
commands/message/google/transcribe.js
View file

@ -1,7 +1,7 @@
const { createEmbed } = require('../../../labscore/utils/embed') const { createEmbed } = require('../../../labscore/utils/embed')
const { editOrReply } = require('../../../labscore/utils/message') const { editOrReply } = require('../../../labscore/utils/message')
const { googleSpeechRecognitionWithLabels } = require('../../../labscore/api'); const { transcribeWithSpeakerLabelsObelisk } = require('../../../labscore/api/obelisk');
const { STATICS } = require('../../../labscore/utils/statics'); const { STATICS } = require('../../../labscore/utils/statics');
const { codeblock, icon } = require('../../../labscore/utils/markdown'); const { codeblock, icon } = require('../../../labscore/utils/markdown');
@ -32,7 +32,7 @@ module.exports = {
if(!msg.attachments.first()) return editOrReply(context, createEmbed("warning", context, "No voice message found.")) if(!msg.attachments.first()) return editOrReply(context, createEmbed("warning", context, "No voice message found."))
if(!msg.attachments.first().url.split('?')[0].endsWith('voice-message.ogg')) return editOrReply(context, createEmbed("warning", context, "No voice message found.")) if(!msg.attachments.first().url.split('?')[0].endsWith('voice-message.ogg')) return editOrReply(context, createEmbed("warning", context, "No voice message found."))
const recog = await googleSpeechRecognitionWithLabels(context, msg.attachments.first().url) const recog = await transcribeWithSpeakerLabelsObelisk(context, msg.attachments.first().url)
return editOrReply(context, createEmbed("default", context, { return editOrReply(context, createEmbed("default", context, {
description: codeblock("md", [ recog.response.body.transcription_with_speakers ]), description: codeblock("md", [ recog.response.body.transcription_with_speakers ]),

1
labscore/api/obelisk/endpoints.js
View file

@ -20,6 +20,7 @@ const ObeliskApi = Object.freeze({
AI_WALLPAPER: "/robin/v1/wallpaper:generate", AI_WALLPAPER: "/robin/v1/wallpaper:generate",
WEBSHOT: "/peacock/v1/screenshot", WEBSHOT: "/peacock/v1/screenshot",
TRANSCRIBE: "/peacock/v1/transcribe",
}) })
module.exports = { module.exports = {

6
labscore/api/obelisk/index.js
View file

@ -110,3 +110,9 @@ module.exports.webshot = async function(context, url, allowAdultContent = false)
allow_adult: allowAdultContent allow_adult: allowAdultContent
}) })
} }
module.exports.transcribeWithSpeakerLabelsObelisk = async function(context, url){
return await request(ObeliskApi.TRANSCRIBE, "POST", {}, {
url
})
}