Adding support for image and table annotation for DocAI

documentai/snippets/handle_response_sample.py

@@ -32,7 +32,8 @@
 # processor_version = "rc" # Refer to https://cloud.google.com/document-ai/docs/manage-processor-versions for more information
 # file_path = "/path/to/local/pdf"
 # mime_type = "application/pdf" # Refer to https://cloud.google.com/document-ai/docs/file-types for supported file types
-
+# enable_image_annotation = False # Set to TRUE to enable processing. Refer to https://cloud.google.com/document-ai/docs/layout-parse-chunk#layout_parser_features
+# enable_table_annotation = False # Set to TRUE to enable processing. Refer to https://cloud.google.com/document-ai/docs/layout-parse-chunk#layout_parser_features
 
 # [END documentai_process_ocr_document]
 # [END documentai_process_form_document]
@@ -463,13 +464,17 @@ def process_document_layout_sample(
     processor_version: str,
     file_path: str,
     mime_type: str,
+    enable_image_annotation: bool = False, # Enable image annotation
+    enable_table_annotation: bool = False, # Enable table annotation
 ) -> documentai.Document:
     process_options = documentai.ProcessOptions(
         layout_config=documentai.ProcessOptions.LayoutConfig(
             chunking_config=documentai.ProcessOptions.LayoutConfig.ChunkingConfig(
                 chunk_size=1000,
                 include_ancestor_headings=True,
-            )
+            ),
+            enable_image_annotation=enable_image_annotation,
+            enable_table_annotation=enable_table_annotation,
         )
     )
 

documentai/snippets/handle_response_sample_test.py

@@ -229,6 +229,8 @@ def test_process_document_layout():
         processor_version="pretrained",
         file_path="resources/superconductivity.pdf",
         mime_type="application/pdf",
+        enable_image_annotation = True,
+        enable_table_annotation = True,
     )
 
     assert document