From 0cac2568d884210d98416891de2d87773cbebfa3 Mon Sep 17 00:00:00 2001
From: derpystuff <3515180-derpystuff@users.noreply.gitlab.com>
Date: Mon, 8 Apr 2024 20:29:59 +0200
Subject: [PATCH] update transcription to use obelisk service

---
 commands/interaction/context/transcribe.js | 4 ++--
 commands/message/google/transcribe.js      | 4 ++--
 labscore/api/obelisk/endpoints.js          | 1 +
 labscore/api/obelisk/index.js              | 6 ++++++
 4 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/commands/interaction/context/transcribe.js b/commands/interaction/context/transcribe.js
index ee98dd9..d1a6aed 100644
--- a/commands/interaction/context/transcribe.js
+++ b/commands/interaction/context/transcribe.js
@@ -1,7 +1,7 @@
 const { Constants } = require('detritus-client');
 const { InteractionCallbackTypes, ApplicationCommandTypes, MessageFlags } = Constants;
 
-const {  googleSpeechRecognitionWithLabels } = require('../../../labscore/api');
+const { transcribeWithSpeakerLabelsObelisk } = require('../../../labscore/api/obelisk');
 
 const { createEmbed } = require('../../../labscore/utils/embed');
 const { codeblock } = require('../../../labscore/utils/markdown');
@@ -34,7 +34,7 @@ module.exports = {
           flags: MessageFlags.EPHEMERAL
         })
         
-        const recog = await googleSpeechRecognitionWithLabels(context, message.attachments.first().url)
+        const recog = await transcribeWithSpeakerLabelsObelisk(context, message.attachments.first().url)
   
         return editOrReply(context, {
           embeds: [createEmbed("default", context, {
diff --git a/commands/message/google/transcribe.js b/commands/message/google/transcribe.js
index 96f7eaa..f5e595c 100644
--- a/commands/message/google/transcribe.js
+++ b/commands/message/google/transcribe.js
@@ -1,7 +1,7 @@
 const { createEmbed } = require('../../../labscore/utils/embed')
 const { editOrReply } = require('../../../labscore/utils/message')
 
-const { googleSpeechRecognitionWithLabels } = require('../../../labscore/api');
+const { transcribeWithSpeakerLabelsObelisk } = require('../../../labscore/api/obelisk');
 
 const { STATICS } = require('../../../labscore/utils/statics');
 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().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, {
         description: codeblock("md", [ recog.response.body.transcription_with_speakers ]),
diff --git a/labscore/api/obelisk/endpoints.js b/labscore/api/obelisk/endpoints.js
index 0d1535d..d5a202e 100644
--- a/labscore/api/obelisk/endpoints.js
+++ b/labscore/api/obelisk/endpoints.js
@@ -20,6 +20,7 @@ const ObeliskApi = Object.freeze({
   AI_WALLPAPER: "/robin/v1/wallpaper:generate",
   
   WEBSHOT: "/peacock/v1/screenshot",
+  TRANSCRIBE: "/peacock/v1/transcribe",
 })
 
 module.exports = {
diff --git a/labscore/api/obelisk/index.js b/labscore/api/obelisk/index.js
index 9270198..0dc9861 100644
--- a/labscore/api/obelisk/index.js
+++ b/labscore/api/obelisk/index.js
@@ -109,4 +109,10 @@ module.exports.webshot = async function(context, url, allowAdultContent = false)
     url,
     allow_adult: allowAdultContent
   })
+}
+
+module.exports.transcribeWithSpeakerLabelsObelisk = async function(context, url){
+  return await request(ObeliskApi.TRANSCRIBE, "POST", {}, {
+    url
+  })
 }
\ No newline at end of file