add option to load data locally instead of AWS

inference.py

@@ -216,10 +216,15 @@ def perform_inference(lon, lat, model, config, debug=False):
     labels_key = 'labels_cog.tif'
     field_ids_key = 'field_ids_cog.tif'
 
-    # Generate pre-signed URLs for the GeoTIFF files
-    features_path = generate_presigned_url(bucket_name, features_key)
-    labels_path = generate_presigned_url(bucket_name, labels_key)
-    field_ids_path = generate_presigned_url(bucket_name, field_ids_key)
+    if os.environ.get('USE_LOCAL_DATA') == "True":
+        features_path = "./data/stacked_features_cog.tif"
+        labels_path = "./data/labels_cog.tif"
+        field_ids_path = "./data/field_ids_cog.tif"
+    else:
+        # Generate pre-signed URLs for the GeoTIFF files
+        features_path = generate_presigned_url(bucket_name, features_key)
+        labels_path = generate_presigned_url(bucket_name, labels_key)
+        field_ids_path = generate_presigned_url(bucket_name, field_ids_key)
 
     # Local paths for stats and dataset info
     stats_path = "./data/chips_stats.yaml"

pages/2_Perform_Crop_Classification.py

@@ -2,6 +2,7 @@ import streamlit as st
 import leafmap.foliumap as leafmap
 from transformers import PretrainedConfig
 from folium import Icon
+import os
 
 from messis.messis import Messis
 from inference import perform_inference
@@ -80,14 +81,24 @@ def perform_inference_step():
         )
         st.success("Inference completed!")
 
-    # Add COG
-    presigned_url = generate_presigned_url('messis-demo', 'stacked_features_cog.tif')
-    m.add_cog_layer(
-        url=presigned_url,
-        name="Sentinel-2 Satellite Imagery",
-        bands=selected_bands,
-        rescale=f"{vmin_vmax[selected_band][0]},{vmin_vmax[selected_band][1]}"
-    )
+    # Add Satellite Imagery
+    if os.environ.get("USE_LOCAL_DATA") == "True":
+        m.add_raster(
+            "./data/stacked_features_cog.tif",
+            layer_name="Satellite Image",
+            bands=selected_bands,
+            fit_bounds=True,
+            vmin=vmin_vmax[selected_band][0],
+            vmax=vmin_vmax[selected_band][1],
+        )
+    else:
+        presigned_url = generate_presigned_url('messis-demo', 'stacked_features_cog.tif')
+        m.add_cog_layer(
+            url=presigned_url,
+            name="Sentinel-2 Satellite Imagery",
+            bands=selected_bands,
+            rescale=f"{vmin_vmax[selected_band][0]},{vmin_vmax[selected_band][1]}"
+        )
 
     # Show the POI on the map
     poi_icon = Icon(color="green", prefix="fa", icon="crosshairs")