1 files changed, 80 insertions, 0 deletions
diff --git a/check/commands/phash/report_html.py b/check/commands/phash/report_html.py
new file mode 100644
index 0000000..61a8b48
--- /dev/null
+++ b/check/commands/phash/report_html.py
@@ -0,0 +1,80 @@
+"""
+Generate a test report from a directory of images
+This demo does not use SQL storage
+
+"""
+
+from pathlib import Path
+
+import click
+
+@click.command('')
+@click.option('--csv', 'opt_fp_in_csv', required=True,
+  help='Path to input CSV')
+@click.option('--images', 'opt_fp_in_img', required=True,
+  help='Path to images')
+@click.option('-o', '--output', 'opt_fp_out_dir', type=click.Path(file_okay=False, dir_okay=True),
+  help='Path to output directory')
+@click.pass_context
+def cli(ctx, opt_fp_in_csv, opt_fp_in_img, opt_fp_out_dir):
+  """Generate HTML report"""
+
+  # ------------------------------------------------
+  # imports
+  import sys
+  from os.path import join
+  from glob import glob
+
+  import pandas as pd
+  from tqdm import tqdm
+  import jinja2
+  from flask import url_for
+  import shutil
+
+  from app.utils import logger_utils, im_utils, file_utils
+  
+  log = logger_utils.Logger.getLogger()
+  log.info(f'Generating HTML report from: {opt_fp_in_csv}')
+
+
+  template_loader = jinja2.FileSystemLoader(searchpath="./static/")
+  template_env = jinja2.Environment(loader=template_loader)
+  TEMPLATE_FILE = "perceptual_hash_report.html"
+  template = template_env.get_template(TEMPLATE_FILE)
+
+  # create project output dir
+  fp_out_dir_assets = join(opt_fp_out_dir, 'assets')
+  fp_out_dir_images = join(opt_fp_out_dir, 'images')
+  
+  file_utils.ensure_dir(opt_fp_out_dir)
+  file_utils.ensure_dir(fp_out_dir_assets)
+  file_utils.ensure_dir(fp_out_dir_images)
+  
+  df_dupes = pd.read_csv(opt_fp_in_csv)
+  image_groups = df_dupes.groupby('fname_a')
+
+  log.info(f'Saving HTML report to: {opt_fp_out_dir}')
+  # im_objs = df_dupes.to_records('dict')
+  fp_out_html = join(opt_fp_out_dir, 'index.html')
+  with open(fp_out_html, 'w') as fp:
+    html_text = template.render(image_groups=image_groups, 
+      dir_ims=Path(fp_out_dir_images).name, dir_assets=Path(fp_out_dir_assets).name)
+    fp.write(html_text)
+  
+  # copy css
+  fp_src = 'static/assets/css.css'
+  fp_dst = join(fp_out_dir_assets, Path(fp_src).name)
+  shutil.copy(fp_src, fp_dst)
+
+  # copy images
+  for fname_a, image_group in image_groups:
+    # get image a
+    for df_im in image_group.itertuples():
+      # image a
+      fp_src = join(opt_fp_in_img, df_im.fname_a)
+      fp_dst = join(fp_out_dir_images, df_im.fname_a)
+      shutil.copy(fp_src, fp_dst)
+      # image b
+      fp_src = join(opt_fp_in_img, df_im.fname_b)
+      fp_dst = join(fp_out_dir_images, df_im.fname_b)
+      shutil.copy(fp_src, fp_dst)
+\ No newline at end of file