Update Counting_Columns_2_1.py

Counting_Columns_2_1.py

@@ -168,7 +168,7 @@ def generate_legend(found_tuple):
   df = pd.DataFrame(data.items(), columns=['Column Type', 'Count'])
   return df
 
-def add_annotations_to_pdf(image, pdf_name, slctd_clm, columns_types_v):
+'''def add_annotations_to_pdf(image, pdf_name, slctd_clm, columns_types_v):
     image_width = image.shape[1]
     image_height = image.shape[0]
     # Create a new PDF document
@@ -189,6 +189,36 @@ def add_annotations_to_pdf(image, pdf_name, slctd_clm, columns_types_v):
         annot.set_colors(stroke=(1, 0, 0), fill=None)  # Set the stroke color to red
         annot.update()
     page.set_rotation(rotationOld)
+    return pdf_document'''
+    
+def add_annotations_to_pdf(image, pdf_name, slctd_clm, columns_types_v):
+    image_width = image.shape[1]
+    image_height = image.shape[0]
+    # Create a new PDF document
+    pdf_document = fitz.open('pdf',pdf_name)
+    page=pdf_document[0]
+    rotationOld=page.rotation
+    derotationMatrix=page.derotation_matrix
+    if page.rotation!=0:
+        rotationangle = page.rotation
+        page.set_rotation(0)
+    for i in range(len(slctd_clm)):
+        x, y = slctd_clm[i]
+        p_midpoint = fitz.Point(x, y) * derotationMatrix
+        annot = page.add_circle_annot(
+                fitz.Rect(x - 10, y - 10, x + 10,y + 10)  # Small circle
+            )
+        # ✅ Assign required Bluebeam metadata
+        annot.set_colors(stroke=(1, 0, 0), fill=(1, 1, 1))  # Set stroke color and fill white
+        annot.set_border(width=2)  # Border thickness
+        annot.set_opacity(1)  # Fully visible
+        text = columns_types_v[i]
+        # ✅ Set annotation properties for Bluebeam Count detection
+        annot.set_info("name", text)  # Unique name for each count
+        annot.set_info("subject", "Count")  # ✅ Bluebeam uses "Count" for Count markups
+        annot.set_info("title", text)  # Optional
+        annot.update()  # Apply changes
+    page.set_rotation(rotationOld)
     return pdf_document
 
 def mainfun(pdf_name,pdfpath,planname):