import pandas as pd

import gradio as gr
from gradio_leaderboard import Leaderboard

from utils import fetch_hf_results, show_output_box
from constants import ASSAY_LIST, ASSAY_RENAME, ASSAY_EMOJIS, ASSAY_DESCRIPTION
from about import ABOUT_TEXT, FAQS
from submit import make_submission


def format_leaderboard_table(df_results: pd.DataFrame, assay: str | None = None):
    # Previous things that were nice in the constellaration leaderboard:
    # Having a submission time column, and a user column where the username is clickable (this is a pro for usability but con for anonymity)
    # full_df.rename(columns={'submission_time': 'submission time', 'problem_type': 'problem type'}, inplace=True)
    # to_show['user'] = to_show['user'].apply(lambda x: make_user_clickable(x)).astype(str)
    column_order = ["model", "property", "spearman", "spearman_cross_val"]
    df = df_results.query("assay.isin(@ASSAY_RENAME.keys())").copy()
    if assay is not None:
        df = df[df["assay"] == assay]
    df = df[column_order]
    return df.sort_values(by="spearman", ascending=False)


def get_leaderboard_object(assay: str | None = None):
    filter_columns = ["model"]
    if assay is None:
        filter_columns.append("property")
    # TODO how to sort filter columns alphabetically?
    Leaderboard(
        # TODO(Lood) check that this actually refreshes using the function
        value=format_leaderboard_table(df_results=fetch_hf_results(), assay=assay),
        datatype=["str", "str", "str", "number"],
        select_columns=["model", "property", "spearman", "spearman_cross_val"],
        search_columns=["model"],
        filter_columns=filter_columns,
        every=60,
        render=True,
    )


with gr.Blocks() as demo:
    gr.Markdown("""
        ## Welcome to the Ginkgo Antibody Developability Benchmark!

        **Beta version, not publicly launched yet**

        Participants can submit their model to the leaderboard by uploading a CSV file (see the "✉️ Submit" tab).
        See more details in the "❔About" tab.
        """)
    with gr.Tabs(elem_classes="tab-buttons"):
        with gr.TabItem("❔About", elem_id="abdev-benchmark-tab-table"):
            gr.Image(
                value="./assets/competition_logo.jpg",
                show_label=False,
                # elem_classes=["resized-image"],
                show_download_button=False,
                width="50vw",  # 50% of the "viewport width"
            )
            gr.Markdown(ABOUT_TEXT)
            for question, answer in FAQS.items():
                with gr.Accordion(question):
                    gr.Markdown(answer)

        # Procedurally make these 5 tabs
        for assay in ASSAY_LIST:
            with gr.TabItem(
                f"{ASSAY_EMOJIS[assay]} {ASSAY_RENAME[assay]}",
                elem_id="abdev-benchmark-tab-table",
            ):
                gr.Markdown(f"# {ASSAY_DESCRIPTION[assay]}")
                get_leaderboard_object(assay=assay)

        with gr.TabItem("🚀 Overall", elem_id="abdev-benchmark-tab-table"):
            gr.Markdown(
                "# Antibody Developability Benchmark Leaderboard over all properties"
            )
            get_leaderboard_object()

        with gr.TabItem("✉️ Submit", elem_id="boundary-benchmark-tab-table"):
            gr.Markdown(
                """
            # Antibody Developability Submission
            Upload a CSV to get a score!

            Please use your Hugging Face account name to submit your model - we use this to track separate submissions, but only Hugging Face/Ginkgo will see these usernames (unless you choose to make them public).
            Your submission will be evaluated and added to the leaderboard.
            """
            )
            filename = gr.State(value=None)
            eval_state = gr.State(value=None)
            user_state = gr.State(value=None)
            anonymous_state = gr.State(value=False)

            login_button = gr.LoginButton(
                value="Sign in with Hugging Face to see account name"
            )  # Note(Lood): Is this mandatory?

            gr.DownloadButton(
                label="📥 Download example submission CSV",
                value="data/example-predictions.csv",
                variant="secondary",
            )

            with gr.Row():
                with gr.Column():
                    username_input = gr.Textbox(
                        label="Username",
                        placeholder="Enter your Hugging Face username",
                        info="This will be displayed on the leaderboard.",
                    )

                    anonymous_checkbox = gr.Checkbox(
                        label="Would you like to keep your submission anonymous?"
                    )  # Can make this ticked by default
                with gr.Column():
                    submission_file = gr.File(label="Submission CSV")

            # TODO(Lood): How do we get the username from the login button instead? Do we want this?
            username_input.change(
                fn=lambda x: x if x.strip() else None,
                inputs=username_input,
                outputs=user_state,
            )

            submit_btn = gr.Button("Evaluate")
            message = gr.Textbox(label="Status", lines=1, visible=False)
            # help message
            gr.Markdown(
                "If you have issues with submission or using the leaderboard, please start a discussion in the Community tab of this Space."
            )

            submit_btn.click(
                make_submission,
                inputs=[submission_file, user_state, anonymous_state],
                outputs=[message],
            ).then(
                fn=show_output_box,
                inputs=[message],
                outputs=[message],
            )
    # Footnote
    gr.Markdown(
        """
        <div style="text-align: center; font-size: 14px; color: gray; margin-top: 2em;">
        📬 For questions or feedback, contact <a href="mailto:datapoints@ginkgobioworks.com">datapoints@ginkgobioworks.com</a> or visit the Community tab at the top of this page.
        </div>
        """,
        elem_id="contact-footer",
    )

if __name__ == "__main__":
    demo.launch()