Update app.py

app.py

@@ -17,10 +17,14 @@ def vote_startup(songs_df):
   except:
     songs_df = pd.DataFrame(columns=['elo_score', 'artist', 'song_title'])
   if len(songs_df)>0:
-      slice_size = 4
-      slice = int(len(songs_df)/slice_size)
-      sample = songs_df[slice:(slice_size-1)*slice].sample(frac=1).iloc[0]
-      song_title, artist = sample["song_title"], sample["artist"]
+      if len(songs_df>10):
+          slice_size = 4
+          slice = int(len(songs_df)/slice_size)
+          sample = songs_df[slice:(slice_size-1)*slice].sample(frac=1).iloc[0]
+          song_title, artist = sample["song_title"], sample["artist"]
+      else:
+          sample = songs_df.sample(frac=1).iloc[0]
+          song_title, artist = sample["song_title"], sample["artist"]
   if len(songs_df) > 1:
       # Randomly select a song to compare with
       sample = songs_df.sample(frac=1)
@@ -73,10 +77,15 @@ def update_ratings_pos(first_string, second_string, songs_df):
 
     songs_df = songs_df.sort_values(by='elo_score', ascending=False)
     songs_df.to_csv("songs_df.csv")
-    slice_size = 4
-    slice = int(len(songs_df)/slice_size)
-    sample = songs_df[slice:(slice_size-1)*slice].sample(frac=1).iloc[0]
-    song_title, artist = sample["song_title"], sample["artist"]
+
+    if len(songs_df>10):
+          slice_size = 4
+          slice = int(len(songs_df)/slice_size)
+          sample = songs_df[slice:(slice_size-1)*slice].sample(frac=1).iloc[0]
+          song_title, artist = sample["song_title"], sample["artist"]
+    else:
+          sample = songs_df.sample(frac=1).iloc[0]
+          song_title, artist = sample["song_title"], sample["artist"]
     if len(songs_df) > 1:
         # Randomly select a song to compare with
         sample = songs_df.sample(frac=1)
@@ -111,10 +120,14 @@ def update_ratings_neg(first_string, second_string, songs_df):
 
     songs_df = songs_df.sort_values(by='elo_score', ascending=False)
     songs_df.to_csv("songs_df.csv")
-    slice_size = 4
-    slice = int(len(songs_df)/slice_size)
-    sample = songs_df[slice:(slice_size-1)*slice].sample(frac=1).iloc[0]
-    song_title, artist = sample["song_title"], sample["artist"]
+    if len(songs_df>10):
+          slice_size = 4
+          slice = int(len(songs_df)/slice_size)
+          sample = songs_df[slice:(slice_size-1)*slice].sample(frac=1).iloc[0]
+          song_title, artist = sample["song_title"], sample["artist"]
+    else:
+          sample = songs_df.sample(frac=1).iloc[0]
+          song_title, artist = sample["song_title"], sample["artist"]
     if len(songs_df) > 1:
         # Randomly select a song to compare with
         sample = songs_df.sample(frac=1)