Search your language

app.py

@@ -44,18 +44,39 @@ def create_piechart():
 
     return fig
 
+def filter_dataframes(search_term):
+    filtered_with_lead = languages_with_lead[languages_with_lead['Language'] == search_term]
+    filtered_without_lead = languages_without_lead[languages_without_lead['Language'] == search_term]
+    return filtered_without_lead, filtered_with_lead
+
+def load_demo():
+    languages_with_lead, languages_without_lead = build_dataframes(get_covered_languages())
+    piechart = create_piechart()
+    return languages_without_lead,languages_with_lead,piechart
 
 with gr.Blocks() as demo:
     gr.Markdown("## Language Leads Dashboard")
     languages_with_lead, languages_without_lead = build_dataframes(get_covered_languages())
-    with gr.Row():
-        piechart = create_piechart()
-        gr.Plot(value=piechart, label="Language Leads")
+    gr_piechart = gr.Plot(label="Language Leads")
+
+    search_box = gr.Textbox(type="text", label="Search your language:")
+    search_button = gr.Button("Search 🔎")
+    
     with gr.Tab("Looking for leads!"):
         gr.Markdown("These languages don't have a lead yet! Would you like to lead one of them? Sign up using [this form](https://forms.gle/mFCMXNRjxvyFvW5q9).")
-        gr.DataFrame(languages_without_lead)
+        gr_languages_without_lead = gr.DataFrame()
     with gr.Tab("Languages with leads"):
         gr.Markdown("We found at least one lead for these languages:")
-        gr.DataFrame(languages_with_lead)
+        gr_languages_with_lead = gr.DataFrame()
+
+    demo.load(
+        load_demo,
+        outputs=[gr_languages_without_lead, gr_languages_with_lead, gr_piechart],
+    )
+    search_button.click(
+        fn=filter_dataframes,
+        inputs=search_box,
+        outputs=[gr_languages_without_lead,gr_languages_with_lead]
+    )
 
 demo.launch()