Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| import numpy as np | |
| import os | |
| import pickle | |
| import spotipy | |
| import spotipy.util as sp_util | |
| dir_path = os.path.dirname(os.path.realpath(__file__)) | |
| # current mess: https://github.com/plamere/spotipy/issues/632 | |
| def centered_button(func, text, n_columns=7, disabled=False, args=None): | |
| columns = st.columns(np.ones(n_columns)) | |
| with columns[n_columns//2]: | |
| if 'button' in str(func): | |
| return func(text, disabled=disabled) | |
| else: | |
| return func(text) | |
| # get credentials | |
| def setup_credentials(): | |
| if 'client_id' in os.environ.keys() and 'client_secret' in os.environ.keys(): | |
| client_info = dict(client_id=os.environ['client_id'], | |
| client_secret=os.environ['client_secret']) | |
| else: | |
| with open(dir_path + "/ids.pk", 'rb') as f: | |
| client_info = pickle.load(f) | |
| os.environ['SPOTIPY_CLIENT_ID'] = client_info['client_id'] | |
| os.environ['SPOTIPY_CLIENT_SECRET'] = client_info['client_secret'] | |
| os.environ['SPOTIPY_REDIRECT_URI'] = 'https://huggingface.co/spaces/ccolas/EmotionPlaylist/' | |
| return client_info | |
| relevant_audio_features = ["danceability", "energy", "loudness", "mode", "valence", "tempo"] | |
| def get_client(): | |
| scope = "playlist-modify-public" | |
| token = sp_util.prompt_for_user_token(scope=scope) | |
| sp = spotipy.Spotify(auth=token) | |
| user_id = sp.me()['id'] | |
| return sp, user_id | |
| def add_button(url, text): | |
| st.write(f''' | |
| <center> | |
| <a style='color:black;' href="{url}"> | |
| <button class='css-1cpxqw2'> | |
| {text} | |
| </button> | |
| </a></center> | |
| ''', | |
| unsafe_allow_html=True | |
| ) | |
| def new_get_client(session): | |
| scope = "playlist-modify-public" | |
| cache_handler = StreamlitCacheHandler(session) | |
| auth_manager = spotipy.oauth2.SpotifyOAuth(scope=scope, | |
| cache_handler=cache_handler, | |
| show_dialog=True) | |
| sp, user_id = None, None | |
| if not auth_manager.validate_token(cache_handler.get_cached_token()): | |
| # Step 1. Display sign in link when no token | |
| auth_url = auth_manager.get_authorize_url() | |
| if 'code' not in st.experimental_get_query_params(): | |
| add_button(auth_url, 'Log in') | |
| # st.markdown(f'<a href="{auth_url}" target="_self">Click here to log in</a>', unsafe_allow_html=True) | |
| # Step 2. Being redirected from Spotify auth page | |
| if 'code' in st.experimental_get_query_params(): | |
| auth_manager.get_access_token(st.experimental_get_query_params()['code']) | |
| sp = spotipy.Spotify(auth_manager=auth_manager) | |
| user_id = sp.me()['id'] | |
| return sp, user_id, auth_manager | |
| def extract_uris_from_links(links, url_type): | |
| assert url_type in ['playlist', 'artist', 'user'] | |
| urls = links.split('\n') | |
| uris = [] | |
| for url in urls: | |
| if 'playlist' in url: | |
| uri = url.split(f'{url_type}/')[-1].split('?')[0] | |
| elif 'user' in url: | |
| uri = url.split(f'{url_type}/')[-1].split('?')[0] | |
| else: | |
| uri = url.split('?')[0] | |
| uris.append(uri) | |
| return uris | |
| def wall_of_checkboxes(labels, max_width=10): | |
| n_labels = len(labels) | |
| n_rows = int(np.ceil(n_labels/max_width)) | |
| checkboxes = [] | |
| for i in range(n_rows): | |
| columns = st.columns(np.ones(max_width)) | |
| row_length = n_labels % max_width if i == n_rows - 1 else max_width | |
| for j in range(row_length): | |
| with columns[j]: | |
| checkboxes.append(st.empty()) | |
| return checkboxes | |
| def find_legit_genre(glabel, legit_genres, verbose=False): | |
| legit_genres_formatted = [lg.replace('-', '').replace(' ', '') for lg in legit_genres] | |
| glabel_formatted = glabel.replace(' ', '').replace('-', '') | |
| if verbose: print('\n', glabel) | |
| best_match = None | |
| best_match_score = 0 | |
| for legit_glabel, legit_glabel_formatted in zip(legit_genres, legit_genres_formatted): | |
| if 'jazz' in glabel_formatted: | |
| best_match = 'jazz' | |
| if verbose: print('\t', 'pop') | |
| break | |
| if 'ukpop' in glabel_formatted: | |
| best_match = 'pop' | |
| if verbose: print('\t', 'pop') | |
| break | |
| if legit_glabel_formatted == glabel_formatted: | |
| if verbose: print('\t', legit_glabel_formatted) | |
| best_match = legit_glabel | |
| break | |
| elif glabel_formatted in legit_glabel_formatted: | |
| if verbose: print('\t', legit_glabel_formatted) | |
| if len(glabel_formatted) > best_match_score: | |
| best_match = legit_glabel | |
| best_match_score = len(glabel_formatted) | |
| elif legit_glabel_formatted in glabel_formatted: | |
| if verbose: print('\t', legit_glabel_formatted) | |
| if len(legit_glabel_formatted) > best_match_score: | |
| best_match = legit_glabel | |
| best_match_score = len(legit_glabel_formatted) | |
| if best_match is None: | |
| return "unknown" | |
| else: | |
| return best_match | |
| # def aggregate_genres(genres, legit_genres, verbose=False): | |
| # genres_output = dict() | |
| # legit_genres_formatted = [lg.replace('-', '').replace(' ', '') for lg in legit_genres] | |
| # for glabel in genres.keys(): | |
| # if verbose: print('\n', glabel) | |
| # glabel_formatted = glabel.replace(' ', '').replace('-', '') | |
| # best_match = None | |
| # best_match_score = 0 | |
| # for legit_glabel, legit_glabel_formatted in zip(legit_genres, legit_genres_formatted): | |
| # if 'jazz' in glabel_formatted: | |
| # best_match = 'jazz' | |
| # if verbose: print('\t', 'pop') | |
| # break | |
| # if 'ukpop' in glabel_formatted: | |
| # best_match = 'pop' | |
| # if verbose: print('\t', 'pop') | |
| # break | |
| # if legit_glabel_formatted == glabel_formatted: | |
| # if verbose: print('\t', legit_glabel_formatted) | |
| # best_match = legit_glabel | |
| # break | |
| # elif glabel_formatted in legit_glabel_formatted: | |
| # if verbose: print('\t', legit_glabel_formatted) | |
| # if len(glabel_formatted) > best_match_score: | |
| # best_match = legit_glabel | |
| # best_match_score = len(glabel_formatted) | |
| # elif legit_glabel_formatted in glabel_formatted: | |
| # if verbose: print('\t', legit_glabel_formatted) | |
| # if len(legit_glabel_formatted) > best_match_score: | |
| # best_match = legit_glabel | |
| # best_match_score = len(legit_glabel_formatted) | |
| # | |
| # if best_match is not None: | |
| # if verbose: print('\t', '-->', best_match) | |
| # if best_match in genres_output.keys(): | |
| # genres_output[best_match] += genres[glabel] | |
| # else: | |
| # genres_output[best_match] = genres[glabel] | |
| # else: | |
| # if "unknown" in genres_output.keys(): | |
| # genres_output["unknown"] += genres[glabel] | |
| # else: | |
| # genres_output["unknown"] = genres[glabel] | |
| # for k in genres_output.keys(): | |
| # genres_output[k] = sorted(set(genres_output[k])) | |
| # return genres_output | |
| def get_all_playlists_uris_from_users(sp, user_ids): | |
| all_uris = [] | |
| all_names = [] | |
| for user_id in user_ids: | |
| print(user_id) | |
| offset = 0 | |
| done = False | |
| while not done: | |
| playlist_list = sp.user_playlists(user_id, offset=offset, limit=50) | |
| these_names = [p['name'] for p in playlist_list['items']] | |
| these_uris = [p['uri'] for p in playlist_list['items']] | |
| for name, uri in zip(these_names, these_uris): | |
| if uri not in all_uris: | |
| all_uris.append(uri) | |
| all_names.append(user_id + '/' + name) | |
| if len(playlist_list['items']) < offset: | |
| done = True | |
| else: | |
| offset += 50 | |
| return all_uris, all_names | |
| class StreamlitCacheHandler(spotipy.cache_handler.CacheHandler): | |
| """ | |
| A cache handler that stores the token info in the session framework | |
| provided by streamlit. | |
| """ | |
| def __init__(self, session): | |
| self.session = session | |
| def get_cached_token(self): | |
| token_info = None | |
| try: | |
| token_info = self.session["token_info"] | |
| except KeyError: | |
| print("Token not found in the session") | |
| return token_info | |
| def save_token_to_cache(self, token_info): | |
| try: | |
| self.session["token_info"] = token_info | |
| except Exception as e: | |
| print("Error saving token to cache: " + str(e)) | |