added netherlands and UK

app.py

@@ -13,19 +13,22 @@ AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
 AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
 
 # Predefined list of datasets
-datasets = ["WayveScenes", "MajorTom-Germany"]
+datasets = ["WayveScenes", "MajorTom-Germany", "MajorTom-Netherlands"]
 description = {
     "WayveScenes": "A large-scale dataset featuring diverse urban driving scenes, captured from vehicles to advance AI perception and navigation in complex environments.",
-    "MajorTom-Germany": "A geospatial dataset containing satellite imagery from across Germany, designed for tasks like land-use classification, environmental monitoring, and earth observation analytics."
+    "MajorTom-Germany": "A geospatial dataset containing satellite imagery from across Germany, designed for tasks like land-use classification, environmental monitoring, and earth observation analytics.",
+    "MajorTom-Netherlands": "A geospatial dataset containing satellite imagery from across Netherlands, designed for tasks like land-use classification, environmental monitoring, and earth observation analytics.",
+    "MajorTom-UK" :"A geospatial dataset containing satellite imagery from across the United Kingdom, designed for tasks like land-use classification, environmental monitoring, and earth observation analytics."
 }
 selection = {
     'WayveScenes': [1, 8],
-    "MajorTom-Germany": [1, 1]
+    "MajorTom-Germany": [1, 1],
+    "MajorTom-Netherlands": [1,1],
+    "MajorTom-UK": [1,1]
 }
-folder_path_dict = {
-        "WayveScenes" : 'WayveScenes/',
-        "MajorTom-Germany": "MajorTom-Europe/"
-    }
+
+
+
 # AWS S3 bucket name
 bucket_name = "datasets-quasara-io"
 
@@ -47,8 +50,11 @@ def main():
 
     # Select dataset from dropdown
     dataset_name = st.selectbox("Select Dataset", datasets)
-
-    folder_path = folder_path_dict[dataset_name]
+    #For Loading from Box
+    if 'MajorTom' in dataset_name:
+        folder_path = MajorTom-Europe/
+    else:
+        folder_path = f'{dataset_name}/'
 
     st.caption(description[dataset_name])
 
@@ -62,12 +68,12 @@ def main():
         st.text("Small Object Search Disabled")
         st.session_state.dataset_number = st.selectbox("Select Subset of Data", list(range(1, selection[dataset_name][0] + 1)))
         st.text(f"You have selected Main Dataset {st.session_state.dataset_number}")
-    if 'MajorTom' in dataset_name:
+    if 'MajorTom' in dataset_name and st.session_state.search_in_small_objects == False:
         df, total_rows = load_dataset_with_limit(dataset_name, st.session_state.dataset_number, st.session_state.search_in_small_objects, limit=1)
         dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=total_rows, value=1500)
         st.text(f'The smaller the dataset the faster the search will work.')
     else:
-        dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=30000, value=10000)
+        dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=100000, value=30000)
         st.text(f'The smaller the dataset the faster the search will work.')
 
     # Load dataset with limit only if not already loaded