import streamlit as st

from frontend.st_utils import get_backend_api_client
from frontend.utils import generate_random_name

backend_api_client = get_backend_api_client()


def get_default_config_loader(controller_name: str):
    all_configs = backend_api_client.get_all_controllers_config()
    existing_configs = [config["id"].split("_")[0] for config in all_configs]
    default_dict = {"id": generate_random_name(existing_configs)}
    default_config = st.session_state.get("default_config", default_dict)
    config_controller_name = default_config.get("controller_name")
    if default_config is None or controller_name != config_controller_name:
        st.session_state["default_config"] = default_dict
    with st.expander("Configurations", expanded=True):
        c1, c2 = st.columns(2)
        with c1:
            use_default_config = st.checkbox("Use default config", value=True)
        with c2:
            if not use_default_config:
                configs = [config for config in all_configs if config["controller_name"] == controller_name]
                if len(configs) > 0:
                    default_config = st.selectbox("Select a config", [config["id"] for config in configs])
                    st.session_state["default_config"] = next(
                        (config for config in all_configs if config["id"] == default_config), None)
                    st.session_state["default_config"]["id"] = st.session_state["default_config"]["id"].split("_")[0]
                else:
                    st.warning("No existing configs found for this controller.")