nataliaElv commited on
Commit
28b6676
Β·
1 Parent(s): 8600eb0

Fix lowercase search

Browse files
Files changed (1) hide show
  1. app.py +15 -5
app.py CHANGED
@@ -44,10 +44,13 @@ def create_piechart():
44
 
45
  return fig
46
 
47
- def filter_dataframes(search_term):
48
- filtered_with_lead = languages_with_lead[languages_with_lead['Language'] == search_term]
49
- filtered_without_lead = languages_without_lead[languages_without_lead['Language'] == search_term]
50
- return filtered_without_lead, filtered_with_lead
 
 
 
51
 
52
  def load_demo():
53
  languages_with_lead, languages_without_lead = build_dataframes(get_covered_languages())
@@ -60,7 +63,9 @@ with gr.Blocks() as demo:
60
  gr_piechart = gr.Plot(label="Language Leads")
61
 
62
  search_box = gr.Textbox(type="text", label="Search your language:")
63
- search_button = gr.Button("Search πŸ”Ž")
 
 
64
 
65
  with gr.Tab("Looking for leads!"):
66
  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).")
@@ -78,5 +83,10 @@ with gr.Blocks() as demo:
78
  inputs=search_box,
79
  outputs=[gr_languages_without_lead,gr_languages_with_lead]
80
  )
 
 
 
 
 
81
 
82
  demo.launch()
 
44
 
45
  return fig
46
 
47
+ def filter_dataframes(search_term=None):
48
+ if search_term:
49
+ filtered_with_lead = languages_with_lead[languages_with_lead['Language'].str.lower() == search_term.lower()]
50
+ filtered_without_lead = languages_without_lead[languages_without_lead['Language'].str.lower() == search_term.lower()]
51
+ return filtered_without_lead, filtered_with_lead
52
+ else:
53
+ return languages_without_lead, languages_with_lead
54
 
55
  def load_demo():
56
  languages_with_lead, languages_without_lead = build_dataframes(get_covered_languages())
 
63
  gr_piechart = gr.Plot(label="Language Leads")
64
 
65
  search_box = gr.Textbox(type="text", label="Search your language:")
66
+ with gr.Row():
67
+ search_button = gr.Button("Search πŸ”Ž")
68
+ reset_button = gr.Button("Reset πŸ”")
69
 
70
  with gr.Tab("Looking for leads!"):
71
  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).")
 
83
  inputs=search_box,
84
  outputs=[gr_languages_without_lead,gr_languages_with_lead]
85
  )
86
+ reset_button.click(
87
+ fn=filter_dataframes,
88
+ inputs=None,
89
+ outputs=[gr_languages_without_lead,gr_languages_with_lead]
90
+ )
91
 
92
  demo.launch()