Update Counting_Columns_2_1.py

Counting_Columns_2_1.py

@@ -137,8 +137,9 @@ def distance(point1, point2):
 
 def getNearestText(point_list, p):
   nearbyy = []
-  selected_clm_point = [] #save the clmn for drawing cirlce on it 
+  selected_clm_point = [] #save the clmn for drawing cirlce on it
   dis = []
+  txt_clmn = []
   for i in range(len(p)):
     nearest_point = min(point_list, key=lambda point: distance(point, p[i]))
     dist = distance(nearest_point, p[i])
@@ -146,7 +147,8 @@ def getNearestText(point_list, p):
     if dist < 44:
       nearbyy.append(nearest_point)
       selected_clm_point.append(p[i])
-  return nearbyy, selected_clm_point
+      txt_clmn.append((nearest_point, p[i]))
+  return nearbyy, selected_clm_point, txt_clmn
 
 
 def getColumnsTypes(nearbyy, x):
@@ -259,7 +261,7 @@ def mainfun(pdf_name,pdfpath,planname):
     
   if len(column_points) > 10:
       # BROWN COLUMNS
-      nearby, slctd_clm = getNearestText(text_points, column_points)
+      nearby, slctd_clm, txt_clmn = getNearestText(text_points, column_points)
       columns_types_v = getColumnsTypesKeyValue(nearby, txtpts_ky_vlu)
       legend = generate_legend(columns_types_v)
       huge_list_clmn_clr_loc = get_drawing_info(txt_clmn,txtpts_ky_vlu,key_colors)
@@ -270,7 +272,7 @@ def mainfun(pdf_name,pdfpath,planname):
       imgResult = segment_blue(img_blue)
       outsu = threshold(imgResult)
       column_points,mask_clmns, mask_walls = get_columns_info(outsu, img)
-      nearby, slctd_clm = getNearestText(text_points, column_points)
+      nearby, slctd_clm, txt_clmn = getNearestText(text_points, column_points)
       columns_types_v = getColumnsTypesKeyValue(nearby, txtpts_ky_vlu)
       legend = generate_legend(columns_types_v)
       huge_list_clmn_clr_loc = get_drawing_info(txt_clmn,txtpts_ky_vlu,key_colors)