nataliaElv commited on
Commit
188a77c
·
1 Parent(s): c2af84e

Improved search

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -46,8 +46,13 @@ def create_piechart():
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
 
46
 
47
  def filter_dataframes(search_term=None):
48
  if search_term:
49
+ search_terms = search_term.lower().split(" ")
50
+ filtered_with_lead = languages_with_lead[
51
+ languages_with_lead.apply(lambda row: any(term in str(row['Language']).lower() or term in str(row['Code']).lower() for term in search_terms), axis=1)
52
+ ]
53
+ filtered_without_lead = languages_without_lead[
54
+ languages_without_lead.apply(lambda row: any(term in str(row['Language']).lower() or term in str(row['Code']).lower() for term in search_terms), axis=1)
55
+ ]
56
  return filtered_without_lead, filtered_with_lead
57
  else:
58
  return languages_without_lead, languages_with_lead