การแยกวิเคราะห์แบบฟอร์มด้วย Document AI (Python)

1. บทนำ

ในโค้ดแล็บนี้ คุณจะได้เรียนรู้วิธีใช้เครื่องมือแยกวิเคราะห์แบบฟอร์มของ Document AI เพื่อแยกวิเคราะห์แบบฟอร์มที่เขียนด้วยลายมือด้วย Python

เราจะใช้แบบฟอร์มการรับข้อมูลทางการแพทย์อย่างง่ายเป็นตัวอย่าง แต่ขั้นตอนนี้จะใช้ได้กับแบบฟอร์มทั่วไปที่ DocAI รองรับ

ข้อกำหนดเบื้องต้น

Codelab นี้สร้างขึ้นจากเนื้อหาที่นำเสนอใน Codelab อื่นๆ ของ Document AI

ขอแนะนำให้คุณทำ Codelab ต่อไปนี้ให้เสร็จก่อนดำเนินการต่อ

สิ่งที่คุณจะได้เรียนรู้

  • วิธีแยกวิเคราะห์และดึงข้อมูลจากแบบฟอร์มที่สแกนโดยใช้เครื่องมือแยกวิเคราะห์แบบฟอร์มของ Document AI

สิ่งที่คุณต้องมี

  • โปรเจ็กต์ Google Cloud
  • เบราว์เซอร์ เช่น Chrome หรือ Firefox
  • ความรู้เกี่ยวกับ Python 3

แบบสำรวจ

คุณจะใช้บทแนะนำนี้อย่างไร

อ่านอย่างเดียว อ่านและทำแบบฝึกหัด

คุณจะให้คะแนนประสบการณ์การใช้งาน Python เท่าใด

ผู้ฝึกหัด ขั้นกลาง ผู้ชำนาญ

คุณจะให้คะแนนประสบการณ์การใช้บริการ Google Cloud เท่าใด

ผู้ฝึกหัด ขั้นกลาง ผู้ชำนาญ

2. การตั้งค่าและข้อกำหนด

Codelab นี้ถือว่าคุณได้ทำตามขั้นตอนการตั้งค่า Document AI ที่ระบุไว้ใน Document AI OCR Codelab เรียบร้อยแล้ว

โปรดทำตามขั้นตอนต่อไปนี้ก่อนดำเนินการต่อ

นอกจากนี้ คุณยังต้องติดตั้ง Pandas ซึ่งเป็นไลบรารีการวิเคราะห์ข้อมูลแบบโอเพนซอร์สสำหรับ Python ด้วย

pip3 install --upgrade pandas

3. สร้างโปรเซสเซอร์ตัวแยกวิเคราะห์แบบฟอร์ม

คุณต้องสร้างอินสแตนซ์ของโปรเซสเซอร์ Form Parser ก่อนเพื่อใช้ใน Document AI Platform สำหรับบทแนะนำนี้

  1. ในคอนโซล ให้ไปที่ภาพรวมแพลตฟอร์ม Document AI
  2. คลิกสร้างโปรเซสเซอร์ แล้วเลือกตัวแยกวิเคราะห์แบบฟอร์มโปรเซสเซอร์
  3. ระบุชื่อโปรเซสเซอร์และเลือกภูมิภาคจากรายการ
  4. คลิกสร้างเพื่อสร้างโปรเซสเซอร์
  5. คัดลอกรหัสโปรเซสเซอร์ คุณต้องใช้รหัสนี้ในโค้ดในภายหลัง

ทดสอบโปรเซสเซอร์ใน Cloud Console

คุณสามารถทดสอบโปรเซสเซอร์ในคอนโซลได้โดยการอัปโหลดเอกสาร คลิกอัปโหลดเอกสาร แล้วเลือกแบบฟอร์มที่จะแยกวิเคราะห์ คุณสามารถดาวน์โหลดและใช้แบบฟอร์มตัวอย่างนี้ได้หากไม่มีแบบฟอร์มที่พร้อมใช้งาน

แบบฟอร์มสุขภาพ

เอาต์พุตควรมีลักษณะดังนี้ แบบฟอร์มที่แยกวิเคราะห์

4. ดาวน์โหลดแบบฟอร์มตัวอย่าง

เรามีเอกสารตัวอย่างที่มีแบบฟอร์มการรับข้อมูลทางการแพทย์อย่างง่าย

คุณดาวน์โหลด PDF ได้โดยใช้ลิงก์ต่อไปนี้ จากนั้นอัปโหลดไปยังอินสแตนซ์ Cloud Shell

หรือจะดาวน์โหลดจากที่เก็บข้อมูล Google Cloud Storage สาธารณะของเราโดยใช้ gsutil ก็ได้

gsutil cp gs://cloud-samples-data/documentai/codelabs/form-parser/intake-form.pdf .

