import pandas as pd
import streamlit as st

data_df = pd.DataFrame(
    {
        "category": [
            ["exploration", "visualization"],
            ["llm", "visualization"],
            ["exploration"],
        ],
    }
)

st.data_editor(
    data_df,
    column_config={
        "category": st.column_config.MultiselectColumn(
            "App Categories",
            options=[
                "exploration",
                "visualization",
                "llm",
            ],
            color=["orange", "red", "#ffc38a"],
            format_func=lambda x: x.capitalize(),
        ),
    },
)

# Or use it for colored tags in a read-only dataframe:

st.dataframe(
    data_df,
    column_config={
        "category": st.column_config.MultiselectColumn(
            "App Categories",
            options=["exploration", "visualization", "llm"],
            color=["orange", "red", "#ffc38a"],
            format_func=lambda x: x.capitalize(),
        ),
    },
)