Update app.py

app.py

@@ -6,7 +6,7 @@ from appStore.prep_data import process_giz_worldwide, remove_duplicates, get_max
 from appStore.prep_utils import create_documents, get_client
 from appStore.embed import hybrid_embed_chunks
 from appStore.search import hybrid_search
-from appStore.region_utils import load_region_data, get_country_name, get_regions
+from appStore.region_utils import load_region_data, clean_country_code, get_country_name, get_regions
 #from appStore.tfidf_extraction import extract_top_keywords  # TF-IDF part commented out
 from torch import cuda
 import json
@@ -295,7 +295,7 @@ def get_country_name_and_region_mapping(_client, collection_name, region_df):
                 country_list = []
         else:
             country_list = [country.strip()]
-        two_digit_codes = [code.upper() for code in country_list if len(code) == 2]
+        two_digit_codes = [clean_country_code(code) for code in country_list if len(clean_country_code(code)) == 2]
         country_set.update(two_digit_codes)
     country_name_to_code = {}
     iso_code_to_sub_region = {}