Update app.py

app.py

@@ -6,19 +6,38 @@ import gradio as gr
 import torch
 import easyocr
 import cv2
+import math
+import numpy as np
 
 torch.hub.download_url_to_file('https://i.pinimg.com/originals/45/d0/30/45d03054e15f4be731781eecba7458a4.jpg', 'korean.png')
+def midpoint(x1, y1, x2, y2):
+    x_mid = int((x1 + x2)/2)
+    y_mid = int((y1 + y2)/2)
+    return (x_mid, y_mid)
 
-def draw_boxes(image, bounds, color='yellow', width=2):
-    draw = ImageDraw.Draw(image)
-    for bound in bounds:
-        p0, p1, p2, p3 = bound[0]
-        draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
-    return image
-def  remove_text(image, bounds):
+def draw_boxes(img, bounds, color='yellow', width=2):
+    draw = ImageDraw.Draw(img)
     for bound in bounds:
         p0, p1, p2, p3 = bound[0]
         
+        #for masking the image
+        x0, y0 = p0[0][1]
+        x1, y1 = p1[0][1]
+        x2, y2 = p2[0][1]
+        x3, y3 = p3[0][1]
+        x_mid0, y_mid0 = midpoint(x1, y1, x2, y2)
+        x_mid1, y_mi1 = midpoint(x0, y0, x3, y3)
+        thickness = int(math.sqrt( (x2 - x1)**2 + (y2 - y1)**2))
+        
+        mask = np.zeros(img.shape[:2], dtype="uint8")
+        cv2.line(mask, (x_mid0, y_mid0), (x_mid1, y_mi1), 255, thickness)
+        masked = cv2.bitwise_and(img, img, mask=mask)
+        
+        
+        
+        #draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
+    return masked
+        
 def inference(img, lang):
     if lang == "english":
         lang = ['en']
@@ -32,7 +51,7 @@ def inference(img, lang):
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
     draw_boxes(im, bounds)
-    remove_text(im, bounds)
+    #remove_text(im, bounds)
     lang = ""
     im.save('result.jpg')
     return ['result.jpg', pd.DataFrame(bounds).iloc[: , 1:]]