Spaces:
Sleeping
Sleeping
| 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) | |