app.py

import streamlit as st

from src import logic as lg
from src import load_or_generate_objective_scores


st.set_page_config(
    page_title="Isle of Musica",
    page_icon="img/avocado-emoji.png",
    initial_sidebar_state="expanded")


lg.initialize_session_state()
# lg.sidebar()
num_voters = st.session_state["num_voters"]
num_songs = st.session_state["num_songs"]
num_winners = st.session_state["num_winners"]
song_df = load_or_generate_objective_scores(num_songs)

# st.dataframe(song_df)


# Simulation Explanation and Interactive Demo
section_title = "The Isle of Musica"
st.title(section_title)
lg.interactive_demo(song_df)
baseline_titles, _ = lg.establish_baseline(song_df)


show_state = 0
st.markdown("---")
_, center, _ = st.columns([3,2,2])
label = "Let's Go"
lg.proceed_button(center, label, show_state)


show_state = 1
if st.session_state['show_state'] >= show_state:
    # section_title = "The Isle of Musica"
    section_title = "Song of the Year Contest on The Isle of Musica"
    lg.write_story(section_title, header_level=2)
    lg.select_num_winners(section_title)
    st.markdown("---")
    _, center, _ = st.columns([2,2,2])
    label = "Let's simulate the first contest!"
    lg.proceed_button(center, label, show_state)


show_state = 2
if lg.this_section_is_viewable(show_state):
    section_title = "Keep it Simple"
    lg.write_story(section_title)
    subtitle = f"Voter cast one vote each for their favorite {num_winners} songs, but no voter has time to listen to every song."
    subtitles = {"current": subtitle}
    sim_alpha, _ = lg.simulation_section(song_df, section_title, 
                                        alphabetical=True,
                                        baseline_results=baseline_titles,
                                        subtitles=subtitles)
    st.markdown("---")
    _, center, _ = st.columns([2,2,2])
    label = "Let's simulate how they can solve this issue?"
    show_state = lg.proceed_button(center, label, show_state)


show_state = 3
if lg.this_section_is_viewable(show_state):
    section_title = "Ensuring Fairness"
    lg.write_story(section_title)
    subtitle = f"All voters take the time to listen and rank all {num_songs} songs."
    subtitles = {"condorcet": subtitle}
    sim1, sim1_chart_df = lg.simulation_section(song_df, section_title,
                                                baseline_results=baseline_titles,
                                                subtitles=subtitles) 
    st.markdown("---")  
    _, center, _ = st.columns([2,2,2])
    label = "Is there a way that's more feasible?"
    lg.proceed_button(center, label, show_state)


# Random Samples
show_state = 4
if lg.this_section_is_viewable(show_state):
    section_title = "The Power of Randomness"
    lg.write_story(section_title)

    col1, col2 = st.columns(2)
    label = "How many songs does each voter get to listen to?"
    listen_limit = col1.slider(label, 
        value=st.session_state["listen_limit"], 
        min_value=50, 
        max_value=num_songs,
        step=50,
        key=section_title+"_listen_limit")
    label = "How many songs does each voter get to rank?"
    initial = st.session_state["ballot_limit"] 
    ballot_limit = col2.slider(label, 
        value=initial if initial < listen_limit else listen_limit,
        min_value=25, 
        max_value=listen_limit,
        step=25,
        key=section_title+"_ballot_limit")
    
    subtitles = {
        "condorcet":    f"Voters listen to {listen_limit} songs and rank their top {ballot_limit}.",
        "current":      f"Voters listen to {listen_limit} songs and cast one vote each for their favorite {num_winners}."
    }

    sim2, _ = lg.simulation_section(song_df, section_title, 
        listen_limit=listen_limit,
        ballot_limit=ballot_limit,
        baseline_results=baseline_titles,
        subtitles=subtitles,
        )

    st.markdown("---")
    _, center, _ = st.columns([2,2,2])
    st.text("")
    label = "Can this help if voters are really biased?"
    lg.proceed_button(center, label, show_state)


