Update app.py

app.py

@@ -21,35 +21,10 @@ def draw_mask(img, bounds):
     
     for bound in bounds:
         p0, p1, p2, p3 = bound[0]
-        draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
+        draw.polygon((*p0, *p1, *p2, *p3, *p0), fill=255)
         
-    return imag
-    """
-    for bound in bounds:
-        p0, p1, p2, p3 = bound[0]
-        
-        x0,y0 = p0
-        x1,y1 = p1
-        x2,y2 = p2
-        x3,y3 = p3
-        
-        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)
-        img = cv2.inpaint(img, mask, 7, cv2.INPAINT_NS)
-
-        draw.polygon((*p0, *p1, *p2, *p3), fill=255) 
-    
-    #masking
-       
-    return img   
-def remove_text(img,imgwmask):
-    masked = cv2.bitwise_and(img, img, mask=imgwmask)
     return img
-"""        
+       
 def inference(img, lang):
     if lang == "english":
         lang = ['en']
@@ -62,10 +37,10 @@ def inference(img, lang):
     reader = easyocr.Reader(lang)
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
-    #mask = PIL.Image.new("L", im1.size, 0)
-    #w, h = im.size 
+    mask = PIL.Image.new("L", im1.size, 0)
+    w, h = im.size 
     
-    #draw_mask(im, bounds)
+    draw_mask(mask, bounds)
     #remove_text(im, mask, bounds)
     lang = ""
     im.save('result.jpg')