Update app.py

app.py

@@ -1,101 +1,112 @@
 import streamlit as st
 import pydeck as pdk
-import pandas as pd
 import geopandas as gpd
-import json
-from shapely.geometry import mapping
+import pandas as pd
+
 
-# Load GeoJSON data function
 @st.cache_data
 def load_geojson_data(file_paths):
     geodata = []
     for path in file_paths:
         gdf = gpd.read_file(path)
+        gdf['lon'] = gdf.geometry.x
+        gdf['lat'] = gdf.geometry.y
         geodata.append(gdf)
     return geodata
 
-# Sample Polygon GeoJSON
-polygon_geojson = json.loads("""
-{
-  "type": "FeatureCollection",
-  "features": [
-    {
-      "type": "Feature",
-      "properties": {"density": 10},
-      "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
 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("Hexagon and Polygon view:")
+
+st.title("Geospatial Data Visualization")
+
+
+geojson_file_paths = ['kathajodi_centroids.geojson','chambal_centroids.geojson']
 
 
-geojson_file_paths = ['chambal_centroids.geojson','kathajodi_centroids.geojson']
 geojson_data = load_geojson_data(geojson_file_paths)
 
 
-if geojson_data:
-    all_geo_data = pd.concat(geojson_data + [gpd.GeoDataFrame.from_features(polygon_geojson["features"])])
-else:
-    all_geo_data = gpd.GeoDataFrame.from_features(polygon_geojson["features"])
-
-# Add centroid coordinates to all_geo_data
-all_geo_data['lon'] = all_geo_data.geometry.centroid.x
-all_geo_data['lat'] = all_geo_data.geometry.centroid.y
-
-# Hexagon Layer
-hexagon_layer = pdk.Layer(
-    "HexagonLayer",
-    all_geo_data,
-    get_position=['lon', 'lat'],
-    elevation_scale=50,
-    radius=2000, 
-    elevation_range=[0, 3000],
-    pickable=True,
-    extruded=True,
-)
-
-polygon_layer = pdk.Layer(
-    "PolygonLayer",
-    all_geo_data,
-    get_polygon="geometry.coordinates",
-    get_fill_color="[200, 30, 0, 160]",
-    get_line_color=[0, 0, 0],
-    pickable=True,
-    auto_highlight=True
-)
-
-# Set viewport location
-view_state = pdk.ViewState(
-    latitude=27.5,
-    longitude=79.5,
-    zoom=6,
-    pitch=40
-)
-
-
-r = pdk.Deck(
-    layers=[hexagon_layer, polygon_layer], 
-    initial_view_state=view_state,
-    map_style='mapbox://styles/mapbox/satellite-v9'
-)
-
-
-st.pydeck_chart(r)
-
-# Add an informational box
-with st.expander("View Information"):
-    st.write(f"Selected Time: {timestamp}")
-    num_points = len(all_geo_data)
-    total_area = all_geo_data.geometry.area.sum() 
-    st.write(f"Number of Features: {num_points}")
-    st.write(f"Total Area: {total_area:.2f} sq km")
+all_geo_data = pd.concat(geojson_data)
+all_geo_data['lon'] = all_geo_data.geometry.x
+all_geo_data['lat'] = all_geo_data.geometry.y
+
+
+if 'view_state' not in st.session_state:
+    st.session_state.view_state = pdk.ViewState(
+        latitude=all_geo_data['lat'].mean(),
+        longitude=all_geo_data['lon'].mean(),
+        zoom=10,
+        pitch=45,
+        bearing=0
+    )
+
+
+tab1, tab2 = st.tabs(["Heatmap", "3D Hexagon Plot"])
+
+
+with tab1:
+    st.subheader("Heatmap View")
+    
+    # Pydeck HeatmapLayer for Points
+    heatmap_layer = pdk.Layer(
+        "HeatmapLayer",
+        all_geo_data,
+        get_position=['lon', 'lat'],
+        aggregation=pdk.types.String("MEAN"),
+        opacity=0.5
+    )
+
+    r = pdk.Deck(
+        layers=[heatmap_layer],
+        initial_view_state=st.session_state.view_state,
+        map_style='mapbox://styles/mapbox/satellite-v9'
+    )
+
+
+    st.session_state.view_state = r.initial_view_state
+
+
+    st.pydeck_chart(r)
+
+    # Add an informational box
+    with st.expander("View Information"):
+        st.write(f"Selected Time: {timestamp}")
+        num_points = len(all_geo_data)
+        st.write(f"Number of Features: {num_points}")
+
+# 3D Hexagon Plot Tab
+with tab2:
+    st.subheader("3D Hexagon Map View")
+    
+    
+    hexagon_layer = pdk.Layer(
+        "HexagonLayer",
+        all_geo_data,
+        get_position=['lon', 'lat'],
+        radius=200,  
+        elevation_scale=20,  
+        elevation_range=[0, 1000], 
+        extruded=True,  
+        coverage=1,  
+        pickable=True
+    )
+
+    
+    r = pdk.Deck(
+        layers=[hexagon_layer],
+        initial_view_state=st.session_state.view_state,
+        map_style='mapbox://styles/mapbox/satellite-v9'
+    )
+
+    
+    st.session_state.view_state = r.initial_view_state
+
+    
+    st.pydeck_chart(r)
+
+    # Add informational box
+    with st.expander("View Information"):
+        st.write(f"Selected Time: {timestamp}")
+        num_points = len(all_geo_data)
+        st.write(f"Number of Features: {num_points}")