Update app.py

app.py

@@ -183,9 +183,13 @@ def filter_results(results, country_filter, region_filter, end_year_range, crs_f
                 end_year_val = 0
         else:
             end_year_val = 0
-
         try:
-            c_list = json.loads(country.replace("'", '"'))
+            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):
+                c_list = [parsed_country]
+            else:
+                c_list = parsed_country
             c_list = [code.upper() for code in c_list if len(code) == 2]
         except json.JSONDecodeError:
             c_list = []
@@ -286,7 +290,12 @@ def get_country_name_and_region_mapping(_client, collection_name, region_df):
     for res in results[0] + results[1]:
         country = res.payload.get('metadata', {}).get('country', "[]")
         try:
-            country_list = json.loads(country.replace("'", '"'))
+            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: