mirror of
https://github.com/open-webui/open-webui.git
synced 2025-12-18 07:15:20 +00:00
refac/enh: include pdf export in chat item
This commit is contained in:
parent
3ec1efb6e0
commit
d041d58bb6
1 changed files with 191 additions and 1 deletions
|
|
@ -1,7 +1,7 @@
|
||||||
<script lang="ts">
|
<script lang="ts">
|
||||||
import { DropdownMenu } from 'bits-ui';
|
import { DropdownMenu } from 'bits-ui';
|
||||||
import { flyAndScale } from '$lib/utils/transitions';
|
import { flyAndScale } from '$lib/utils/transitions';
|
||||||
import { getContext, createEventDispatcher } from 'svelte';
|
import { getContext, createEventDispatcher, tick } from 'svelte';
|
||||||
|
|
||||||
import fileSaver from 'file-saver';
|
import fileSaver from 'file-saver';
|
||||||
const { saveAs } = fileSaver;
|
const { saveAs } = fileSaver;
|
||||||
|
|
@ -31,6 +31,7 @@
|
||||||
import { downloadChatAsPDF } from '$lib/apis/utils';
|
import { downloadChatAsPDF } from '$lib/apis/utils';
|
||||||
import Download from '$lib/components/icons/Download.svelte';
|
import Download from '$lib/components/icons/Download.svelte';
|
||||||
import Folder from '$lib/components/icons/Folder.svelte';
|
import Folder from '$lib/components/icons/Folder.svelte';
|
||||||
|
import Messages from '$lib/components/chat/Messages.svelte';
|
||||||
|
|
||||||
const i18n = getContext('i18n');
|
const i18n = getContext('i18n');
|
||||||
|
|
||||||
|
|
@ -48,6 +49,9 @@
|
||||||
let show = false;
|
let show = false;
|
||||||
let pinned = false;
|
let pinned = false;
|
||||||
|
|
||||||
|
let chat = null;
|
||||||
|
let showFullMessages = false;
|
||||||
|
|
||||||
const pinHandler = async () => {
|
const pinHandler = async () => {
|
||||||
await toggleChatPinnedStatusById(localStorage.token, chatId);
|
await toggleChatPinnedStatusById(localStorage.token, chatId);
|
||||||
dispatch('change');
|
dispatch('change');
|
||||||
|
|
@ -81,6 +85,162 @@
|
||||||
saveAs(blob, `chat-${chat.chat.title}.txt`);
|
saveAs(blob, `chat-${chat.chat.title}.txt`);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
const downloadPdf = async () => {
|
||||||
|
chat = await getChatById(localStorage.token, chatId);
|
||||||
|
if (!chat) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ($settings?.stylizedPdfExport ?? true) {
|
||||||
|
showFullMessages = true;
|
||||||
|
await tick();
|
||||||
|
|
||||||
|
const containerElement = document.getElementById('full-messages-container');
|
||||||
|
if (containerElement) {
|
||||||
|
try {
|
||||||
|
const isDarkMode = document.documentElement.classList.contains('dark');
|
||||||
|
const virtualWidth = 800; // px, fixed width for cloned element
|
||||||
|
|
||||||
|
// Clone and style
|
||||||
|
const clonedElement = containerElement.cloneNode(true);
|
||||||
|
clonedElement.classList.add('text-black');
|
||||||
|
clonedElement.classList.add('dark:text-white');
|
||||||
|
clonedElement.style.width = `${virtualWidth}px`;
|
||||||
|
clonedElement.style.position = 'absolute';
|
||||||
|
clonedElement.style.left = '-9999px';
|
||||||
|
clonedElement.style.height = 'auto';
|
||||||
|
document.body.appendChild(clonedElement);
|
||||||
|
|
||||||
|
// Wait for DOM update/layout
|
||||||
|
await new Promise((r) => setTimeout(r, 100));
|
||||||
|
|
||||||
|
// Render entire content once
|
||||||
|
const canvas = await html2canvas(clonedElement, {
|
||||||
|
backgroundColor: isDarkMode ? '#000' : '#fff',
|
||||||
|
useCORS: true,
|
||||||
|
scale: 2, // increase resolution
|
||||||
|
width: virtualWidth
|
||||||
|
});
|
||||||
|
|
||||||
|
document.body.removeChild(clonedElement);
|
||||||
|
|
||||||
|
const pdf = new jsPDF('p', 'mm', 'a4');
|
||||||
|
const pageWidthMM = 210;
|
||||||
|
const pageHeightMM = 297;
|
||||||
|
|
||||||
|
// Convert page height in mm to px on canvas scale for cropping
|
||||||
|
// Get canvas DPI scale:
|
||||||
|
const pxPerMM = canvas.width / virtualWidth; // width in px / width in px?
|
||||||
|
// Since 1 page width is 210 mm, but canvas width is 800 px at scale 2
|
||||||
|
// Assume 1 mm = px / (pageWidthMM scaled)
|
||||||
|
// Actually better: Calculate scale factor from px/mm:
|
||||||
|
// virtualWidth px corresponds directly to 210mm in PDF, so pxPerMM:
|
||||||
|
const pxPerPDFMM = canvas.width / pageWidthMM; // canvas px per PDF mm
|
||||||
|
|
||||||
|
// Height in px for one page slice:
|
||||||
|
const pagePixelHeight = Math.floor(pxPerPDFMM * pageHeightMM);
|
||||||
|
|
||||||
|
let offsetY = 0;
|
||||||
|
let page = 0;
|
||||||
|
|
||||||
|
while (offsetY < canvas.height) {
|
||||||
|
// Height of slice
|
||||||
|
const sliceHeight = Math.min(pagePixelHeight, canvas.height - offsetY);
|
||||||
|
|
||||||
|
// Create temp canvas for slice
|
||||||
|
const pageCanvas = document.createElement('canvas');
|
||||||
|
pageCanvas.width = canvas.width;
|
||||||
|
pageCanvas.height = sliceHeight;
|
||||||
|
|
||||||
|
const ctx = pageCanvas.getContext('2d');
|
||||||
|
|
||||||
|
// Draw the slice of original canvas onto pageCanvas
|
||||||
|
ctx.drawImage(
|
||||||
|
canvas,
|
||||||
|
0,
|
||||||
|
offsetY,
|
||||||
|
canvas.width,
|
||||||
|
sliceHeight,
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
canvas.width,
|
||||||
|
sliceHeight
|
||||||
|
);
|
||||||
|
|
||||||
|
const imgData = pageCanvas.toDataURL('image/jpeg', 0.7);
|
||||||
|
|
||||||
|
// Calculate image height in PDF units keeping aspect ratio
|
||||||
|
const imgHeightMM = (sliceHeight * pageWidthMM) / canvas.width;
|
||||||
|
|
||||||
|
if (page > 0) pdf.addPage();
|
||||||
|
|
||||||
|
if (isDarkMode) {
|
||||||
|
pdf.setFillColor(0, 0, 0);
|
||||||
|
pdf.rect(0, 0, pageWidthMM, pageHeightMM, 'F'); // black bg
|
||||||
|
}
|
||||||
|
|
||||||
|
pdf.addImage(imgData, 'JPEG', 0, 0, pageWidthMM, imgHeightMM);
|
||||||
|
|
||||||
|
offsetY += sliceHeight;
|
||||||
|
page++;
|
||||||
|
}
|
||||||
|
|
||||||
|
pdf.save(`chat-${chat.chat.title}.pdf`);
|
||||||
|
|
||||||
|
showFullMessages = false;
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Error generating PDF', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
console.log('Downloading PDF');
|
||||||
|
|
||||||
|
const chatText = await getChatAsText(chat);
|
||||||
|
|
||||||
|
const doc = new jsPDF();
|
||||||
|
|
||||||
|
// Margins
|
||||||
|
const left = 15;
|
||||||
|
const top = 20;
|
||||||
|
const right = 15;
|
||||||
|
const bottom = 20;
|
||||||
|
|
||||||
|
const pageWidth = doc.internal.pageSize.getWidth();
|
||||||
|
const pageHeight = doc.internal.pageSize.getHeight();
|
||||||
|
const usableWidth = pageWidth - left - right;
|
||||||
|
const usableHeight = pageHeight - top - bottom;
|
||||||
|
|
||||||
|
// Font size and line height
|
||||||
|
const fontSize = 8;
|
||||||
|
doc.setFontSize(fontSize);
|
||||||
|
const lineHeight = fontSize * 1; // adjust if needed
|
||||||
|
|
||||||
|
// Split the markdown into lines (handles \n)
|
||||||
|
const paragraphs = chatText.split('\n');
|
||||||
|
|
||||||
|
let y = top;
|
||||||
|
|
||||||
|
for (let paragraph of paragraphs) {
|
||||||
|
// Wrap each paragraph to fit the width
|
||||||
|
const lines = doc.splitTextToSize(paragraph, usableWidth);
|
||||||
|
|
||||||
|
for (let line of lines) {
|
||||||
|
// If the line would overflow the bottom, add a new page
|
||||||
|
if (y + lineHeight > pageHeight - bottom) {
|
||||||
|
doc.addPage();
|
||||||
|
y = top;
|
||||||
|
}
|
||||||
|
doc.text(line, left, y);
|
||||||
|
y += lineHeight * 0.5;
|
||||||
|
}
|
||||||
|
// Add empty line at paragraph breaks
|
||||||
|
y += lineHeight * 0.1;
|
||||||
|
}
|
||||||
|
|
||||||
|
doc.save(`chat-${chat.chat.title}.pdf`);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
const downloadJSONExport = async () => {
|
const downloadJSONExport = async () => {
|
||||||
const chat = await getChatById(localStorage.token, chatId);
|
const chat = await getChatById(localStorage.token, chatId);
|
||||||
|
|
||||||
|
|
@ -97,6 +257,27 @@
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
|
{#if chat && showFullMessages}
|
||||||
|
<div class="hidden w-full h-full flex-col">
|
||||||
|
<div id="full-messages-container">
|
||||||
|
<Messages
|
||||||
|
className="h-full flex pt-4 pb-8 w-full"
|
||||||
|
chatId={`chat-preview-${chat?.id ?? ''}`}
|
||||||
|
user={$user}
|
||||||
|
readOnly={true}
|
||||||
|
history={chat.chat.history}
|
||||||
|
messages={chat.chat.messages}
|
||||||
|
autoScroll={true}
|
||||||
|
sendMessage={() => {}}
|
||||||
|
continueResponse={() => {}}
|
||||||
|
regenerateResponse={() => {}}
|
||||||
|
messagesCount={null}
|
||||||
|
editCodeBlock={false}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
{/if}
|
||||||
|
|
||||||
<Dropdown
|
<Dropdown
|
||||||
bind:show
|
bind:show
|
||||||
on:change={(e) => {
|
on:change={(e) => {
|
||||||
|
|
@ -161,6 +342,15 @@
|
||||||
>
|
>
|
||||||
<div class="flex items-center line-clamp-1">{$i18n.t('Plain text (.txt)')}</div>
|
<div class="flex items-center line-clamp-1">{$i18n.t('Plain text (.txt)')}</div>
|
||||||
</DropdownMenu.Item>
|
</DropdownMenu.Item>
|
||||||
|
|
||||||
|
<DropdownMenu.Item
|
||||||
|
class="flex gap-2 items-center px-3 py-1.5 text-sm cursor-pointer hover:bg-gray-50 dark:hover:bg-gray-800 rounded-xl select-none w-full"
|
||||||
|
on:click={() => {
|
||||||
|
downloadPdf();
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<div class="flex items-center line-clamp-1">{$i18n.t('PDF document (.pdf)')}</div>
|
||||||
|
</DropdownMenu.Item>
|
||||||
</DropdownMenu.SubContent>
|
</DropdownMenu.SubContent>
|
||||||
</DropdownMenu.Sub>
|
</DropdownMenu.Sub>
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue