diff --git a/miniprogram/pages/batch/index.js b/miniprogram/pages/batch/index.js index 79bba95..3c4d090 100644 --- a/miniprogram/pages/batch/index.js +++ b/miniprogram/pages/batch/index.js @@ -79,5 +79,83 @@ Page({ showCancel: false, confirmText: '关闭' }) + }, + + generateCSV() { + const items = this.data.items + if (!items.length) return '' + + const headers = ['文本', '判定结果', '置信度', '垃圾概率', '正常概率', '风险关键词'] + const rows = items.map((item) => { + const prediction = item.prediction === 'spam' ? '垃圾信息' : '正常信息' + const confidence = item.confidence_text || '0%' + const spamProb = this.formatPercent(item.spam_probability, 4) + const hamProb = this.formatPercent(item.ham_probability, 4) + const tokens = (item.reason_tokens || []).map((t) => t.token || t).join('; ') + // CSV 转义:文本中的逗号和换行需要处理 + const text = (item.text || '').replace(/"/g, '""') + const tokensEscaped = tokens.replace(/"/g, '""') + return `"${text}","${prediction}","${confidence}","${spamProb}","${hamProb}","${tokensEscaped}"` + }) + + return [headers.join(','), ...rows].join('\n') + }, + + exportCSV() { + const items = this.data.items + if (!items.length) { + wx.showToast({ title: '暂无识别结果可导出', icon: 'none' }) + return + } + + const csvContent = this.generateCSV() + const timestamp = new Date().toISOString().slice(0, 19).replace(/[T:]/g, '-') + const filename = `batch_detect_${timestamp}.csv` + + // 写入临时文件 + const fs = wx.getFileSystemManager() + const tempPath = `${wx.env.USER_DATA_PATH}/${filename}` + + try { + fs.writeFileSync(tempPath, csvContent, 'utf8') + wx.showModal({ + title: '导出成功', + content: `CSV文件已生成,是否打开查看?\n文件名:${filename}`, + confirmText: '打开', + cancelText: '关闭', + success: (res) => { + if (res.confirm) { + wx.openDocument({ + filePath: tempPath, + fileType: 'csv', + showMenu: true, + fail: (err) => { + console.error('打开文件失败', err) + wx.showToast({ title: '打开失败,请检查文件管理器', icon: 'none' }) + } + }) + } + } + }) + } catch (err) { + console.error('写入文件失败', err) + wx.showToast({ title: '导出失败', icon: 'none' }) + } + }, + + copyCSVToClipboard() { + const items = this.data.items + if (!items.length) { + wx.showToast({ title: '暂无识别结果可复制', icon: 'none' }) + return + } + + const csvContent = this.generateCSV() + wx.setClipboardData({ + data: csvContent, + success: () => { + wx.showToast({ title: 'CSV内容已复制到剪贴板', icon: 'success' }) + } + }) } }) diff --git a/miniprogram/pages/batch/index.wxml b/miniprogram/pages/batch/index.wxml index 9475202..ea6d8e0 100644 --- a/miniprogram/pages/batch/index.wxml +++ b/miniprogram/pages/batch/index.wxml @@ -56,6 +56,10 @@ 明细结果 + + + + {{item.text}}