diff --git a/web/src/components/MemoContent/MermaidBlock.tsx b/web/src/components/MemoContent/MermaidBlock.tsx index 064d4608..ac5f377c 100644 --- a/web/src/components/MemoContent/MermaidBlock.tsx +++ b/web/src/components/MemoContent/MermaidBlock.tsx @@ -6,16 +6,14 @@ interface Props { } const MermaidBlock: React.FC = ({ content }: Props) => { + const { mode: colorMode } = useColorScheme(); const mermaidDockBlock = useRef(null); - const { mode } = useColorScheme(); - - const mermaidTheme = mode == "dark" ? "dark" : "default"; useEffect(() => { // Dynamically import mermaid to ensure compatibility with Vite const initializeMermaid = async () => { const mermaid = (await import("mermaid")).default; - mermaid.initialize({ startOnLoad: false, theme: mermaidTheme }); + mermaid.initialize({ startOnLoad: false, theme: colorMode == "dark" ? "dark" : "default" }); if (mermaidDockBlock.current) { mermaid.run({ nodes: [mermaidDockBlock.current],