Update app.py

app.py

@@ -34,18 +34,11 @@ def clean_string(string):
   string = " ".join([x[0].upper()+x[1:] for x in string.split()])
   return string
 
-# Function to add a song and compare it with a random existing song
-def add_and_compare(artist, song_title, songs_df):
-    if artist != "" and song_title != "":
-      artist = clean_string(artist)
-      song_title = clean_string(song_title)
-      new_song = {'artist': artist, 'song_title': song_title, 'elo_score': 1000}
-      songs_df = songs_df.append(new_song, ignore_index=True)
-      songs_df.to_csv("songs_df.csv")
-    songs_df = songs_df[["elo_score",	"artist",	"song_title"]]
-    return "", "", display_rankings(songs_df)
-
 def add_and_compare(artist, song_title, songs_df):
+    try:
+      songs_df = songs_df[["elo_score",	"artist",	"song_title"]]
+    except:
+      songs_df = pd.DataFrame(columns=['elo_score', 'artist', 'song_title'])
     if artist != "" and song_title != "":
         artist = clean_string(artist)
         song_title = clean_string(song_title)
@@ -57,6 +50,10 @@ def add_and_compare(artist, song_title, songs_df):
 
 # Function to update Elo ratings based on user's choice
 def update_ratings_pos(first_string, second_string, songs_df):
+    try:
+      songs_df = songs_df[["elo_score",	"artist",	"song_title"]]
+    except:
+      songs_df = pd.DataFrame(columns=['elo_score', 'artist', 'song_title'])
     if len(songs_df)==0:
       return "Add some songs to start voting!", "", "", display_rankings(songs_df)
     if first_string != "":
@@ -88,6 +85,10 @@ def update_ratings_pos(first_string, second_string, songs_df):
 
 # Function to update Elo ratings based on user's choice
 def update_ratings_neg(first_string, second_string, songs_df):
+    try:
+      songs_df = songs_df[["elo_score",	"artist",	"song_title"]]
+    except:
+      songs_df = pd.DataFrame(columns=['elo_score', 'artist', 'song_title'])
     if len(songs_df)==0:
       return "Add some songs to start voting!", "", "", display_rankings(songs_df)
     if first_string != "":
@@ -118,7 +119,7 @@ def update_ratings_neg(first_string, second_string, songs_df):
     else:
         return "Add some songs to start voting!", "", "", display_rankings(songs_df)
 
-def display_rankings(songs_df=pd.DataFrame()):
+def display_rankings(songs_df=pd.DataFrame(columns=['elo_score', 'artist', 'song_title'])):
   songs_df = songs_df.sort_values(by='elo_score', ascending=False)
   songs_df = songs_df[["elo_score",	"artist",	"song_title"]]
   songs_df.to_csv("songs_df.csv")