ยืนยันว่าระบบได้ดาวน์โหลดไฟล์ไปยัง Cloud Shell โดยใช้คำสั่งด้านล่าง

ls -ltr intake-form.pdf

5. แยกคู่คีย์/ค่าของแบบฟอร์ม

ในขั้นตอนนี้ คุณจะใช้ API การประมวลผลออนไลน์เพื่อเรียกโปรเซสเซอร์ตัวแยกวิเคราะห์แบบฟอร์มที่สร้างไว้ก่อนหน้านี้ จากนั้นคุณจะแยกคู่คีย์-ค่าที่พบในเอกสาร

การประมวลผลออนไลน์ใช้สำหรับการส่งเอกสารเดียวและรอการตอบกลับ นอกจากนี้ คุณยังใช้การประมวลผลเป็นชุดได้หากต้องการส่งหลายไฟล์หรือหากไฟล์มีขนาดเกินจำนวนหน้าสูงสุดสำหรับการประมวลผลออนไลน์ คุณดูวิธีดำเนินการนี้ได้ใน OCR Codelab

โค้ดสำหรับการส่งคำขอประมวลผลจะเหมือนกันสำหรับตัวประมวลผลทุกประเภท ยกเว้นรหัสตัวประมวลผล

ออบเจ็กต์การตอบกลับ Document มีรายการหน้าจากเอกสารอินพุต

ออบเจ็กต์ page แต่ละรายการมีรายการช่องแบบฟอร์มและตำแหน่งของช่องเหล่านั้นในข้อความ

โค้ดต่อไปนี้จะวนซ้ำในแต่ละหน้าและดึงคีย์ ค่า และคะแนนความเชื่อมั่นแต่ละรายการ ซึ่งเป็นข้อมูลที่มีโครงสร้างที่จัดเก็บในฐานข้อมูลหรือใช้ในแอปพลิเคชันอื่นๆ ได้ง่ายขึ้น

สร้างไฟล์ชื่อ form_parser.py แล้วใช้โค้ดด้านล่าง

form_parser.py

import pandas as pd
from google.cloud import documentai_v1 as documentai


def online_process(
    project_id: str,
    location: str,
    processor_id: str,
    file_path: str,
    mime_type: str,
) -> documentai.Document:
    """
    Processes a document using the Document AI Online Processing API.
    """

    opts = {"api_endpoint": f"{location}-documentai.googleapis.com"}

    # Instantiates a client
    documentai_client = documentai.DocumentProcessorServiceClient(client_options=opts)

    # The full resource name of the processor, e.g.:
    # projects/project-id/locations/location/processor/processor-id
    # You must create new processors in the Cloud Console first
    resource_name = documentai_client.processor_path(project_id, location, processor_id)

    # Read the file into memory
    with open(file_path, "rb") as image:
        image_content = image.read()

        # Load Binary Data into Document AI RawDocument Object
        raw_document = documentai.RawDocument(
            content=image_content, mime_type=mime_type
        )

        # Configure the process request
        request = documentai.ProcessRequest(
            name=resource_name, raw_document=raw_document
        )

        # Use the Document AI client to process the sample form
        result = documentai_client.process_document(request=request)

        return result.document


def trim_text(text: str):
    """
    Remove extra space characters from text (blank, newline, tab, etc.)
    """
    return text.strip().replace("\n", " ")


PROJECT_ID = "YOUR_PROJECT_ID"
LOCATION = "YOUR_PROJECT_LOCATION"  # Format is 'us' or 'eu'
PROCESSOR_ID = "FORM_PARSER_ID"  # Create processor in Cloud Console

# The local file in your current working directory
FILE_PATH = "intake-form.pdf"
# Refer to https://cloud.google.com/document-ai/docs/processors-list
# for supported file types
MIME_TYPE = "application/pdf"

document = online_process(
    project_id=PROJECT_ID,
    location=LOCATION,
    processor_id=PROCESSOR_ID,
    file_path=FILE_PATH,
    mime_type=MIME_TYPE,
)

names = []
name_confidence = []
values = []
value_confidence = []

for page in document.pages:
    for field in page.form_fields:
        # Get the extracted field names
        names.append(trim_text(field.field_name.text_anchor.content))
        # Confidence - How "sure" the Model is that the text is correct
        name_confidence.append(field.field_name.confidence)

        values.append(trim_text(field.field_value.text_anchor.content))
        value_confidence.append(field.field_value.confidence)

# Create a Pandas Dataframe to print the values in tabular format.
df = pd.DataFrame(
    {
        "Field Name": names,
        "Field Name Confidence": name_confidence,
        "Field Value": values,
        "Field Value Confidence": value_confidence,
    }
)

print(df)

