Update app.py

app.py

@@ -34,24 +34,34 @@ def setup():
 #print('about to setup')
 setup()
 
-def load_lang(langname):
+# display some of the selected corpus
+# (because gradio pagination is currently broken)
+# and reset all filter menus
+def pick_lang(langname):
     if langname=="Icelandic":
         df = datas.ds_i
-        lang_aligner = datas.a_i
+        
     elif langname =="Faroese":
         df = datas.ds_f
-        lang_aligner = datas.a_f
 
     df = df.data.to_pandas()
     df = df.drop(columns=['audio', 'speaker_id','duration'])
-    return (df[:15], lang_aligner) #(df, df[:50])
+    return (df[:15],"all") #(df, df[:50])
+
 
 
-def f1(langname,lang_aligner):
+
+def f1(langname,gender):
     if langname=="Icelandic":
         ds = datas.ds_i
+        lang_aligner = datas.a_i
     elif langname =="Faroese":
         ds = datas.ds_f
+        lang_aligner = datas.a_f
+
+
+    if gender != "all":
+        ds = ds.filter(lambda x: x["gender"].startswith(gender))
 
     maxdat=len(ds)
 
@@ -65,6 +75,7 @@ def f1(langname,lang_aligner):
     return (graph.align_and_graph(sound_path,transcript,lang_aligner),sound_path,rec_info)
 
 
+
 bl = gr.Blocks()
 
 with bl:
@@ -76,7 +87,7 @@ with bl:
     ### 3. Click the button below to view time-aligned prosody information for a random example
     """ )
     with gr.Row():
-        lloadr = gr.Dropdown(["Faroese", "Icelandic"], label="Language")#, info="Loading the dataset takes some time")
+        langmenu = gr.Dropdown(["Faroese", "Icelandic"], label="Language")#, info="Loading the dataset takes some time")
         gr.Markdown(
         """
         Pitch is shown in dark blue and loudness is the light orange line. The pitch estimation, and the time-alignment of words to audio, are completely automated and there will be some inaccuracy.
@@ -84,7 +95,13 @@ with bl:
         """ )
 
     align_func = gr.State()#value=ctcalign.aligner(model_path="carlosdanielhernandezmena/wav2vec2-large-xlsr-53-icelandic-ep10-1000h",model_word_separator = '|',model_blank_token = '[PAD]'))
-    
+
+
+    with gr.Row():
+        gmenu = gr.Dropdown(["all", "f", "m"], label="Gender", value="all")
+        amenu = gr.Dropdown(["all", 0, 1, 2, 3], label="Age", value="0")
+
+
     with gr.Row():
         #invisidata = gr.DataFrame(interactive=False, visible=False)
         databrowser = gr.DataFrame(wrap=True, max_rows=50, interactive=False, overflow_row_behaviour='paginate')
@@ -102,12 +119,16 @@ with bl:
 
     pl1 = gr.Plot()
 
-    btn1.click(f1, [lloadr,align_func], [pl1,audio1,ainfo])
-
-
-
     
-    lloadr.change(load_lang,lloadr,[databrowser,align_func])
+    # when user selects a language,
+    # display some data
+    # and reset all filter menus
+    langmenu.change(pick_lang,langmenu,[databrowser,gmenu])
+    
+    
+    # 
+    btn1.click(f1, [langmenu,gmenu], [pl1,audio1,ainfo])
+
 
 
     gr.Markdown(