import gradio as gr
from utilities.constants import FD_DROPDOWN_CHOICES
from controllers.fd import fd_controller
from services.state import app_state


def fd_interface():
    return gr.Interface(
        title="Face Detection",
        fn=lambda model, image, color: fd_controller(
            image, model, color, app_state.get('fw_option')),
        inputs=[
            gr.Dropdown(choices=FD_DROPDOWN_CHOICES,
                        value=FD_DROPDOWN_CHOICES[0], label="Select Model. *Not affect in forwarding mode*"),
            gr.Image(type="pil", label="Input Image"),
            gr.ColorPicker(label="Box Color")
        ],
        outputs=[
            gr.Image(type="pil", label="Output Image"),
            gr.JSON(label="Output Boxes")
        ],
        flagging_options=[]
    )