Update app.py

app.py

@@ -9,6 +9,7 @@ import geojson
 from shapely.geometry import Polygon, MultiPolygon, shape, Point
 from io import BytesIO
 import fiona
+from shapely import wkb
 
 
 # Enable fiona driver
@@ -26,22 +27,11 @@ with open(os.path.expanduser("~/.config/earthengine/credentials"), "w") as f:
 ee.Initialize(project='in793-aq-nb-24330048')
 
 # Functions
-def convert_to_2d_geometry(geom): #Handles Polygon Only
+def convert_to_2d_geometry(geom): 
   if geom is None:
     return None
   elif geom.has_z:
-    # Extract exterior coordinates and convert to 2D
-    exterior_coords = geom.exterior.coords[:]  # Get all coordinates of the exterior ring
-    exterior_coords_2d = [(x, y) for x, y, *_ in exterior_coords] # Keep only the x and y coordinates, ignoring z
-
-    # Handle interior rings (holes) if any
-    interior_coords_2d = []
-    for interior in geom.interiors:
-        interior_coords = interior.coords[:]
-        interior_coords_2d.append([(x, y) for x, y, *_ in interior_coords])
-
-    # Create a new Polygon with 2D coordinates
-    return type(geom)(exterior_coords_2d, interior_coords_2d)
+    return df.geometry.transform(lambda geom: wkb.loads(wkb.dumps(geom, output_dimension=2)))
   else:
     return geom
   
@@ -61,7 +51,7 @@ def validate_KML_file(gdf):
     polygon_info = {}
 
     # Check if it's a single polygon or multipolygon
-    if isinstance(gdf.iloc[0].geometry, Polygon):
+    if isinstance(gdf.iloc[0].geometry, Polygon) and len(gdf)==1:
       polygon_info['is_single_polygon'] = True
 
       polygon = convert_to_2d_geometry(gdf.geometry.iloc[0])
@@ -130,16 +120,19 @@ st.title("Mean NDVI Calculator")
 col = st.columns(2)
 start_date = col[0].date_input("Start Date", value=pd.to_datetime('2021-01-01'))
 end_date = col[1].date_input("End Date", value=pd.to_datetime('2021-01-30'))
+# Check if start and end dates are valid
+if start_date>end_date:
+    st.stop()
+
 start_date = start_date.strftime("%Y-%m-%d")
 end_date = end_date.strftime("%Y-%m-%d")
 
+
 max_cloud_cover = st.number_input("Max Cloud Cover", value=20)
 
 # Get the geojson file from the user
 uploaded_file = st.file_uploader("Upload KML/GeoJSON file", type=["geojson", "kml"])
 
-
-
 if uploaded_file is not None:
     try:
         if uploaded_file.name.endswith("kml"):
@@ -192,11 +185,26 @@ if uploaded_file is not None:
         # Re-order the columns of the resultant dataframe
         resultant_df = resultant_df[['Date', 'Imagery', 'AvgNDVI_Inside', 'Avg_NDVI_Buffer', 'Ratio']]
 
+        # Write the final table
         st.write(resultant_df)
         
         # plot the time series
         st.write("Time Series Plot")
         st.line_chart(resultant_df[['AvgNDVI_Inside', 'Avg_NDVI_Buffer', 'Date']].set_index('Date'))
+
+        # Visualize map on ESRI basemap
+        # Set the title of the app
+        st.write("Map Visualization")
+        
+        # Create a Leafmap object
+        m = leafmap.Map()
+        
+        # Add ESRI latest imagery basemap and polygon
+        m.add_basemap("ESRI/WorldImagery")
+        m.add_polygon(gdf.iloc[0].geometry, "Polygon Example", color="red", fill_opacity=0.5)
+        
+        # Display the map
+        m.to_streamlit(height=600)
     
     else:
         st.write('ValueError: "Input must have single polygon geometry"')