import streamlit as st

from common.basket_analysis import results, intro_text


st.subheader("Market basket analysis.")
st.write("This view currently includes data from all territories combined per artist.")
data_basket = st.session_state["basket_assoc_df"]
data_basket_detailed = st.session_state["basket_assoc_item_df"]


with st.sidebar:
    selected_artist = st.selectbox(
        label="Choose artist",
        options=st.session_state["artists"],
        index=st.session_state["index"],
        key="artist",
    )
# Update the index. It is used in the select box.
st.session_state["index"] = st.session_state["artists"].index(st.session_state["artist"])

explain_basket_analysis = st.toggle("Show detailed explanation")
if explain_basket_analysis:
    intro_text()

st.divider()
# st.write("Product category view:")
# results(data_basket.drop_duplicates(subset=["lift", "support"]), st.session_state["artist"])
# st.divider()


results(
    data_basket_detailed.drop_duplicates(subset=["lift", "support"]), st.session_state["artist"]
)