Googleドライブに保存したレシートを自動リネームする その4 – フォルダ内のすべてのPDF・画像ファイルをリネームする

/**
 * 
 * 画像ファイルをOCRでテキストに変換し、その情報をもとにファイル名を変更する関数
 * この関数は、特定のフォルダ内の画像ファイルを対象に、各ファイルをOCRでテキストに変換し、
 * ChatGPTを使用してテキストから特定の情報を抽出します。その後、抽出した情報をもとにファイル名を変更します。
 */

function reNameImgFiles() {
  const idReceiptFolder = 'レシートを保存しているフォルダのID':
  const systemRole = 'あなたは、領収書やレシートを仕訳入力する事業会社の会計責任者です。';

  const prompt =
    '#制約条件に従いつつ、以下の領収書・レシートをOCRしたテキストから支払日, 支払総額, 支払先を抽出し'
    + '{"payment_date": 支払日,"total_amount": 支払総額,"payment_to": 支払先}'
    + 'のJSON形式の文字列を出力してください。' + '\n'
    + '\n'
    + '#制約条件' + '\n'
    + '- 支払日はyyyyMMdd形式の文字列' + '\n'
    + '- 支払総額は数値のみ' + '\n'
    + '- 支払先はスペースなしの文字列' + '\n'
    + '\n'
    + '#テキスト' + '\n'
    + '###' + '\n';

  // 指定フォルダから画像ファイルのIDを取得
  const aryImgId = getImgPdfIds_(idReceiptFolder);

  // 処理すべき画像ファイルがない場合は処理を中断
  if (aryImgId.length === 0) { return };

  // 各画像ファイルを処理
  const aryObjReceipt = aryImgId.map(idImg => {
    // 画像をテキストに変換
    const ocrText = img2Text_(idImg);
    // ChatGPTに問い合わせ
    const objChat = askChatGPT_(systemRole, prompt + ocrText, 'gpt-4-1106-preview'); // gpt-3.5-turbo-1106 or gpt-4-1106-preview
    // レスポンスから必要な情報を抽出
    const answer = objChat.choices[0].message.content;
    const objJson = JSON.parse(answer);
    objJson.idFile = idImg;
    return objJson;
  });
  // 各ファイルの名前を変更
  aryObjReceipt.forEach(objReceipt => {
    const file = DriveApp.getFileById(objReceipt.idFile);
    const payment_date = objReceipt.payment_date.replace(/[ 　.,]/g, '');
    const total_amount = objReceipt.total_amount;
    if (typeof (total_amount) === 'string') { total_amount.replace(/[ 　.,]/g, '') }; // 文字列の場合のみ不要なスペース等を削除
    const payment_to = objReceipt.payment_to.replace(/[ 　.,]/g, '');
    // 新しいファイル名を設定
    const newFileName = `${payment_date}_${total_amount}_${payment_to}`;
    file.setName(newFileName);
  });
}

/**
 * 指定されたフォルダ内のPDFおよび画像ファイルのIDを取得する関数
 * 
 * @param {string} idFolder - ファイルを検索するフォルダのID
 * @return {Array.<string>} PDFおよび画像ファイルのIDの配列
 */

function getImgPdfIds_(idFolder) {
  // 指定されたIDのフォルダからファイルを取得
  const folder = DriveApp.getFolderById(idFolder);
  const files = folder.getFiles();
  const aryFileId = new Array();

  while (files.hasNext()) {
    const file = files.next();
    const mimeType = file.getMimeType();

    // PDFまたは画像ファイル（JPEG、PNGなど）の場合、そのIDを配列に追加
    if (mimeType === MimeType.PDF ||
        mimeType === MimeType.JPEG ||
        mimeType === MimeType.PNG ||
        mimeType === MimeType.GIF ||
        mimeType === MimeType.BMP ||
        mimeType === MimeType.TIFF) {
      aryFileId.push(file.getId());
    }
  }

  return aryFileId;
}

  // 各画像ファイルを処理
  const aryObjReceipt = aryImgId.map(idImg => {
    // 画像をテキストに変換
    const ocrText = img2Text_(idImg);
    // ChatGPTに問い合わせ
    const objChat = askChatGPT_(systemRole, prompt + ocrText, 'gpt-3.5-turbo-1106'); // gpt-3.5-turbo-1106 or gpt-4-1106-preview
    // レスポンスから必要な情報を抽出
    const answer = objChat.choices[0].message.content;
    const objJson = JSON.parse(answer);
    objJson.idFile = idImg;
    return objJson;
  });

  // 各ファイルの名前を変更
  aryObjReceipt.forEach(objReceipt => {
    const file = DriveApp.getFileById(objReceipt.idFile);
    const payment_date = objReceipt.payment_date.replace(/[ 　.,]/g, '');
    const total_amount = objReceipt.total_amount;
    if (typeof (total_amount) === 'string') { total_amount.replace(/[ 　.,]/g, '') }; // 文字列の場合のみ不要なスペース等を削除
    const payment_to = objReceipt.payment_to.replace(/[ 　.,]/g, '');
    // 新しいファイル名を設定
    const newFileName = `${payment_date}_${total_amount}_${payment_to}`;
    file.setName(newFileName);
  });

/**
 * 指定したGoogleドライブのフォルダから別のフォルダへすべてのファイルを移動する関数
 * 
 * @param {string} idFolderFrom - ファイルを移動させたい元のフォルダのID
 * @param {string} idFolderTo - ファイルを移動させる先のフォルダのID
 * 
 */

function moveFilesToAnotherFolder_(idFolderFrom, idFolderTo) {
  // 元のフォルダを取得
  const folderFrom = DriveApp.getFolderById(idFolderFrom);
  // 移動先のフォルダを取得
  const folderTo = DriveApp.getFolderById(idFolderTo);

  // 元のフォルダ内のすべてのファイルを取得
  const files = folderFrom.getFiles();

  // すべてのファイルを移動先のフォルダへ移動
  while (files.hasNext()) {
    const file = files.next();
    file.moveTo(folderTo);
  }
}

/**
 * 指定したGoogleドライブのフォルダ内にあるすべてのファイルの名前を
 * 「フォルダ名_元のファイル名」 にリネームする関数
 * 
 * @param {string} folderId - ファイル名をリネームしたいフォルダのID
 * 
 */

function renameFilesInFolder_(folderId) {
  // フォルダIDからフォルダを取得
  const folder = DriveApp.getFolderById(folderId);
  const folderName = folder.getName();

  // フォルダ内のすべてのファイルを取得
  const files = folder.getFiles();

  // フォルダ内の各ファイルについて処理を行う
  while (files.hasNext()) {
    const file = files.next();
    const fileName = file.getName();
    // ファイル名を "フォルダ名_ファイル名" にリネーム
    const newFileName = `${folderName}_${fileName}`;
    file.setName(newFileName);
  }
}