Update app.py

app.py

@@ -42,6 +42,7 @@ def extract_track_id_from_url(url):
 
 def app():
 
+    
     # 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 = {
@@ -52,10 +53,7 @@ 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']
+        auth_code = st.session_state['code']
 
             token_data = {
                 'grant_type': 'authorization_code',
@@ -67,7 +65,7 @@ def app():
             token_r = requests.post(TOKEN_URL, data=token_data)
             token_response = token_r.json()
             st.session_state['access_token'] = token_response.get('access_token')
-
+    else:#
         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", "")