import os
from os.path import join
import collections
import logging

from dotenv import load_dotenv
import yaml

from app.models import types
from pathlib import Path

import codecs
codecs.register(lambda name: codecs.lookup('utf8') if name == 'utf8mb4' else None)

# -----------------------------------------------------------------------------
# Click config
# -----------------------------------------------------------------------------

CLICK_GROUPS = {
  'process': 'commands/process',
}

# -----------------------------------------------------------------------------
# File I/O
# -----------------------------------------------------------------------------

SELF_CWD = os.path.dirname(os.path.realpath(__file__))  # Script CWD
DIR_APP = str(Path(SELF_CWD).parent.parent.parent)
DIR_IMAGENET = join(DIR_APP, 'data_store/imagenet')
DIR_OUTPUTS = join(DIR_APP, 'data_store/outputs')
FP_MODELZOO = join(DIR_APP, 'modelzoo/modelzoo.yaml')

# -----------------------------------------------------------------------------
# Model config
# -----------------------------------------------------------------------------

with open(FP_MODELZOO, 'r') as fp:
  MODELZOO_CFG = yaml.load(fp, Loader=yaml.Loader)

# -----------------------------------------------------------------------------
# Imagenet
# -----------------------------------------------------------------------------

IMAGENET_IMAGES_PER_CLASS = 200
FP_IMAGENET_WORDS = join(DIR_IMAGENET, 'words.txt')
FP_IMAGENET_CLASSES = join(DIR_IMAGENET, 'classes_in_imagenet.csv')