fix: get bounds for blocks instead of pages (GoogleCloudPlatform#2705)

busunkim96 · leahecole · leahecole · commit a8d3a1b69756 · 2020-01-10T11:09:18.000-08:00
* fix: use `page.bounding_box` when feature is page Closes GoogleCloudPlatform#2702 * fix: outline blocks instead of pages Co-authored-by: Leah E. Cole <6719667+leahecole@users.noreply.github.com>
diff --git a/vision/cloud-client/document_text/doctext.py b/vision/cloud-client/document_text/doctext.py
@@ -85,17 +85,14 @@ def get_document_bounds(image_file, feature):
             if (feature == FeatureType.BLOCK):
                 bounds.append(block.bounding_box)
 
-        if (feature == FeatureType.PAGE):
-            bounds.append(block.bounding_box)
-
     # The list `bounds` contains the coordinates of the bounding boxes.
     # [END vision_document_text_tutorial_detect_bounds]
     return bounds
 
 
 def render_doc_text(filein, fileout):
     image = Image.open(filein)
-    bounds = get_document_bounds(filein, FeatureType.PAGE)
+    bounds = get_document_bounds(filein, FeatureType.BLOCK)
     draw_boxes(image, bounds, 'blue')
     bounds = get_document_bounds(filein, FeatureType.PARA)
     draw_boxes(image, bounds, 'red')
