import gradio as gr

from src.nlp_circle_demo.interface import GradioInterfaceWrapper, GradioTabWrapper


robertaGer = GradioInterfaceWrapper.from_yaml(
    "resources/legal_german_roberta_interface.yml"
)
gbert = GradioInterfaceWrapper.from_yaml("resources/gbert_interface.yml")

legalInterface = GradioTabWrapper.from_interface_list("Legal", [robertaGer, gbert])


ner = GradioInterfaceWrapper.from_yaml("resources/ner_interface.yml")

zeroShot = GradioInterfaceWrapper.from_yaml("resources/zero_shot_interface.yml")

qaInterface = GradioInterfaceWrapper.from_yaml("resources/qa_interface.yml")
simplicationInterface = GradioInterfaceWrapper.from_yaml(
    "resources/simplification_interface.yml"
)
gptInterface = GradioInterfaceWrapper.from_yaml("resources/gpt2_interface.yml")
summarizationInterface = GradioInterfaceWrapper.from_yaml(
    "resources/summarization_interface.yml"
)


demo = GradioTabWrapper.from_interface_list(
    "NLP Demo-Seite",
    [
        gptInterface,
        legalInterface,
        qaInterface,
        summarizationInterface,
        simplicationInterface,
        ner,
        zeroShot,
    ],
)
demo.launch()