Update appStore/region_utils.py

appStore/region_utils.py

@@ -6,11 +6,18 @@ import pandas as pd
 def load_region_data(file_path):
     return pd.read_csv(file_path)
 
+def clean_country_code(code):
+    # Remove non-alphabetical characters and convert to uppercase.
+    return ''.join(filter(str.isalpha, code)).upper()
+    
+
 def get_country_name(alpha2_code, region_df):
     """Map ISO country code to country name (case-insensitive)."""
-    alpha2_code = alpha2_code.upper()  # Ensure the code is in uppercase
-    row = region_df[region_df['alpha-2'] == alpha2_code]
-    return row['name'].values[0] if not row.empty else alpha2_code
+    # Clean the code before processing.
+    code = clean_country_code(alpha2_code)
+    row = region_df[region_df['alpha-2'] == code]
+    return row['name'].values[0] if not row.empty else code
+
 
 def get_regions(region_df):
     """Get unique regions and sub-regions."""