Upload app_utils.py

app_utils.py

@@ -0,0 +1,155 @@
+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, args=None):
+    columns = st.columns(np.ones(n_columns))
+    with columns[n_columns//2]:
+        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'] = 'http://localhost:8080/'
+    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 new_get_client(session):
+    scope = "playlist-modify-public"
+    print('Here3')
+
+    cache_handler = StreamlitCacheHandler(session)
+    auth_manager = spotipy.oauth2.SpotifyOAuth(scope=scope,
+                                               cache_handler=cache_handler,
+                                               show_dialog=True)
+    print('Here4')
+
+    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()
+        # st.markdown(f'[Click here to log in]({auth_url})', unsafe_allow_html=True)
+        login = centered_button(st.button, 'Log in', n_columns=5)
+        print('Here6')
+        if login:
+            print('Here7')
+            sp = spotipy.Spotify(auth_manager=auth_manager)
+            user_id = sp.me()['id']
+        else:
+            sp, user_id = None, None
+    return sp, user_id
+
+
+
+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]
+        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 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]
+    return genres_output
+
+
+
+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))