// ★ 先に Google Cloud で Gemini APIキーを取得し、以下に設定してください
const API_KEY = "YOUR_API_KEY";
const MODEL = "gemini-1.5-flash"; // 軽量モデルでOK

// カスタムメニューを追加
function onOpen() {
  SpreadsheetApp.getUi()
    .createMenu("AIチェック")
    .addItem("請求書・見積書をチェック", "checkInvoices")
    .addToUi();
}

function checkInvoices() {
  const sheet = SpreadsheetApp.getActiveSpreadsheet().getActiveSheet();
  const lastRow = sheet.getLastRow();
  
  // データは 2 行目以降を対象にする
  let records = [];
  for (let row = 2; row <= lastRow; row++) {
    records[row-2] = {
      管理番号: sheet.getRange(row, 1).getValue(),
      顧客名: sheet.getRange(row, 2).getValue(),
      請求日: sheet.getRange(row, 3).getValue(),
      支払期日: sheet.getRange(row, 4).getValue(),
      品目: sheet.getRange(row, 5).getValue(),
      単価: sheet.getRange(row, 6).getValue(),
      数量: sheet.getRange(row, 7).getValue(),
      金額: sheet.getRange(row, 8).getValue(),
      消費税: sheet.getRange(row, 9).getValue(),
      合計金額: sheet.getRange(row, 10).getValue(),
    }; 
  }

  // プロンプト: 過去行との比較も含めたチェック依頼
  const prompt = `
以下の請求書データについて、次に記載するルールに対して、前行までの履歴を踏まえて不自然な点をチェックしてください。
各行についてJSON形式で「行番号」「チェック結果」を返してください。
ルール:
- 消費税に関して、消費税率は10%とし、金額×0.1(端数がある場合は切り捨て)と等しいかどうか、正しければ「問題なし」です
- 合計金額に関して、金額と消費税の合計が合計金額になっているかどうか、正しければ「問題なし」です
- 支払期日に関して、支払期日が入力されているか、入力されていれば「問題なし」です
- 単価に関して、単価が同一顧客名、同一品目の過去の履歴があれば、その単価と同じかどうか、過去の履歴が無い場合は「問題なし」です
4つのチェックすべて「問題なし」の場合は「OK」と答えてください。問題点がある場合は、問題点のみ具体的に指摘してください。

データ:
${JSON.stringify(records)}
    `;

  // Geminiにアクセス
  const checkResults = callGeminiAPI(prompt);
  if (checkResults === false) {
    Logger.log("エラー: 結果を取得できませんでした");
    return;
  }

  // シートに反映
  checkResults.forEach(r => {
    let rowIndex = r.行番号 +1;
    let resultText = r.チェック結果;
    let resultCell = sheet.getRange(rowIndex, 11); // 「結果」列（K列=11列目）
    resultCell.setValue(resultText);

    if (resultText !== "OK") {
      sheet.getRange(rowIndex, 1, 1, 11).setBackground("#FFCCCC"); // 行全体を赤色に
    } else {
      sheet.getRange(rowIndex, 1, 1, 11).setBackground(null); // 色リセット
    }
  });
}

// Gemini API 呼び出し
function callGeminiAPI(prompt) {
  const url = `https://generativelanguage.googleapis.com/v1beta/models/${MODEL}:generateContent?key=${API_KEY}`;
  
  const payload = {
    contents: [{ parts: [{ text: prompt }] }]
  };

  const options = {
    method: "post",
    contentType: "application/json",
    payload: JSON.stringify(payload),
    muteHttpExceptions: true,
  };

  try {
    const response = UrlFetchApp.fetch(url, options);
    const result = JSON.parse(response.getContentText());
    // Geminiの返答からJSON部分を抽出
    let text = result.candidates?.[0]?.content?.parts?.[0]?.text || "";
    // 不要な部分を空文字列に置き換える
    text = text.replace("```json\n", "").replace("\n```\n", "");
    let checkResults;
    try {
      checkResults = JSON.parse(text);
      return checkResults;
    } catch (e) {
      Logger.log("JSON parse error: " + e);
      return false;
    }
  } catch (e) {
    Logger.log("エラー: " + e.message);
    return false;
  }
}