| 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) |
|
|
