readd the feature to add the filename to the top of the output image

Author: SeriousGuy888

main.py

@@ -18,7 +18,7 @@
 
 FONT = ImageFont.truetype("./fonts/UbuntuMono-Regular.ttf", 24)
 
-# ENABLE_INFO_ON_IMAGE = False
+ENABLE_INFO_ON_IMAGE = True
 
 
 def main():
@@ -52,8 +52,8 @@ def main():
 
     img = Image.open(img_path)
     img = crop_img(img, top_left, bottom_right)
-    # if ENABLE_INFO_ON_IMAGE:
-    #   img = add_img_info(img, file_name)
+    if ENABLE_INFO_ON_IMAGE:
+      img = add_img_info(img, file_name)
 
     if not os.path.exists(OUTPUT_DIR):
       os.mkdir(OUTPUT_DIR)
@@ -117,20 +117,20 @@ def get_template() -> cv.Mat:
   return cv.imread(template_path, 0)
 
 
-# def add_img_info(img: Image, info_text: str, info_section_height: int = 35) -> Image:
-#   """
-#   Add a section at the top of the image with space to add some
-#   extra text.
-#   """
+def add_img_info(img: Image, info_text: str, info_section_height: int = 35) -> Image:
+  """
+  Add a section at the top of the image with space to add some
+  extra text.
+  """
 
-#   dimensions = (img.width, img.height + info_section_height)
-#   new_img = Image.new("RGBA", dimensions, "black")
-#   new_img.paste(img, (0, info_section_height))
+  dimensions = (img.width, img.height + info_section_height)
+  new_img = Image.new("RGBA", dimensions, "black")
+  new_img.paste(img, (0, info_section_height))
 
-#   d = ImageDraw.Draw(new_img)
-#   d.text((10, 5), info_text, font=FONT, fill="lightgray")
+  d = ImageDraw.Draw(new_img)
+  d.text((10, 5), info_text, font=FONT, fill="lightgray")
 
-#   return new_img
+  return new_img
 
 
 def crop_img(img: Image, top_left: tuple[int, int], bottom_right: tuple[int, int]) -> Image: