This commit is contained in:
Julian Freeman
2026-03-11 13:21:34 -04:00
parent dbbf62b793
commit d50a9a1b86
2 changed files with 53 additions and 24 deletions

View File

@@ -26,31 +26,33 @@ function doPost(e) {
var headers = [];
// 2. If sheet is brand new (no rows), create headers first
// 2. Determine headers
if (sheet.getLastRow() === 0) {
// Extract keys from data, excluding 'sheetName' which is a control parameter
// Use provided headers if available
if (data.headers && Array.isArray(data.headers)) {
headers = data.headers;
} else {
// Fallback: Use keys but exclude internal control keys
headers = Object.keys(data).filter(function(key) {
return key !== "sheetName";
return !["sheetName", "headers"].includes(key);
});
}
sheet.appendRow(headers);
} else {
// 3. If sheet already exists, get existing headers from the first row
// Get existing headers from the first row
var lastCol = sheet.getLastColumn();
if (lastCol > 0) {
headers = sheet.getRange(1, 1, 1, lastCol).getValues()[0];
} else {
// Fallback for edge cases where lastRow > 0 but lastCol is 0
headers = Object.keys(data).filter(function(key) {
return key !== "sheetName";
});
sheet.appendRow(headers);
}
}
// 4. Map data to the determined headers
// 3. Map data to headers
var row = headers.map(function(header) {
// data[header] might be undefined for some rows if schema changes
return data[header] !== undefined ? data[header] : "N/A";
var val = data[header];
if (val === undefined || val === null) return "N/A";
// If value is an object or array, stringify it to avoid [Ljava.lang.Object]
if (typeof val === 'object') return JSON.stringify(val);
return val;
});
sheet.appendRow(row);

View File

@@ -157,29 +157,56 @@ async function handleExtraction(type, presetObj) {
resultsArea.textContent = '正在发送至 Gemini...';
const systemPrompt = `你是一个专业的采购助手。
请从以下文本中提取产品数据。
将技术术语翻译成中文。
请从以下网页文本中提取产品数据。
仅返回有效的 JSON 格式。
如果可能,将本地货币转换为美元,或者保留原始符号。
URL: ${pageData.url}
Title: ${pageData.title}`;
const userPrompt = `提取以下字段:${presetObj.fields}仅返回 JSON 格式。`;
const userPrompt = `提取以下字段:${presetObj.fields}
【重要约束】:
1. 必须使用我提供的字段名称作为 JSON 的键名Key不要翻译成英文。
2. 将提取到的具体内容翻译成中文。
3. 仅返回 JSON 格式,不要包含任何多余的解释文字。`;
const geminiResult = await callGemini(geminiApiKey, selectedModel, systemPrompt, pageData.text, userPrompt);
const cleanedJson = parseGeminiJson(geminiResult);
let cleanedJson = parseGeminiJson(geminiResult);
// 核心修复:处理 AI 返回数组的情况
if (Array.isArray(cleanedJson)) {
cleanedJson = cleanedJson[0] || {};
}
resultsArea.textContent = JSON.stringify(cleanedJson, null, 2);
if (googleScriptUrl) {
updateStatus('保存至表格...', 'bg-purple-500 text-white');
await sendToGoogleScript(googleScriptUrl, {
...cleanedJson,
preset_type: presetObj.name || 'custom',
// 1. 获取用户预设的业务表头
const businessHeaders = presetObj.fields.split(',').map(f => f.trim()).filter(f => f !== "");
// 2. 定义元数据表头(可选,方便您追踪来源)
const metaHeaders = ["来源链接", "提取时间"];
// 3. 合并总表头顺序
const allHeaders = [...businessHeaders, ...metaHeaders];
// 4. 构建发送给后端的最终数据
const payload = {
sheetName: presetObj.sheetName || '',
source_url: pageData.url,
timestamp: new Date().toISOString()
headers: allHeaders
};
// 将提取的数据填入 payload
businessHeaders.forEach(h => {
payload[h] = cleanedJson[h] || "N/A";
});
// 填入元数据
payload["来源链接"] = pageData.url;
payload["提取时间"] = new Date().toLocaleString();
await sendToGoogleScript(googleScriptUrl, payload);
updateStatus('成功', 'bg-green-500 text-white');
} else {
updateStatus('完成 (未配置脚本链接)', 'bg-yellow-500 text-white');