Update app.py

app.py

@@ -12,54 +12,25 @@ ee.Initialize(credentials)
 # خواندن فایل 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]]]
-)
-
-# تعریف پالت‌های رنگی برای شاخص‌ها
-palettes = {
-    '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']
-    }
-}
-
-# توابع برای هر شاخص
-def get_index_map(index, start_date, end_date):
+# تعریف ناحیه مورد مطالعه
+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]]])
+
+# تابع برای دریافت تصویر NDVI
+def get_ndvi_image(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()
-        vis_params = palettes['NDVI']
-    elif index == "NDMI":
-        image = s2.map(lambda image: image.normalizedDifference(['B8', 'B11']).rename('NDMI')).median()
-        vis_params = palettes['NDMI']
-    return image, vis_params
-
-# توابع برای دریافت داده‌های باد از Google Earth Engine
-def get_wind_data():
-    # استفاده از نام صحیح باندها در مجموعه داده ERA5
-    era5 = ee.ImageCollection('ECMWF/ERA5_LAND/HOURLY').select(['u_component_of_wind_10m', 'v_component_of_wind_10m']).filterDate('2023-01-01', '2023-12-31').mean()
-    u10 = era5.select('u_component_of_wind_10m').reduceRegion(ee.Reducer.mean(), region, scale=1000).getInfo()
-    v10 = era5.select('v_component_of_wind_10m').reduceRegion(ee.Reducer.mean(), region, scale=1000).getInfo()
-    
-    # محاسبه سرعت و جهت باد
-    wind_speed = (u10['u_component_of_wind_10m'] ** 2 + v10['v_component_of_wind_10m'] ** 2) ** 0.5
-    wind_direction = (180 / 3.14159) * ee.Number(ee.Image(u10['u_component_of_wind_10m']).atan2(ee.Image(v10['v_component_of_wind_10m']))).getInfo()
-    return wind_speed, wind_direction
-
+    ndvi = s2.map(lambda image: image.normalizedDifference(['B8', 'B4']).rename('NDVI')).median()
+    ndvi_params = {'min': -0.2, 'max': 0.8, 'palette': ['#0000FF', '#8A2BE2', '#FFFF00', '#00FF00', '#008000', '#006400']}
+    return ndvi, ndvi_params
 
 # ساخت برنامه Streamlit
-st.title("نقشه شاخص‌های کشاورزی برای مزارع شرکت دهخدا")
+st.title("نقشه NDVI و NDMI برای مزارع شرکت دهخدا")
 
 # انتخاب بازه زمانی
 start_date = st.date_input("تاریخ شروع", datetime.date(2023, 1, 1))
@@ -71,44 +42,34 @@ index_option = st.selectbox("شاخص مورد نظر را انتخاب کنید
 # انتخاب مزرعه از فایل CSV
 farm_name = st.sidebar.selectbox("نام مزرعه را انتخاب کنید:", farms_data['farm_name'].unique())
 
-# پیدا کردن مختصات و اطلاعات مزرعه انتخاب شده
+# پیدا کردن مختصات مزرعه انتخاب شده
 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]
 
 # دکمه برای نمایش نقشه
 if st.button("نمایش نقشه"):
-    # بارگذاری نقشه شاخص
-    index_map, vis_params = get_index_map(index_option, start_date.isoformat(), end_date.isoformat())
+    # بسته به شاخص انتخاب ش��ه، نقشه را بارگذاری کنید
+    if index_option == "NDVI":
+        index_image, vis_params = get_ndvi_image(start_date.isoformat(), end_date.isoformat())
+    else:
+        st.error("شاخص انتخاب‌شده پشتیبانی نمی‌شود")
+        st.stop()
 
     # ایجاد نقشه با Geemap
     map_ = geemap.Map(center=[latitude, longitude], zoom=12)
-    map_.addLayer(index_map, vis_params, index_option)
+    map_.addLayer(index_image, vis_params, index_option)
 
     # افزودن نوار رنگ به نقشه
     map_.add_colorbar(vis_params, label=index_option)
 
-    # افزودن نشان مزرعه به نقشه
-    map_.add_marker([latitude, longitude], popup=f"نام: {farm_name}\nسن: {farm_age}\nواریته: {farm_variety}")
+    # افزودن مزرعه به نقشه
+    map_.add_marker([latitude, longitude], popup=f"نام: {farm_name}")
 
     # نمایش نقشه در Streamlit
     map_.to_streamlit()
 
-    # امکان دانلود نقشه
-    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)
+    # ایجاد لینک دانلود تصویر
+    image_url = index_image.getThumbURL(vis_params)
+    st.markdown(f"[دانلود نقشه با کیفیت بالا]({image_url})")
+