Update flare-ui/src/app/services/conversation-manager.service.ts

flare-ui/src/app/services/conversation-manager.service.ts

@@ -296,44 +296,6 @@ export class ConversationManagerService implements OnDestroy {
     }
   }
 
-  private handleTTSAudio(message: any): void {
-    try {
-      // Validate audio data
-      if (!message['data']) {
-        console.warn('TTS audio message missing data');
-        return;
-      }
-      
-      // Accumulate audio chunks
-      this.audioQueue.push(message['data']);
-      
-      if (message['is_last']) {
-        // All chunks received, create audio blob
-        const audioData = this.audioQueue.join('');
-        const audioBlob = this.base64ToBlob(audioData, message['mime_type'] || 'audio/mpeg');
-        const audioUrl = URL.createObjectURL(audioBlob);
-        
-        // Update last message with audio URL
-        const messages = this.messagesSubject.value;
-        if (messages.length > 0 && messages[messages.length - 1].role === 'assistant') {
-          messages[messages.length - 1].audioUrl = audioUrl;
-          this.messagesSubject.next([...messages]);
-        }
-        
-        // Clear queue
-        this.audioQueue = [];
-        
-        // Auto-play if welcome message
-        if (message['is_welcome']) {
-          setTimeout(() => this.playAudio(audioUrl), 100);
-        }
-      }
-    } catch (error) {
-      console.error('Error handling TTS audio:', error);
-      this.audioQueue = []; // Clear queue on error
-    }
-  }
-
   private playQueuedAudio(): void {
     const messages = this.messagesSubject.value;
     const lastMessage = messages[messages.length - 1];
@@ -474,9 +436,72 @@ export class ConversationManagerService implements OnDestroy {
     this.messagesSubject.next([...messages]);
   }
 
+  private handleTTSAudio(message: any): void {
+    try {
+      // Validate audio data
+      if (!message['data']) {
+        console.warn('TTS audio message missing data');
+        return;
+      }
+      
+      // Check if we need to accumulate chunks
+      if (!message['is_last']) {
+        // Accumulate audio chunks
+        this.audioQueue.push(message['data']);
+        return;
+      }
+      
+      // All chunks received, create audio blob
+      this.audioQueue.push(message['data']); // Add last chunk
+      const audioData = this.audioQueue.join('');
+      
+      try {
+        const audioBlob = this.base64ToBlob(audioData, message['mime_type'] || 'audio/mpeg');
+        const audioUrl = URL.createObjectURL(audioBlob);
+        
+        // Update last message with audio URL
+        const messages = this.messagesSubject.value;
+        if (messages.length > 0 && messages[messages.length - 1].role === 'assistant') {
+          messages[messages.length - 1].audioUrl = audioUrl;
+          this.messagesSubject.next([...messages]);
+        }
+        
+        // Clear queue
+        this.audioQueue = [];
+        
+        // Auto-play if in playing_audio state
+        if (this.currentStateSubject.value === 'playing_audio') {
+          this.playQueuedAudio();
+        }
+      } catch (error) {
+        console.error('Error creating audio blob:', error);
+        this.audioQueue = []; // Clear queue on error
+        throw error;
+      }
+    } catch (error) {
+      console.error('Error handling TTS audio:', error);
+      this.audioQueue = []; // Clear queue on error
+    }
+  }
+  
   private base64ToBlob(base64: string, mimeType: string): Blob {
     try {
-      const byteCharacters = atob(base64);
+      // Clean base64 string - remove any whitespace or newlines
+      const cleanBase64 = base64.replace(/\s/g, '');
+      
+      // Validate base64 string
+      if (!cleanBase64 || cleanBase64.length === 0) {
+        throw new Error('Empty base64 string');
+      }
+      
+      // Check if base64 string is valid
+      const base64Regex = /^[A-Za-z0-9+/]*={0,2}$/;
+      if (!base64Regex.test(cleanBase64)) {
+        throw new Error('Invalid base64 string format');
+      }
+      
+      // Decode base64
+      const byteCharacters = atob(cleanBase64);
       const byteNumbers = new Array(byteCharacters.length);
       
       for (let i = 0; i < byteCharacters.length; i++) {
@@ -487,6 +512,7 @@ export class ConversationManagerService implements OnDestroy {
       return new Blob([byteArray], { type: mimeType });
     } catch (error) {
       console.error('Error converting base64 to blob:', error);
+      console.error('Base64 preview:', base64.substring(0, 100) + '...');
       throw new Error('Failed to convert audio data');
     }
   }