Update app.py

app.py

@@ -129,10 +129,17 @@ def postprocessor(prediction, orig_img):
             color = (255, 0, 0)
             thickness = 1
             result_image = cv2.circle(result_image, (x,y), radius=0, color=color, thickness=thickness)
-            result_image = cv2.putText(result_image,text[idx], (x-10,y+10), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (255,0,0), thickness, cv2.LINE_AA)
-    
+            result_image = cv2.putText(result_image,text[idx], (x-10,y+10), cv2.FONT_HERSHEY_SIMPLEX, 0.10, (255,0,0), thickness, cv2.LINE_AA)
+
+    distance = distanceCalculate(points[0], points[1])
     result_image = cv2.line(result_image, points[0], points[1], (0,0,255), 1)
+    result_image = cv2.putText(result_image,distance, (points[0][0] - 10 ,points[0][1] + 10 ), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,0,255), thickness, cv2.LINE_AA)
     return result_image 
+
+def distanceCalculate(p1, p2):
+    """p1 and p2 in format (x1,y1) and (x2,y2) tuples"""
+    dis = ((p2[0] - p1[0]) ** 2 + (p2[1] - p1[1]) ** 2) ** 0.5
+    return dis    
     
 
 def infer(preprocessed):