Update app.py

app.py

@@ -1,82 +1,91 @@
 import streamlit as st
 import ee
-import geemap
+import geemap.foliumap as geemap
+import pandas as pd
 import datetime
 
-# Authenticate Earth Engine
+# اعتبار سنجی و اتصال به Google 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')
-}
+# خواندن فایل CSV مزارع
+farms_data = pd.read_csv('Farm_Details_Export.csv')
+
+# تعریف ناحیه مورد مطالعه با مختصات جدید
+region = ee.Geometry.Polygon(
+    [[[48.681879, 31.417603], [48.721447, 31.413209], [48.724279, 31.420826], [48.726768, 31.427418],
+      [48.728228, 31.435694], [48.736382, 31.42837], [48.739557, 31.435657], [48.742261, 31.441772],
+      [48.752303, 31.452243], [48.75226, 31.459784], [48.759127, 31.473657], [48.766809, 31.472413],
+      [48.773203, 31.491188], [48.77758, 31.534579], [48.785563, 31.540797], [48.792601, 31.59696],
+      [48.694668, 31.60756], [48.691921, 31.603466], [48.697586, 31.534067], [48.69381, 31.507727],
+      [48.685226, 31.468496], [48.681879, 31.417603]]]
+)
 
-# Define color palettes for each index
-COLOR_PALETTES = {
+# تعریف پالت‌های رنگی برای هر شاخص
+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')
+# توابع برای هر شاخص
+def get_index_map(index, start_date, end_date):
+    s2 = ee.ImageCollection('COPERNICUS/S2').filterBounds(region).filterDate(start_date, end_date)
+    if index == "NDVI":
+        image = s2.map(lambda image: image.normalizedDifference(['B8', 'B4']).rename('NDVI')).median()
+    elif index == "EVI":
+        image = s2.map(lambda image: image.expression('2.5 * ((B8 - B4) / (B8 + 6 * B4 - 7.5 * B2 + 1))', {
+            'B8': image.select('B8'), 'B4': image.select('B4'), 'B2': image.select('B2')
+        }).rename('EVI')).median()
+    elif index == "SAVI":
+        image = s2.map(lambda image: image.expression('(1 + 0.5) * ((B8 - B4) / (B8 + B4 + 0.5))', {
+            'B8': image.select('B8'), 'B4': image.select('B4')
+        }).rename('SAVI')).median()
+    elif index == "NDMI":
+        image = s2.map(lambda image: image.normalizedDifference(['B8', 'B11']).rename('NDMI')).median()
+    
+    vis_params = {'min': -1, 'max': 1, 'palette': palettes[index]}
+    return image, vis_params
+
+# ساخت برنامه Streamlit
+st.title("نقشه شاخص‌های کشاورزی برای مزارع شرکت دهخدا")
 
-# 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))
+# انتخاب بازه زمانی
+start_date = st.date_input("تاریخ شروع", datetime.date(2023, 1, 1))
+end_date = st.date_input("تاریخ پایان", datetime.date(2023, 12, 31))
 
-# Convert dates to ee.Date objects
-ee_start_date = ee.Date(start_date.strftime('%Y-%m-%d'))
-ee_end_date = ee.Date(end_date.strftime('%Y-%m-%d'))
+# انتخاب شاخص
+index_option = st.selectbox("شاخص مورد نظر را انتخاب کنید:", ["NDVI", "EVI", "SAVI", "NDMI"])
 
-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)
+# انتخاب مزرعه از فایل CSV
+farm_name = st.sidebar.selectbox("نام مزرعه را انتخاب کنید:", farms_data['farm_name'].unique())
 
-    # Get Sentinel-2 imagery
-    s2 = ee.ImageCollection('COPERNICUS/S2_SR') \
-        .filterBounds(aoi) \
-        .filterDate(ee_start_date, ee_end_date) \
-        .filter(ee.Filter.lt('CLOUDY_PIXEL_PERCENTAGE', 20)) \
-        .sort('CLOUDY_PIXEL_PERCENTAGE')
+# پیدا کردن مختصات و اطلاعات مزرعه انتخاب شده
+selected_farm = farms_data[farms_data['farm_name'] == farm_name]
+latitude = selected_farm['latitude'].values[0]
+longitude = selected_farm['longitude'].values[0]
+farm_age = selected_farm['age'].values[0]
+farm_variety = selected_farm['variety'].values[0]
 
-    # Calculate indices
-    for index in selected_indices:
-        s2 = s2.map(lambda img: img.addBands(INDICES[index](img)))
+# دکمه برای نمایش نقشه
+if st.button("نمایش نقشه"):
+    # بارگذاری نقشه شاخص
+    index_map, vis_params = get_index_map(index_option, start_date.isoformat(), end_date.isoformat())
 
-    # Reduce the collection to a single image (median)
-    image = s2.select(selected_indices).median()
+    # ایجاد نقشه با Geemap
+    map_ = geemap.Map(center=[latitude, longitude], zoom=12)
+    map_.addLayer(index_map, vis_params, index_option)
 
-    # Create a map
-    m = geemap.Map()
-    m.centerObject(aoi, 12)
+    # افزودن نوار رنگ به نقشه
+    map_.add_colorbar(vis_params, label=index_option)
 
-    # Add layers to the map
-    for index in selected_indices:
-        m.addLayer(image.select(index), {'min': 0, 'max': 1, 'palette': COLOR_PALETTES[index]}, index)
+    # افزودن نشان مزرعه به نقشه
+    map_.add_marker([latitude, longitude], popup=f"نام: {farm_name}\nسن: {farm_age}\nواریته: {farm_variety}")
 
-    # Add the map to the Streamlit app
-    m.to_streamlit(height=600)
+    # نمایش نقشه در Streamlit
+    map_.to_streamlit()
 
-else:
-    st.warning('Please select two indices and ensure the start date is before the end date.')
+    # امکان دانلود نقشه
+    st.download_button(label="دانلود نقشه", data=map_.to_image(), file_name="map.png", mime="image/png")