DashboardLLM / app.py
Esmaeilkiani's picture
Update app.py
81fc5b0 verified
raw
history blame
3.02 kB
import streamlit as st
import ee
import geemap
import datetime
# Authenticate Earth Engine
service_account = 'earth-engine-service-account@ee-esmaeilkiani1387.iam.gserviceaccount.com'
credentials = ee.ServiceAccountCredentials(service_account, 'ee-esmaeilkiani1387-1b2c5e812a1d.json')
ee.Initialize(credentials)
# Define the indices and their corresponding functions
INDICES = {
'NDVI': lambda img: img.normalizedDifference(['B8', 'B4']).rename('NDVI'),
'EVI': lambda img: img.expression(
'2.5 * ((NIR - RED) / (NIR + 6 * RED - 7.5 * BLUE + 1))',
{
'NIR': img.select('B8'),
'RED': img.select('B4'),
'BLUE': img.select('B2')
}
).rename('EVI'),
'SAVI': lambda img: img.expression(
'((NIR - RED) / (NIR + RED + 0.5)) * (1.5)',
{
'NIR': img.select('B8'),
'RED': img.select('B4')
}
).rename('SAVI'),
'NDMI': lambda img: img.normalizedDifference(['B8', 'B11']).rename('NDMI')
}
# Define color palettes for each index
COLOR_PALETTES = {
'NDVI': ['FFFFFF', 'CE7E45', 'DF923D', 'F1B555', 'FCD163', '99B718', '74A901', '66A000', '529400', '3E8601', '207401', '056201', '004C00', '023B01', '012E01', '011D01', '011301'],
'EVI': ['FFFFFF', 'CE7E45', 'DF923D', 'F1B555', 'FCD163', '99B718', '74A901', '66A000', '529400', '3E8601', '207401', '056201', '004C00', '023B01', '012E01', '011D01', '011301'],
'SAVI': ['#D73027', '#F46D43', '#FDAE61', '#FEE08B', '#FFFFBF', '#D9EF8B', '#A6D96A', '#66BD63', '#1A9850'],
'NDMI': ['#D73027', '#F46D43', '#FDAE61', '#FEE08B', '#FFFFBF', '#D9EF8B', '#A6D96A', '#66BD63', '#1A9850']
}
# Streamlit app
st.title('Google Earth Engine Indices Viewer')
# User inputs
selected_indices = st.multiselect('Select two indices', list(INDICES.keys()), max_selections=2)
start_date = st.date_input('Start date', datetime.date(2022, 1, 1))
end_date = st.date_input('End date', datetime.date(2022, 12, 31))
if len(selected_indices) == 2 and start_date < end_date:
# Define the area of interest
aoi = ee.Geometry.Point([48.7312815, 31.5200749]).buffer(5000)
# Get Sentinel-2 imagery
s2 = ee.ImageCollection('COPERNICUS/S2_SR') \
.filterBounds(aoi) \
.filterDate(start_date, end_date) \
.filter(ee.Filter.lt('CLOUDY_PIXEL_PERCENTAGE', 20)) \
.sort('CLOUDY_PIXEL_PERCENTAGE')
# Calculate indices
for index in selected_indices:
s2 = s2.map(lambda img: img.addBands(INDICES[index](img)))
# Reduce the collection to a single image (median)
image = s2.select(selected_indices).median()
# Create a map
m = geemap.Map()
m.centerObject(aoi, 12)
# Add layers to the map
for index in selected_indices:
m.addLayer(image.select(index), {'min': 0, 'max': 1, 'palette': COLOR_PALETTES[index]}, index)
# Add the map to the Streamlit app
m.to_streamlit(height=600)
else:
st.warning('Please select two indices and ensure the start date is before the end date.')