import os

import gradio as gr

from detection import EyesDetection, FaceDetection, HeadDetection, PersonDetection, HandDetection, CensorDetection, \
    HalfBodyDetection

_GLOBAL_CSS = """
.limit-height {
    max-height: 55vh;
}
"""

if __name__ == '__main__':
    with gr.Blocks(css=_GLOBAL_CSS) as demo:
        with gr.Tabs():
            with gr.Tab('Face Detection'):
                FaceDetection().make_ui()
            with gr.Tab('Head Detection'):
                HeadDetection().make_ui()
            with gr.Tab('Person Detection'):
                PersonDetection().make_ui()
            with gr.Tab('Half Body Detection'):
                HalfBodyDetection().make_ui()
            with gr.Tab('Eyes Detection'):
                EyesDetection().make_ui()
            with gr.Tab('Hand Detection'):
                HandDetection().make_ui()
            with gr.Tab('Censor Point Detection'):
                CensorDetection().make_ui()
            with gr.Tab('Manbits Detection\n(Deprecated)'):
                pass

    demo.queue(os.cpu_count()).launch()