Update app.py

app.py

@@ -42,18 +42,7 @@ def extract_track_id_from_url(url):
 
 def app():
 
-    if st.session_state.get('code') is not None:
-        auth_code = st.session_state['code']
-        token_data = {
-            'grant_type': 'authorization_code',
-            'code': auth_code,
-            'redirect_uri': REDIRECT_URI,
-            'client_id': CLIENT_ID,
-            'client_secret': CLIENT_SECRET,
-        }
-        token_r = requests.post(TOKEN_URL, data=token_data)
-        token_response = token_r.json()
-        st.session_state['access_token'] = token_response.get('access_token')
+    
     # Step 1: User Authorization
     if st.session_state.get('code') is None or ('access_token' not in st.session_state or st.session_state['access_token'] is None):
         auth_params = {
@@ -65,6 +54,21 @@ def app():
         st.write('Please log in to Spotify')
         st.markdown(f"[Log In]({AUTH_URL}?{urlencode(auth_params)})", unsafe_allow_html=True)
     else:#
+        # Step 2: Request Access Token
+        if 'access_token' not in st.session_state or (st.session_state['access_token'] is None):
+            auth_code = st.session_state['code']
+
+            token_data = {
+                'grant_type': 'authorization_code',
+                'code': auth_code,
+                'redirect_uri': REDIRECT_URI,
+                'client_id': CLIENT_ID,
+                'client_secret': CLIENT_SECRET,
+            }
+            token_r = requests.post(TOKEN_URL, data=token_data)
+            token_response = token_r.json()
+            st.session_state['access_token'] = token_response.get('access_token')
+            
         if 'access_token' in st.session_state:
             st.write("Enter a Spotify Track URL or URI to get its audio features.")
             track_input = st.text_input("Spotify Track URL/URI", "")