import click

from app.settings import types
from app.models.dataset import Dataset
from app.utils import click_utils
from app.settings import app_cfg as cfg
from app.utils.logger_utils import Logger

@click.command()
@click.option('--data_store', 'opt_data_store',
  type=cfg.DataStoreVar,
  default=click_utils.get_default(types.DataStore.NAS),
  show_default=True,
  help=click_utils.show_help(types.Dataset))
@click.option('--dataset', 'opt_dataset',
  type=cfg.DatasetVar,
  required=True,
  show_default=True,
  help=click_utils.show_help(types.Dataset))
@click.pass_context
def cli(ctx, opt_index, opt_data_store, opt_dataset):
  """Display image info"""
  
  import sys
  from glob import glob
  from os.path import join
  from pathlib import Path
  import time

  import pandas as pd
  import cv2 as cv
  from tqdm import tqdm
  
  from app.utils import file_utils, im_utils, path_utils

  log = Logger.getLogger()

  dataset = Dataset(opt_dataset).load(opt_data_store)
  # find image records
  image_record = dataset.roi_idx_to_record(opt_index)
  # debug
  image_record.summarize()
  # load image
  fp_im = image_record.filepath
  im = cv.imread(fp_im)
  # display
  cv.imshow('', im)
  # cv gui
  while True:
    k = cv.waitKey(1) & 0xFF
    if k == 27 or k == ord('q'):  # ESC
      cv.destroyAllWindows()
      sys.exit()
    elif k != 255:
      # any key to continue
      break