# Bloc 
show_state = 5
if lg.this_section_is_viewable(show_state):
    section_title = "Nullifying Bloc Voting"
    lg.write_story(section_title)

    col1, col2 = st.columns(2)
    label = "How many songs does each voter get to listen to?"
    listen_limit = col1.slider(label, 
        value=st.session_state["listen_limit"],
        min_value=50, 
        max_value=num_songs,
        step=10,
        key=section_title+"_listen_limit")
    label = "How many songs does each voter get to rank?"
    initial = st.session_state["ballot_limit"] 
    ballot_limit = col2.slider(label, 
        value=initial if initial < listen_limit else listen_limit,
        min_value=10, 
        max_value=listen_limit,
        step=10,
        key=section_title+"_ballot_limit")

    label = "How many corrupt artists are there?"
    num_mafiosos = col1.number_input(label,
        value=2,
        min_value=0,
        max_value=10,
        step=1)
    label = "How large are each of their mafias?"
    max_mafia_size = num_voters//10
    mafia_size = col2.slider(label, 
        value=max_mafia_size//2, 
        min_value=0, 
        max_value=max_mafia_size,
        step=5)
    
    subtitles = {
        "condorcet":    f"Voters listen to {listen_limit} songs and rank their top {ballot_limit}.",
        "current":      f"Voters listen to {listen_limit} songs and cast one vote each for their favorite {num_winners}."
    }
    
    sim3, _ = lg.simulation_section(song_df, section_title, 
        listen_limit=listen_limit, ballot_limit=ballot_limit,
        baseline_results=baseline_titles,
        num_mafiosos=num_mafiosos, mafia_size=mafia_size,
        subtitles=subtitles,
        )
    
    st.markdown("---")
    _, center, _ = st.columns([2,1,2])
    st.text("")
    label = "Wrap Things Up"
    lg.proceed_button(center, label, show_state)


# Conclusion
show_state = 6
if lg.this_section_is_viewable(show_state):
    lg.write_story("Conclusion", header_level=2)
    st.write("")
    lg.write_story("Some Notes on Simulations", header_level=3)

    st.markdown("")
    st.markdown("---")
    section_title = "Sandbox"
    lg.write_story(section_title, header_level=3)

    col1, col2 = st.columns(2)

    with col1:
        label = "Choose the number of contest finalists."
        options = st.session_state["finalist_options"]
        num_winners = st.selectbox(label, options)

    with col2:
        label = "How many voters particiapte in the contest?"
        num_voters = st.slider(label,
                            value=st.session_state["num_voters"],
                            min_value=250,
                            max_value=5000,
                            step=50)
    
    # Initializing these again ensures the two widgets are aligned
    col1, col2 = st.columns(2)

    with col1:
        label = "How many songs does each voter get to listen to?"
        listen_limit = col1.slider(label, 
            value=st.session_state["listen_limit"],
            min_value=50, 
            max_value=num_songs,
            step=10,
            key=section_title+"_listen_limit")

    with col2:
        label = "How many songs does each voter get to rank?"
        initial = st.session_state["ballot_limit"] 
        ballot_limit = col2.slider(label, 
            value=initial if initial < listen_limit else listen_limit,
            min_value=10, 
            max_value=listen_limit,
            step=10,
            key=section_title+"_ballot_limit")
        
    col1, col2 = st.columns(2)

    with col1:
        label = "How many corrupt artists are there?"
        num_mafiosos = col1.number_input(label,
            value=2,
            min_value=0,
            max_value=10,
            step=1,
            key=f"num_mafiosos_{section_title}")

    with col2:
        label = "How large are each of their mafias?"
        max_mafia_size = num_voters//10
        mafia_size = col2.slider(label, 
            value=max_mafia_size//2, 
            min_value=0, 
            max_value=max_mafia_size,
            step=5,
            key=f"mafia_size_{section_title}")
        
    baseline_titles, _ = lg.establish_baseline(song_df, num_winners=num_winners, no_story=True)    
    subtitles = {
        "condorcet":    f"{num_voters} voters listen to {listen_limit} songs and rank their top {ballot_limit}.",
        "current":      f"{num_voters} voters listen to {listen_limit} songs and cast one vote each for their favorite {num_winners}."
    }
    sandbox, _ = lg.simulation_section(song_df, section_title,
        num_voters=num_voters, num_winners=num_winners,
        listen_limit=listen_limit, ballot_limit=ballot_limit,
        baseline_results=baseline_titles,
        num_mafiosos=num_mafiosos, mafia_size=mafia_size,
        subtitles=subtitles,
        )