Update Counting_Columns_2_1.py

Counting_Columns_2_1.py

@@ -171,19 +171,24 @@ def generate_legend(found_tuple):
 def add_annotations_to_pdf(image, pdf_name, slctd_clm, columns_types_v):
     image_width = image.shape[1]
     image_height = image.shape[0]
-
+    rotationOld=page.rotation
+    derotationMatrix=page.derotation_matrix
+    if page.rotation!=0:
+        rotationangle = page.rotation
+        page.set_rotation(0)
     # Create a new PDF document
     pdf_document = fitz.open('pdf',pdf_name)
     page=pdf_document[0]
     for i in range(len(slctd_clm)):
         x, y = slctd_clm[i]
+        p_midpoint = fitz.Point(x, y) * derotationMatrix
         text = columns_types_v[i]
         # Create an annotation (sticky note)
         annot = page.add_text_annot((x, y), text)
         annot.set_border(width=0.2, dashes=(1, 2))  # Optional border styling
         annot.set_colors(stroke=(1, 0, 0), fill=None)  # Set the stroke color to red
         annot.update()
-    page.set_rotation(90)
+    page.set_rotation(rotationOld)
     return pdf_document
 
 def mainfun(pdf_name,pdfpath,planname):