Update Counting_Columns_2_1.py

Counting_Columns_2_1.py

@@ -167,6 +167,23 @@ 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):
+    image_width, image_height = image.size
+
+    # 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]
+        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(rotationOld)
+    return pdf_document
+
 def mainfun(plan):
   pdf_document = fitz.open(plan)
   page = pdf_document[0]
@@ -199,7 +216,9 @@ def mainfun(plan):
       nearby, slctd_clm = getNearestText(text_points, column_points)
       columns_types_v = getColumnsTypesKeyValue(nearby, txtpts_ky_vlu)
       legend = generate_legend(columns_types_v)
-  return legend, slctd_clm
+  pdf_document = add_annotations_to_pdf(image, pdf_name, slctd_clm, columns_types_v)
+
+  return legend
 
 '''def mainfun(plan):
   texts_from_pdf = get_text_from_pdf(plan)