refactor: enhance markdown processing and code structure

- Updated the ConduitMarkdown class to streamline markdown rendering, improving maintainability and clarity.
- Refactored the markdown configuration to utilize new methods for building markdown blocks and handling LaTeX syntax.
- Improved the StreamingMarkdownWidget to leverage the updated markdown processing logic, ensuring a cohesive user experience.
- Enhanced the handling of Mermaid diagrams and LaTeX rendering, providing better support for complex markdown content.

lib/core/auth/auth_state_manager.dart

@@ -610,12 +610,11 @@ class AuthStateManager extends _$AuthStateManager {
 
   /// Prime the conversations list so navigation drawers show real data after login.
   void _prefetchConversations() {
-    Future.microtask(() async {
+    Future.microtask(() {
       if (!ref.mounted) return;
       try {
         refreshConversationsCache(ref, includeFolders: true);
-        await ref.read(conversationsProvider.future);
+        DebugLogger.auth('Conversations prefetch scheduled');
-        DebugLogger.auth('Conversations prefetch requested');
       } catch (e) {
         if (!ref.mounted) return;
         DebugLogger.warning(

lib/shared/widgets/markdown/code_block_header.dart

@@ -0,0 +1,49 @@
+import 'package:flutter/material.dart';
+import '../../theme/theme_extensions.dart';
+
+class CodeBlockHeader extends StatelessWidget {
+  const CodeBlockHeader({
+    super.key,
+    required this.language,
+    required this.onCopy,
+  });
+
+  final String language;
+  final VoidCallback onCopy;
+
+  @override
+  Widget build(BuildContext context) {
+    final theme = context.conduitTheme;
+    final label = language.isEmpty ? 'code' : language;
+    return Container(
+      padding: const EdgeInsets.symmetric(
+        horizontal: Spacing.sm,
+        vertical: Spacing.xs,
+      ),
+      decoration: BoxDecoration(
+        color: theme.surfaceContainer.withValues(alpha: 0.35),
+        borderRadius: const BorderRadius.vertical(
+          top: Radius.circular(AppBorderRadius.sm),
+        ),
+      ),
+      child: Row(
+        children: [
+          Text(
+            label,
+            style: AppTypography.codeStyle.copyWith(
+              color: theme.textSecondary,
+              fontWeight: FontWeight.w600,
+            ),
+          ),
+          const Spacer(),
+          IconButton(
+            icon: const Icon(Icons.copy_rounded, size: 18),
+            color: theme.iconPrimary,
+            tooltip: 'Copy code',
+            onPressed: onCopy,
+          ),
+        ],
+      ),
+    );
+  }
+}

lib/shared/widgets/markdown/latex_block_widget.dart

@@ -0,0 +1,41 @@
+import 'package:flutter/material.dart';
+import 'package:flutter_math_fork/flutter_math.dart';
+
+import '../../theme/theme_extensions.dart';
+
+class LatexBlockWidget extends StatelessWidget {
+  const LatexBlockWidget({
+    super.key,
+    required this.content,
+    required this.isInline,
+    required this.style,
+    required this.isDark,
+  });
+
+  final String content;
+  final bool isInline;
+  final TextStyle style;
+  final bool isDark;
+
+  @override
+  Widget build(BuildContext context) {
+    final mathWidget = Math.tex(
+      content,
+      mathStyle: MathStyle.text,
+      textStyle: style,
+      textScaleFactor: 1,
+      onErrorFallback: (error) {
+        return Text(content, style: style.copyWith(color: Colors.red));
+      },
+    );
+
+    if (isInline) {
+      return mathWidget;
+    }
+
+    return Padding(
+      padding: const EdgeInsets.symmetric(vertical: Spacing.xs),
+      child: Center(child: mathWidget),
+    );
+  }
+}

lib/shared/widgets/markdown/markdown_config.dart

@@ -6,53 +6,73 @@ import 'package:flutter/foundation.dart';
 import 'package:flutter/gestures.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter/services.dart';
+import 'package:flutter_highlight/flutter_highlight.dart';
 import 'package:flutter_highlight/themes/a11y-dark.dart';
 import 'package:flutter_highlight/themes/a11y-light.dart';
-import 'package:markdown/markdown.dart' as m;
 import 'package:markdown_widget/markdown_widget.dart';
 import 'package:webview_flutter/webview_flutter.dart';
 
 import '../../theme/color_tokens.dart';
 import '../../theme/theme_extensions.dart';
+import 'code_block_header.dart';
 import 'markdown_latex.dart';
 
-/// Callback invoked when a markdown link is tapped.
 typedef MarkdownLinkTapCallback = void Function(String url, String title);
 
-/// Bundles markdown configuration and generator metadata for the app theme.
+class ConduitMarkdown {
-class ConduitMarkdownTheme {
+  const ConduitMarkdown._();
-  const ConduitMarkdownTheme({
-    required this.config,
-    required this.inlineSyntaxes,
-    required this.blockSyntaxes,
-    required this.generators,
-    required this.linesMargin,
-  });
 
-  final MarkdownConfig config;
+  static MarkdownWidget build({
-  final List<m.InlineSyntax> inlineSyntaxes;
+    required BuildContext context,
-  final List<m.BlockSyntax> blockSyntaxes;
+    required String data,
-  final List<SpanNodeGeneratorWithTag> generators;
+    MarkdownLinkTapCallback? onTapLink,
-  final EdgeInsets linesMargin;
+    bool selectable = true,
-
+    bool shrinkWrap = false,
-  MarkdownGenerator createGenerator() {
+    ScrollPhysics? physics,
-    return MarkdownGenerator(
+  }) {
-      inlineSyntaxList: inlineSyntaxes,
+    final components = prepare(context, onTapLink: onTapLink);
-      blockSyntaxList: blockSyntaxes,
+    return MarkdownWidget(
-      linesMargin: linesMargin,
+      data: data,
-      generators: generators,
+      selectable: selectable,
+      config: components.config,
+      markdownGenerator: components.generator,
+      shrinkWrap: shrinkWrap,
+      physics: physics,
+      padding: EdgeInsets.zero,
     );
   }
+
+  static MarkdownBlock buildBlock({
+    required BuildContext context,
+    required String data,
+    MarkdownLinkTapCallback? onTapLink,
+    bool selectable = true,
+  }) {
+    final components = prepare(context, onTapLink: onTapLink);
+    return MarkdownBlock(
+      data: data,
+      selectable: selectable,
+      config: components.config,
+      generator: components.generator,
+    );
   }
 
-class ConduitMarkdownConfig {
+  static ({MarkdownConfig config, MarkdownGenerator generator}) prepare(
-  static ConduitMarkdownTheme resolve(
+    BuildContext context, {
+    MarkdownLinkTapCallback? onTapLink,
+  }) {
+    final config = _buildConfig(context, onTapLink: onTapLink);
+    final generator = _buildGenerator(context);
+    return (config: config, generator: generator);
+  }
+
+  static MarkdownConfig _buildConfig(
     BuildContext context, {
     MarkdownLinkTapCallback? onTapLink,
   }) {
     final theme = context.conduitTheme;
-    final materialTheme = Theme.of(context);
+    final material = Theme.of(context);
-    final isDark = materialTheme.brightness == Brightness.dark;
+    final isDark = material.brightness == Brightness.dark;
 
     final baseBody = AppTypography.bodyMediumStyle.copyWith(
       color: theme.textPrimary,
@@ -65,11 +85,9 @@ class ConduitMarkdownConfig {
 
     final codeBackground = theme.surfaceContainer.withValues(alpha: 0.55);
     final borderColor = theme.cardBorder.withValues(alpha: 0.25);
-    final latex = const ConduitLatex();
+    final highlightTheme = _codeHighlightTheme(theme, isDark: isDark);
 
-    final markdownConfig =
+    return MarkdownConfig(
-        (isDark ? MarkdownConfig.darkConfig : MarkdownConfig.defaultConfig)
-            .copy(
       configs: [
         PConfig(textStyle: baseBody),
         H1Config(
@@ -96,9 +114,9 @@ class ConduitMarkdownConfig {
         H6Config(style: secondaryBody),
         LinkConfig(
           style: baseBody.copyWith(
-                    color: materialTheme.colorScheme.primary,
+            color: material.colorScheme.primary,
             decoration: TextDecoration.underline,
-                    decorationColor: materialTheme.colorScheme.primary,
+            decorationColor: material.colorScheme.primary,
           ),
           onTap: (url) => onTapLink?.call(url, url),
         ),
@@ -109,31 +127,40 @@ class ConduitMarkdownConfig {
           ),
         ),
         PreConfig(
-                  padding: const EdgeInsets.all(Spacing.sm),
+          textStyle: AppTypography.codeStyle.copyWith(color: theme.codeText),
-                  margin: const EdgeInsets.symmetric(vertical: Spacing.xs),
-                  decoration: BoxDecoration(
-                    color: codeBackground,
-                    borderRadius: BorderRadius.circular(AppBorderRadius.sm),
-                    border: Border.all(
-                      color: borderColor,
-                      width: BorderWidth.micro,
-                    ),
-                  ),
-                  textStyle: AppTypography.codeStyle.copyWith(
-                    color: theme.codeText,
-                  ),
           styleNotMatched: AppTypography.codeStyle.copyWith(
             color: theme.codeText,
           ),
-                  theme: _codeHighlightTheme(theme, isDark: isDark),
+          theme: highlightTheme,
-                  language: 'plaintext',
+          builder: (code, language) {
+            final normalizedLanguage = language.trim().isEmpty
+                ? 'plaintext'
+                : language.trim();
+            final highlight = HighlightView(
+              code,
+              language: normalizedLanguage == 'plaintext'
+                  ? null
+                  : normalizedLanguage,
+              theme: highlightTheme,
+              textStyle: AppTypography.codeStyle.copyWith(
+                color: theme.codeText,
+              ),
+              padding: EdgeInsets.zero,
+            );
+            return _buildCodeWrapper(
+              context: context,
+              child: highlight,
+              backgroundColor: codeBackground,
+              borderColor: borderColor,
+              language: normalizedLanguage,
+              rawCode: code,
+            );
+          },
         ),
         BlockquoteConfig(
-                  sideColor: materialTheme.colorScheme.primary.withValues(
+          sideColor: material.colorScheme.primary.withValues(alpha: 0.35),
-                    alpha: 0.35,
-                  ),
           textColor: theme.textSecondary,
-                  sideWith: BorderWidth.micro,
+          sideWith: BorderWidth.small,
           padding: const EdgeInsets.symmetric(
             horizontal: Spacing.md,
             vertical: Spacing.sm,
@@ -142,10 +169,7 @@ class ConduitMarkdownConfig {
         ),
         ListConfig(marginLeft: Spacing.lg, marginBottom: Spacing.xs),
         TableConfig(
-                  border: TableBorder.all(
+          border: TableBorder.all(color: borderColor, width: BorderWidth.micro),
-                    color: borderColor,
-                    width: BorderWidth.micro,
-                  ),
           headPadding: const EdgeInsets.symmetric(
             horizontal: Spacing.sm,
             vertical: Spacing.xs,
@@ -154,9 +178,7 @@ class ConduitMarkdownConfig {
             horizontal: Spacing.sm,
             vertical: Spacing.xs,
           ),
-                  headerStyle: secondaryBody.copyWith(
+          headerStyle: secondaryBody.copyWith(fontWeight: FontWeight.w600),
-                    fontWeight: FontWeight.w600,
-                  ),
           bodyStyle: secondaryBody,
           headerRowDecoration: BoxDecoration(
             color: theme.surfaceBackground.withValues(alpha: 0.35),
@@ -167,30 +189,25 @@ class ConduitMarkdownConfig {
         ),
         HrConfig(color: theme.dividerColor, height: BorderWidth.small),
         ImgConfig(
-                  builder: (url, _) {
+          builder: (url, attributes) {
-                    return Builder(
-                      builder: (context) {
             final uri = Uri.tryParse(url);
             if (uri == null) {
-                          return _buildImageError(
+              return _buildImageError(context, theme);
-                            context,
-                            context.conduitTheme,
-                          );
             }
             return _buildImage(context, uri);
           },
-                    );
-                  },
         ),
       ],
     );
+  }
 
-    return ConduitMarkdownTheme(
+  static MarkdownGenerator _buildGenerator(BuildContext context) {
-      config: markdownConfig,
+    final isDark = Theme.of(context).brightness == Brightness.dark;
-      inlineSyntaxes: [latex.syntax()],
+    final latex = ConduitLatex();
-      blockSyntaxes: const [],
+    return MarkdownGenerator(
+      inlineSyntaxList: latex.syntaxes(),
       generators: [latex.generator(isDark: isDark)],
-      linesMargin: const EdgeInsets.only(bottom: Spacing.sm),
+      linesMargin: const EdgeInsets.symmetric(vertical: Spacing.xs),
     );
   }
 
@@ -212,6 +229,67 @@ class ConduitMarkdownConfig {
     };
   }
 
+  static Widget _buildCodeWrapper({
+    required BuildContext context,
+    required Widget child,
+    required Color backgroundColor,
+    required Color borderColor,
+    required String language,
+    required String rawCode,
+  }) {
+    return LayoutBuilder(
+      builder: (context, constraints) {
+        final width = constraints.maxWidth.isFinite
+            ? constraints.maxWidth
+            : MediaQuery.sizeOf(context).width;
+
+        return Container(
+          margin: const EdgeInsets.symmetric(vertical: Spacing.xs),
+          decoration: BoxDecoration(
+            color: backgroundColor,
+            borderRadius: BorderRadius.circular(AppBorderRadius.sm),
+            border: Border.all(color: borderColor, width: BorderWidth.micro),
+          ),
+          child: ClipRect(
+            child: Column(
+              mainAxisSize: MainAxisSize.min,
+              crossAxisAlignment: CrossAxisAlignment.stretch,
+              children: [
+                CodeBlockHeader(
+                  language: language,
+                  onCopy: () async {
+                    await Clipboard.setData(ClipboardData(text: rawCode));
+                    if (!context.mounted) {
+                      return;
+                    }
+                    ScaffoldMessenger.of(context).hideCurrentSnackBar();
+                    ScaffoldMessenger.of(context).showSnackBar(
+                      const SnackBar(
+                        content: Text('Code copied to clipboard.'),
+                      ),
+                    );
+                  },
+                ),
+                SingleChildScrollView(
+                  scrollDirection: Axis.horizontal,
+                  child: IntrinsicWidth(
+                    child: ConstrainedBox(
+                      constraints: BoxConstraints(minWidth: width),
+                      child: Padding(
+                        padding: const EdgeInsets.all(Spacing.sm),
+                        child: child,
+                      ),
+                    ),
+                  ),
+                ),
+              ],
+            ),
+          ),
+        );
+      },
+    );
+  }
+
   static Widget buildMermaidBlock(BuildContext context, String code) {
     final conduitTheme = context.conduitTheme;
     final materialTheme = Theme.of(context);
@@ -299,6 +377,13 @@ class ConduitMarkdownConfig {
         ),
       ),
       errorWidget: (context, url, error) => _buildImageError(context, theme),
+      imageBuilder: (context, imageProvider) => Container(
+        margin: const EdgeInsets.symmetric(vertical: Spacing.sm),
+        decoration: BoxDecoration(
+          borderRadius: BorderRadius.circular(AppBorderRadius.md),
+          image: DecorationImage(image: imageProvider, fit: BoxFit.contain),
+        ),
+      ),
     );
   }
 
@@ -486,61 +571,47 @@ class _MermaidDiagramState extends State<MermaidDiagram> {
 
   String _buildHtml(String code, String script) {
     final theme = widget.brightness == Brightness.dark ? 'dark' : 'default';
-    final encoded = jsonEncode(code);
     final primary = _toHex(widget.tokens.brandTone60);
     final secondary = _toHex(widget.tokens.accentTeal60);
     final background = _toHex(widget.tokens.codeBackground);
     final onBackground = _toHex(widget.tokens.codeText);
-    final lineColor = _toHex(widget.tokens.codeAccent);
-    final errorColor = _toHex(widget.tokens.statusError60);
 
     return '''
 <!DOCTYPE html>
 <html>
 <head>
-    <meta charset="utf-8">
+<meta charset="utf-8" />
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
 <style>
-      html, body { margin: 0; padding: 0; background: transparent; }
+  body {
-      body { color: $onBackground; font-family: -apple-system, sans-serif; }
+    margin: 0;
-      #diagram { padding: 8px; overflow: auto; }
+    background-color: transparent;
-      svg { height: auto; display: block; }
+  }
+  #container {
+    width: 100%;
+    height: 100%;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    background-color: transparent;
+  }
 </style>
-    <script type="text/javascript">
-$script
-    </script>
 </head>
 <body>
-    <div id="diagram"></div>
+<div id="container">
-    <script type="text/javascript">
+  <div class="mermaid">$code</div>
-      const graphDefinition = $encoded;
+</div>
-      const themeConfig = {
+<script>$script</script>
-        startOnLoad: false,
+<script>
+  mermaid.initialize({
     theme: '$theme',
-        securityLevel: 'loose',
     themeVariables: {
       primaryColor: '$primary',
-          secondaryColor: '$secondary',
+      primaryTextColor: '$onBackground',
-          background: '$background',
+      primaryBorderColor: '$secondary',
-          textColor: '$onBackground',
+      background: '$background'
-          lineColor: '$lineColor'
+    },
-        }
+  });
-      };
+  mermaid.contentLoaded();
-
-      (async () => {
-        const target = document.getElementById('diagram');
-        try {
-          mermaid.initialize(themeConfig);
-          const { svg, bindFunctions } = await mermaid.render('graphDiv', graphDefinition);
-          target.innerHTML = svg;
-          if (typeof bindFunctions === 'function') {
-            bindFunctions(target);
-          }
-        } catch (error) {
-          target.innerHTML = '<pre style="color:$errorColor">' + String(error) + '</pre>';
-          console.error('Mermaid render failed', error);
-        }
-      })();
 </script>
 </body>
 </html>
@@ -548,11 +619,22 @@ $script
   }
 
   String _toHex(Color color) {
-    final value = color.toARGB32();
+    int channel(double value) {
-    return '#'
+      final scaled = (value * 255).round();
-            '${((value >> 16) & 0xFF).toRadixString(16).padLeft(2, '0')}'
+      if (scaled < 0) {
-            '${((value >> 8) & 0xFF).toRadixString(16).padLeft(2, '0')}'
+        return 0;
-            '${(value & 0xFF).toRadixString(16).padLeft(2, '0')}'
+      }
-        .toUpperCase();
+      if (scaled > 255) {
+        return 255;
+      }
+      return scaled;
+    }
+
+    final argb =
+        (channel(color.a) << 24) |
+        (channel(color.r) << 16) |
+        (channel(color.g) << 8) |
+        channel(color.b);
+    return '#${argb.toRadixString(16).padLeft(8, '0')}';
   }
 }

lib/shared/widgets/markdown/markdown_latex.dart

@@ -1,9 +1,8 @@
 import 'package:flutter/material.dart';
-import 'package:flutter_math_fork/flutter_math.dart';
 import 'package:markdown/markdown.dart' as m;
 import 'package:markdown_widget/markdown_widget.dart';
 
-import '../../theme/theme_extensions.dart';
+import 'latex_block_widget.dart';
 
 const String _latexTag = 'latex';
 
@@ -12,7 +11,10 @@ class ConduitLatex {
   const ConduitLatex();
 
   /// Returns the inline syntax used to identify LaTeX segments.
-  m.InlineSyntax syntax() => _LatexSyntax();
+  List<m.InlineSyntax> syntaxes() => [
+    _LatexDollarSyntax(),
+    _LatexEscapedSyntax(),
+  ];
 
   /// Returns the span generator that renders LaTeX expressions.
   SpanNodeGeneratorWithTag generator({required bool isDark}) {
@@ -30,27 +32,56 @@ class ConduitLatex {
   }
 }
 
-class _LatexSyntax extends m.InlineSyntax {
+class _LatexDollarSyntax extends m.InlineSyntax {
-  _LatexSyntax() : super(r'(\$\$[\s\S]+?\$\$)|(\$[^\n]+?\$)');
+  _LatexDollarSyntax()
+    : super(
+        r'(\$\$[\s\S]+?\$\$)|(\$[^\n]+?\$)',
+        startCharacter: r'$'.codeUnitAt(0),
+      );
 
   @override
   bool onMatch(m.InlineParser parser, Match match) {
-    final raw = match.input.substring(match.start, match.end);
+    return _handleMatch(parser, match.input.substring(match.start, match.end));
-    final element = m.Element.text(_latexTag, raw);
-    if (raw.startsWith(r'$$') && raw.endsWith(r'$$') && raw.length > 4) {
-      element.attributes['content'] = raw.substring(2, raw.length - 2);
-      element.attributes['isInline'] = 'false';
-    } else if (raw.startsWith(r'$') && raw.endsWith(r'$') && raw.length > 2) {
-      element.attributes['content'] = raw.substring(1, raw.length - 1);
-      element.attributes['isInline'] = 'true';
-    } else {
-      element.attributes['content'] = raw;
-      element.attributes['isInline'] = 'true';
   }
+}
+
+class _LatexEscapedSyntax extends m.InlineSyntax {
+  _LatexEscapedSyntax()
+    : super(
+        r'(\\\\\([\s\S]+?\\\\\))|(\\\\\[[\s\S]+?\\\\\])',
+        startCharacter: r'\'.codeUnitAt(0),
+      );
+
+  @override
+  bool onMatch(m.InlineParser parser, Match match) {
+    return _handleMatch(parser, match.input.substring(match.start, match.end));
+  }
+}
+
+bool _handleMatch(m.InlineParser parser, String raw) {
+  final element = m.Element.text(_latexTag, raw);
+  String content = raw;
+  var isInline = true;
+
+  if (raw.startsWith(r'$$') && raw.endsWith(r'$$') && raw.length > 4) {
+    content = raw.substring(2, raw.length - 2);
+    isInline = false;
+  } else if (raw.startsWith(r'$') && raw.endsWith(r'$') && raw.length > 2) {
+    content = raw.substring(1, raw.length - 1);
+    isInline = true;
+  } else if (raw.startsWith(r'\\(') && raw.endsWith(r'\\)') && raw.length > 4) {
+    content = raw.substring(2, raw.length - 2);
+    isInline = true;
+  } else if (raw.startsWith(r'\\[') && raw.endsWith(r'\\]') && raw.length > 4) {
+    content = raw.substring(2, raw.length - 2);
+    isInline = false;
+  }
+
+  element.attributes['content'] = content;
+  element.attributes['isInline'] = '$isInline';
   parser.addNode(element);
   return true;
 }
-}
 
 class _LatexNode extends SpanNode {
   _LatexNode({
@@ -79,23 +110,14 @@ class _LatexNode extends SpanNode {
       return TextSpan(text: rawText, style: baseStyle);
     }
 
-    final latexWidget = Math.tex(
+    return WidgetSpan(
-      content,
+      alignment: PlaceholderAlignment.middle,
-      mathStyle: MathStyle.text,
+      child: LatexBlockWidget(
-      textStyle: baseStyle,
+        content: content,
-      textScaleFactor: 1,
+        isInline: isInline,
-      onErrorFallback: (error) {
+        style: baseStyle,
-        return Text(rawText, style: baseStyle.copyWith(color: Colors.red));
+        isDark: isDark,
-      },
+      ),
     );
-
-    final widget = isInline
-        ? latexWidget
-        : Padding(
-            padding: const EdgeInsets.symmetric(vertical: Spacing.xs),
-            child: Center(child: latexWidget),
-          );
-
-    return WidgetSpan(alignment: PlaceholderAlignment.middle, child: widget);
   }
 }

lib/shared/widgets/markdown/streaming_markdown_widget.dart

@@ -1,5 +1,7 @@
 import 'package:flutter/material.dart';
 
+import 'package:markdown_widget/markdown_widget.dart';
+
 import '../../theme/theme_extensions.dart';
 import 'markdown_config.dart';
 import 'markdown_preprocessor.dart';
@@ -23,20 +25,23 @@ class StreamingMarkdownWidget extends StatelessWidget {
     }
 
     final normalized = ConduitMarkdownPreprocessor.normalize(content);
-    final markdownTheme = ConduitMarkdownConfig.resolve(
+    final mermaidRegex = RegExp(r'```mermaid\s*([\s\S]*?)```', multiLine: true);
+    final matches = mermaidRegex.allMatches(normalized).toList();
+    final renderComponents = ConduitMarkdown.prepare(
       context,
       onTapLink: onTapLink,
     );
-    final generator = markdownTheme.createGenerator();
-    final mermaidRegex = RegExp(r'```mermaid\s*([\s\S]*?)```', multiLine: true);
-    final matches = mermaidRegex.allMatches(normalized).toList();
 
-    List<Widget> buildMarkdownBlocks(String data) {
+    Widget buildMarkdown(String data) {
-      return generator.buildWidgets(data, config: markdownTheme.config);
+      return MarkdownBlock(
+        data: data,
+        selectable: false,
+        config: renderComponents.config,
+        generator: renderComponents.generator,
+      );
     }
 
     if (matches.isEmpty) {
-      final blocks = buildMarkdownBlocks(normalized);
       return SelectionArea(
         child: Theme(
           data: Theme.of(context).copyWith(
@@ -44,10 +49,7 @@ class StreamingMarkdownWidget extends StatelessWidget {
               cursorColor: context.conduitTheme.buttonPrimary,
             ),
           ),
-          child: Column(
+          child: buildMarkdown(normalized),
-            crossAxisAlignment: CrossAxisAlignment.stretch,
-            children: blocks,
-          ),
         ),
       );
     }
@@ -57,12 +59,12 @@ class StreamingMarkdownWidget extends StatelessWidget {
     for (final match in matches) {
       final before = normalized.substring(currentIndex, match.start);
       if (before.trim().isNotEmpty) {
-        children.addAll(buildMarkdownBlocks(before));
+        children.add(buildMarkdown(before));
       }
 
       final code = match.group(1)?.trim() ?? '';
       if (code.isNotEmpty) {
-        children.add(ConduitMarkdownConfig.buildMermaidBlock(context, code));
+        children.add(ConduitMarkdown.buildMermaidBlock(context, code));
       }
 
       currentIndex = match.end;
@@ -70,7 +72,7 @@ class StreamingMarkdownWidget extends StatelessWidget {
 
     final tail = normalized.substring(currentIndex);
     if (tail.trim().isNotEmpty) {
-      children.addAll(buildMarkdownBlocks(tail));
+      children.add(buildMarkdown(tail));
     }
 
     return SelectionArea(