Update plotting functions

plotting_functions.py

@@ -3,7 +3,7 @@ from PIL import Image, ImageDraw
 
 class PlotHTR:
     def __init__(self, region_alpha = 0.3, line_alpha = 0.3):
-        self.region_colors = {'marginalia': (0,201,87), 'page-number': (0,0,255), 'paragraph': (238,18,137), 'case-start': (238,18,137)}
+        self.polygon_color = (238,18,137)
         self.region_alpha = region_alpha 
         self.line_alpha = line_alpha 
 
@@ -12,14 +12,10 @@ class PlotHTR:
         img = image.copy()
         draw = ImageDraw.Draw(img)
         for region in region_data:
-            region_type = region['region_name']
             img_h, img_w = region['img_shape']
             region_polygon = list(map(tuple, region['region_coords']))
-            
-            color = self.region_colors[region_type]
-            
-            draw.polygon(region_polygon, fill = color, outline='black')
-            # Lower alpha value is applied to filled polygons to make them transparent
+            draw.polygon(region_polygon, fill = self.polygon_color, outline='black')
+        # Lower alpha value is applied to filled polygons to make them transparent
         blended_img = Image.blend(image, img, self.region_alpha)
         return blended_img
 
@@ -30,10 +26,10 @@ class PlotHTR:
         for region in region_data:
             img_h, img_w = region['img_shape']
             line_polygons = region['lines']
-            region_type = region['region_name']
-            color = self.region_colors[region_type]
+            print(f'{len(line_polygons)} text lines')
             for i, polygon in enumerate(line_polygons):
-                draw.polygon(polygon, fill = color)
+                line_polygon = list(map(tuple, polygon))
+                draw.polygon(line_polygon, fill = self.polygon_color, outline='black')
         # Lower alpha value is applied to filled polygons to make them transparent
         blended_img = Image.blend(image, img, self.line_alpha)
-        return blended_img
+        return blended_img