import numpy as np
import leafmap.foliumap as leafmap
import streamlit as st
import geojson

file_map = {"Lucknow 80x80 km": "lucknow_sarath_grid.geojson", "Delhi 80x80 km": "delhi_sarath_grid.geojson", "West Bengal 20x20 km": "wb_20x20_grid.geojson", "Uttar Pradesh": "m1_labels_zero_threshold.geojson"}
st.title("Brick Kilns Interactive Map")
region = st.selectbox("Select a region", list(file_map.keys()))

def style_function(feature):
    class_name = feature['properties']['class_name']
    if class_name == "Zigzag":
        return {'color': 'green', 'weight': 3, 'fillColor': 'green', 'fillOpacity': 0}
    elif class_name == "FCBK":
        return {'color': 'orange', 'weight': 3, 'fillColor': 'orange', 'fillOpacity': 0}
    elif class_name == "CFCBK":
        return {'color': 'red', 'weight': 3, 'fillColor': 'red', 'fillOpacity': 0}
    else:
        return {'color': 'black', 'weight': 3, 'fillColor': 'black', 'fillOpacity': 0}

m = leafmap.Map()
m.add_basemap("SATELLITE")
legend_dict = {"CFCBK": "red", "FCBK": "orange", "Zigzag": "green"}
m.add_geojson(file_map[region], style_function=style_function, layer_name="Kilns")
m.add_legend(legend_dict=legend_dict, title="Brick Kilns")
m.to_streamlit()