From 40abedbe4479aeadcb055f57b35c5bc189a2ccd0 Mon Sep 17 00:00:00 2001
From: Laurent Denoue <ldenoue@google.com>
Date: Sat, 7 Dec 2024 10:44:55 +0100
Subject: [PATCH] fixed language detection

---
 code.js | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/code.js b/code.js
index 1c87b40..44ea202 100644
--- a/code.js
+++ b/code.js
@@ -32,7 +32,7 @@ if (DEEP_INFRA_API_KEY) {
 const params = new URLSearchParams(window.location.search)
 let currentProvider = params.get('model') || geminiModel
 
-const languageCode = params.get('language') || 'en'
+let languageCode = params.get('language')// || 'en'
 const followingAudio = true
 const chapterDelta = 1000
 let chapters = []
@@ -1197,7 +1197,7 @@ function getTranscriptURLAndLanguage(yt, preferredLanguage = 'en') {
     }
     return { defaultLanguage, transcripts: obj, translationLanguages }
 }
-async function getLocal(videoId, languageCode = 'en') {
+async function getLocal(videoId, languageCode) {
     const data = await getUserData(videoId)
     if (data && data[languageCode]) {
         return data
@@ -1267,7 +1267,6 @@ async function getLocal(videoId, languageCode = 'en') {
     obj.thumbnail = `https://img.youtube.com/vi/${obj.videoId}/mqdefault.jpg`
     const { defaultLanguage, transcripts, translationLanguages } = getTranscriptURLAndLanguage(json, languageCode)
     obj.translationLanguages = translationLanguages
-    const languageCodes = Object.keys(transcripts)
     obj.defaultLanguage = defaultLanguage ?? 'en'
     for (let languageCode in transcripts) {
         const chunks = await getChunks(transcripts[languageCode])
@@ -1333,14 +1332,18 @@ function showError(msg) {
 
 let transcript = null
 let vocab = null
-async function punctuate(videoId, languageCode = 'en') {
+async function punctuate(videoId, preferedLanguageCode) {
     apiCalls = 0
     totalPrice = 0
     inputTokens = 0
     outputTokens = 0
     totalTokens = 0
     updateEstimatedPrice(inputTokens, outputTokens, totalTokens, totalPrice)
-    let json = await getLocal(videoId, languageCode)
+    let json = await getLocal(videoId, preferedLanguageCode)
+    if (!preferedLanguageCode && json && json.defaultLanguage) {
+      languageCode = json.defaultLanguage
+    }
+    console.log('languageCode',languageCode, preferedLanguageCode)
     window.json = json
     if (json.error) {
         container.style.display = 'none'