feat: add dynamic example loading based on language selection

app.py

@@ -117,6 +117,8 @@ def load_f5tts(ckpt_path, vocab_path, old=False, fp16=False):
 OmegaConf.register_new_resolver("load_f5tts", load_f5tts)
 
 models_config = OmegaConf.to_object(OmegaConf.load("configs/models.yaml"))
+refs_config = OmegaConf.to_object(OmegaConf.load("configs/refs.yaml"))
+examples_config = OmegaConf.to_object(OmegaConf.load("configs/examples.yaml"))
 
 
 DEFAULT_MODEL_ID = list(models_config.keys())[0]
@@ -322,34 +324,30 @@ with demo:
         ],
         outputs=[audio_output, spectrogram_output],
     )
-    gr.Examples(
-        [
-            [
-                "阿美_秀姑巒",
-                "./ref_wav/E-PV001-0001.wav",
-                "o pakafanaʼ ni akong to pinangan no romiʼad.",
-                "Mafanaʼ kiso a misanoPangcah haw?",
-            ],
-            [
-                "阿美_秀姑巒",
-                "./ref_wav/E-PV001-0001.wav",
-                "o pakafanaʼ ni akong to pinangan no romiʼad.",
-                "Kering sa masoni⌃ to ko pipahanhanan a tatokian, o fe:soc no niyam a tayra i piondoan.",
-            ],
-            [
-                "阿美_秀姑巒",
-                "./ref_wav/cu_practice-0016849.wav",
-                "ano cikasoan to, ano o falangaw to i, malecaday to a matira.",
-                "Pafelien cingra to misapoeneray a falocoʼ, nanay madaʼoc matilid i falocoʼ nira konini.",
-            ],
-        ],
+
+    def get_examples_by_language(language):
+        if language not in examples_config:
+            return []
+        return [
+            [refs_config[ex["ref"]]["wav"], refs_config[ex["ref"]]["text"], ex["text"]]
+            for ex in examples_config[language]
+        ]
+
+    examples = gr.Examples(
+        get_examples_by_language(language.value),
         label="範例",
         inputs=[
-            language,
             ref_audio_input,
             ref_text_input,
             gen_text_input,
         ],
     )
 
+    language.change(
+        lambda language: gr.Dataset(samples=get_examples_by_language(language)),
+        language,
+        examples.dataset,
+    )
+
+
 demo.launch()

configs/examples.yaml

@@ -0,0 +1,7 @@
+阿美_秀姑巒:
+  - ref: 阿美_秀姑巒_1
+    text: Mafanaʼ kiso a misanoPangcah haw?
+  - ref: 阿美_秀姑巒_1
+    text: Kering sa masoni⌃ to ko pipahanhanan a tatokian, o fe:soc no niyam a tayra i piondoan.
+  - ref: 阿美_秀姑巒_2
+    text: Pafelien cingra to misapoeneray a falocoʼ, nanay madaʼoc matilid i falocoʼ nira konini.

configs/refs.yaml

@@ -0,0 +1,7 @@
+阿美_秀姑巒_1:
+  text: o pakafanaʼ ni akong to pinangan no romiʼad.
+  wav: ./assets/ref_wav/E-PV001-0001.wav
+
+阿美_秀姑巒_2:
+  text: ano cikasoan to, ano o falangaw to i, malecaday to a matira.
+  wav: ./assets/ref_wav/cu_practice-0016849.wav