Update app.py

app.py

@@ -43,7 +43,7 @@ def get_country_name_mapping(_client, collection_name, region_df):
         countries = res.payload.get('metadata', {}).get('countries', "[]")
         try:
             country_list = json.loads(countries.replace("'", '"'))
-            country_set.update(country_list)
+            country_set.update([code.upper() for code in country_list])  # Normalize to uppercase
         except json.JSONDecodeError:
             pass
     
@@ -71,23 +71,23 @@ button = st.button("Search")
 if button:
     results = hybrid_search(client, var, collection_name)
 
-    # Filter results based on the user's input
     def filter_results(results, country_filter, end_year_range):
         filtered = []
         for res in results:
             metadata = res.payload.get('metadata', {})
             countries = metadata.get('countries', "[]")
             end_year = float(metadata.get('end_year', 0))
-
+    
             # Process countries string to a list
             try:
                 country_list = json.loads(countries.replace("'", '"'))
+                country_list = [code.upper() for code in country_list]  # Convert to uppercase
             except json.JSONDecodeError:
                 country_list = []
-
+    
             # Translate selected country name back to ISO code
             selected_iso_code = country_name_mapping.get(country_filter, None)
-
+    
             # Apply country and year filters
             if (country_filter == "All" or selected_iso_code in country_list) and (end_year_range[0] <= end_year <= end_year_range[1]):
                 filtered.append(res)