Update app.py

app.py

@@ -284,17 +284,19 @@ def get_country_name_and_region_mapping(_client, collection_name, region_df):
     country_set = set()
     for res in results[0] + results[1]:
         country = res.payload.get('metadata', {}).get('country', "[]")
-        try:
-            parsed_country = json.loads(country.replace("'", '"'))
-            # If the parsed result is a string (not a list), wrap it in a list.
-            if isinstance(parsed_country, str):
-                country_list = [parsed_country]
-            else:
-                country_list = parsed_country
-            two_digit_codes = [code.upper() for code in country_list if len(code) == 2]
-            country_set.update(two_digit_codes)
-        except json.JSONDecodeError:
-            pass
+        if country.strip().startswith("["):
+            try:
+                parsed_country = json.loads(country.replace("'", '"'))
+                if isinstance(parsed_country, str):
+                    country_list = [parsed_country]
+                else:
+                    country_list = parsed_country
+            except json.JSONDecodeError:
+                country_list = []
+        else:
+            country_list = [country.strip()]
+        two_digit_codes = [code.upper() for code in country_list if len(code) == 2]
+        country_set.update(two_digit_codes)
     country_name_to_code = {}
     iso_code_to_sub_region = {}
     for code in country_set:
@@ -490,10 +492,14 @@ else:
                     total_volume = metadata.get('total_volume', "Unknown")
                     formatted_project_budget = format_currency(total_project)
                     formatted_total_volume = format_currency(total_volume)
-                    try:
-                        c_list = json.loads(metadata.get('country', "[]").replace("'", '"'))
-                    except json.JSONDecodeError:
-                        c_list = []
+                    country_value = metadata.get('country', "").strip()
+                    if country_value.startswith("["):
+                        try:
+                            c_list = json.loads(country_value.replace("'", '"'))
+                        except json.JSONDecodeError:
+                            c_list = []
+                    else:
+                        c_list = [country_value]
                     matched_country = []
                     for code in c_list:
                         if len(code) == 2: