refactor(streaming): Optimize image collection and debounce mechanism

lib/core/services/streaming_helper.dart

@@ -277,6 +277,13 @@ ActiveSocketStream attachUnifiedChunkedStreaming({
     )..start();
   }
 
+  Timer? imageCollectionDebounce;
+  String? pendingImageContent;
+  String? pendingImageMessageId;
+  String? pendingImageSignature;
+  String? lastProcessedImageSignature;
+  int imageCollectionRequestId = 0;
+
   void disposeSocketSubscriptions() {
     if (socketSubscriptions.isEmpty) {
       return;
@@ -287,56 +294,119 @@ ActiveSocketStream attachUnifiedChunkedStreaming({
       } catch (_) {}
     }
     socketSubscriptions.clear();
+    imageCollectionDebounce?.cancel();
+    imageCollectionDebounce = null;
+    pendingImageContent = null;
+    pendingImageMessageId = null;
+    pendingImageSignature = null;
+    lastProcessedImageSignature = null;
+    imageCollectionRequestId = 0;
     socketWatchdog?.stop();
   }
 
   bool isSearching = false;
 
+  void runPendingImageCollection() {
+    imageCollectionDebounce?.cancel();
+    imageCollectionDebounce = null;
+
+    final content = pendingImageContent;
+    final targetMessageId = pendingImageMessageId;
+    final signature = pendingImageSignature;
+    if (content == null || targetMessageId == null || signature == null) {
+      return;
+    }
+
+    pendingImageContent = null;
+    pendingImageMessageId = null;
+    pendingImageSignature = null;
+
+    final requestId = ++imageCollectionRequestId;
+    unawaited(
+      workerManager
+          .schedule<String, List<Map<String, dynamic>>>(
+            _collectImageReferencesWorker,
+            content,
+            debugLabel: 'stream_collect_images',
+          )
+          .then((collected) {
+            if (requestId != imageCollectionRequestId) {
+              return;
+            }
+
+            final currentMessages = getMessages();
+            if (currentMessages.isEmpty) {
+              return;
+            }
+            final last = currentMessages.last;
+            if (last.id != targetMessageId || last.role != 'assistant') {
+              return;
+            }
+
+            lastProcessedImageSignature = signature;
+
+            if (collected.isEmpty) {
+              return;
+            }
+
+            final existing = last.files ?? <Map<String, dynamic>>[];
+            final seen = <String>{
+              for (final f in existing)
+                if (f['url'] is String) (f['url'] as String) else '',
+            }..removeWhere((e) => e.isEmpty);
+
+            final merged = <Map<String, dynamic>>[...existing];
+            for (final f in collected) {
+              final url = f['url'] as String?;
+              if (url != null && url.isNotEmpty && !seen.contains(url)) {
+                merged.add({'type': 'image', 'url': url});
+                seen.add(url);
+              }
+            }
+
+            if (merged.length != existing.length) {
+              updateLastMessageWith((m) => m.copyWith(files: merged));
+            }
+          })
+          .catchError((_) {}),
+    );
+  }
+
   void updateImagesFromCurrentContent() {
     try {
       final msgs = getMessages();
       if (msgs.isEmpty || msgs.last.role != 'assistant') return;
-      final content = msgs.last.content;
+      final last = msgs.last;
+      final content = last.content;
       if (content.isEmpty) return;
 
-      final targetMessageId = msgs.last.id;
+      final targetMessageId = last.id;
-      unawaited(
+      final signature =
-        workerManager
+          '$targetMessageId:${content.hashCode}:${content.length}';
-            .schedule<String, List<Map<String, dynamic>>>(
-              _collectImageReferencesWorker,
-              content,
-              debugLabel: 'stream_collect_images',
-            )
-            .then((collected) {
-              if (collected.isEmpty) return;
-              final currentMessages = getMessages();
-              if (currentMessages.isEmpty) return;
-              final last = currentMessages.last;
-              if (last.id != targetMessageId || last.role != 'assistant') {
-                return;
-              }
 
-              final existing = last.files ?? <Map<String, dynamic>>[];
+      if (signature == lastProcessedImageSignature &&
-              final seen = <String>{
+          pendingImageSignature == null) {
-                for (final f in existing)
+        return;
-                  if (f['url'] is String) (f['url'] as String) else '',
+      }
-              }..removeWhere((e) => e.isEmpty);
+      if (signature == pendingImageSignature) {
+        return;
+      }
 
-              final merged = <Map<String, dynamic>>[...existing];
+      pendingImageMessageId = targetMessageId;
-              for (final f in collected) {
+      pendingImageContent = content;
-                final url = f['url'] as String?;
+      pendingImageSignature = signature;
-                if (url != null && url.isNotEmpty && !seen.contains(url)) {
-                  merged.add({'type': 'image', 'url': url});
-                  seen.add(url);
-                }
-              }
 
-              if (merged.length != existing.length) {
+      final shouldDelay = last.isStreaming;
-                updateLastMessageWith((m) => m.copyWith(files: merged));
+
-              }
+      imageCollectionDebounce?.cancel();
-            })
+      if (shouldDelay) {
-            .catchError((_) {}),
+        imageCollectionDebounce = Timer(
-      );
+          const Duration(milliseconds: 200),
+          runPendingImageCollection,
+        );
+      } else {
+        runPendingImageCollection();
+      }
     } catch (_) {}
   }
 

lib/features/chat/providers/chat_providers.dart

@@ -647,13 +647,6 @@ class ChatMessagesNotifier extends Notifier<List<ChatMessage>> {
       return;
     }
 
-    // Log content replacement for debugging
-    DebugLogger.log(
-      'Replacing message content: messageId=${lastMessage.id}, '
-      'oldLength=${lastMessage.content.length}, newLength=${content.length}',
-      scope: 'chat/providers',
-    );
-
     _ensureFormatterForMessage(lastMessage);
 
     // Defensive check: ensure the formatter is for the correct message

lib/features/chat/widgets/assistant_message_widget.dart

@@ -71,6 +71,11 @@ class _AssistantMessageWidgetState extends ConsumerState<AssistantMessageWidget>
   bool _allowTypingIndicator = false;
   Timer? _typingGateTimer;
   String _ttsPlainText = '';
+  Timer? _ttsPlainTextDebounce;
+  Map<String, dynamic>? _pendingTtsPlainTextPayload;
+  String? _pendingTtsPlainTextSource;
+  String? _lastAppliedTtsPlainTextSource;
+  int _ttsPlainTextRequestId = 0;
   // Active version index (-1 means current/live content)
   int _activeVersionIndex = -1;
   // press state handled by shared ChatActionButton
@@ -162,13 +167,11 @@ class _AssistantMessageWidgetState extends ConsumerState<AssistantMessageWidget>
     final rSegs = ReasoningParser.segments(raw);
 
     final out = <MessageSegment>[];
-    final textBuf = StringBuffer();
     final textSegments = <String>[];
     if (rSegs == null || rSegs.isEmpty) {
       final tSegs = ToolCallsParser.segments(raw);
       if (tSegs == null || tSegs.isEmpty) {
         out.add(MessageSegment.text(raw));
-        textBuf.write(raw);
         textSegments.add(raw);
       } else {
         for (final s in tSegs) {
@@ -176,7 +179,6 @@ class _AssistantMessageWidgetState extends ConsumerState<AssistantMessageWidget>
             out.add(MessageSegment.tool(s.entry!));
           } else if ((s.text ?? '').isNotEmpty) {
             out.add(MessageSegment.text(s.text!));
-            textBuf.write(s.text);
             textSegments.add(s.text!);
           }
         }
@@ -190,7 +192,6 @@ class _AssistantMessageWidgetState extends ConsumerState<AssistantMessageWidget>
           final tSegs = ToolCallsParser.segments(t);
           if (tSegs == null || tSegs.isEmpty) {
             out.add(MessageSegment.text(t));
-            textBuf.write(t);
             textSegments.add(t);
           } else {
             for (final s in tSegs) {
@@ -198,7 +199,6 @@ class _AssistantMessageWidgetState extends ConsumerState<AssistantMessageWidget>
                 out.add(MessageSegment.tool(s.entry!));
               } else if ((s.text ?? '').isNotEmpty) {
                 out.add(MessageSegment.text(s.text!));
-                textBuf.write(s.text);
                 textSegments.add(s.text!);
               }
             }
@@ -208,23 +208,15 @@ class _AssistantMessageWidgetState extends ConsumerState<AssistantMessageWidget>
     }
 
     final segments = out.isEmpty ? [MessageSegment.text(raw)] : out;
-    String speechText;
-    try {
-      final worker = ref.read(workerManagerProvider);
-      speechText = await worker.schedule<Map<String, dynamic>, String>(
-        _buildTtsPlainTextWorker,
-        {'segments': textSegments, 'fallback': raw},
-        debugLabel: 'tts_plain_text',
-      );
-    } catch (_) {
-      speechText = _buildTtsPlainTextFallback(textSegments, raw);
-    }
 
     if (!mounted) return;
     setState(() {
       _segments = segments;
-      _ttsPlainText = speechText;
     });
+    _scheduleTtsPlainTextBuild(
+      List<String>.from(textSegments, growable: false),
+      raw,
+    );
     _updateTypingIndicatorGate();
   }
 