เรียกใช้โค้ดตอนนี้ แล้วคุณจะเห็นข้อความที่แยกออกมาและพิมพ์ในคอนโซล

คุณควรเห็นเอาต์พุตต่อไปนี้หากใช้เอกสารตัวอย่างของเรา

$ python3 form_parser.py
                                           Field Name  Field Name Confidence                                        Field Value  Field Value Confidence
0                                            Phone #:               0.999982                                     (906) 917-3486                0.999982
1                                  Emergency Contact:               0.999972                                         Eva Walker                0.999972
2                                     Marital Status:               0.999951                                             Single                0.999951
3                                             Gender:               0.999933                                                  F                0.999933
4                                         Occupation:               0.999914                                  Software Engineer                0.999914
5                                        Referred By:               0.999862                                               None                0.999862
6                                               Date:               0.999858                                            9/14/19                0.999858
7                                                DOB:               0.999716                                         09/04/1986                0.999716
8                                            Address:               0.999147                                     24 Barney Lane                0.999147
9                                               City:               0.997718                                             Towaco                0.997718
10                                              Name:               0.997345                                       Sally Walker                0.997345
11                                             State:               0.996944                                                 NJ                0.996944
...

6. แยกวิเคราะห์ตาราง

นอกจากนี้ ตัวแยกวิเคราะห์แบบฟอร์มยังดึงข้อมูลจากตารางภายในเอกสารได้ด้วย ในขั้นตอนนี้ เราจะดาวน์โหลดเอกสารตัวอย่างใหม่และดึงข้อมูลจากตาราง เนื่องจากเราโหลดข้อมูลลงใน Pandas ข้อมูลนี้จึงสามารถส่งออกไปยังไฟล์ CSV และรูปแบบอื่นๆ อีกมากมายได้ด้วยการเรียกใช้เมธอดเพียงครั้งเดียว

ดาวน์โหลดแบบฟอร์มตัวอย่างที่มีตาราง

เรามีเอกสารตัวอย่างที่มีแบบฟอร์มและตารางตัวอย่าง

คุณดาวน์โหลด PDF ได้โดยใช้ลิงก์ต่อไปนี้ จากนั้นอัปโหลดไปยังอินสแตนซ์ Cloud Shell

หรือจะดาวน์โหลดจากที่เก็บข้อมูล Google Cloud Storage สาธารณะของเราโดยใช้ gsutil ก็ได้

gsutil cp gs://cloud-samples-data/documentai/codelabs/form-parser/form_with_tables.pdf .

ยืนยันว่าระบบได้ดาวน์โหลดไฟล์ไปยัง Cloud Shell โดยใช้คำสั่งด้านล่าง

ls -ltr form_with_tables.pdf

ดึงข้อมูลตาราง

คำขอประมวลผลข้อมูลตารางจะเหมือนกับการแยกคู่คีย์-ค่าทุกประการ ความแตกต่างคือฟิลด์ที่เราดึงข้อมูลจากในการตอบกลับ ระบบจะจัดเก็บข้อมูลตารางไว้ในฟิลด์ pages[].tables[]

ตัวอย่างนี้จะดึงข้อมูลเกี่ยวกับแถวส่วนหัวของตารางและแถวเนื้อหาของตารางสำหรับแต่ละตารางและหน้า จากนั้นจะพิมพ์ตารางและบันทึกตารางเป็นไฟล์ CSV

สร้างไฟล์ชื่อ table_parsing.py แล้วใช้โค้ดด้านล่าง

table_parsing.py

# type: ignore[1]
"""
Uses Document AI online processing to call a form parser processor
Extracts the tables and data in the document.
"""
from os.path import splitext
from typing import List, Sequence

import pandas as pd
from google.cloud import documentai


def online_process(
    project_id: str,
    location: str,
    processor_id: str,
    file_path: str,
    mime_type: str,
) -> documentai.Document:
    """
    Processes a document using the Document AI Online Processing API.
    """

    opts = {"api_endpoint": f"{location}-documentai.googleapis.com"}

    # Instantiates a client
    documentai_client = documentai.DocumentProcessorServiceClient(client_options=opts)

    # The full resource name of the processor, e.g.:
    # projects/project-id/locations/location/processor/processor-id
    # You must create new processors in the Cloud Console first
    resource_name = documentai_client.processor_path(project_id, location, processor_id)

    # Read the file into memory
    with open(file_path, "rb") as image:
        image_content = image.read()

        # Load Binary Data into Document AI RawDocument Object
        raw_document = documentai.RawDocument(
            content=image_content, mime_type=mime_type
        )

        # Configure the process request
        request = documentai.ProcessRequest(
            name=resource_name, raw_document=raw_document
        )

        # Use the Document AI client to process the sample form
        result = documentai_client.process_document(request=request)

        return result.document


