Spaces:
Running
Running
| import os | |
| import leafmap.foliumap as leafmap | |
| import streamlit as st | |
| import palettable | |
| 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:]] | |
| 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() | |