created new generic class WrappedGradioObjects, and GradioTabWrapper

app.py

@@ -1,44 +1,32 @@
 import gradio as gr
 
-from src.nlp_circle_demo.interface import GradioElementWrapper
+from src.nlp_circle_demo.interface import GradioInterfaceWrapper, GradioTabWrapper
 
-GradioElementWrapper.interface_from_yaml("resources/qa_interface.yml")
 
-
-robertaGer = GradioElementWrapper.interface_from_yaml(
+robertaGer = GradioInterfaceWrapper.from_yaml(
     "resources/legal_german_roberta_interface.yml"
-).gradio_element
-gbert = GradioElementWrapper.interface_from_yaml(
-    "resources/gbert_interface.yml"
-).gradio_element
-
-ner = GradioElementWrapper.interface_from_yaml(
-    "resources/ner_interface.yml"
-).gradio_element
+)
+gbert = GradioInterfaceWrapper.from_yaml("resources/gbert_interface.yml")
 
+legalInterface = GradioTabWrapper.from_interface_list("Legal", [robertaGer, gbert])
 
-zeroShot = GradioElementWrapper.interface_from_yaml(
-    "resources/zero_shot_interface.yml"
-).gradio_element
 
-legalInterface = gr.TabbedInterface([robertaGer, gbert], ["Roberta Legal", "Bert"])
+ner = GradioInterfaceWrapper.from_yaml("resources/ner_interface.yml")
 
+zeroShot = GradioInterfaceWrapper.from_yaml("resources/zero_shot_interface.yml")
 
-qaInterface = GradioElementWrapper.interface_from_yaml(
-    "resources/qa_interface.yml"
-).gradio_element
-simplicationInterface = GradioElementWrapper.interface_from_yaml(
+qaInterface = GradioInterfaceWrapper.from_yaml("resources/qa_interface.yml")
+simplicationInterface = GradioInterfaceWrapper.from_yaml(
     "resources/simplification_interface.yml"
-).gradio_element
-gptInterface = GradioElementWrapper.interface_from_yaml(
-    "resources/gpt2_interface.yml"
-).gradio_element
-summarizationInterface = GradioElementWrapper.interface_from_yaml(
+)
+gptInterface = GradioInterfaceWrapper.from_yaml("resources/gpt2_interface.yml")
+summarizationInterface = GradioInterfaceWrapper.from_yaml(
     "resources/summarization_interface.yml"
-).gradio_element
+)
 
 
-demo = gr.TabbedInterface(
+demo = GradioTabWrapper.from_interface_list(
+    "NLP Demo-Seite",
     [
         gptInterface,
         legalInterface,
@@ -48,15 +36,5 @@ demo = gr.TabbedInterface(
         ner,
         zeroShot,
     ],
-    [
-        "GPT",
-        "Legal",
-        "Question Answering",
-        "Summarization",
-        "Simplification",
-        "Named Entity Recognition",
-        "Zero-Shot-Klassifizierung",
-    ],
 )
-
 demo.launch()

src/nlp_circle_demo/interface.py

@@ -1,15 +1,19 @@
 import gradio as gr
-import json
 import yaml
 
 
-class GradioElementWrapper:
+class WrappedGradioObject:
     def __init__(self, title, gradio_element):
         self.title = title
         self.gradio_element = gradio_element
 
+    def launch(self):
+        return self.gradio_element.launch()
+
+
+class GradioInterfaceWrapper(WrappedGradioObject):
     @classmethod
-    def interface_from_yaml(cls, path):
+    def from_yaml(cls, path):
         """Initializes Interface from YAML file."""
         with open(path) as f:
             content_dict = yaml.safe_load(f)
@@ -19,16 +23,31 @@ class GradioElementWrapper:
     def create_interface(cls, name, title, description, examples=None):
         """Creates Gradio-Element containing an interface."""
         description = cls._prepend_link_to_description(name, title, description)
-        element = gr.load(
+        interface = gr.load(
             name,
             title=None,  # Having the Tab-Name is sufficient.
             description=description,
             examples=examples,
         )
-        return cls(title, element)
+        return cls(title, interface)
 
     @staticmethod
     def _prepend_link_to_description(name, title, description):
         without_huggingface = name.removeprefix("huggingface/")
         link = f"https://huggingface.co/{without_huggingface}"
         return f'<a href="{link}">{title}</a> </br> {description}'
+
+
+class GradioTabWrapper(WrappedGradioObject):
+    @classmethod
+    def from_interface_list(cls, title, gradio_objects):
+        """Constructs a GradioTabWrapper from a title and a list of WrappedGradioObjects."""
+        interface = gr.TabbedInterface(
+            [obj.gradio_element for obj in gradio_objects],
+            [obj.title for obj in gradio_objects],
+        )
+        return cls(title, interface)
+
+    @classmethod
+    def from_yaml(cls, path):
+        pass