update app.py

app.py

@@ -177,13 +177,6 @@ def calculate_custom_formula(image, geometry, formula):
         scale=30
     )
 
-# Function to mask clouds based on QA60 (Sentinel-2)
-def mask_clouds(image, cloud_threshold):
-    # Cloud mask based on the QA60 band (sentinel-2)
-    qa60 = image.select(['QA60'])
-    cloud_mask = qa60.bitwiseAnd(1 << 10).eq(0)
-    return image.updateMask(cloud_mask).clip(image.geometry())
-
 # Process each point or polygon
 results = []
 if locations_df is not None:
@@ -198,13 +191,10 @@ if locations_df is not None:
         roi = ee.Geometry.Point([longitude, latitude])
 
         # Load the image collection based on the user's selection
-        collection = ee.ImageCollection(sub_options[sub_selection]) \
+        collection = ee.ImageCollection('COPERNICUS/S2') \
             .filterDate(ee.Date(start_date_str), ee.Date(end_date_str)) \
-            .filterBounds(roi)
-
-        # Apply cloud masking if the index is NDVI or NDWI
-        if index_choice in ['NDVI', 'NDWI'] and cloud_threshold is not None:
-            collection = collection.map(lambda image: mask_clouds(image, cloud_threshold))
+            .filterBounds(roi) \
+            .filter(ee.Filter.lt('system:cloud_coverage', cloud_threshold))  # Apply cloud coverage filter
 
         # Check if the collection has images for the selected date range
         image_count = collection.size().getInfo()