Update helpers/functional.py

helpers/functional.py

@@ -27,15 +27,23 @@ else:
     DATASET_NAME = 'Major-TOM/Core-S1RTC'
     rtc_meta_path = 'https://huggingface.co/datasets/{}/resolve/main/metadata.parquet'.format(DATASET_NAME)
 
+if os.path.isfile('helpers/dem_metadata.parquet'):
+    dem_meta_path = 'helpers/dem_metadata.parquet'
+else:
+    DATASET_NAME = 'Major-TOM/Core-DEM'
+    dem_meta_path = 'https://huggingface.co/datasets/{}/resolve/main/metadata.parquet'.format(DATASET_NAME)
+
 grid = Grid(10, latitude_range=(-90,90), longitude_range=(-180,180))
 l2a_df = pd.read_parquet(l2a_meta_path)
 l1c_df = pd.read_parquet(l1c_meta_path)
 rtc_df = pd.read_parquet(rtc_meta_path)
+dem_df = pd.read_parquet(dem_meta_path)
 
 df_dict = {
-    'S2-L2A' : l2a_df,
-    'S2-L1C' : l1c_df,
-    'S1-RTC' : rtc_df
+    'Sentinel-2 L2A' : l2a_df,
+    'Sentinel-2 L1C' : l1c_df,
+    'Sentinel-1 RTC' : rtc_df,
+    'COP-DEM' : dem_df
 }
 
 def pretty_date(input):
@@ -84,7 +92,7 @@ def cell2row(grid_string, meta_df, return_row = False):
     else:
         return None
 
-def map_to_image(map, return_centre=False, return_gridcell=False, return_timestamp=False, source='S2-L2A'):
+def map_to_image(map, return_centre=False, return_gridcell=False, return_timestamp=False, source='Sentinel-2 L2A'):
 
     try:
         # 1. get bounds