Update app.py

app.py

@@ -1,112 +1,191 @@
 import streamlit as st
-import pydeck as pdk
 import geopandas as gpd
 import pandas as pd
+import os
+import zipfile
+import tempfile
+from keplergl import KeplerGl
+from shapely.geometry import Polygon, MultiPolygon
+import pydeck as pdk
 
+def main():
+    polygons_zip_path = 'polygons-20240817T212638Z-001.zip'
 
-@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
-
-st.sidebar.title("Geospatial Data Visualization")
-timestamp = st.sidebar.select_slider("Time", options=[f"{year}-01" for year in range(2016, 2024)])
-
-
-st.title("Geospatial Data Visualization")
-
-
-geojson_file_paths = ['kathajodi_centroids.geojson','chambal_centroids.geojson']
-
-
-geojson_data = load_geojson_data(geojson_file_paths)
-
-
-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'
-    )
+    valid_polygon_gdfs = []
 
+   
+    with tempfile.TemporaryDirectory() as tmpdirname:
+        with zipfile.ZipFile(polygons_zip_path, 'r') as zip_ref:
+            zip_ref.extractall(tmpdirname)
+        
+        polygons_dir = os.path.join(tmpdirname, 'polygons')
+        geojson_files = [f for f in os.listdir(polygons_dir) if f.endswith('.geojson')]
 
-    st.session_state.view_state = r.initial_view_state
+        geojson_files = geojson_files[:2]
 
+        for file in geojson_files:
+            try:
+                gdf = gpd.read_file(os.path.join(polygons_dir, file)).to_crs(epsg=4326)
 
-    st.pydeck_chart(r)
+                if gdf.empty:
+                    st.warning(f"Skipping empty file: {file}")
+                    continue
 
-    # 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}")
+                gdf = gdf[gdf.geometry.apply(lambda geom: isinstance(geom, (Polygon, MultiPolygon)))]
 
-# 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=1000,  
-        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'
-    )
+                if gdf.empty:
+                    st.warning(f"No valid geometries found in file: {file}")
+                    continue
 
-    
-    st.session_state.view_state = r.initial_view_state
+                valid_polygon_gdfs.append(gdf)
+            
+            except Exception as e:
+                st.error(f"Error processing file {file}: {str(e)}")
+                continue
 
     
-    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}")
+    if valid_polygon_gdfs:
+        tab1, tab2 = st.tabs(["Editable Point based map", "Pydeck Heatmap"])
+
+        combined_polygon_gdf = gpd.GeoDataFrame(pd.concat(valid_polygon_gdfs, ignore_index=True))
+
+        with tab1:
+            kepler_data = {
+                "polygons": combined_polygon_gdf
+            }
+
+            kepler_map = KeplerGl(height=600)
+
+            kepler_map.add_data(data=kepler_data["polygons"], name="Polygons")
+
+            config = {
+            "version": "v1",
+            "config": {
+                "visState": {
+                    "layers": [
+                        {
+                            "id": "polygon-layer",
+                            "type": "geojson",
+                            "config": {
+                                "dataId": "Polygons",
+                                "label": "Polygons",
+                                "color": [255, 0, 0],
+                                "highlightColor": [252, 242, 26, 255],
+                                "columns": {
+                                    "geojson": "geometry"
+                                },
+                                "isVisible": True,
+                                "visConfig": {
+                                    "opacity": 0.8,
+                                    "thickness": 1.5,
+                                    "strokeColor": [255, 0, 0],
+                                    "colorRange": {
+                                        "name": "Custom",
+                                        "type": "sequential",
+                                        "category": "Custom",
+                                        "colors": [
+                                            "#000004", "#3b0f70", "#8c2981", "#de4968", "#fe9f6d", "#fcfdbf"  # Magma
+                                        ]
+                                    },
+                                    "filled": True,
+                                    "stroked": True,
+                                    "enable3d": False,
+                                    "wireframe": False
+                                },
+                                "visualChannels": {
+                                    "colorField": {
+                                        "name": "class",  
+                                        "type": "real"
+                                    },
+                                    "colorScale": "quantile"
+                                }
+                            }
+                        }
+                    ]
+                },
+                "mapStyle": {
+                    "styleType": "satellite",  
+                    "topLayerGroups": {},     
+                    "visibleLayerGroups": {
+                        "label": True,
+                        "road": True,
+                        "border": False,
+                        "building": True,
+                        "water": True,
+                        "land": True,
+                        "3d building": False
+                    },
+                    "threeDBuildingColor": [9.665468314072013, 17.18305478057247, 31.1442867897876],
+                    "mapStyles": {}
+                },
+                "mapState": {
+                    "bearing": 0,
+                    "dragRotate": True,
+                    "latitude": combined_polygon_gdf.geometry.centroid.y.mean(),
+                    "longitude": combined_polygon_gdf.geometry.centroid.x.mean(),
+                    "pitch": 0,
+                    "zoom": 10
+                },
+                "interactionConfig": {
+                    "brush": {
+                        "enabled": True
+                    },
+                    "tooltip": {
+                        "enabled": True,
+                        "fieldsToShow": {
+                            "Polygons": [
+                                {
+                                    "name": "class", 
+                                    "format": None
+                                }
+                            ]
+                        }
+                    }
+                },
+                "animationConfig": {
+                    "enabled": True
+                }
+            }
+        }
+
+            kepler_map = KeplerGl(height=600, config=config)
+            kepler_map.add_data(data=kepler_data["polygons"], name="Polygons")
+            kepler_map.save_to_html(file_name="kepler_map.html")
+            st.components.v1.html(open("kepler_map.html", 'r').read(), height=600)
+            st.sidebar.title("Summary Stats")
+            st.sidebar.write(combined_polygon_gdf.describe())
+
+        with tab2:
+            combined_polygon_gdf['centroid'] = combined_polygon_gdf.geometry.centroid
+            combined_polygon_gdf['longitude'] = combined_polygon_gdf.centroid.x
+            combined_polygon_gdf['latitude'] = combined_polygon_gdf.centroid.y
+            data = combined_polygon_gdf[['longitude', 'latitude', 'class']]
+            heatmap_layer = pdk.Layer(
+                "HeatmapLayer",
+                data=data,
+                get_position='[longitude, latitude]',
+                get_weight="class",
+                radius_pixels=50,
+            )
+
+            view_state = pdk.ViewState(
+                latitude=combined_polygon_gdf.geometry.centroid.y.mean(),
+                longitude=combined_polygon_gdf.geometry.centroid.x.mean(),
+                zoom=10,
+                bearing=0,
+                pitch=0
+            )
+
+            r = pdk.Deck(
+                layers=[heatmap_layer],
+                initial_view_state=view_state,
+                map_style='mapbox://styles/mapbox/satellite-v9'
+            )
+
+            st.pydeck_chart(r)
+    else:
+        st.error("No valid GeoDataFrames could be combined.")
+
+if __name__ == "__main__":
+    main()