en ch

youtube_sub.js

@@ -661,6 +661,12 @@ GM_addStyle(`
             this.currentTheme = this.currentTheme === 'dark' ? 'light' : 'dark';
             this.container.dataset.theme = this.currentTheme;
         }
+        updateTabText(tabId, text) {
+            const tab = this.container.querySelector(`button[data-tab="${tabId}"]`);
+            if (tab) {
+                tab.textContent = text;
+            }
+        }
 
         setupUI() {
             debug.log('Setting up UI');
@@ -863,9 +869,9 @@ GM_addStyle(`
                                             RETURN_TRUSTED_TYPE: true
                                         });
 
-                                        try{
+                                        try {
                                             temp.innerHTML = sanitizedHtml;
-                                        }catch(e){
+                                        } catch (e) {
                                             console.error('DOMPurify error:', e);
                                         }
 
@@ -1213,6 +1219,81 @@ GM_addStyle(`
                 }
             }
 
+            this.updateSubtitleDisplay();
+
+            // If no Chinese subtitles, get translation
+            if (this.subtitles.english.length > 0 && !this.subtitles.chinese.length) {
+                // Format English subtitles as JSON
+                const subtitleData = this.subtitles.english.map(sub => ({
+                    text: sub.words.map(word => word.text).join(' '),
+                    time: sub.startTime
+                }));
+
+                this.updateTabText('chinese', '翻译中...');
+
+                try {
+                    const response = await new Promise((resolve, reject) => {
+                        GM_xmlhttpRequest({
+                            method: 'POST',
+                            url: 'https://sonygod-flash.hf.space/ask',
+                            headers: {
+                                'Content-Type': 'application/json',
+                                'Accept': 'application/json'
+                            },
+                            data: JSON.stringify({
+                                prompt: `请将以下英文字幕翻译成中文，保持JSON格式返回:
+                    ${JSON.stringify(subtitleData, null, 2)}
+                    请返回格式:
+                    {
+                        "translations": [
+                            {
+                                "text": "中文翻译",
+                                "time": 原时间戳
+                            }
+                        ]
+                    }`,
+                                model: 'GEMINI'
+                            }),
+                            onload: response => {
+                                if (response.status === 200) {
+                                    resolve(JSON.parse(response.responseText));
+                                } else {
+                                    reject(new Error(`HTTP error: ${response.status}`));
+                                }
+                            },
+                            onerror: error => reject(error)
+                        });
+                    });
+
+                    const content = response.data.response;
+                   // const jsonMatch = content.match(/```json\s*([\s\S]*?)\s*```/);
+                   const jsonMatch = content.match(/(?:")?```json\s*([\s\S]*?)\s*```(?:")?/);
+                   
+                    if (!jsonMatch) {
+                        throw new Error('No JSON content found in response');
+                    }
+
+                    const parsedResponse = JSON.parse(jsonMatch[1]);
+                    const translations = parsedResponse.translations || [];
+
+                    // Create Chinese subtitles array matching English structure
+                    this.subtitles.chinese = this.subtitles.english.map((sub, index) => ({
+                        startTime: sub.startTime,
+                        endTime: sub.endTime,
+                        words: [{
+                            text: translations[index]?.text || '',
+                            startTime: sub.startTime,
+                            endTime: sub.endTime
+                        }]
+                    }));
+
+                    this.updateTabText('chinese', '中文');
+                } catch (error) {
+                    this.updateTabText('chinese', '翻译失败');
+                    debug.error('Translation failed:', error);
+                }
+            }
+
             // 4. Display subtitles
             this.updateSubtitleDisplay();
         }