Update visualizer.py

visualizer.py

@@ -83,19 +83,19 @@ def _get_dynamic_font_size(
 
     total_area = 0
     valid_count = 0
-    
+
     for points in polygons:
         area = _calculate_polygon_area(points)
         if area > 0:
             total_area += area
             valid_count += 1
-    
+
     if valid_count == 0:
         return 16
-    
+
     avg_area = total_area / valid_count
     font_size = int(math.sqrt(avg_area) * 0.2)
-    
+
     return max(12, min(72, font_size))
 
 
@@ -107,11 +107,11 @@ def _get_font(size: int) -> Optional[ImageFont.FreeTypeFont]:
             "/System/Library/Fonts/Helvetica.ttc",
             "C:\\Windows\\Fonts\\arial.ttf",
         ]
-        
+
         for font_path in font_paths:
             if os.path.exists(font_path):
                 return ImageFont.truetype(font_path, size)
-        
+
         return ImageFont.load_default()
     except:
         return ImageFont.load_default()
@@ -130,30 +130,30 @@ def _visualize_page_xml(
     for elem in root.iter():
         if elem.tag.endswith("TextLine"):
             text_lines.append(elem)
-    
+
     line_data = []
     all_polygons = []
-    
+
     for text_line in text_lines:
         coords_elem = None
         for child in text_line:
             if child.tag.endswith("Coords"):
                 coords_elem = child
                 break
-        
+
         if coords_elem is not None:
             points_str = coords_elem.get("points", "")
             points = _parse_points(points_str)
-            
+
             if len(points) >= 3:
                 text_content = ""
                 confidence = None
-                
+
                 for te in text_line.iter():
                     if te.tag.endswith("Unicode") and te.text:
                         text_content = te.text.strip()
                         break
-                
+
                 for te in text_line.iter():
                     if te.tag.endswith("TextEquiv"):
                         conf_str = te.get("conf")
@@ -163,30 +163,34 @@ def _visualize_page_xml(
                             except:
                                 pass
                         break
-                
+
                 display_text = text_content
                 if confidence is not None:
                     display_text = f"{text_content} ({confidence:.3f})"
-                
+
                 line_data.append((points, display_text))
                 all_polygons.append(points)
-    
+
     font_size = _get_dynamic_font_size(all_polygons, image_size)
     font = _get_font(font_size)
-    
+
     for i, (points, text) in enumerate(line_data):
         color = "red" if i % 2 == 0 else "blue"
         draw.polygon(points, outline=color, width=2)
-        
+
         if text:
             centroid_x = sum(p[0] for p in points) // len(points)
             centroid_y = sum(p[1] for p in points) // len(points)
-            
+
             if font != ImageFont.load_default():
-                bbox = draw.textbbox((centroid_x, centroid_y), text, font=font, anchor="mm")
+                bbox = draw.textbbox(
+                    (centroid_x, centroid_y), text, font=font, anchor="mm"
+                )
                 bbox = (bbox[0] - 2, bbox[1] - 2, bbox[2] + 2, bbox[3] + 2)
                 draw.rectangle(bbox, fill=(255, 255, 255, 200), outline="black")
-                draw.text((centroid_x, centroid_y), text, fill="black", font=font, anchor="mm")
+                draw.text(
+                    (centroid_x, centroid_y), text, fill="black", font=font, anchor="mm"
+                )
             else:
                 draw.text((centroid_x, centroid_y), text, fill="black")
 
@@ -195,15 +199,15 @@ def _visualize_alto_xml(
     draw: ImageDraw.Draw, root: ET.Element, image_size: Tuple[int, int]
 ):
     namespace = _get_namespace(root)
-    
+
     text_lines = []
     for elem in root.iter():
         if elem.tag.endswith("TextLine"):
             text_lines.append(elem)
-    
+
     line_data = []
     all_polygons = []
-    
+
     for text_line in text_lines:
         points = []
         for shape in text_line.iter():
@@ -214,11 +218,11 @@ def _visualize_alto_xml(
                         points = _parse_points(points_str)
                         break
                 break
-        
+
         if len(points) >= 3:
             text_content = ""
             confidence = None
-            
+
             for string_elem in text_line.iter():
                 if string_elem.tag.endswith("String"):
                     text_content = string_elem.get("CONTENT", "")
@@ -229,29 +233,33 @@ def _visualize_alto_xml(
                         except:
                             pass
                     break
-            
+
             display_text = text_content
             if confidence is not None:
                 display_text = f"{text_content} ({confidence:.3f})"
-            
+
             line_data.append((points, display_text))
             all_polygons.append(points)
-    
+
     font_size = _get_dynamic_font_size(all_polygons, image_size)
     font = _get_font(font_size)
-    
+
     for i, (points, text) in enumerate(line_data):
         color = "red" if i % 2 == 0 else "blue"
         draw.polygon(points, outline=color, width=2)
-        
+
         if text:
             centroid_x = sum(p[0] for p in points) // len(points)
             centroid_y = sum(p[1] for p in points) // len(points)
-            
+
             if font != ImageFont.load_default():
-                bbox = draw.textbbox((centroid_x, centroid_y), text, font=font, anchor="mm")
+                bbox = draw.textbbox(
+                    (centroid_x, centroid_y), text, font=font, anchor="mm"
+                )
                 bbox = (bbox[0] - 2, bbox[1] - 2, bbox[2] + 2, bbox[3] + 2)
                 draw.rectangle(bbox, fill=(255, 255, 255, 200), outline="black")
-                draw.text((centroid_x, centroid_y), text, fill="black", font=font, anchor="mm")
+                draw.text(
+                    (centroid_x, centroid_y), text, fill="black", font=font, anchor="mm"
+                )
             else:
-                draw.text((centroid_x, centroid_y), text, fill="black")
+                draw.text((centroid_x, centroid_y), text, fill="black")