Update app.py

app.py

@@ -253,7 +253,7 @@ def find_best_songs_for_mood(all_tracks_audio_features, genre_selected_indexes,
     return min_dist_indexes, n_candidates
 
 @st.cache
-def run_exploration(selected_tracks_uris, selected_tracks_genres, playlist_length, exploration, all_tracks_uris, target_mood):
+def run_exploration(selected_tracks_uris, selected_tracks_genres, playlist_length, exploration, all_tracks_uris, target_mood, selected_genres):
     # sample exploration songs
     if exploration > 0:
         n_known = int(playlist_length * (1 - exploration))
@@ -276,19 +276,23 @@ def run_exploration(selected_tracks_uris, selected_tracks_genres, playlist_lengt
             dict_args_loose[f'max_{m}'] = min(1, target_mood[i_m] + 0.3)
         new_songs = []
         counter_seed = 0
+        print(selected_genres)
         while len(new_songs) < n_new:
             try:
                 print(seed_songs[counter_seed])
                 print(dict_args)
-                reco = sp.recommendations(seed_tracks=[seed_songs[counter_seed]], seed_genres=seed_genres[counter_seed],
+                np.random.shuffle(selected_genres)
+                reco = sp.recommendations(seed_tracks=[seed_songs[counter_seed]], seed_genres=selected_genres,
                                           market="from_token", country='from_token', **dict_args)['tracks']
                 if len(reco) == 0:
                     print('Using loose bounds')
-                    reco = sp.recommendations(seed_tracks=[seed_songs[counter_seed]], seed_genres=seed_genres[counter_seed],
+                    np.random.shuffle(selected_genres)
+                    reco = sp.recommendations(seed_tracks=[seed_songs[counter_seed]], seed_genres=selected_genres,
                                               market="from_token", country='from_token', **dict_args_loose)['tracks']
                     if len(reco) == 0:
                         print('Using looser bounds')
-                        reco = sp.recommendations(seed_tracks=[seed_songs[counter_seed]], seed_genres=seed_genres[counter_seed],
+                        np.random.shuffle(selected_genres)
+                        reco = sp.recommendations(seed_tracks=[seed_songs[counter_seed]], seed_genres=selected_genres,
                                                   market="from_token", country='from_token', **dict_args_looser)['tracks']
                         if len(reco) == 0:
                             print('Removing bounds')
@@ -298,6 +302,7 @@ def run_exploration(selected_tracks_uris, selected_tracks_genres, playlist_lengt
                     if r['uri'] not in all_tracks_uris and r['uri'] not in new_songs:
                         new_songs.append(r['uri'])
                         break
+
             except:
                 pass
             print(counter_seed, len(new_songs))
@@ -348,6 +353,7 @@ def run_app():
             if custom_button or 'run_custom' in st.session_state.keys() or debug:
                 st.session_state['run_custom'] = True
                 checkboxes = st.session_state['checkboxes'].copy()
+                selected_genres = [genres_labels[i] for i in range(len(genres_labels)) if checkboxes[i] and genres_labels[i] != 'unknown']
                 init_time = time.time()
                 genre_selected_indexes = filter_songs_by_genre(checkboxes, genres_labels, indexes_by_genre)
                 if len(genre_selected_indexes) < 10:
@@ -380,7 +386,7 @@ def run_app():
                             generation_button = centered_button(st.button, 'Generate playlist', n_columns=5)
                             if generation_button:
                                 selected_tracks_uris = run_exploration(selected_tracks_uris, selected_tracks_genres, playlist_length, exploration, all_tracks_uris,
-                                                                       target_mood.flatten())
+                                                                       target_mood.flatten(), selected_genres)
                                 print(f'9. run exploration: {time.time() - init_time:.2f}')
                                 init_time = time.time()