import os
import leafmap.foliumap as leafmap
import streamlit as st
import palettable


@st.cache(allow_output_mutation=True)
def load_cog_list():
    print(os.getcwd())
    in_txt = os.path.join(os.getcwd(), "data/cog_files.txt")
    with open(in_txt) as f:
        return [line.strip() for line in f.readlines()[1:]]


@st.cache(allow_output_mutation=True)
def get_palettes():
    palettes = dir(palettable.matplotlib)[:-16]
    return ["matplotlib." + p for p in palettes]


def app():

    st.title("Visualize Raster Datasets")
    st.markdown(
        """
    An interactive web app for visualizing local raster datasets and Cloud Optimized GeoTIFF ([COG](https://www.cogeo.org)). The app was built using [streamlit](https://streamlit.io), [leafmap](https://leafmap.org), and [localtileserver](https://github.com/banesullivan/localtileserver).


    """
    )

    row1_col1, row1_col2 = st.columns([2, 1])

    with row1_col1:
        cog_list = load_cog_list()
        cog = st.selectbox("Select a sample Cloud Opitmized GeoTIFF (COG)", cog_list)

    with row1_col2:
        empty = st.empty()

        url = empty.text_input(
            "Enter a HTTP URL to a Cloud Optimized GeoTIFF (COG)",
            cog,
        )

        data = st.file_uploader("Upload a raster dataset", type=["tif", "img"])

        if data:
            url = empty.text_input(
                "Enter a URL to a Cloud Optimized GeoTIFF (COG)",
                "",
            )

        add_palette = st.checkbox("Add a color palette")
        if add_palette:
            palette = st.selectbox("Select a color palette", get_palettes())
        else:
            palette = None

        submit = st.button("Submit")

    m = leafmap.Map(latlon_control=False)

    if submit:
        if data or url:
            try:
                if data:
                    file_path = leafmap.save_data(data)
                    m.add_local_tile(file_path, palette=palette, debug=True)
                elif url:
                    m.add_remote_tile(url, palette=palette, debug=True)
            except Exception as e:
                with row1_col2:
                    st.error("Work in progress. Try it again later.")

    with row1_col1:
        m.to_streamlit()