Update Counting_Columns_2_1.py

Counting_Columns_2_1.py

@@ -148,6 +148,9 @@ def mainfun(plan):
     # BROWN COLUMNS
     text_points = getTextsPoints(texts_from_pdf)
     nearby = getNearestText(text_points, column_points)
+    if rotation != 0:
+      if rotation ==90:
+        nearby = fix_rotation_90(pc_coordinates)
     columns_types = getColumnsTypes(nearby, texts_from_pdf)
     legend = generate_legend(columns_types)
   else:
@@ -158,6 +161,9 @@ def mainfun(plan):
     column_points,mask_clmns, mask_walls = get_columns_info(outsu, img)
     text_points = getTextsPoints(texts_from_pdf)
     nearby = getNearestText(text_points, column_points)
+    if rotation != 0:
+      if rotation ==90:
+        nearby = fix_rotation_90(pc_coordinates)
     columns_types = getColumnsTypes(nearby, texts_from_pdf)
     legend = generate_legend(columns_types)
   return legend