import os
import streamlit as st
import pandas as pd
import leafmap.foliumap as leafmap

st.set_page_config(layout="wide")

def random_color(feature):
    return {
        'color': 'black',
        'fillColor': random.choice(['red', 'yellow', 'green', 'orange']),
    }

url = 'https://open.gishub.org/maxar-open-data'
repo = 'https://github.com/opengeos/maxar-open-data/blob/master/datasets'

os.environ['GOOGLE_MAPS_API_KEY'] = 'API-KEY'
m = leafmap.Map()
m.add_basemap('SATELLITE')
m.add_basemap('ROADMAP')


@st.cache_data
def get_datasets():
    datasets = f'{url}/datasets.csv'
    df = pd.read_csv(datasets)
    return df


@st.cache_data
def get_catalogs(name, date_selected):
    dataset = f'{url}/datasets/{name}.tsv'

    dataset_df = pd.read_csv(dataset, sep='\t')
    
    dataset_df['as_datetime'] = dataset_df['datetime'].apply(pd.to_datetime)
    dataset_df['as_date'] = dataset_df['datetime'].apply(pd.to_datetime).apply(pd.Timestamp.date)
    filtered_to_date = dataset_df.loc[(dataset_df['as_date'] == date_selected), 'catalog_id' ]
    catalog_ids = filtered_to_date.unique().tolist()
    return catalog_ids

@st.cache_data
def get_catalogs_dates(name):
    dataset = f'{url}/datasets/{name}.tsv'

    dataset_df = pd.read_csv(dataset, sep='\t')
    catalog_dates_as_datetime = pd.to_datetime(dataset_df['datetime']).sort_values(ascending=False)
    catalog_dates_ids = catalog_dates_as_datetime.map(pd.Timestamp.date).unique().tolist()
    return catalog_dates_ids


st.title('Visualizing Maxar Open Data')
# st.markdown(get_catalogs('Morocco-Earthquake-Sept-2023', '2023-09-11'))

col1, col2 = st.columns([1.2, 3.8])

with col1:
    default = 'Morocco-Earthquake-Sept-2023'
    datasets = get_datasets()['dataset'].tolist()
    dataset = st.selectbox('Select a dataset', datasets, index=datasets.index(default))
    datepicker = st.selectbox('Select Date', get_catalogs_dates(dataset))
    # st.markdown(get_catalogs(dataset, datepicker))
    catalog = st.selectbox('Select a COG mosaic', get_catalogs(dataset, datepicker), index=get_catalogs(dataset, datepicker).index(get_catalogs(dataset, datepicker)[0]))
    geojson = f'{url}/datasets/{dataset}.geojson'
    mosaic = f'{url}/datasets/{dataset}/{catalog}.json'
    tsv = f'{repo}/{dataset}/{catalog}.tsv'
    st.markdown(f'View metadata: [{catalog}.tsv]({tsv})')

    with st.expander("Python code snippets"):
        markdown = f"""
import leafmap.foliumap as leafmap
m = leafmap.Map()
geojson = '{geojson}'
mosaic = '{mosaic}'
m.add_geojson(geojson, layer_name='{dataset}', info_mode='on_click')
m.add_stac_layer(mosaic, name='{catalog}')
m
        """
        st.code(markdown)


    style = {
        'weight': 1,
        'fillOpacity': 0
    }
    
    callback = lambda feat: print(feat["properties"]["datetime"])
    m.add_geojson(geojson, layer_name=dataset, style=style, info_mode='on_click', style_callback=random_color)
    m.add_stac_layer(mosaic, name=catalog)

    st.info('About')
    markdown = f"""
    - [Web App Source Code](https://github.com/opengeos/maxar-open-data/blob/master/streamlit_app.py)
    - [GitHub Repo](https://github.com/opengeos/maxar-open-data)
    - [Notebook Example](https://github.com/opengeos/maxar-open-data/blob/master/examples/maxar_open_data.ipynb)
    - [Maxar Open Data Program](https://www.maxar.com/open-data)
    - [Maxar Open Data on AWS](https://registry.opendata.aws/maxar-open-data/)
    - [Maxar Open Data on STAC Index](https://stacindex.org/catalogs/maxar-open-data-catalog-ard-format#/)
    - [Maxar Open Data on STAC Browser](https://radiantearth.github.io/stac-browser/#/external/maxar-opendata.s3.amazonaws.com/events/catalog.json?.language=en)
    - Contact: [Qiusheng Wu](https://github.com/giswqs)
    """
    st.markdown(markdown)

with col2:
    m.to_streamlit(height=780)