agentgui 1.0.175 → 1.0.176

package/lib/speech.js

@@ -116,13 +116,17 @@ let sttLoading = false;
 let ttsLoading = false;
 let speakerEmbeddingLoading = false;
 let ttsLoadError = null;
+let ttsLoadErrorTime = 0;
 let sttLoadError = null;
 const voiceEmbeddingsCache = new Map();
 const SAMPLE_RATE_STT = 16000;
 const SAMPLE_RATE_TTS = 16000;
+const TTS_ERROR_RETRY_MS = 30000;
 
-const TTS_CACHE_MAX = 100;
+const TTS_CACHE_MAX_BYTES = 10 * 1024 * 1024;
+let ttsCacheBytes = 0;
 const ttsCache = new Map();
+const ttsInflight = new Map();
 
 async function loadTransformers() {
   if (transformersModule) return transformersModule;
@@ -295,7 +299,11 @@ async function getSTT() {
 
 async function getTTS() {
   if (ttsPipeline) return ttsPipeline;
-  if (ttsLoadError) throw ttsLoadError;
+  if (ttsLoadError) {
+    if (Date.now() - ttsLoadErrorTime < TTS_ERROR_RETRY_MS) throw ttsLoadError;
+    ttsLoadError = null;
+    ttsLoadErrorTime = 0;
+  }
   if (ttsLoading) {
     while (ttsLoading) await new Promise(r => setTimeout(r, 100));
     if (ttsLoadError) throw ttsLoadError;
@@ -312,10 +320,12 @@ async function getTTS() {
     });
     await ensureSpeakerEmbeddings();
     ttsLoadError = null;
+    ttsLoadErrorTime = 0;
     return ttsPipeline;
   } catch (err) {
     ttsPipeline = null;
     ttsLoadError = new Error('TTS model load failed: ' + err.message);
+    ttsLoadErrorTime = Date.now();
     throw ttsLoadError;
   } finally {
     ttsLoading = false;
@@ -443,11 +453,17 @@ function splitSentences(text) {
 }
 
 function cachePut(key, buf) {
-  if (ttsCache.size >= TTS_CACHE_MAX) {
+  if (ttsCache.has(key)) {
+    ttsCacheBytes -= ttsCache.get(key).length;
+    ttsCache.delete(key);
+  }
+  while (ttsCacheBytes + buf.length > TTS_CACHE_MAX_BYTES && ttsCache.size > 0) {
     const oldest = ttsCache.keys().next().value;
+    ttsCacheBytes -= ttsCache.get(oldest).length;
     ttsCache.delete(oldest);
   }
   ttsCache.set(key, buf);
+  ttsCacheBytes += buf.length;
 }
 
 async function synthesize(text, voiceId) {
@@ -458,12 +474,18 @@ async function synthesize(text, voiceId) {
     ttsCache.set(cacheKey, cached);
     return cached;
   }
-  const tts = await getTTS();
-  const embeddings = await loadVoiceEmbedding(voiceId);
-  const result = await tts(text, { speaker_embeddings: embeddings });
-  const wav = encodeWav(result.audio, result.sampling_rate || SAMPLE_RATE_TTS);
-  cachePut(cacheKey, wav);
-  return wav;
+  const inflight = ttsInflight.get(cacheKey);
+  if (inflight) return inflight;
+  const promise = (async () => {
+    const tts = await getTTS();
+    const embeddings = await loadVoiceEmbedding(voiceId);
+    const result = await tts(text, { speaker_embeddings: embeddings });
+    const wav = encodeWav(result.audio, result.sampling_rate || SAMPLE_RATE_TTS);
+    cachePut(cacheKey, wav);
+    return wav;
+  })();
+  ttsInflight.set(cacheKey, promise);
+  try { return await promise; } finally { ttsInflight.delete(cacheKey); }
 }
 
 async function* synthesizeStream(text, voiceId) {
@@ -497,4 +519,18 @@ function getStatus() {
   };
 }
 
-export { transcribe, synthesize, synthesizeStream, getSTT, getTTS, getStatus, getVoices };
+function preloadTTS() {
+  getTTS().catch(err => console.error('[TTS] Preload failed:', err.message));
+}
+
+function ttsCacheKey(text, voiceId) {
+  return (voiceId || 'default') + ':' + text;
+}
+
+function ttsCacheGet(key) {
+  const cached = ttsCache.get(key);
+  if (cached) { ttsCache.delete(key); ttsCache.set(key, cached); }
+  return cached || null;
+}
+
+export { transcribe, synthesize, synthesizeStream, getSTT, getTTS, getStatus, getVoices, preloadTTS, ttsCacheKey, ttsCacheGet, splitSentences };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentgui",
-  "version": "1.0.175",
+  "version": "1.0.176",
   "description": "Multi-agent ACP client with real-time communication",
   "type": "module",
   "main": "server.js",

package/server.js

@@ -15,6 +15,49 @@ async function getSpeech() {
   return speechModule;
 }
 
+function eagerTTS(text, conversationId, sessionId) {
+  getSpeech().then(speech => {
+    const status = speech.getStatus();
+    if (!status.ttsReady || status.ttsError) return;
+    const voices = new Set();
+    for (const ws of syncClients) {
+      const vid = ws.ttsVoiceId || 'default';
+      const convKey = `conv-${conversationId}`;
+      if (ws.subscriptions && (ws.subscriptions.has(sessionId) || ws.subscriptions.has(convKey))) {
+        voices.add(vid);
+      }
+    }
+    if (voices.size === 0) return;
+    const sentences = speech.splitSentences(text);
+    for (const vid of voices) {
+      for (const sentence of sentences) {
+        const cacheKey = speech.ttsCacheKey(sentence, vid);
+        const cached = speech.ttsCacheGet(cacheKey);
+        if (cached) {
+          pushTTSAudio(cacheKey, cached, conversationId, sessionId, vid);
+          continue;
+        }
+        speech.synthesize(sentence, vid).then(wav => {
+          pushTTSAudio(cacheKey, wav, conversationId, sessionId, vid);
+        }).catch(() => {});
+      }
+    }
+  }).catch(() => {});
+}
+
+function pushTTSAudio(cacheKey, wav, conversationId, sessionId, voiceId) {
+  const b64 = wav.toString('base64');
+  broadcastSync({
+    type: 'tts_audio',
+    cacheKey,
+    audio: b64,
+    voiceId,
+    conversationId,
+    sessionId,
+    timestamp: Date.now()
+  });
+}
+
 const require = createRequire(import.meta.url);
 const express = require('express');
 const Busboy = require('busboy');
@@ -852,6 +895,10 @@ async function processMessageWithStreaming(conversationId, messageId, sessionId,
             blockIndex: allBlocks.length - 1,
             timestamp: Date.now()
           });
+
+          if (block.type === 'text' && block.text) {
+            eagerTTS(block.text, conversationId, sessionId);
+          }
         }
       } else if (parsed.type === 'user' && parsed.message?.content) {
         for (const block of parsed.message.content) {
@@ -900,6 +947,11 @@ async function processMessageWithStreaming(conversationId, messageId, sessionId,
           timestamp: Date.now()
         });
 
+        if (parsed.result) {
+          const resultText = typeof parsed.result === 'string' ? parsed.result : JSON.stringify(parsed.result);
+          if (resultText) eagerTTS(resultText, conversationId, sessionId);
+        }
+
         if (parsed.result && allBlocks.length === 0) {
           allBlocks.push({ type: 'text', text: String(parsed.result) });
         }
@@ -1128,6 +1180,8 @@ wss.on('connection', (ws, req) => {
             subscriptions: Array.from(ws.subscriptions),
             timestamp: Date.now()
           }));
+        } else if (data.type === 'set_voice') {
+          ws.ttsVoiceId = data.voiceId || 'default';
         } else if (data.type === 'ping') {
           ws.send(JSON.stringify({
             type: 'pong',

package/static/js/voice.js

@@ -15,6 +15,8 @@
   var spokenChunks = new Set();
   var isLoadingHistory = false;
   var selectedVoiceId = localStorage.getItem('voice-selected-id') || 'default';
+  var ttsAudioCache = new Map();
+  var TTS_CLIENT_CACHE_MAX = 50;
 
   function init() {
     setupTTSToggle();
@@ -69,6 +71,7 @@
     selector.addEventListener('change', function() {
       selectedVoiceId = selector.value;
       localStorage.setItem('voice-selected-id', selectedVoiceId);
+      sendVoiceToServer();
     });
   }
 
@@ -295,6 +298,22 @@
     processQueue();
   }
 
+  function cacheTTSAudio(cacheKey, b64) {
+    if (ttsAudioCache.size >= TTS_CLIENT_CACHE_MAX) {
+      var oldest = ttsAudioCache.keys().next().value;
+      ttsAudioCache.delete(oldest);
+    }
+    var binary = atob(b64);
+    var bytes = new Uint8Array(binary.length);
+    for (var i = 0; i < binary.length; i++) bytes[i] = binary.charCodeAt(i);
+    ttsAudioCache.set(cacheKey, new Blob([bytes], { type: 'audio/wav' }));
+  }
+
+  function getCachedTTSBlob(text) {
+    var key = selectedVoiceId + ':' + text;
+    return ttsAudioCache.get(key) || null;
+  }
+
   var audioChunkQueue = [];
   var isPlayingChunk = false;
   var streamDone = false;
@@ -344,6 +363,15 @@
     audioChunkQueue = [];
     isPlayingChunk = false;
 
+    var cachedBlob = getCachedTTSBlob(text);
+    if (cachedBlob) {
+      ttsConsecutiveFailures = 0;
+      audioChunkQueue.push(cachedBlob);
+      streamDone = true;
+      if (!isPlayingChunk) playNextChunk();
+      return;
+    }
+
     function onTtsSuccess() {
       ttsConsecutiveFailures = 0;
     }
@@ -532,11 +560,23 @@
     }
   }
 
+  function sendVoiceToServer() {
+    if (typeof agentGUIClient !== 'undefined' && agentGUIClient && agentGUIClient.wsManager && agentGUIClient.wsManager.isConnected) {
+      agentGUIClient.wsManager.sendMessage({ type: 'set_voice', voiceId: selectedVoiceId });
+    }
+  }
+
   function setupStreamingListener() {
     window.addEventListener('ws-message', function(e) {
-      if (!voiceActive) return;
       var data = e.detail;
       if (!data) return;
+      if (data.type === 'tts_audio' && data.audio && data.voiceId === selectedVoiceId) {
+        cacheTTSAudio(data.cacheKey, data.audio);
+      }
+      if (data.type === 'sync_connected') {
+        sendVoiceToServer();
+      }
+      if (!voiceActive) return;
       if (data.type === 'streaming_progress' && data.block) {
         handleVoiceBlock(data.block, true);
       }