feat: 前端解析 docx/xlsx/pptx,内联文本给模型

This commit is contained in:
zimk
2026-03-16 23:52:26 +08:00
parent ea967fc220
commit c6f2c99a02
2 changed files with 86 additions and 1 deletions

View File

@@ -322,6 +322,50 @@ async function addPendingFiles(files) {
continue;
}
const kind = getAttachmentKind(file);
let parsedText = null;
// Office 文件:前端解析提取文本
if (kind === 'docx') {
try {
const arrayBuffer = await readFileAsArrayBuffer(file);
const result = await mammoth.extractRawText({ arrayBuffer });
parsedText = result.value || '';
} catch (err) {
console.warn('mammoth 解析失败', err);
}
} else if (kind === 'xlsx') {
try {
const arrayBuffer = await readFileAsArrayBuffer(file);
const workbook = XLSX.read(arrayBuffer, { type: 'array' });
const lines = [];
workbook.SheetNames.forEach((sheetName) => {
const sheet = workbook.Sheets[sheetName];
const csv = XLSX.utils.sheet_to_csv(sheet);
if (csv.trim()) lines.push(`## Sheet: ${sheetName}\n${csv}`);
});
parsedText = lines.join('\n\n');
} catch (err) {
console.warn('SheetJS 解析失败', err);
}
} else if (kind === 'pptx') {
try {
const arrayBuffer = await readFileAsArrayBuffer(file);
// pptx 本质是 zip用 XLSX 的 zip 工具提取文本节点
const zip = XLSX.read(arrayBuffer, { type: 'array' });
const textParts = [];
Object.keys(zip.Strings || {}).forEach((k) => {
const v = zip.Strings[k];
if (typeof v === 'string' && v.trim()) textParts.push(v.trim());
});
// 更可靠的方式:直接用 JSZip-like 解包XLSX 内置 CFB/ZIP
// 若 zip.Strings 为空则给提示
parsedText = textParts.length ? textParts.join('\n') : 'PPT 文本提取失败,内容可能为空)';
} catch (err) {
console.warn('pptx 解析失败', err);
}
}
const dataUrl = await readFileAsDataURL(file);
const base64 = dataUrl.split(',')[1] || '';
state.pendingAttachments.push({
@@ -329,15 +373,25 @@ async function addPendingFiles(files) {
name: file.name,
type: file.type || 'application/octet-stream',
size: file.size,
kind: getAttachmentKind(file),
kind,
dataUrl,
base64,
parsedText,
});
}
renderAttachments();
}
function readFileAsArrayBuffer(file) {
return new Promise((resolve, reject) => {
const reader = new FileReader();
reader.onload = () => resolve(reader.result);
reader.onerror = () => reject(new Error(`读取文件失败:${file.name}`));
reader.readAsArrayBuffer(file);
});
}
function renderAttachments() {
attachmentListEl.innerHTML = '';
if (!state.pendingAttachments.length) return;
@@ -878,6 +932,35 @@ function loadConversations() {
function getAttachmentKind(file) {
if (file.type?.startsWith('image/')) return 'image';
const textTypes = [
'text/',
'application/json',
'application/xml',
'application/javascript',
'application/typescript',
'application/x-yaml',
'application/x-sh',
'application/x-python',
];
if (textTypes.some((t) => file.type?.startsWith(t))) return 'text';
const textExts = /\.(txt|md|markdown|csv|json|xml|yaml|yml|toml|ini|cfg|conf|log|sh|bash|zsh|py|js|ts|jsx|tsx|java|c|cpp|h|hpp|cs|go|rs|rb|php|swift|kt|scala|r|sql|html|htm|css|scss|sass|less|vue|svelte|astro|diff|patch)$/i;
if (textExts.test(file.name)) return 'text';
if (file.type === 'application/pdf') return 'pdf';
const docxTypes = [
'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
'application/msword',
];
const xlsxTypes = [
'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
'application/vnd.ms-excel',
];
const pptxTypes = [
'application/vnd.openxmlformats-officedocument.presentationml.presentation',
'application/vnd.ms-powerpoint',
];
if (docxTypes.includes(file.type) || /\.(docx|doc)$/i.test(file.name)) return 'docx';
if (xlsxTypes.includes(file.type) || /\.(xlsx|xls)$/i.test(file.name)) return 'xlsx';
if (pptxTypes.includes(file.type) || /\.(pptx|ppt)$/i.test(file.name)) return 'pptx';
return 'file';
}

View File

@@ -89,6 +89,8 @@
</div>
<script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
<script src="https://cdn.jsdelivr.net/npm/mammoth@1.8.0/mammoth.browser.min.js"></script>
<script src="https://cdn.jsdelivr.net/npm/xlsx@0.18.5/dist/xlsx.full.min.js"></script>
<script src="/config.js"></script>
<script src="/app.js" defer></script>
</body>