Update app.py

app.py

@@ -187,14 +187,14 @@ def filter_results(results, country_filter, region_filter, end_year_range, crs_f
             try:
                 parsed_country = json.loads(country.replace("'", '"'))
                 if isinstance(parsed_country, str):
-                    c_list = [parsed_country]
+                    country_list = [parsed_country]
                 else:
-                    c_list = parsed_country
-                c_list = [code.upper() for code in c_list if len(code) == 2]
+                    country_list = parsed_country
+                c_list = [clean_country_code(code) for code in country_list if len(clean_country_code(code)) == 2]
             except json.JSONDecodeError:
                 c_list = []
         else:
-            c_list = [country.strip().upper()]
+            c_list = [clean_country_code(country)]
         crs_key = metadata.get("crs_key", "").strip()
         # Convert crs_key to a string and remove trailing ".0"
         crs_key_clean = re.sub(r'\.0$', '', str(crs_key))
@@ -435,9 +435,10 @@ else:
                         c_list = [country_value]
                     matched_country = []
                     for code in c_list:
-                        if len(code) == 2:
-                            resolved_name = get_country_name(code.upper(), region_df)
-                            if resolved_name.upper() != code.upper():
+                        cleaned = clean_country_code(code)
+                        if len(cleaned) == 2:
+                            resolved_name = get_country_name(cleaned, region_df)
+                            if resolved_name.upper() != cleaned.upper():
                                 matched_country.append(resolved_name)
                     crs_key = metadata.get("crs_key", "").strip()
                     # Convert to string and remove trailing ".0"
@@ -518,9 +519,10 @@ else:
                         c_list = [country_value]
                     matched_country = []
                     for code in c_list:
-                        if len(code) == 2:
-                            resolved_name = get_country_name(code.upper(), region_df)
-                            if resolved_name.upper() != code.upper():
+                        cleaned = clean_country_code(code)
+                        if len(cleaned) == 2:
+                            resolved_name = get_country_name(cleaned, region_df)
+                            if resolved_name.upper() != cleaned.upper():
                                 matched_country.append(resolved_name)
                     crs_key = metadata.get("crs_key", "").strip()
                     # Convert to string and remove trailing ".0"