Update app.py

app.py

@@ -22,12 +22,16 @@ region = ee.Geometry.Polygon(
       [48.685226, 31.468496], [48.681879, 31.417603]]]
 )
 
-# تعریف پالت‌های رنگی برای هر شاخص
+# تعریف پالت‌های رنگی برای شاخص‌ها
 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']
+    'NDVI': {
+        'min': -0.2, 'max': 0.8,
+        'palette': ['#0000FF', '#8A2BE2', '#FFFF00', '#00FF00', '#008000', '#006400']
+    },
+    'NDMI': {
+        'min': -0.1, 'max': 0.5,
+        'palette': ['#FF0000', '#FFA500', '#FFFF00', '#008000', '#0000FF']
+    }
 }
 
 # توابع برای هر شاخص
@@ -35,20 +39,22 @@ 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()
+        vis_params = palettes['NDVI']
     elif index == "NDMI":
         image = s2.map(lambda image: image.normalizedDifference(['B8', 'B11']).rename('NDMI')).median()
-    
-    vis_params = {'min': -1, 'max': 1, 'palette': palettes[index]}
+        vis_params = palettes['NDMI']
     return image, vis_params
 
+# توابع برای دریافت داده‌های باد از Google Earth Engine
+def get_wind_data():
+    # استفاده از مجموعه داده ERA5 برای دریافت سرعت و جهت باد
+    era5 = ee.ImageCollection('ECMWF/ERA5_LAND/HOURLY').select(['u10', 'v10']).filterDate('2023-01-01', '2023-12-31').mean()
+    u10 = era5.select('u10').reduceRegion(ee.Reducer.mean(), region, scale=1000).getInfo()
+    v10 = era5.select('v10').reduceRegion(ee.Reducer.mean(), region, scale=1000).getInfo()
+    wind_speed = (u10['u10'] ** 2 + v10['v10'] ** 2) ** 0.5  # محاسبه سرعت باد
+    wind_direction = (180 / 3.14159) * ee.Number(ee.Image(u10['u10']).atan2(ee.Image(v10['v10']))).getInfo()  # محاسبه جهت باد
+    return wind_speed, wind_direction
+
 # ساخت برنامه Streamlit
 st.title("نقشه شاخص‌های کشاورزی برای مزارع شرکت دهخدا")
 
@@ -57,7 +63,7 @@ start_date = st.date_input("تاریخ شروع", datetime.date(2023, 1, 1))
 end_date = st.date_input("تاریخ پایان", datetime.date(2023, 12, 31))
 
 # انتخاب شاخص
-index_option = st.selectbox("شاخص مورد نظر را انتخاب کنید:", ["NDVI", "EVI", "SAVI", "NDMI"])
+index_option = st.selectbox("شاخص مورد نظر را انتخاب کنید:", ["NDVI", "NDMI"])
 
 # انتخاب مزرعه از فایل CSV
 farm_name = st.sidebar.selectbox("نام مزرعه را انتخاب کنید:", farms_data['farm_name'].unique())
@@ -89,3 +95,17 @@ if st.button("نمایش نقشه"):
 
     # امکان دانلود نقشه
     st.download_button(label="دانلود نقشه", data=map_.to_image(), file_name="map.png", mime="image/png")
+
+# نمایش سرعت و جهت باد در پنل کناری
+st.sidebar.header("اطلاعات باد")
+wind_speed, wind_direction = get_wind_data()
+st.sidebar.write(f"سرعت باد: {wind_speed:.2f} متر بر ثانیه")
+st.sidebar.write(f"جهت باد: {wind_direction:.2f} درجه")
+
+# نمایش قطب‌نما برای جهت باد
+compass_html = f"""
+<div style='width: 100px; height: 100px; border-radius: 50%; background: conic-gradient(red {wind_direction}deg, lightgray 0);'>
+  <div style='transform: rotate({wind_direction}deg); color: white; font-size: 20px; text-align: center; position: relative; top: 35px; font-weight: bold;'>⬆</div>
+</div>
+"""
+st.sidebar.markdown(compass_html, unsafe_allow_html=True)