Add LAYOUTS to the CLI

Author: Belval

textractor/cli/cli.py

@@ -427,6 +427,8 @@ def textractor_cli():
                 print(out.signatures.pretty_print())
             if "ALL" in args.print or "IDS" in args.print:
                 print(out.identity_documents.pretty_print())
+            if "ALL" in args.print or "LAYOUTS" in args.print:
+                print(out.layouts.pretty_print())
 
         if args.linearize is not None:
             if args.linearize_config_path is not None:
@@ -450,6 +452,8 @@ def textractor_cli():
                 entity_list += out.queries
             if "ALL" in args.overlay or "SIGNATURES" in args.overlay:
                 entity_list += out.signatures
+            if "ALL" in args.overlay or "LAYOUTS" in args.overlay:
+                entity_list += out.layouts
             image = entity_list.visualize(
                 with_text=True,
                 font_size_ratio=args.font_size_ratio,

textractor/data/constants.py

@@ -319,7 +319,7 @@ class CLIPrint(Enum):
     EXPENSES = 5
     SIGNATURES = 6
     IDS = 7
-
+    LAYOUTS = 8
 
 class CLIOverlay(Enum):
     ALL = 0
@@ -328,4 +328,5 @@ class CLIOverlay(Enum):
     TABLES = 3
     FORMS = 4
     QUERIES = 5
-    SIGNATURE = 6
+    SIGNATURES = 6
+    LAYOUTS = 7