Update app.py

app.py

@@ -1,47 +1,85 @@
 import streamlit as st
 import leafmap.foliumap as leafmap
 import geopandas as gpd
+import json
 
 # Function to load GeoJSON data
-@st.cache_data
-def load_geojson_data(file_path):
-    gdf = gpd.read_file(file_path)
-    return gdf
+@st.cache
+def load_geojson_data(file_paths):
+    geodata = []
+    for path in file_paths:
+        gdf = gpd.read_file(path)
+        geodata.append(gdf)
+    return geodata
 
-# Load the provided GeoJSON data
-geojson_file_path = 'kathajodi_centroids.geojson'
-geojson_data = load_geojson_data(geojson_file_path)
+# Function to generate heatmap data from polygons
+def generate_heatmap_data(gdfs):
+    heatmap_data = []
+    for gdf in gdfs:
+        for _, row in gdf.iterrows():
+            if row.geometry.type == 'Polygon':
+                x, y = row.geometry.centroid.x, row.geometry.centroid.y
+                heatmap_data.append([y, x])
+            elif row.geometry.type == 'Point':
+                x, y = row.geometry.x, row.geometry.y
+                heatmap_data.append([y, x])
+    return heatmap_data
 
-# Sidebar for user input
+# Placeholder for polygon GeoJSON
+polygon_geojson = json.loads("""
+{
+  "type": "FeatureCollection",
+  "features": [
+    {
+      "type": "Feature",
+      "properties": {},
+      "geometry": {
+        "type": "Polygon",
+        "coordinates": [[[79.486114,27.577006],[79.5157554,27.5646727],[79.5405574,27.5461701],[79.5695939,27.519081],[79.5756432,27.4938634],[79.5593101,27.4871556],[79.5308786,27.5257868],[79.5003298,27.5319559],[79.4849041,27.5456337],[79.4761327,27.5652089],[79.486114,27.577006]]]
+      }
+    }
+  ]
+}
+""")
+
+# Sidebar for input
 st.sidebar.title("Geospatial Data Visualization")
 timestamp = st.sidebar.select_slider("Time", options=[f"{year}-01" for year in range(2016, 2024)])
 
 # Main title
 st.title("Heatmap view:")
 
+# Load your GeoJSON files
+geojson_file_paths = ['/mnt/data/kathajodi_centroids.geojson']
+geojson_data = load_geojson_data(geojson_file_paths)
+
 # Create a Leafmap map with satellite basemap
-m = leafmap.Map(center=[20, 78], zoom=5, basemap="SATELLITE")
+m = leafmap.Map(center=[27.5, 79.5], zoom=6, basemap="SATELLITE")
 
-# Function to generate heatmap data
-def generate_heatmap_data(gdf):
-    return [(point.y, point.x) for point in gdf.geometry]
+# Add GeoJSON data to the map
+for gdf in geojson_data:
+    m.add_gdf(gdf, layer_name="GeoData")
 
-# Add heatmap data
-heatmap_data = generate_heatmap_data(geojson_data)
-m.add_heatmap(heatmap_data, radius=15)
+# Add the polygon GeoJSON
+polygon_gdf = gpd.GeoDataFrame.from_features(polygon_geojson["features"])
+m.add_gdf(polygon_gdf, layer_name="Polygon")
 
-# Display the map
+# Generate heatmap data and add heatmap
+heatmap_data = generate_heatmap_data(geojson_data + [polygon_gdf])
+m.add_heatmap(heatmap_data, radius=20)
+
+# Display the map 
 m.to_streamlit(height=600)
 
 # Add an informational box
 with st.expander("View Information"):
     st.write(f"Selected Time: {timestamp}")
-    num_points = len(geojson_data)
-    total_area = geojson_data.geometry.area.sum()  
-    st.write(f"Number of Points: {num_points}")
+    num_points = sum(len(gdf) for gdf in geojson_data)
+    total_area = sum(gdf.geometry.area.sum() for gdf in geojson_data)  # Customize as needed.
+    st.write(f"Number of Features: {num_points}")
     st.write(f"Total Area: {total_area:.2f} sq km")
 
-# Display the summary stats in a fixed position (simulate top-right box)
+
 css_code = """
 <style>
 .fixed-box {
@@ -62,7 +100,7 @@ st.markdown(
     f"""
     <div class="fixed-box">
         <strong>Summary Stats:</strong>
-        <p>Number of Points: {num_points}</p>
+        <p>Number of Features: {num_points}</p>
         <p>Total Area: {total_area:.2f} sq km</p>
     </div>
     """, unsafe_allow_html=True