add utlis.py

utlis.py

@@ -0,0 +1,75 @@
+
+import re
+import cv2
+
+def read_custum_html(file_name):
+    with open(file_name) as f:
+        return f.read()
+
+def get_lines(result):
+  lines_dict = {}
+  l=0
+  lines_dict[0]=[]
+  cord = result[0][0]
+  x_min, y_min = [int(min(idx)) for idx in zip(*cord)]
+  x_max, y_max = [int(max(idx)) for idx in zip(*cord)]
+  lines_dict[0].append([[x_max, y_min], result[0][1]])
+  y_min_prev = lines_dict[0][0][0][1]
+  for i in range(1, len(result)):
+    cord = result[i][0]
+    x_min, y_min = [int(min(idx)) for idx in zip(*cord)]
+    x_max, y_max = [int(max(idx)) for idx in zip(*cord)]
+    if y_min-y_min_prev<18:
+      lines_dict[l].append([[x_max, y_min], result[i][1]])
+      y_min_prev = y_min
+    else:
+      l= l+1
+      lines_dict[l]=[]
+      lines_dict[l].append([[x_max, y_min], result[i][1]])
+      y_min_prev = y_max
+  return lines_dict
+
+def annotate_image(image, result): 
+  for i in range(len(result)):
+    cord = result[i][0]
+    x_min, y_min = [int(min(idx)) for idx in zip(*cord)]
+    x_max, y_max = [int(max(idx)) for idx in zip(*cord)]
+    cv2.rectangle(image,(x_min,y_min),(x_max,y_max),(0,0,255),1)
+  return image
+
+def arrange_words_in_line(lines_dict):
+  if isinstance(lines_dict, dict):
+    arranged_dict = {}
+    for key, values in lines_dict.items():
+      line = lines_dict[key]
+      sorted_line = sorted(line,key=lambda x:x[0][0], reverse=True)
+      arranged_dict[key] = sorted_line
+    return arranged_dict
+  else:
+    raise TypeError("The arg must be dict of lines")
+
+def get_raw_text(result):
+  lines_dict = get_lines(result)
+  arranged_lines_dict = arrange_words_in_line(lines_dict)
+  text_list = []
+  for i in range(len(arranged_lines_dict.keys())):
+    for j in range (len(arranged_lines_dict[i])):
+      line_text = arranged_lines_dict[i][j][1]
+      text_list.append(line_text)
+    text_list.append('\n')
+    raw_text = ' '.join(text_list)
+    raw_text = replace_en_num(raw_text)
+  return raw_text
+
+def replace_en_num(text):
+   text = re.sub("0", "\u0660", text)
+   text = re.sub("1", "\u0661", text)
+   text = re.sub("2", "\u0662", text)
+   text = re.sub("3", "\u0663", text)
+   text = re.sub("4", "\u0664", text)
+   text = re.sub("5", "\u0665", text)
+   text = re.sub("6", "\u0666", text)
+   text = re.sub("6", "\u0667", text)
+   text = re.sub("8", "\u0668", text)
+   text = re.sub("9", "\u0669", text)
+   return text