Update helpers/functional.py

helpers/functional.py

@@ -140,16 +140,22 @@ def text_to_image(text, return_centre=False, return_gridcell=False, return_times
 
 def cell_to_image(cell, return_centre=False, return_gridcell=False, return_timestamp=False, source='Sentinel-2 L2A'):
 
-    # AEF is special - it's read live
-    if source == 'AlphaEarth Embeddings':
-        return cell_to_aef(cell, return_centre=return_centre, return_gridcell=return_gridcell, return_timestamp=return_timestamp)
-
-    # 3. translate major-tom cell to row in parquet
-    df = df_dict[source]
+    try:
+        df = df_dict[source]
+    except:
+        # fallback to default (e.g. for metadata for AEF)
+        df = df_dict['Sentinel-2 L2A']
+        
     row = cell2row(cell, df, return_row = True)
-
+    
     if row is not None:
         parquet_url, parquet_row, meta_row = row
+        year = int(meta_row.iloc[0].timestamp[:4])
+
+        # AEF is special - it's read live
+        if source == 'AlphaEarth Embeddings':
+            return cell_to_aef(cell, year, return_centre=return_centre, return_gridcell=return_gridcell, return_timestamp=return_timestamp)
+
 
         try:
             img = row2image(parquet_url, parquet_row)