Update app.py

app.py

@@ -18,10 +18,16 @@ def midpoint(x1, y1, x2, y2):
 
 def draw_mask(img, bounds):
     draw = ImageDraw.Draw(img)
+    
+    for bound in bounds:
+        p0, p1, p2, p3 = bound[0]
+        draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
+        
+    return imag
+    """
     for bound in bounds:
         p0, p1, p2, p3 = bound[0]
         
-        """
         x0,y0 = p0
         x1,y1 = p1
         x2,y2 = p2
@@ -34,14 +40,12 @@ def draw_mask(img, bounds):
         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
-"""    
+    return img   
 def remove_text(img,imgwmask):
     masked = cv2.bitwise_and(img, img, mask=imgwmask)
     return img
@@ -58,7 +62,7 @@ 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)
+    #mask = PIL.Image.new("L", im1.size, 0)
     #w, h = im.size 
     
     draw_mask(im, bounds)
@@ -66,7 +70,7 @@ def inference(img, lang):
     lang = ""
     im.save('result.png')
     #mask.save('mask.png')
-    return ['result.png', 'mask.png', pd.DataFrame(bounds). iloc[: , 1:2]]
+    return ['result.png', 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.'
@@ -82,7 +86,7 @@ choices = [
 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.Dataframe()],
+    [gr.outputs.Image(type='file', label='Output'), gr.outputs.Dataframe()],
     title=title,
     description=description,
     article=article,