import click

from app.settings import types
from app.utils import click_utils
from app.settings import app_cfg as cfg


@click.command()
@click.option('-i', '--input', 'opt_fp_in', default=None, required=True,
  help='Image filepath')
@click.option('-o', '--output', 'opt_fp_out', default=None,
  help='GIF output path')
@click.option('--size', 'opt_size', 
  type=(int, int), default=(300, 300),
  help='Output image size')
@click.option('-g', '--gpu', 'opt_gpu', default=0,
  help='GPU index')
@click.option('-f', '--force', 'opt_force', is_flag=True,
  help='Force overwrite file')
@click.option('--display/--no-display', 'opt_display', is_flag=True, default=True,
  help='Display detections to debug')
@click.pass_context
def cli(ctx, opt_fp_in, opt_fp_out, opt_gpu, opt_size, opt_force, opt_display):
  """Face detector demo"""
  
  import sys
  import os
  from os.path import join
  from pathlib import Path
  import time
  
  from tqdm import tqdm
  import numpy as np
  import pandas as pd
  import cv2 as cv
  import dlib

  from app.utils import logger_utils, file_utils, im_utils, display_utils, draw_utils
  from app.utils import plot_utils
  from app.processors import face_detector, face_age_gender
  from app.models.data_store import DataStore
  

  log = logger_utils.Logger.getLogger()

  # -------------------------------------------------
  # load image

  im = cv.imread(opt_fp_in)
  im_resized = im_utils.resize(im, width=opt_size[0], height=opt_size[1])

  # ----------------------------------------------------------------------------
  # detect face

  face_detector = face_detector.DetectorCVDNN()
  bboxes = face_detector.detect(im_resized, largest=True)
  bbox_norm = bboxes[0]
  dim = im_resized.shape[:2][::-1]
  bbox_dim = bbox_norm.to_dim(dim)
  if not bbox_norm:
    log.error('no face detected')
    return
  else:
    log.info(f'face detected: {bbox_dim.to_xyxy()}')


  # ----------------------------------------------------------------------------
  # age

  # real
  age_real_predictor = face_age_gender.FaceAgeReal()
  st = time.time()
  age_real = age_real_predictor.predict(im_resized, bbox_norm)
  log.info(f'age real took: {(time.time()-st)/1000:.5f}s')
  
  # apparent
  age_apparent_predictor = face_age_gender.FaceAgeApparent()
  st = time.time()
  age_apparent = age_apparent_predictor.predict(im_resized, bbox_norm)
  log.info(f'age apparent took: {(time.time()-st)/1000:.5f}s')

  # gender
  gender_predictor = face_age_gender.FaceGender()
  st = time.time()
  gender = gender_predictor.predict(im_resized, bbox_norm)
  log.info(f'gender took: {(time.time()-st)/1000:.5f}s')

  # ethnicity 
  # TODO
  
  # ----------------------------------------------------------------------------
  # output
  
  log.info(f'Face coords: {bbox_dim} face')
  log.info(f'Age (real): {(age_real):.2f}')
  log.info(f'Age (apparent): {(age_apparent):.2f}')
  log.info(f'gender: {gender}')


  # ----------------------------------------------------------------------------
  # draw

  # draw real age
  im_age_real = im_resized.copy()
  im_age_real = draw_utils.draw_bbox(im_age_real, bbox_norm)
  txt = f'{(age_real):.2f}'
  im_age_real = draw_utils.draw_text(im_age_real, bbox_norm.pt_tl, txt)

  # apparent age
  im_age_apparent = im_resized.copy()
  im_age_apparent = draw_utils.draw_bbox(im_age_apparent, bbox_norm)
  txt = f'{(age_apparent):.2f}'
  im_age_apparent = draw_utils.draw_text(im_age_apparent, bbox_norm.pt_tl, txt)

  # gender
  im_gender = im_resized.copy()
  im_gender = draw_utils.draw_bbox(im_gender, bbox_norm)
  txt = f"M: {gender['m']:.2f}, F: {gender['f']:.2f}"
  im_gender = draw_utils.draw_text(im_gender, (.1, .9), txt)


  # ----------------------------------------------------------------------------
  # save
  
  if opt_fp_out:
    # save pose only
    fpp_out = Path(opt_fp_out)
    
    fp_out = join(fpp_out.parent, f'{fpp_out.stem}_real{fpp_out.suffix}')
    cv.imwrite(fp_out, im_age_real)

    fp_out = join(fpp_out.parent, f'{fpp_out.stem}_apparent{fpp_out.suffix}')
    cv.imwrite(fp_out, im_age_apparent)

    fp_out = join(fpp_out.parent, f'{fpp_out.stem}_gender{fpp_out.suffix}')
    cv.imwrite(fp_out, im_age_apparent)


  # ----------------------------------------------------------------------------
  # display

  if opt_display:
    # show all images here
    cv.imshow('real', im_age_real)
    cv.imshow('apparent', im_age_apparent)
    cv.imshow('gender', im_gender)
    display_utils.handle_keyboard()