Update InitialMarkups.py

InitialMarkups.py

@@ -8,7 +8,7 @@ Original file is located at
 """
 baselink='https://findconsole-initialmarkups.hf.space/view-pdf?'
 
-
+  
 
 from io import BytesIO
 import re
@@ -65,7 +65,7 @@ def is_header(span, most_common_font_size, most_common_color, most_common_font):
         (
             span["size"] > most_common_font_size or
             span["font"].lower() != most_common_font.lower() or
-            is_bold
+            (is_bold and span["size"] > most_common_font_size )
         )
     )
 
@@ -536,32 +536,36 @@ def highlight_boxes(doc, highlights, stringtowrite, fixed_width=500):  # Set you
         # Get original rect for vertical coordinates
         orig_rect = fitz.Rect(bbox)
         rect_height = orig_rect.height
-
-        if orig_rect.width > 10:
-            # Center horizontally using fixed width
-            center_x = page_width / 2
-            new_x0 = center_x - fixed_width / 2
-            new_x1 = center_x + fixed_width / 2
-            new_rect = fitz.Rect(new_x0, orig_rect.y0, new_x1, orig_rect.y1)
-
-            # Add highlight rectangle
-            annot = page.add_rect_annot(new_rect)
-            annot.set_colors(stroke=(1, 1, 0), fill=(1, 1, 0))
-            annot.set_opacity(0.3)
-            annot.update()
-
-            # Add right-aligned freetext annotation inside the fixed-width box
-            text = '['+stringtowrite +']'
-            annot1 = page.add_freetext_annot(
-                new_rect,
-                text,
-                fontsize=15,
-                fontname='helv',
-                text_color=(1, 0, 0),
-                rotate=page.rotation,
-                align=2  # right alignment
-            )
-            annot1.update()
+        if rect_height > 30:
+            if orig_rect.width > 10:
+                # Center horizontally using fixed width
+                center_x = page_width / 2
+                new_x0 = center_x - fixed_width / 2
+                new_x1 = center_x + fixed_width / 2
+                new_rect = fitz.Rect(new_x0, orig_rect.y0, new_x1, orig_rect.y1)
+    
+                # Add highlight rectangle
+                annot = page.add_rect_annot(new_rect)
+                if stringtowrite.startswith('Not'):
+                    annot.set_colors(stroke=(0.5, 0.5, 0.5), fill=(0.5, 0.5, 0.5))
+                else:
+                    annot.set_colors(stroke=(1, 1, 0), fill=(1, 1, 0))
+ 
+                annot.set_opacity(0.3)
+                annot.update()
+    
+                # Add right-aligned freetext annotation inside the fixed-width box
+                text = '['+stringtowrite +']'
+                annot1 = page.add_freetext_annot(
+                    new_rect,
+                    text,
+                    fontsize=15,
+                    fontname='helv',
+                    text_color=(1, 0, 0),
+                    rotate=page.rotation,
+                    align=2  # right alignment
+                )
+                annot1.update()
 
 def get_leaf_headers_with_paths(listtoloop, path=None, output=None):
     if path is None: