import gradio as gr
from prompt_parser import Parse_Prompt
from scoreboard import Score
import warnings

warnings.filterwarnings("ignore")

arena = Parse_Prompt()
score = Score()

with gr.Blocks(fill_height=True) as app:
    with gr.Tab("🪖 Battle Field"):
        gr.Markdown('''## ⚔️ LLM: Large Language Mayhem
                    - Voting should be fair and based on the performance of the models.
                    - No cheating or manipulating the outcomes.
                    - Press 🎲 Random to change the models.
                    - Everything else except the Random button will only clear the screen, model being the same.
                    - Have fun and enjoy the language mayhem!
                    - Warrior names will be visible after your first query (after random also you will have to give a query to see changes)
                    - Don't See Warrior names before voting
                    ''')
        with gr.Row():
            with gr.Accordion("🥷 Warriors", open=False):
                gr.Dataframe([[model] for model in arena.models], col_count=1, headers=["🥷"])
        with gr.Group():
            with gr.Row():
                with gr.Column():
                    chatbox1 = gr.Chatbot(label="Warrior A", show_copy_button=True)
                with gr.Column():
                    chatbox2 = gr.Chatbot(label="Warrior B", show_copy_button=True)
            textbox = gr.Textbox(show_label=False, placeholder="👉 Enter your prompt")
        with gr.Row():
            with gr.Accordion("🥷 Current Warriors", open=False):
                with gr.Row():
                    war1 = gr.Textbox(arena.model1, interactive=False, show_label=False)
                    war2 = gr.Textbox(arena.model2, interactive=False, show_label=False)
        with gr.Row():
            with gr.Accordion("👆 Vote", open=False):
                with gr.Row():
                    vote_a = gr.ClearButton([textbox, chatbox1, chatbox2], value="👈 Warrior A Wins")
                    vote_b = gr.ClearButton([textbox, chatbox1, chatbox2], value="👉 Warrior B Wins")
                    vote_tie = gr.ClearButton([textbox, chatbox1, chatbox2], value="🤝 Both Won")

        submit_button = gr.Button("Submit")
        with gr.Row():
            new_round = gr.ClearButton([textbox, chatbox1, chatbox2], value="🎲New Round🎲")
            clear = gr.ClearButton([textbox, chatbox1, chatbox2], value="🧹 Clear")
        with gr.Row():
            with gr.Accordion("🔩 Parameters", open=False):
                temp_slider = gr.Slider(0, 1, value=0.7, step=0.1, label="Temprature")

        textbox.submit(
            fn=arena.gen_output,
            inputs=[temp_slider, textbox],
            outputs=[chatbox1, chatbox2]
        )
        textbox.submit(fn=arena.current_model2, outputs=war2)
        textbox.submit(fn=arena.current_model1, outputs=war1)

        submit_button.click(fn=arena.gen_output, inputs=[temp_slider, textbox], outputs=[chatbox1, chatbox2])
        submit_button.click(fn=arena.current_model1, outputs=war1)
        submit_button.click(fn=arena.current_model2, outputs=war2)

        vote_a.click(fn=lambda: score.update(arena.model1, score.df))
        vote_b.click(fn=lambda: score.update(arena.model2, score.df))
        vote_tie.click(fn=arena.change_models)
        new_round.click(fn=arena.change_models)
        clear.click(fn=arena.clear_history)

    with gr.Tab("💯 Score Board") as data_tab:
        gr.Markdown('''## ⚔️ LLM: Large Language Mayhem
                    - Voting should be fair and based on the performance of the models.
                    - No cheating or manipulating the outcomes.
                    - Click on Generate button to Update the 💯 Scoreboard.
                    ''')
        generate_btn = gr.Button("🔄 Generate Scoreboard")
        scoreboard_df = gr.Dataframe(type="pandas", label="Scoreboard", headers=["", "", ""])

        generate_btn.click(
            fn=score.df_show,
            outputs=scoreboard_df
        )

app.launch(share=True)