Update app.py

app.py

@@ -1,31 +1,52 @@
-import ee
 import streamlit as st
+import ee
 import pandas as pd
+from datetime import date
+from google.oauth2 import service_account
+import geemap
 
 # 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) 
 
-# آپلود فایل مختصات
-uploaded_file = st.file_uploader("Farm_Details_Export (1).csv", type="csv")
-if uploaded_file:
-    farms_df = pd.read_csv(uploaded_file)
-    
-    # پردازش اطلاعات هر مزرعه
-    for idx, row in farms_df.iterrows():
-        lat, lon = row['latitude'], row['longitude']
-        location = ee.Geometry.Point(lon, lat)
-        
-        # دریافت تصویر ماهواره‌ای و محاسبه NDVI
-        image = ee.ImageCollection('COPERNICUS/S2') \
-                  .filterBounds(location) \
-                  .filterDate('2023-01-01', '2023-12-31') \
-                  .mean()
-        ndvi = image.normalizedDifference(['B8', 'B4']).reduceRegion(
-            reducer=ee.Reducer.mean(),
-            geometry=location,
-            scale=30
-        )
+@st.cache_data
+def get_farm_coordinates(farm_name, dataset_url):
+    # بارگذاری دیتاست
+    df = pd.read_csv(dataset_url)
+    # جستجوی مزرعه
+    farm_data = df[df['Farm'] == farm_name]
+    if farm_data.empty:
+        return None
+    return farm_data[['latitude', 'longitude']].values[0]
+
+# تابع برای محاسبه شاخص NDVI در بازه زمانی مشخص
+def get_ndvi_map(latitude, longitude, start_date, end_date):
+    point = ee.Geometry.Point([longitude, latitude])
+    collection = (ee.ImageCollection("COPERNICUS/S2")
+                  .filterBounds(point)
+                  .filterDate(start_date, end_date)
+                  .map(lambda img: img.normalizedDifference(['B8', 'B4']).rename('NDVI')))
+    ndvi_image = collection.median()
+    vis_params = {'min': 0, 'max': 1, 'palette': ['blue', 'white', 'green']}
+    return geemap.ee_to_url(ndvi_image.clip(point.buffer(10000)), vis_params)
+
+# رابط کاربری با Streamlit
+st.title("نمایش نقشه شاخص NDVI مزارع نیشکر")
+farm_name = st.text_input("نام مزرعه را وارد کنید:")
+dataset_url = "https://huggingface.co/datasets/Esmaeilkianii/SugarcaneDataSet/embed/viewer/default/train"
+
+if farm_name:
+    coords = get_farm_coordinates(farm_name, dataset_url)
+    if coords is not None:
+        st.success(f"مزرعه یافت شد! مختصات: {coords}")
+        start_date = st.date_input("تاریخ شروع:", date(2023, 1, 1))
+        end_date = st.date_input("تاریخ پایان:", date.today())
         
-        st.write(f"شاخص NDVI برای مزرعه {idx}: {ndvi.getInfo()['nd']}")
+        if start_date < end_date:
+            map_url = get_ndvi_map(coords[0], coords[1], start_date.isoformat(), end_date.isoformat())
+            st.image(map_url, caption=f"نقشه NDVI برای مزرعه {farm_name}")
+        else:
+            st.error("تاریخ شروع باید قبل از تاریخ پایان باشد.")
+    else:
+        st.error("مزرعه‌ای با این نام یافت نشد.")