update app.py

app.py

@@ -14,24 +14,40 @@ div.stButton > button:first-child {
     unsafe_allow_html=True,
 )
 
-
-import streamlit as st
-import json
-
-# Load dataset categories and sub-options from JSON file
-with open("datasets.json") as f:
-    sentinel_categories = json.load(f)
+# Load JSON data from file
+with open("sentinel_datasets.json") as f:
+    data = json.load(f)
 
 # Display title
 st.title("Sentinel Dataset Selector")
 
-# First-level selection box for the main category
-main_selection = st.selectbox("Select Sentinel Dataset Category", list(sentinel_categories.keys()))
+# Step 1: Select main dataset category
+main_selection = st.selectbox("Select Sentinel Dataset Category", list(data.keys()))
 
-# Check the main selection and display the corresponding sub-options
+# Step 2: Display sub-options based on main selection
 if main_selection:
-    sub_options = sentinel_categories[main_selection]
-    sub_selection = st.selectbox("Select Specific Dataset", sub_options)
-    
-    # Display the selected sub-option
-    st.write(f"**Selected Dataset ID:** {sub_selection}")
+    sub_options = data[main_selection]["sub_options"]
+    sub_selection = st.selectbox("Select Specific Dataset ID", list(sub_options.keys()))
+
+    # Step 3: Display band selection based on sub-selection with max 3 selections
+    if sub_selection:
+        bands = sub_options[sub_selection]
+        st.write("Select up to 3 bands:")
+        
+        # Create a list to store selected bands
+        selected_bands = []
+        
+        # Display checkboxes for each band with a maximum selection limit of 3
+        for band in bands:
+            if st.checkbox(band, key=band):
+                selected_bands.append(band)
+            # Limit selection to a maximum of 3 bands
+            if len(selected_bands) == 3:
+                break
+        
+        # Show a warning if more than 3 bands are selected
+        if len(selected_bands) > 3:
+            st.warning("You can select a maximum of 3 bands.")
+        
+        # Display selected bands
+        st.write("Selected Bands:", selected_bands)