Update app.py

app.py

@@ -24,11 +24,23 @@ def midpoint(x1, y1, x2, y2):
     return (x_mid, y_mid)
 
 def draw_mask(img, bounds):
-    draw = ImageDraw.Draw(img)
-    for bound in bounds:
-        p0, p1, p2, p3 = bound[0]
-        draw.polygon((*p0, *p1, *p2, *p3, *p0), fill=1)
-    return img
+    mask = np.zeros(img.shape[:2], dtype="uint8")
+    for box in bounds:
+        
+        x0, y0 = box[0]
+        x1, y1 = box[1] 
+        x2, y2 = box[2]
+        x3, y3 = box[3] 
+        
+        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))
+        
+        cv2.line(mask, (x_mid0, y_mid0), (x_mid1, y_mi1), 255, thickness)
+        img = cv2.inpaint(img, mask, 7, cv2.INPAINT_NS)
+                 
+    return(img)
     
 def masking(img, mask):
     #img = cv.inpaint(img, mask, 3, cv.INPAINT_TELEA)
@@ -50,16 +62,16 @@ def inference(img, lang):
     img_array = np.array(im)
     
     #mask = PIL.Image.open(img.name)
-    mask = Image.new("1", im.size, 0)
-    draw_mask(mask, bounds)
+    #mask = Image.new("L", im.size, 0)
+    draw_mask(img_array, bounds)
     #
     #masking(img_array, mask)
-    img_inpainted = cv.inpaint(img_array, mask, 7, cv.INPAINT_NS)
+    #img_inpainted = cv.inpaint(img_array, mask, 7, cv.INPAINT_NS)
     #remove_text(im, mask, bounds)
     lang = ""
     #cv.imshow('dst', dst)
     im.save('result.jpg')
-    mask.save('mask.png')
+    #mask.save('mask.png')
     
     #img = cv.imread('result.jpg')
     #mask = cv.imread('mask2.png',0)
@@ -71,7 +83,7 @@ def inference(img, lang):
     #img_inpainted = cv2.inpaint(im, mask, 7, cv2.INPAINT_TELEA)
     #cv2.imwrite('dst.png', dst)
     #cv.imwrite('result.jpg',dst)
-    return ['result.jpg', 'mask.png', pd.DataFrame(bounds). iloc[: , 1:2]]
+    return ['result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
 
 title = 'EasyOCR'
 description = 'Gradio demo for EasyOCR. EasyOCR demo supports 80+ languages.To use it, simply upload your image and choose a language from the dropdown menu, or click one of the examples to load them. Read more at the links below.'
@@ -88,7 +100,7 @@ gr.Interface(
     inference,
     [gr.inputs.Image(type='file', label='Input'),gr.inputs.Dropdown(choices, type="value", default="korean", label='language')],
     [gr.outputs.Image(type='file', label='Output'), 
-    gr.outputs.Image(type='file', label='Output'),
+    #gr.outputs.Image(type='file', label='Output'),
     #gr.outputs.Image(type='file', label='Output'), 
     gr.outputs.Dataframe()],
     title=title,