import gradio as gr
from fer import FaceExpressionRecognizer

fer = FaceExpressionRecognizer()

webcam_interface = gr.Interface(
    fer.handle_frame,
    inputs=gr.Image(type='pil', sources=['webcam'], streaming=True, label='Input webcam'),
    outputs=gr.Image(label='Output video'),
    live=True,
    title='Webcam mode',
    description='Created by Czarna Magia AI Student Club',
    theme=gr.themes.Soft(),
)

img_interface = gr.Interface(
    fer.handle_frame,
    inputs=gr.Image(type='pil', sources=['webcam', 'upload'], label='Input image'),
    outputs=gr.Image(label='Output image'),
    title='Image upload mode',
    description='Created by Czarna Magia AI Student Club',
    theme=gr.themes.Soft(),
)

app = gr.TabbedInterface(
    interface_list=[webcam_interface, img_interface],
    tab_names=['Webcam', 'Image upload'],
    title='Face Expression Recognizer',
    theme=gr.themes.Soft(),
)


if __name__ == '__main__':
    app.launch()