@@ -290,6 +282,96 @@ class _AssistantMessageWidgetState extends ConsumerState<AssistantMessageWidget>
     return result;
   }
 
+  void _scheduleTtsPlainTextBuild(List<String> segments, String raw) {
+    final hasContent =
+        segments.any((segment) => segment.trim().isNotEmpty) ||
+        raw.trim().isNotEmpty;
+    if (!hasContent) {
+      _pendingTtsPlainTextPayload = null;
+      _pendingTtsPlainTextSource = null;
+      _lastAppliedTtsPlainTextSource = '';
+      if (_ttsPlainText.isNotEmpty && mounted) {
+        setState(() {
+          _ttsPlainText = '';
+        });
+      }
+      return;
+    }
+
+    if (_pendingTtsPlainTextPayload == null &&
+        raw == _lastAppliedTtsPlainTextSource) {
+      return;
+    }
+    if (raw == _pendingTtsPlainTextSource &&
+        _pendingTtsPlainTextPayload != null) {
+      return;
+    }
+
+    final pendingSegments = List<String>.from(segments, growable: false);
+    _pendingTtsPlainTextPayload = {
+      'segments': pendingSegments,
+      'fallback': raw,
+    };
+    _pendingTtsPlainTextSource = raw;
+
+    final delay = widget.isStreaming
+        ? const Duration(milliseconds: 250)
+        : Duration.zero;
+
+    _ttsPlainTextDebounce?.cancel();
+    if (delay == Duration.zero) {
+      _runPendingTtsPlainTextBuild();
+    } else {
+      _ttsPlainTextDebounce = Timer(delay, _runPendingTtsPlainTextBuild);
+    }
+  }
+
+  void _runPendingTtsPlainTextBuild() {
+    _ttsPlainTextDebounce?.cancel();
+    _ttsPlainTextDebounce = null;
+
+    final payload = _pendingTtsPlainTextPayload;
+    final source = _pendingTtsPlainTextSource;
+    if (payload == null || source == null) {
+      return;
+    }
+
+    _pendingTtsPlainTextPayload = null;
+    _pendingTtsPlainTextSource = null;
+    final requestId = ++_ttsPlainTextRequestId;
+    unawaited(_executeTtsPlainTextBuild(payload, source, requestId));
+  }
+
+  Future<void> _executeTtsPlainTextBuild(
+    Map<String, dynamic> payload,
+    String raw,
+    int requestId,
+  ) async {
+    final segments = (payload['segments'] as List).cast<String>();
+    String speechText;
+    try {
+      final worker = ref.read(workerManagerProvider);
+      speechText = await worker.schedule<Map<String, dynamic>, String>(
+        _buildTtsPlainTextWorker,
+        payload,
+        debugLabel: 'tts_plain_text',
+      );
+    } catch (_) {
+      speechText = _buildTtsPlainTextFallback(segments, raw);
+    }
+
+    if (!mounted || requestId != _ttsPlainTextRequestId) {
+      return;
+    }
+
+    _lastAppliedTtsPlainTextSource = raw;
+    if (_ttsPlainText != speechText) {
+      setState(() {
+        _ttsPlainText = speechText;
+      });
+    }
+  }
+
   // No streaming-specific markdown fixes needed here; handled by Markdown widget
 
   Widget _buildToolCallTile(ToolCallEntry tc) {
@@ -622,6 +704,9 @@ class _AssistantMessageWidgetState extends ConsumerState<AssistantMessageWidget>
   @override
   void dispose() {
     _typingGateTimer?.cancel();
+    _ttsPlainTextDebounce?.cancel();
+    _pendingTtsPlainTextPayload = null;
+    _pendingTtsPlainTextSource = null;
     _fadeController.dispose();
     _slideController.dispose();
     super.dispose();

lib/features/profile/views/app_customization_page.dart

@@ -871,19 +871,13 @@ class AppCustomizationPage extends ConsumerWidget {
                 duration: const Duration(milliseconds: 200),
                 child: Text(
                   ttsDescription,
-                  key: ValueKey<String>(
+                  key: ValueKey<String>('tts-desc-${settings.ttsEngine.name}'),
-                    'tts-desc-${settings.ttsEngine.name}',
-                  ),
                   style:
                       theme.bodyMedium?.copyWith(
-                        color: theme.sidebarForeground.withValues(
+                        color: theme.sidebarForeground.withValues(alpha: 0.9),
-                          alpha: 0.9,
-                        ),
                       ) ??
                       TextStyle(
-                        color: theme.sidebarForeground.withValues(
+                        color: theme.sidebarForeground.withValues(alpha: 0.9),
-                          alpha: 0.9,
-                        ),
                         fontSize: 14,
                       ),
                 ),