update app.py

app.py

@@ -55,23 +55,127 @@ with open(os.path.expanduser("~/.config/earthengine/credentials"), "w") as f:
     f.write(earthengine_credentials)
 
 ee.Initialize(project='ee-yashsacisro24')
+# # Imagery base selection
+# imagery_base = st.selectbox("Select Imagery Base", ["Sentinel", "Landsat"], index=0)
+
+# # Load the appropriate dataset based on imagery base
+# if imagery_base == "Sentinel":
+#     dataset_file = "sentinel_datasets.json"
+# else:
+#     dataset_file = "landsat_datasets.json"
+
+# with open(dataset_file) as f:
+#     data = json.load(f)
+
+# # Display the title for the Streamlit app
+# st.title(f"{imagery_base} Dataset")
+
+# # Select dataset category (main selection)
+# main_selection = st.selectbox(f"Select {imagery_base} Dataset Category", list(data.keys()))
+
+# # Initialize sub_selection and dataset_id as None
+# sub_selection = None
+# dataset_id = None
+
+# # If a category is selected, display the sub-options (specific datasets)
+# if main_selection:
+#     sub_options = data[main_selection]["sub_options"]
+#     sub_selection = st.selectbox(f"Select Specific {imagery_base} Dataset ID", list(sub_options.keys()))
+
+#     # Display the selected dataset ID based on user input
+#     if sub_selection:
+#         st.write(f"You selected: {main_selection} -> {sub_options[sub_selection]}")
+#         st.write(f"Dataset ID: {sub_selection}")
+#         dataset_id = sub_selection  # Use the key directly as the dataset ID
+
+# # Earth Engine Index Calculator Section
+# st.header("Earth Engine Index Calculator")
+
+# # Load band information based on selected dataset
+# if main_selection and sub_selection:  # Now safe because sub_selection is initialized
+#     dataset_bands = data[main_selection]["bands"].get(sub_selection, [])
+#     st.write(f"Available Bands for {sub_options[sub_selection]}: {', '.join(dataset_bands)}")
+
+#     # Allow user to select 1 or 2 bands
+#     selected_bands = st.multiselect(
+#         "Select 1 or 2 Bands for Calculation",
+#         options=dataset_bands,
+#         default=[dataset_bands[0]] if dataset_bands else [],
+#         help=f"Select 1 or 2 bands from: {', '.join(dataset_bands)}"
+#     )
+
+#     # Ensure minimum 1 and maximum 2 bands are selected
+#     if len(selected_bands) < 1:
+#         st.warning("Please select at least one band.")
+#         st.stop()
+#     elif len(selected_bands) > 2:
+#         st.warning("You can select a maximum of 2 bands.")
+#         st.stop()
+
+#     # Show custom formula input if bands are selected
+#     if selected_bands:
+#         # Provide a default formula based on the number of selected bands
+#         if len(selected_bands) == 1:
+#             default_formula = f"{selected_bands[0]}"
+#             example = f"'{selected_bands[0]} * 2' or '{selected_bands[0]} + 1'"
+#         else:  # len(selected_bands) == 2
+#             default_formula = f"({selected_bands[0]} - {selected_bands[1]}) / ({selected_bands[0]} + {selected_bands[1]})"
+#             example = f"'{selected_bands[0]} * {selected_bands[1]} / 2' or '({selected_bands[0]} - {selected_bands[1]}) / ({selected_bands[0]} + {selected_bands[1]})'"
+
+#         custom_formula = st.text_input(
+#             "Enter Custom Formula",
+#             value=default_formula,
+#             help=f"Use only these bands: {', '.join(selected_bands)}. Examples: {example}"
+#         )
+
 # Imagery base selection
-imagery_base = st.selectbox("Select Imagery Base", ["Sentinel", "Landsat"], index=0)
+imagery_base = st.selectbox("Select Imagery Base", ["Sentinel", "Landsat", "MODIS", "Custom Input"], index=0)
 
 # Load the appropriate dataset based on imagery base
 if imagery_base == "Sentinel":
     dataset_file = "sentinel_datasets.json"
-else:
+    with open(dataset_file) as f:
+        data = json.load(f)
+elif imagery_base == "Landsat":
     dataset_file = "landsat_datasets.json"
-
-with open(dataset_file) as f:
-    data = json.load(f)
+    with open(dataset_file) as f:
+        data = json.load(f)
+elif imagery_base == "MODIS":
+    dataset_file = "modis_datasets.json"
+    with open(dataset_file) as f:
+        data = json.load(f)
+elif imagery_base == "Custom Input":
+    custom_dataset_id = st.text_input("Enter Custom Earth Engine Dataset ID (e.g., ee.ImageCollection('AHN/AHN4'))", value="")
+    if custom_dataset_id:
+        try:
+            # Remove potential "ee.ImageCollection()" wrapper for simplicity
+            if custom_dataset_id.startswith("ee.ImageCollection("):
+                custom_dataset_id = custom_dataset_id.replace("ee.ImageCollection('", "").replace("')", "")
+            # Fetch dataset info from GEE
+            collection = ee.ImageCollection(custom_dataset_id)
+            band_names = collection.first().bandNames().getInfo()
+            data = {
+                f"Custom Dataset: {custom_dataset_id}": {
+                    "sub_options": {custom_dataset_id: f"Custom Dataset ({custom_dataset_id})"},
+                    "bands": {custom_dataset_id: band_names}
+                }
+            }
+            st.write(f"Fetched bands for {custom_dataset_id}: {', '.join(band_names)}")
+        except Exception as e:
+            st.error(f"Error fetching dataset: {str(e)}. Please check the dataset ID and ensure it's valid in Google Earth Engine.")
+            data = {}
+    else:
+        st.warning("Please enter a custom dataset ID to proceed.")
+        data = {}
 
 # Display the title for the Streamlit app
 st.title(f"{imagery_base} Dataset")
 
 # Select dataset category (main selection)
-main_selection = st.selectbox(f"Select {imagery_base} Dataset Category", list(data.keys()))
+if data:
+    main_selection = st.selectbox(f"Select {imagery_base} Dataset Category", list(data.keys()))
+else:
+    main_selection = None
 
 # Initialize sub_selection and dataset_id as None
 sub_selection = None
@@ -92,7 +196,7 @@ if main_selection:
 st.header("Earth Engine Index Calculator")
 
 # Load band information based on selected dataset
-if main_selection and sub_selection:  # Now safe because sub_selection is initialized
+if main_selection and sub_selection:
     dataset_bands = data[main_selection]["bands"].get(sub_selection, [])
     st.write(f"Available Bands for {sub_options[sub_selection]}: {', '.join(dataset_bands)}")