Update app_utils.py

app_utils.py

@@ -4,6 +4,7 @@ import os
 import pickle
 import spotipy
 import spotipy.util as sp_util
+import requests
 
 dir_path = os.path.dirname(os.path.realpath(__file__))
 
@@ -47,19 +48,21 @@ def new_get_client(session):
                                                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)
-        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