Update app.py

app.py

@@ -168,43 +168,31 @@ collection_name = "giz_worldwide"
 ################### Hybrid Search #####################################################
 client = get_client()
 print(client.get_collections())
-
-# Get the maximum end_year across the entire collection
 max_end_year = get_max_end_year(client, collection_name)
-
-# Get all unique sub-regions
 _, unique_sub_regions = get_regions(region_df)
 
-# Fetch unique country codes and map to country names
 @st.cache_data
 def get_country_name_and_region_mapping(_client, collection_name, region_df):
     results = hybrid_search(_client, "", collection_name)
     country_set = set()
-    
     for res in results[0] + results[1]:
         countries = res.payload.get('metadata', {}).get('countries', "[]")
         try:
             country_list = json.loads(countries.replace("'", '"'))
-            # Only add codes of length 2
             two_digit_codes = [code.upper() for code in country_list if len(code) == 2]
             country_set.update(two_digit_codes)
         except json.JSONDecodeError:
             pass
-
-    # Create a mapping of {CountryName -> ISO2Code} and {ISO2Code -> SubRegion}
     country_name_to_code = {}
     iso_code_to_sub_region = {}
-
     for code in country_set:
         name = get_country_name(code, region_df)
         sub_region_row = region_df[region_df['alpha-2'] == code]
         sub_region = sub_region_row['sub-region'].values[0] if not sub_region_row.empty else "Not allocated"
         country_name_to_code[name] = code
         iso_code_to_sub_region[code] = sub_region
-
     return country_name_to_code, iso_code_to_sub_region
 
-# Get country name and region mappings
 client = get_client()
 country_name_mapping, iso_code_to_sub_region = get_country_name_and_region_mapping(client, collection_name, region_df)
 unique_country_names = sorted(country_name_mapping.keys())  # List of country names