Update draw_bboxes.py

draw_bboxes.py

@@ -13,9 +13,8 @@ def is_overlapping(rect1, rect2):
     return not (x2 < x3 or x1 > x4 or y2 < y3 or y1 > y4)
 
 
-def draw_entity_boxes_on_image(image, entities):
+def draw_entity_boxes_on_image(image, entities, show=False, save_path=None):
     """_summary_
-
     Args:
         image (_type_): image or image path
         collect_entity_location (_type_): _description_
@@ -112,6 +111,9 @@ def draw_entity_boxes_on_image(image, entities):
             previous_bboxes.append((text_bg_x1, text_bg_y1, text_bg_x2, text_bg_y2))
 
     pil_image = Image.fromarray(new_image[:, :, [2, 1, 0]])
-    pil_image.save("new_snowman.jpg")
+    if save_path:
+        pil_image.save(save_path)
+    if show:
+        pil_image.show()
 
-    return new_image
+    return new_image