Update Doors_Schedule.py

Doors_Schedule.py

@@ -386,25 +386,46 @@ def add_bluebeam_count_annotations(pdf_bytes, locations):
     pdf_document = fitz.open("pdf", pdf_stream.read())  # Open PDF in memory
 
     page = pdf_document[0]  # First page
-    for loc in locations:
-        coor, lbl, clr,w = loc
-        clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
-        for cor in coor:
-            #Create a Circle annotation (Count Markup)
-            annot = page.add_circle_annot(
-                fitz.Rect(cor[0] - 10, cor[1] - 10, cor[0] + 10, cor[1] + 10)  # Small circle
-            )
-
-            #Assign required Bluebeam metadata
-            annot.set_colors(stroke=clr, fill=(1, 1, 1))  # Set stroke color and fill white
-            annot.set_border(width=2)  # Border thickness
-            annot.set_opacity(1)  # Fully visible
-
-            #Set annotation properties for Bluebeam Count detection
-            annot.set_info("name", lbl)  # Unique name for each count
-            annot.set_info("subject", "Count")  #Bluebeam uses "Count" for Count markups
-            annot.set_info("title", lbl)  # Optional
-            annot.update()  # Apply changes
+    if len(locations[0]) == 4:
+        for loc in locations:
+            coor, lbl, clr,w = loc
+            clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
+            for cor in coor:
+                #Create a Circle annotation (Count Markup)
+                annot = page.add_circle_annot(
+                    fitz.Rect(cor[0] - 10, cor[1] - 10, cor[0] + 10, cor[1] + 10)  # Small circle
+                )
+    
+                #Assign required Bluebeam metadata
+                annot.set_colors(stroke=clr, fill=(1, 1, 1))  # Set stroke color and fill white
+                annot.set_border(width=2)  # Border thickness
+                annot.set_opacity(1)  # Fully visible
+    
+                #Set annotation properties for Bluebeam Count detection
+                annot.set_info("name", lbl)  # Unique name for each count
+                annot.set_info("subject", "Count")  #Bluebeam uses "Count" for Count markups
+                annot.set_info("title", lbl)  # Optional
+                annot.update()  # Apply changes
+    if len(locations[0]) == 5:
+        for loc in locations:
+            coor, lbl, clr,w,h = loc
+            clr = (clr[0] / 255, clr[1] / 255, clr[2] / 255)
+            for cor in coor:
+                #Create a Circle annotation (Count Markup)
+                annot = page.add_circle_annot(
+                    fitz.Rect(cor[0] - 10, cor[1] - 10, cor[0] + 10, cor[1] + 10)  # Small circle
+                )
+    
+                #Assign required Bluebeam metadata
+                annot.set_colors(stroke=clr, fill=(1, 1, 1))  # Set stroke color and fill white
+                annot.set_border(width=2)  # Border thickness
+                annot.set_opacity(1)  # Fully visible
+    
+                #Set annotation properties for Bluebeam Count detection
+                annot.set_info("name", lbl)  # Unique name for each count
+                annot.set_info("subject", "Count")  #Bluebeam uses "Count" for Count markups
+                annot.set_info("title", lbl)  # Optional
+                annot.update()  # Apply changes
 
     #Save modified PDF to a variable instead of a file
     output_stream = io.BytesIO()