スプレッドシートとGASでシフト表を自動割り当て！ChatGPTを活用した業務効率化

const OPENAI_API_KEY = PropertiesService.getScriptProperties().getProperty('OPENAI_API_KEY');

function assignShiftsWithChatGPT() {
  const sheet = SpreadsheetApp.getActiveSpreadsheet().getSheetByName("シフト表");
  
  // A6以降の従業員名を取得
  const members = sheet.getRange("A6:A" + sheet.getLastRow()).getValues()
    .flat()
    .filter(name => name);

  // B4:AF4の範囲から日付リストを取得
  const dates = sheet.getRange("B4:AF4").getValues()[0].filter(date => date);

  // プロンプトを生成
  const prompt = createShiftPrompt(members, dates);

  // ChatGPT APIにリクエストを送信
  const chatGPTResponse = callChatGPTAPI(prompt);

  // ChatGPTからの応答をシフト表に反映
  if (chatGPTResponse) {
    applyShiftAssignments(sheet, chatGPTResponse, 2, 6); // 日付の開始列（B列=2）と従業員の開始行（6行目）
  }
}

function createShiftPrompt(members, dates) {
  return `以下の従業員リストと日付に基づいて、シフト表を厳密にJSON形式でのみ出力してください。以下の指示を厳密に守り、余分な説明を含めないでください。

` +
         `条件:
- 各日には必ず「早番」が3名、「遅番」が3名、それ以外の従業員は「休み」として割り当ててください。
` +
         `- シフトは以下の3種類です：「早」（早番）、「遅」（遅番）、「休」（休み）。
` +
         `- 各従業員は、できるだけ公平に「早」「遅」「休」のシフトが割り当てられるようにしてください。
` +
         `- 同じ従業員が連続して「早」または「遅」シフトに入らないようにしてください。
` +
         `- 全員のシフトが均等になるようにし、できるだけランダムに割り当ててください。
` +
         `- 各日付のシフトには、毎回異なる従業員が含まれるようにしてください。
` +
         `- 各従業員が各シフト（「早」、「遅」、「休」）をできるだけ均等に受けるようにしてください。

` +
         `出力は必ず有効なJSON形式で次のようにしてください:
` +
         `{
  "10/1": {"従業員名1": "早", "従業員名2": "早", "従業員名3": "早", "従業員名4": "遅", "従業員名5": "遅", "従業員名6": "遅", "従業員名7": "休", "従業員名8": "休"},
  "10/2": {"従業員名1": "遅", "従業員名2": "早", "従業員名3": "早", "従業員名4": "早", "従業員名5": "遅", "従業員名6": "遅", "従業員名7": "休", "従業員名8": "休"}
}` +
         `

従業員リスト: ${members.join(", ")}
` +
         `日付リスト: ${dates.join(",")}

` +
         `注意: 必ず出力は厳密にJSON形式で出力してください。余分な説明は一切含めないでください。途中で切れることなく完全なJSONを返してください。`;
}

function callChatGPTAPI(prompt) {
  const url = 'https://api.openai.com/v1/chat/completions';
  const options = {
    method: 'post',
    contentType: 'application/json',
    headers: {
      'Authorization': `Bearer ${OPENAI_API_KEY}`
    },
    payload: JSON.stringify({
      model: 'gpt-4o',
      messages: [
        { role: 'system', content: 'You are an AI that only outputs valid JSON based on user prompts.' },
        { role: 'user', content: prompt }
      ],
      max_tokens: 4500,
      temperature: 0.7
    })
  };

  try {
    const response = UrlFetchApp.fetch(url, options);
    const json = JSON.parse(response.getContentText());

    // ChatGPTの応答を取得
    let content = json.choices[0].message.content.trim();

    // JSON形式の部分のみを抽出する処理
    const jsonStart = content.indexOf('{');
    const jsonEnd = content.lastIndexOf('}') + 1;
    if (jsonStart !== -1 && jsonEnd !== -1) {
      content = content.slice(jsonStart, jsonEnd);
    }

    // JSONパースの試行
    try {
      return JSON.parse(content); // JSONとしてパースを試みる
    } catch (e) {
      Logger.log("Error: The response is not valid JSON.");
      Logger.log("Received content: " + content);
      Logger.log("Prompt used: " + prompt);
      return null; // JSONとしてパースできない場合
    }
  } catch (e) {
    Logger.log("Error: " + e.toString());
    return null;
  }
}

function applyShiftAssignments(sheet, shiftAssignments, startDateColumn, startRow) {
  // ChatGPTの応答をシフト表に反映
  Object.keys(shiftAssignments).forEach((date, dateIndex) => {
    const col = startDateColumn + dateIndex;
    const assignments = shiftAssignments[date];
    let rowIndex = startRow;

    // 各従業員のシフトを対応するセルに配置
    Object.keys(assignments).forEach((member) => {
      const shift = assignments[member];
      sheet.getRange(rowIndex, col).setValue(shift);
      rowIndex++;
    });
  });
}