Update app.py

app.py

@@ -43,14 +43,22 @@ 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([code.upper() for code in country_list])  # Normalize to uppercase
+            # ADD: only add codes of length 2
+            two_digit_codes = [code.upper() for code in country_list if len(code) == 2]
+            country_set.update(two_digit_codes)
         except json.JSONDecodeError:
             pass
     
-    # Create a mapping of country names to ISO codes
-    country_name_to_code = {get_country_name(code, region_df): code for code in country_set}
+    # Create a mapping of {CountryName -> ISO2Code} 
+    # so you can display the name in the selectbox but store the 2-digit code
+    country_name_to_code = {}
+    for code in country_set:
+        name = get_country_name(code, region_df)
+        country_name_to_code[name] = code
+
     return country_name_to_code
 
+
 # Get country name mapping
 client = get_client()
 country_name_mapping = get_country_name_mapping(client, collection_name, region_df)