First commit

.gitignore

@@ -0,0 +1,2 @@
+.venv
+__pycache__

app.py

@@ -0,0 +1,134 @@
+import itertools
+from datetime import datetime
+from pathlib import Path
+
+import folium
+import streamlit as st
+from loguru import logger as log
+from matplotlib import colors
+from streamlit_folium import st_folium
+from utils.folium import (
+    draw_gdf,
+    get_clean_rendering_container,
+    get_map_crop,
+    get_parquet,
+)
+
+COLOR_MAP = {
+    "Grassland": "lime",
+    "Cereals": "saddlebrown",
+    "Other": "white",
+    "Flowers": "cyan",
+    "Vegetables": "purple",
+    "Fruit trees": "darkgreen",
+}
+
+
+id_class = {(i + 1): k for i, k in enumerate(COLOR_MAP.keys())}
+color_map_hex = {k: colors.to_hex(v) for k, v in COLOR_MAP.items()}
+id_map_hex = {(i + 1): v for i, v in enumerate(color_map_hex.values())}
+color_map_hex_r = {v: k for k, v in color_map_hex.items()}
+merged_list = list(itertools.chain(*list(color_map_hex_r.items())))
+
+
+def style_call(feat):
+    di = {
+        "fillColor": id_map_hex[feat["properties"]["class"]],
+        "fillOpacity": 0.7,
+        "color": id_map_hex[feat["properties"]["class"]],
+        "border-width": "thin",
+        "border-color": "#ffffff",
+        "weight": 1.2,
+    }
+    return di
+
+
+# Page configs
+st.set_page_config(page_title="AIDA", page_icon="🌐", layout="wide")
+
+hide_button_style = """
+    <style>
+    div.stButton > button:first-child {
+        display: none;
+    }
+    </style>
+"""
+st.markdown(hide_button_style, unsafe_allow_html=True)
+
+# base paths
+base_path = Path("data")
+inference_path = base_path / "inference"
+
+# list examples and predictions
+inference_paths = sorted(list(inference_path.iterdir()))
+
+example_names = [p.stem for p in inference_paths]
+inference_dict = {p.stem: p for p in inference_paths}
+
+# Create selection menu
+container_predictions = st.sidebar.container(border=True)
+with container_predictions:
+    prediction_selectbox = st.selectbox(
+        "Select an example",
+        options=example_names,
+        index=None,
+        key="selectbox_pred",
+    )
+    is_prediction_selected = prediction_selectbox is not None
+    if is_prediction_selected:
+        try:
+            # add loading of the selected parquet prediction
+            chosen_tile_path = inference_dict[prediction_selectbox]
+            with open(chosen_tile_path, "rb") as f:
+                prediction_file = f.read()
+        except:
+            st.warning("File not found")
+            chosen_tile_path = None
+            prediction_file = None
+    else:
+        prediction_file = None
+        chosen_tile_path = None
+
+container = get_clean_rendering_container(prediction_selectbox)
+
+
+# Stange Hack to always update the map height I guess
+def change_key():
+    st.session_state["key_map"] = str(datetime.now())
+
+
+height_value = st.sidebar.radio(
+    "Map height",
+    options=[800, 500],
+    horizontal=True,
+    key="height_map",
+    on_change=change_key,
+)
+container = get_clean_rendering_container(height_value)
+
+# draw map
+interactive_map = get_map_crop()
+
+if prediction_selectbox is not None:
+    # draw prediction
+    draw_gdf(interactive_map, chosen_tile_path, "Prediction", style_call, id_class)
+
+
+with container.form(key="form1", clear_on_submit=True):
+    folium.LayerControl().add_to(interactive_map)
+    output_map = st_folium(
+        interactive_map,
+        width=None,
+        height=height_value,
+        returned_objects=["all_drawings"],
+        key=st.session_state.get("key_map", "key_map"),
+    )
+    submit = st.form_submit_button("Recenter map")
+
+# Update messages
+
+
+for name, path in inference_dict.items():
+    if path not in st.session_state.keys():
+        log.info(f"Loading parquet {name}")
+        st.session_state[path] = get_parquet(path, id_class)

data/inference/France_2022.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ce8babda3c49b070f2770d9d08257f5d7f3499fe959e3a31cb17ebb6d996604e
+size 4754525

data/inference/Germany_2021.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6328e97907549ddca90a3b11430fabfed5dbfb16c2febdb96d22f8c367c59ad8
+size 2679827

data/inference/Netherlands_2023.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:92fba1cc19cd9b73adabc4a4c15be15a8fa266d240083cfbc07c6aafaa518344
+size 5718056

requirements.txt

