Update app_utils.py

app_utils.py

@@ -4,15 +4,20 @@ import os
 import pickle
 import spotipy
 import spotipy.util as sp_util
+from streamlit import button
 import requests
+from bokeh.models.widgets import Div
 
 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):
+def centered_button(func, text, n_columns=7, disabled=False, args=None):
     columns = st.columns(np.ones(n_columns))
     with columns[n_columns//2]:
-        return func(text)
+        if 'button' in str(func):
+            return func(text, disabled=disabled)
+        else:
+            return func(text)
 
 # get credentials
 def setup_credentials():
@@ -25,7 +30,7 @@ def setup_credentials():
 
     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/'
+    os.environ['SPOTIPY_REDIRECT_URI'] = 'http://localhost:8501/'
     return client_info
 
 relevant_audio_features = ["danceability", "energy", "loudness", "mode", "valence", "tempo"]
@@ -38,33 +43,41 @@ def get_client():
     user_id = sp.me()['id']
     return sp, user_id
 
+def add_button(url, text):
+    st.write(f'''
+        <center>
+        <a target="_self" 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"
-    print('Here3')
 
     cache_handler = StreamlitCacheHandler(session)
     auth_manager = spotipy.oauth2.SpotifyOAuth(scope=scope,
                                                cache_handler=cache_handler,
                                                show_dialog=True)
-    print('Here4')
     sp, user_id = None, None
 
-    if requests.get("code"):
-        # Step 2. Being redirected from Spotify auth page
-        auth_manager.get_access_token(requests.get("code"))
-        print('Here7')
-        sp = spotipy.Spotify(auth_manager=auth_manager)
-        user_id = sp.me()['id']
-
-
     if not auth_manager.validate_token(cache_handler.get_cached_token()):
-        print('Here6')
         # 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)
-    return sp, user_id
+        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
 
 
 def extract_uris_from_links(links, url_type):
@@ -131,6 +144,29 @@ def aggregate_genres(genres, legit_genres, verbose=False):
                 genres_output[best_match] = genres[glabel]
     return genres_output
 
+def get_all_playlists_uris_from_users(sp, user_ids):
+    all_uris = []
+    all_names = []
+    for user_id in user_ids:
+        offset = 0
+        done = False
+        all_uris = []
+        playlist_names = []
+        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):