import streamlit as st

from .enums import Page


def clear_session_state():
    st.session_state.page = Page.ARTIST_SEARCH


def artist_select(artists: list[str]) -> str | None:
    artist_name = st.session_state.artist_name
    if artist_name and artist_name in artists:
        index = artists.index(artist_name)
    else:
        index = None
    selected_artist = st.selectbox(
        "Choose an Artist",
        options=artists,
        accept_new_options=False,
        index=index,
        key="selected_artist",
    )
    return selected_artist