import streamlit as st
import opendashboards.utils.plotting as plotting

# @st.cache_data
def uid_diversty(df, rm_failed=True):
    return st.plotly_chart(
        plotting.plot_uid_diversty(
            df,
            remove_unsuccessful=rm_failed
        ),
        use_container_width=True
    )

# @st.cache_data
def leaderboard(df, ntop, group_on, agg_col, agg, alias=False):
    return st.plotly_chart(
        plotting.plot_leaderboard(
            df,
            ntop=ntop,
            group_on=group_on,
            agg_col=agg_col,
            agg=agg,
            alias=alias
        ),
        use_container_width=True
    )

# @st.cache_data
def completion_rewards(df, completion_col, reward_col, uid_col, ntop, completions=None, completion_regex=None):
    return st.plotly_chart(
        plotting.plot_completion_rewards(
            df,
            msg_col=completion_col,
            reward_col=reward_col,
            uid_col=uid_col,
            ntop=ntop,
            completions=completions,
            completion_regex=completion_regex
        ),
        use_container_width=True
    )
    
def weights(df, uids, ntop=10):
    return st.plotly_chart(
        plotting.plot_weights(
            df,
            uids=[f'UID-{i}' for i in uids],
            ntop=ntop
        ),
        use_container_width=True
    )