def get_table_data(
    rows: Sequence[documentai.Document.Page.Table.TableRow], text: str
) -> List[List[str]]:
    """
    Get Text data from table rows
    """
    all_values: List[List[str]] = []
    for row in rows:
        current_row_values: List[str] = []
        for cell in row.cells:
            current_row_values.append(
                text_anchor_to_text(cell.layout.text_anchor, text)
            )
        all_values.append(current_row_values)
    return all_values


def text_anchor_to_text(text_anchor: documentai.Document.TextAnchor, text: str) -> str:
    """
    Document AI identifies table data by their offsets in the entirety of the
    document's text. This function converts offsets to a string.
    """
    response = ""
    # If a text segment spans several lines, it will
    # be stored in different text segments.
    for segment in text_anchor.text_segments:
        start_index = int(segment.start_index)
        end_index = int(segment.end_index)
        response += text[start_index:end_index]
    return response.strip().replace("\n", " ")


PROJECT_ID = "YOUR_PROJECT_ID"
LOCATION = "YOUR_PROJECT_LOCATION"  # Format is 'us' or 'eu'
PROCESSOR_ID = "FORM_PARSER_ID"  # Create processor before running sample

# The local file in your current working directory
FILE_PATH = "form_with_tables.pdf"
# Refer to https://cloud.google.com/document-ai/docs/file-types
# for supported file types
MIME_TYPE = "application/pdf"

document = online_process(
    project_id=PROJECT_ID,
    location=LOCATION,
    processor_id=PROCESSOR_ID,
    file_path=FILE_PATH,
    mime_type=MIME_TYPE,
)

header_row_values: List[List[str]] = []
body_row_values: List[List[str]] = []

# Input Filename without extension
output_file_prefix = splitext(FILE_PATH)[0]

for page in document.pages:
    for index, table in enumerate(page.tables):
        header_row_values = get_table_data(table.header_rows, document.text)
        body_row_values = get_table_data(table.body_rows, document.text)

        # Create a Pandas Dataframe to print the values in tabular format.
        df = pd.DataFrame(
            data=body_row_values,
            columns=pd.MultiIndex.from_arrays(header_row_values),
        )

        print(f"Page {page.page_number} - Table {index}")
        print(df)

        # Save each table as a CSV file
        output_filename = f"{output_file_prefix}_pg{page.page_number}_tb{index}.csv"
        df.to_csv(output_filename, index=False)

เรียกใช้โค้ดตอนนี้ แล้วคุณจะเห็นข้อความที่แยกออกมาและพิมพ์ในคอนโซล

คุณควรเห็นเอาต์พุตต่อไปนี้หากใช้เอกสารตัวอย่างของเรา

$ python3 table_parsing.py
Page 1 - Table 0
     Item    Description
0  Item 1  Description 1
1  Item 2  Description 2
2  Item 3  Description 3
Page 1 - Table 1
  Form Number:     12345678
0   Form Date:   2020/10/01
1        Name:   First Last
2     Address:  123 Fake St

นอกจากนี้ คุณควรมีไฟล์ CSV ใหม่ 2 ไฟล์ในไดเรกทอรีที่คุณเรียกใช้โค้ด

$ ls
form_with_tables_pg1_tb0.csv form_with_tables_pg1_tb1.csv table_parsing.py

7. ขอแสดงความยินดี

ขอแสดงความยินดี คุณใช้ Document AI API เพื่อดึงข้อมูลจากแบบฟอร์มที่เขียนด้วยลายมือได้สำเร็จแล้ว เราขอแนะนำให้คุณลองใช้เอกสารแบบฟอร์มอื่นๆ

ล้างข้อมูล

โปรดดำเนินการดังนี้เพื่อเลี่ยงไม่ให้เกิดการเรียกเก็บเงินกับบัญชี Google Cloud สำหรับทรัพยากรที่ใช้ในบทแนะนำนี้

  • ใน Cloud Console ให้ไปที่หน้าจัดการทรัพยากร
  • เลือกโปรเจ็กต์ในรายการโปรเจ็กต์ แล้วคลิกลบ
  • ในกล่องโต้ตอบ ให้พิมพ์รหัสโปรเจ็กต์ แล้วคลิกปิดเพื่อลบโปรเจ็กต์

ดูข้อมูลเพิ่มเติม

เรียนรู้เพิ่มเติมเกี่ยวกับ Document AI ด้วย Codelab ที่เกี่ยวข้องต่อไปนี้

แหล่งข้อมูล

ใบอนุญาต

ผลงานนี้ได้รับอนุญาตภายใต้สัญญาอนุญาตครีเอทีฟคอมมอนส์สำหรับยอมรับสิทธิของผู้สร้าง (Creative Commons Attribution License) 2.0 แบบทั่วไป