update app.py

app.py

@@ -6,6 +6,7 @@ import pandas as pd
 import geopandas as gpd
 from datetime import datetime
 from shapely.geometry import Point
+import leafmap.foliumap as leafmap
 
 # Set up the page layout
 st.set_page_config(layout="wide")
@@ -114,7 +115,63 @@ if parameters_changed():
         'end_date_str': end_date_str
     }
 
-# Function to perform index calculations
+# Process file and display map
+locations_df = None  # Initialize locations_df to None
+polygons_df = None   # Initialize polygons_df to None
+
+if file_upload:
+    file_extension = os.path.splitext(file_upload.name)[1].lower()
+
+    # Read file based on shape type
+    if shape_type == 'Point':
+        if file_extension == '.csv':
+            locations_df = read_csv(file_upload)
+        elif file_extension == '.geojson':
+            locations_df = read_geojson(file_upload)
+        elif file_extension == '.kml':
+            locations_df = read_kml(file_upload)
+        else:
+            st.error("Unsupported file type. Please upload a CSV, GeoJSON, or KML file for points.")
+    elif shape_type == 'Polygon':
+        if file_extension == '.geojson':
+            polygons_df = read_geojson(file_upload)
+        elif file_extension == '.kml':
+            polygons_df = read_kml(file_upload)
+        else:
+            st.error("Unsupported file type. Please upload a GeoJSON or KML file for polygons.")
+
+    # If there are points, display them on the map
+    if locations_df is not None:
+        st.write("Preview of the uploaded points data:")
+        st.dataframe(locations_df.head())
+
+        # Create a LeafMap object to display the points
+        m = leafmap.Map(center=[locations_df['latitude'].mean(), locations_df['longitude'].mean()], zoom=10)
+
+        for _, row in locations_df.iterrows():
+            m.add_marker(location=[row['latitude'], row['longitude']], popup=row.get('name', 'No Name'))
+
+        # Display map
+        st.write("Map of Uploaded Points:")
+        m.to_streamlit()
+
+    # If there are polygons, display them on the map
+    if polygons_df is not None:
+        st.write("Preview of the uploaded polygons data:")
+        st.dataframe(polygons_df.head())
+
+        # Create a LeafMap object to display the polygons
+        m = leafmap.Map(center=[polygons_df.geometry.centroid.y.mean(), polygons_df.geometry.centroid.x.mean()], zoom=10)
+
+        # Plot polygons on the map
+        for _, row in polygons_df.iterrows():
+            m.add_geojson(geojson=row['geometry'].__geo_interface__)
+
+        # Display map
+        st.write("Map of Uploaded Polygons:")
+        m.to_streamlit()
+
+# Function to perform index calculations (as before)
 def calculate_ndvi(image, geometry):
     ndvi = image.normalizedDifference(['B8', 'B4']).rename('NDVI')
     return ndvi.reduceRegion(