Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import pandas as pd | |
| import re | |
| import csv | |
| import pandas as pd | |
| def update_scores(winner, loser, k_factor=100): | |
| score_difference = int(k_factor/(winner/loser)) | |
| winner += score_difference | |
| loser -= score_difference | |
| return winner, loser | |
| def vote_startup(opponents_df): | |
| try: | |
| opponents_df = opponents_df[["elo_score", "descriptor", "opponent"]] | |
| except: | |
| opponents_df = pd.DataFrame(columns=['elo_score', 'descriptor', 'opponent']) | |
| if len(opponents_df)>0: | |
| if len(opponents_df)>10: | |
| slice_size = 4 | |
| slice = int(len(opponents_df)/slice_size) | |
| sample = opponents_df[slice:(slice_size-1)*slice].sample(frac=1).iloc[0] | |
| opponent, descriptor = sample["opponent"], sample["descriptor"] | |
| else: | |
| sample = opponents_df.sample(frac=1).iloc[0] | |
| opponent, descriptor = sample["opponent"], sample["descriptor"] | |
| if len(opponents_df) > 1: | |
| # Randomly select a opponent to compare with | |
| sample = opponents_df.sample(frac=1) | |
| comparison_opponent = sample.iloc[0] | |
| if comparison_opponent['opponent'] == opponent and comparison_opponent['descriptor'] == descriptor: | |
| comparison_opponent = sample.iloc[1] | |
| first_df = opponents_df[opponents_df["opponent"]==opponent][opponents_df["descriptor"]==descriptor] | |
| first_string = first_df["opponent"].tolist()[0]+" - "+first_df["descriptor"].tolist()[0] | |
| second_df = comparison_opponent | |
| second_string = second_df["opponent"]+" - "+second_df["descriptor"] | |
| return f"Do you like '{descriptor} - {opponent}' better than '{comparison_opponent['descriptor']} - {comparison_opponent['opponent']}'?", first_string, second_string, display_rankings(opponents_df) | |
| else: | |
| return "Add some opponents to start voting!", "", "", display_rankings(opponents_df) | |
| def clean_string(string): | |
| string = string.strip().replace(" "," ").lower() | |
| string = " ".join([x[0].upper()+x[1:] for x in string.split()]) | |
| return string | |
| def add_and_compare(descriptor, opponent, opponents_df): | |
| try: | |
| opponents_df = opponents_df[["elo_score", "descriptor", "opponent"]] | |
| except: | |
| opponents_df = pd.DataFrame(columns=['elo_score', 'descriptor', 'opponent']) | |
| if descriptor != "" and opponent != "": | |
| descriptor = clean_string(descriptor) | |
| opponent = clean_string(opponent) | |
| new_opponent = pd.DataFrame({'descriptor': [descriptor], 'opponent': [opponent], 'elo_score': [1000]}) | |
| opponents_df = pd.concat([opponents_df, new_opponent], ignore_index=True) | |
| opponents_df.to_csv("opponents_df.csv") | |
| opponents_df = opponents_df[["elo_score", "descriptor", "opponent"]] | |
| return "", "", display_rankings(opponents_df) | |
| # Function to update Elo ratings based on user's choice | |
| def update_ratings_pos(first_string, second_string, opponents_df): | |
| try: | |
| opponents_df = opponents_df[["elo_score", "descriptor", "opponent"]] | |
| except: | |
| opponents_df = pd.DataFrame(columns=['elo_score', 'descriptor', 'opponent']) | |
| if len(opponents_df)==0: | |
| return "Add some opponents to start voting!", "", "", display_rankings(opponents_df) | |
| if first_string != "": | |
| opponents_df["combined"] = opponents_df["opponent"] + " - " + opponents_df["descriptor"] | |
| loser = opponents_df[opponents_df["combined"] == second_string] | |
| winner = opponents_df[opponents_df["combined"] == first_string] | |
| # Update Elo scores | |
| winner_score, loser_score = update_scores(winner['elo_score'].values[0], loser['elo_score'].values[0]) | |
| opponents_df.at[winner.index[0], 'elo_score'] = winner_score | |
| opponents_df.at[loser.index[0], 'elo_score'] = loser_score | |
| opponents_df = opponents_df.sort_values(by='elo_score', ascending=False) | |
| opponents_df.to_csv("opponents_df.csv") | |
| if len(opponents_df)>10: | |
| slice_size = 4 | |
| slice = int(len(opponents_df)/slice_size) | |
| sample = opponents_df[slice:(slice_size-1)*slice].sample(frac=1).iloc[0] | |
| opponent, descriptor = sample["opponent"], sample["descriptor"] | |
| else: | |
| sample = opponents_df.sample(frac=1).iloc[0] | |
| opponent, descriptor = sample["opponent"], sample["descriptor"] | |
| if len(opponents_df) > 1: | |
| # Randomly select a opponent to compare with | |
| sample = opponents_df.sample(frac=1) | |
| comparison_opponent = sample.iloc[0] | |
| if comparison_opponent['opponent'] == opponent and comparison_opponent['descriptor'] == descriptor: | |
| comparison_opponent = sample.iloc[1] | |
| first_df = opponents_df[opponents_df["opponent"]==opponent][opponents_df["descriptor"]==descriptor] | |
| first_string = first_df["opponent"].tolist()[0]+" - "+first_df["descriptor"].tolist()[0] | |
| second_df = comparison_opponent | |
| second_string = second_df["opponent"]+" - "+second_df["descriptor"] | |
| return f"Do you like '{descriptor} - {opponent}' better than '{comparison_opponent['descriptor']} - {comparison_opponent['opponent']}'?", first_string, second_string, display_rankings(opponents_df) | |
| else: | |
| return "Add some opponents to start voting!", "", "", display_rankings(opponents_df) | |
| # Function to update Elo ratings based on user's choice | |
| def update_ratings_neg(first_string, second_string, opponents_df): | |
| try: | |
| opponents_df = opponents_df[["elo_score", "descriptor", "opponent"]] | |
| except: | |
| opponents_df = pd.DataFrame(columns=['elo_score', 'descriptor', 'opponent']) | |
| if len(opponents_df)==0: | |
| return "Add some opponents to start voting!", "", "", display_rankings(opponents_df) | |
| if first_string != "": | |
| opponents_df["combined"] = opponents_df["opponent"] + " - " + opponents_df["descriptor"] | |
| loser = opponents_df[opponents_df["combined"] == first_string] | |
| winner = opponents_df[opponents_df["combined"] == second_string] | |
| # Update Elo scores | |
| winner_score, loser_score = update_scores(winner['elo_score'].values[0], loser['elo_score'].values[0]) | |
| opponents_df.at[winner.index[0], 'elo_score'] = winner_score | |
| opponents_df.at[loser.index[0], 'elo_score'] = loser_score | |
| opponents_df = opponents_df.sort_values(by='elo_score', ascending=False) | |
| opponents_df.to_csv("opponents_df.csv") | |
| if len(opponents_df)>10: | |
| slice_size = 4 | |
| slice = int(len(opponents_df)/slice_size) | |
| sample = opponents_df[slice:(slice_size-1)*slice].sample(frac=1).iloc[0] | |
| opponent, descriptor = sample["opponent"], sample["descriptor"] | |
| else: | |
| sample = opponents_df.sample(frac=1).iloc[0] | |
| opponent, descriptor = sample["opponent"], sample["descriptor"] | |
| if len(opponents_df) > 1: | |
| # Randomly select a opponent to compare with | |
| sample = opponents_df.sample(frac=1) | |
| comparison_opponent = sample.iloc[0] | |
| if comparison_opponent['opponent'] == opponent and comparison_opponent['descriptor'] == descriptor: | |
| comparison_opponent = sample.iloc[1] | |
| first_df = opponents_df[opponents_df["opponent"]==opponent][opponents_df["descriptor"]==descriptor] | |
| first_string = first_df["opponent"].tolist()[0]+" - "+first_df["descriptor"].tolist()[0] | |
| second_df = comparison_opponent | |
| second_string = second_df["opponent"]+" - "+second_df["descriptor"] | |
| return f"Do you like '{descriptor} - {opponent}' better than '{comparison_opponent['descriptor']} - {comparison_opponent['opponent']}'?", first_string, second_string, display_rankings(opponents_df) | |
| else: | |
| return "Add some opponents to start voting!", "", "", display_rankings(opponents_df) | |
| def display_rankings(opponents_df=pd.DataFrame(columns=['elo_score', 'descriptor', 'opponent'])): | |
| opponents_df = opponents_df.sort_values(by='elo_score', ascending=False) | |
| opponents_df = opponents_df[["elo_score", "descriptor", "opponent"]] | |
| opponents_df.to_csv("opponents_df.csv") | |
| return opponents_df | |
| def export_csv(opponents_df): | |
| # Function to export DataFrame to CSV | |
| save_df = opponents_df | |
| save_df.to_csv("opponents_df.csv") | |
| return "opponents_df.csv" | |
| def import_csv(file, opponents_df): | |
| if file is not None: | |
| #file_content = file.decode('utf-8') | |
| new_df = pd.read_csv(file) | |
| try: | |
| opponents_df = opponents_df[["elo_score", "descriptor", "opponent"]] | |
| except: | |
| opponents_df = pd.DataFrame(columns=['elo_score', 'descriptor', 'opponent']) | |
| new_df = new_df[["elo_score", "descriptor", "opponent"]] | |
| opponents_df = pd.concat([opponents_df,new_df]) | |
| opponents_df = opponents_df.drop_duplicates(subset=['descriptor', 'opponent']) | |
| return opponents_df | |
| # Function to remove a opponent | |
| def remove_opponent(descriptor, opponent, opponents_df): | |
| # Find and remove the opponent from the DataFrame | |
| descriptor = clean_string(descriptor) | |
| opponent = clean_string(opponent) | |
| opponents_df = opponents_df[~((opponents_df["descriptor"] == descriptor) & (opponents_df["opponent"] == opponent))] | |
| return opponents_df[["elo_score", "descriptor", "opponent"]] | |
| def reset_rankings(opponents_df): | |
| opponents_df["elo_score"] = [1000]*len(opponents_df) | |
| opponents_df = opponents_df[["elo_score", "descriptor", "opponent"]] | |
| return display_rankings(opponents_df) | |
| def clear_rankings(opponents_df): | |
| opponents_df = pd.DataFrame(columns=['elo_score', 'descriptor', 'opponent']) | |
| return display_rankings(opponents_df) | |
| # theme='Taithrah/Minimal' | |
| # Gradio interface | |
| theme = gr.themes.Soft(primary_hue="red", secondary_hue="blue") | |
| with gr.Blocks(theme=theme) as app: | |
| gr.Markdown( | |
| """## Preference-based Elo Ranker | |
| This tool helps you create **accurate rankings** of things based on your personal preferences. | |
| It does this by asking you questions comparing a random pair of your inpys, and then using your | |
| answers to calculate Elo scores for ranking. | |
| """ | |
| ) | |
| with gr.Row(): | |
| previews_df = pd.DataFrame(columns=['elo_score', 'descriptor', 'opponent']) | |
| previews = gr.DataFrame(value=previews_df, interactive=False, visible=False) | |
| with gr.Column(): | |
| gr.Markdown( | |
| """### Vote to Rank | |
| """ | |
| ) | |
| with gr.Row(): | |
| compare_output = gr.Textbox("Add some options to start voting!", label="Comparison", interactive=False, scale=3) | |
| with gr.Row(): | |
| yes_button = gr.Button("Yes", variant="secondary") | |
| no_button = gr.Button("No", variant="primary") | |
| new_vote = gr.Button("New Vote") | |
| with gr.Row(): | |
| with gr.Column(): | |
| compare_index_1 = gr.Textbox(label="",interactive=False, visible=False) | |
| with gr.Column(): | |
| compare_index_2 = gr.Textbox(label="",interactive=False, visible=False) | |
| with gr.Column(): | |
| gr.Markdown( | |
| """### Rankings | |
| """ | |
| ) | |
| opponents_df = pd.DataFrame(columns=['elo_score', 'descriptor', 'opponent']) | |
| rankings = gr.DataFrame(value=opponents_df, interactive=False, headers=["Score","descriptor", "opponent"]) | |
| gr.Markdown( | |
| """### Add Opponents | |
| """ | |
| ) | |
| with gr.Row(): | |
| descriptor_input = gr.Textbox(label="Type") | |
| opponent_input = gr.Textbox(label="Opponent") | |
| add_button = gr.Button("Add Opponent") | |
| add_button.click(add_and_compare, inputs=[descriptor_input, opponent_input, rankings], outputs=[descriptor_input, opponent_input, rankings]) | |
| gr.Markdown( | |
| """### Remove Opponents | |
| """ | |
| ) | |
| with gr.Row(): | |
| remove_descriptor_input = gr.Textbox(label="Type") | |
| remove_opponent_input = gr.Textbox(label="Opponent") | |
| remove_button = gr.Button("Remove opponent") | |
| remove_button.click(remove_opponent, inputs=[remove_descriptor_input, remove_opponent_input, rankings], outputs=rankings) | |
| gr.Markdown( | |
| """### Import and Export Rankings | |
| """ | |
| ) | |
| with gr.Row(): | |
| # Import CSV file to replace the existing DataFrame | |
| import_button = gr.File(label="Import CSV", file_count="single") | |
| import_button.change(fn=import_csv, inputs=[import_button, rankings], outputs=[rankings]) | |
| with gr.Column(): | |
| # Export button to download the DataFrame as CSV | |
| export_link = gr.File(label="Download CSV", file_count="single") | |
| export_button = gr.Button("Export as CSV") | |
| export_button.click(fn=export_csv, inputs=[rankings], outputs=export_link) | |
| gr.Markdown("### Reset Data") | |
| with gr.Row(): | |
| reset_button = gr.Button("Reset Scores") | |
| reset_button.click(reset_rankings, inputs=[rankings], outputs=rankings) | |
| clear_button = gr.Button("Clear Table", variant="primary") | |
| clear_button.click(clear_rankings, inputs=[rankings], outputs=rankings) | |
| # add_button.click(add_and_compare, inputs=[descriptor_input, opponent_input, rankings], outputs=[descriptor_input, opponent_input, rankings]) | |
| yes_button.click(update_ratings_pos, inputs=[compare_index_1, compare_index_2, rankings], outputs=[compare_output, compare_index_1, compare_index_2, rankings]) | |
| no_button.click(update_ratings_neg, inputs=[compare_index_1, compare_index_2, rankings], outputs=[compare_output, compare_index_1, compare_index_2, rankings]) | |
| new_vote.click(vote_startup, inputs=[rankings],outputs=[compare_output, compare_index_1, compare_index_2, rankings]) | |
| app.launch(share=False) |