Update app.py

app.py

@@ -20,6 +20,13 @@ def midpoint(x1, y1, x2, y2):
     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 draw_mask(img, bounds):
     mask = np.zeros(img.shape[:2], dtype="uint8")
     for bound in bounds:
@@ -53,12 +60,14 @@ def inference(img, lang):
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
     img_array = np.array(im)
+    img = draw_box(img_bounds)
     im = draw_mask(img_array, bounds)
     im = Image.fromarray(im, 'RGB')
     lang = ""
+    img.save('box.jpg')
     im.save('result.jpg')
     
-    return ['result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
+    return ['box.jpg', 'result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
 
 title = 'Manga Image Cleaner'
 description = 'Image inpainting and text detection demo with the use of EasyOCR and CV2. To use it, simply upload your image and choose a language from the dropdown menu, or click one of the examples to test the program.'
@@ -75,7 +84,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.Dataframe()],
     title=title,
     description=description,