@@ -0,0 +1,8 @@
+streamlit==1.33.0
+pathlib==1.0.1
+geopandas==0.14.2
+matplotlib==3.8.2
+folium==0.15.1
+streamlit-folium==0.18.0
+loguru==0.7.2
+shapely==2.0.3

utils/folium.py

@@ -0,0 +1,128 @@
+import branca
+import folium
+import geopandas as gpd
+import streamlit as st
+from folium.plugins import Draw
+
+
+def get_map_crop():
+    m = folium.Map(
+        location=[50, 32],
+        zoom_start=4,
+        tiles="Esri.WorldImagery",
+        attributionControl=False,
+        prefer_canvas=True,
+    )
+    Draw(
+        export=False,
+        show_geometry_on_click=False,
+        draw_options={
+            "polyline": False,
+            "polygon": False,
+            "circle": False,
+            "circlemarker": False,
+            "marker": False,
+        },
+    ).add_to(m)
+    legend_html = """
+    {% macro html(this, kwargs) %}
+    <div style="
+        position: fixed; 
+        bottom: 35px;
+        left: 30px;
+        width: 130px;
+        line-height: 3px;
+        height: 120px;
+        z-index:9999;
+        font-size:14px;
+        ">
+        <p><a style="color:#00ff00;font-size:150%;margin-left:20px;">&FilledSmallSquare;</a>&emsp;Grassland</p>
+        <p><a style="color:#8b4513;font-size:150%;margin-left:20px;">&FilledSmallSquare;</a>&emsp;Cereals</p>
+        <p><a style="color:#ffffff;font-size:150%;margin-left:20px;">&FilledSmallSquare;</a>&emsp;Other</p>
+        <p><a style="color:#00ffff;font-size:150%;margin-left:20px;">&FilledSmallSquare;</a>&emsp;Flowers</p>
+        <p><a style="color:#800080;font-size:150%;margin-left:20px;">&FilledSmallSquare;</a>&emsp;Vegetables</p>
+        <p><a style="color:#006400;font-size:150%;margin-left:20px;">&FilledSmallSquare;</a>&emsp;Fruit trees</p>
+    </div>
+    <div style="
+        position: fixed; 
+        bottom: 50px;
+        left: 30px;
+        width: 130px;
+        line-height: 3px;
+        height: 120px; 
+        z-index:9998;
+        font-size:14px;
+        background-color: #ffffff;
+
+        opacity: 0.85;
+        ">
+    </div>
+    {% endmacro %}
+    """
+    legend = branca.element.MacroElement()
+    legend._template = branca.element.Template(legend_html)
+    m.get_root().add_child(legend)
+    return m
+
+
+def get_parquet(tile_path, id_classes):
+    data = gpd.read_parquet(tile_path).to_crs(epsg="4326")
+    data["class_name"] = [id_classes[i] for i in data["class"]]
+    data["geometry"] = data["geometry"].simplify(0.0001)
+    return data
+
+
+def draw_gdf(
+    _map,
+    tile_path,
+    name,
+    _style_call,
+    id_classes,
+):
+    if tile_path not in st.session_state.keys():
+        tile_gdf = get_parquet(tile_path, id_classes)
+    else:
+        tile_gdf = st.session_state[tile_path]
+    feature_group = folium.FeatureGroup(f"{name} layer")
+    tooltip = folium.GeoJsonTooltip(
+        fields=["class_name", "area"],
+        aliases=["Crop type: \t", "Area (m²): \t"],
+        localize=True,
+        sticky=False,
+        labels=True,
+        style="""
+                background-color: #F0EFEF;
+                border: 2px solid black;
+                border-radius: 3px;
+                box-shadow: 3px;
+            """,
+        max_width=800,
+    )
+
+    folium.GeoJson(tile_gdf, style_function=_style_call, tooltip=tooltip).add_to(
+        feature_group
+    )
+    feature_group.add_to(_map)
+    bound = feature_group.get_bounds()
+    _map.fit_bounds(bound)
+
+
+def get_clean_rendering_container(app_state: str):
+    """Makes sure we can render from a clean slate on state changes."""
+    slot_in_use = st.session_state.slot_in_use = st.session_state.get(
+        "slot_in_use", "a"
+    )
+    if app_state != st.session_state.get("previous_state", app_state):
+        if slot_in_use == "a":
+            slot_in_use = st.session_state.slot_in_use = "b"
+        else:
+            slot_in_use = st.session_state.slot_in_use = "a"
+
+    st.session_state.previous_state = app_state
+
+    slot = {
+        "a": st.empty(),
+        "b": st.empty(),
+    }[slot_in_use]
+
+    return slot.container()