map-predict_predict.py

def predict(max_branch: int = 3,
            max_length: int = 10) -> FeatureGroup | None:
    fg = folium.FeatureGroup(name="polylines")

    if "token_list" in st.session_state:
        token_list = st.session_state["token_list"]
        seed = token_list[-3:-1]
        if len(seed) > 1:
            paths = expand_seed(seed[0], seed[1],
                                max_branch=max_branch,
                                max_length=max_length)

            for path in paths[:max_branch]:
                fg.add_child(path.get_polyline())
    return fg