ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-24 02:05:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refactor: optimize PDF export logic

Browse source
This commit is contained in:
Shirasawa 2025-12-03 09:15:06 +00:00
parent d6048bcfc3
commit 2cde5022b4
1 changed files with 14 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
src/lib/utils/pdf.ts
View file

@ -44,9 +44,9 @@ interface ChatPdfOptions {
/** Whether to use stylized PDF export (default: true) */
stylizedPdfExport?: boolean;
/** Optional callback before rendering (for showing full messages) */
onBeforeRender?: () => Promise<void>;
onBeforeRender?: () => Promise<void> | void;
/** Optional callback after rendering (for hiding full messages) */
onAfterRender?: () => void;
onAfterRender?: () => Promise<void> | void;
}
// ==================== Shared Constants ====================
@ -384,10 +384,10 @@ export const downloadNotePdf = async (note: NoteData): Promise<void> => {
* @throws Error if PDF generation fails or if chatText is missing in plain text mode
*/
export const downloadChatPdf = async (options: ChatPdfOptions): Promise<void> => {
console.log('Downloading PDF', options);
if (options.stylizedPdfExport ?? true) {
if (options.onBeforeRender) {
await options.onBeforeRender();
}
await options.onBeforeRender?.();
const containerElement = document.getElementById(
options.containerElementId || 'full-messages-container'
@ -415,25 +415,18 @@ export const downloadChatPdf = async (options: ChatPdfOptions): Promise<void> =>
canvasToPdfWithSlicing(pdf, canvas, virtualWidth, A4_PAGE_WIDTH_MM, A4_PAGE_HEIGHT_MM);
pdf.save(`chat-${options.title}.pdf`);
if (options.onAfterRender) {
options.onAfterRender();
}
} catch (error) {
console.error('Error generating PDF', error);
if (options.onAfterRender) {
options.onAfterRender();
}
}
}
} else {
console.log('Downloading PDF');
if (!options.chatText) {
console.error('chatText is required for plain text PDF export');
return;
}
await exportPlainTextToPdf(options.chatText, `chat-${options.title}.pdf`);
await options.onAfterRender?.();
return;
}
if (!options.chatText) {
console.error('chatText is required for plain text PDF export');
return;
}
await exportPlainTextToPdf(options.chatText, `chat-${options.title}.pdf`);
};