diff --git a/src/renderer/src/pages/home/Chat.tsx b/src/renderer/src/pages/home/Chat.tsx index e5e98df4..6f68e6d4 100644 --- a/src/renderer/src/pages/home/Chat.tsx +++ b/src/renderer/src/pages/home/Chat.tsx @@ -52,8 +52,12 @@ const Container = styled.div` height: 100%; flex: 1; justify-content: space-between; + background-color: var(--color-background); &.bubble { background-color: var(--chat-background); + #messages { + background-color: var(--chat-background); + } .system-prompt { background-color: var(--chat-background-assistant); } diff --git a/src/renderer/src/pages/home/Messages/Messages.tsx b/src/renderer/src/pages/home/Messages/Messages.tsx index ad8c764d..fefdbe06 100644 --- a/src/renderer/src/pages/home/Messages/Messages.tsx +++ b/src/renderer/src/pages/home/Messages/Messages.tsx @@ -219,6 +219,7 @@ const Container = styled(Scrollbar)` padding: 10px 0; padding-bottom: 20px; overflow-x: hidden; + background-color: var(--color-background); ` export default Messages diff --git a/src/renderer/src/utils/index.ts b/src/renderer/src/utils/index.ts index 9916ab26..b019b771 100644 --- a/src/renderer/src/utils/index.ts +++ b/src/renderer/src/utils/index.ts @@ -264,7 +264,7 @@ export const captureScrollableDiv = async (divRef: React.RefObject { + // Ensure all images in cloned document are loaded + const images = clonedDoc.getElementsByTagName('img') + return Promise.all( + Array.from(images).map((img) => { + if (img.complete) { + return Promise.resolve() + } + return new Promise((resolve) => { + img.onload = resolve + img.onerror = resolve + }) + }) + ) + } }) - // 恢复原始样式 + // Restore original styles div.style.height = originalStyle.height div.style.maxHeight = originalStyle.maxHeight div.style.overflow = originalStyle.overflow @@ -294,7 +313,7 @@ export const captureScrollableDiv = async (divRef: React.RefObject { div.scrollTop = originalScrollTop }, 0)