from gradio import AnnotatedImage, Blocks, Button, Column, Dropdown, Examples, Gallery, HTML, Image, Row, SelectData, Tab, Textbox

import db
import deeperface
import os.path

################################################################################


def delete(id):
    db.delete_by_id(id)


def emotions():
    return deeperface.Emotion.labels


def process(path):
    img = deeperface.Image.read(path)

    ############################################################################

    if img.nsfw():
        img.pixelate().write(path)

    ############################################################################

    if db.exists(path):
        id, metadata = db.get(path)
    else:
        metadata = deeperface.Metadata(img)

        id = db.update(path, metadata)

    ############################################################################

    annotations = []
    for face in metadata:
        annotations.extend(img.annotate(face, metadata[face]['emotion']))

    ############################################################################

    verified_paths = []
    for test_id, test_path, test_metadata in db.tuples():
        if test_path != path:
            if deeperface.verify(metadata.representations(),
                                 test_metadata.representations()):
                verified_paths.append((test_path, os.path.basename(test_id)))

    ############################################################################

    return id, (path, annotations), verified_paths


def search(filter='All'):
    filtered_paths = []
    for id, path, metadata in db.tuples():
        if filter == 'All' or filter in metadata.emotions():
            filtered_paths.append((path, os.path.basename(id)))

    return filtered_paths


################################################################################

with Blocks(title='Face recognition and facial attribute analysis') as blocks:
    HTML(
        '<h1><p align="center">Face recognition and facial attribute analysis</p></h1>'
    )
    with Row():
        with Column(scale=3):
            with Row():
                with Column():
                    textbox = Textbox(visible=False)
                    annotated_image = AnnotatedImage(color_map={
                        'face': '#f97316',
                        'emotion': '#f3f4f6'
                    },
                                                     show_legend=False,
                                                     visible=False)
                    button = Button('Delete', visible=False)
                with Column():
                    gallery_1 = Gallery(columns=3,
                                        container=False,
                                        show_download_button=False,
                                        show_share_button=False,
                                        visible=False)
                    html = HTML(visible=False)
        with Column(scale=2):
            with Tab(label='Upload / Camera'):
                image = Image(container=False,
                              sources=['upload', 'webcam'],
                              type='filepath')
                Examples('examples', image)
            with Tab(label='Gallery') as tab:
                dropdown = Dropdown(['All'] + emotions(),
                                    container=False,
                                    filterable=False,
                                    value=0)
                gallery_2 = Gallery(allow_preview=False,
                                    columns=3,
                                    container=False,
                                    show_share_button=False)

    ############################################################################

    def on_button_click(textbox, dropdown):
        if not textbox or not dropdown:
            return AnnotatedImage(), Button(), Gallery(), HTML(), Gallery()

        delete(textbox)
        gallery_2 = search(dropdown)

        return AnnotatedImage(visible=False), Button(visible=False), Gallery(
            visible=False), HTML(visible=False), Gallery(gallery_2,
                                                         selected_index=None)

    button.click(on_button_click, [textbox, dropdown],
                 [annotated_image, button, gallery_1, html, gallery_2],
                 show_progress='hidden')

    ############################################################################


    def on_image_change_or_select(image, dropdown):
        if not image or not dropdown:
            return Textbox(), AnnotatedImage(), Button(), Gallery(), HTML(
            ), Gallery()

        textbox, annotated_image, gallery_1 = process(image)
        gallery_2 = search(dropdown)

        if len(gallery_1) > 1:
            return textbox, AnnotatedImage(
                annotated_image, label=textbox, visible=True
            ), Button(visible=True), Gallery(gallery_1, visible=True), HTML(
                f'<i><p align="center">{len(gallery_1)} Similar Images in Gallery</p></i>',
                visible=True), Gallery(gallery_2, selected_index=None)
        elif len(gallery_1) > 0:
            return textbox, AnnotatedImage(
                annotated_image, label=textbox, visible=True
            ), Button(visible=True), Gallery(gallery_1, visible=True), HTML(
                '<i><p align="center">1 Similar Image in Gallery</p></i>',
                visible=True), Gallery(gallery_2, selected_index=None)
        else:
            return textbox, AnnotatedImage(
                annotated_image, label=textbox,
                visible=True), Button(visible=True), Gallery(
                    visible=False), HTML(visible=False), Gallery(
                        gallery_2, selected_index=None)

    image.change(
        on_image_change_or_select, [image, dropdown],
        [textbox, annotated_image, button, gallery_1, html, gallery_2],
        show_progress='hidden')

    image.select(
        on_image_change_or_select, [image, dropdown],
        [textbox, annotated_image, button, gallery_1, html, gallery_2],
        show_progress='hidden')

    ############################################################################


    def on_tab_select(dropdown):
        if not dropdown:
            return Gallery()

        gallery_2 = search(dropdown)

        return Gallery(gallery_2, selected_index=None)

    tab.select(on_tab_select, dropdown, gallery_2, show_progress='hidden')

    ############################################################################


    def on_dropdown_select_event(event: SelectData):
        dropdown = event.value

        gallery_2 = on_tab_select(dropdown)

        return gallery_2

    dropdown.select(on_dropdown_select_event,
                    outputs=gallery_2,
                    show_progress='hidden')

    ############################################################################


    def on_gallery_2_select_event(event: SelectData, dropdown):
        image = event.value['image']['path']

        textbox, annotated_image, button, gallery_1, html, gallery_2 = on_image_change_or_select(
            image, dropdown)

        return textbox, annotated_image, button, gallery_1, html, gallery_2

    gallery_2.select(
        on_gallery_2_select_event,
        dropdown,
        [textbox, annotated_image, button, gallery_1, html, gallery_2],
        show_progress='hidden')

    ############################################################################

blocks.launch(show_api=False)