Update app.py

app.py

@@ -43,13 +43,14 @@ def extract_track_id_from_url(url):
 def app():
 
     # Step 1: User Authorization
-    if st.session_state.get('code') is None:
+    if st.session_state.get('code') is None or st.session_state.get('reset') == 1:
         auth_params = {
             'client_id': CLIENT_ID,
             'response_type': 'code',
             'redirect_uri': REDIRECT_URI,
             'scope': 'user-read-private',  # Modify as per your required scopes
         }
+        st.session_state['reset'] = 0
         st.write('Please log in to Spotify')
         st.markdown(f"[Log In]({AUTH_URL}?{urlencode(auth_params)})", unsafe_allow_html=True)
     else:#
@@ -131,8 +132,7 @@ def app():
 
 
 if 'code' not in st.session_state:
-    if st.session_state.get('code') is not None:
-        query_params = st.experimental_get_query_params()
-        st.session_state['code'] = query_params.get('code', [None])[0]
+    query_params = st.experimental_get_query_params()
+    st.session_state['code'] = query_params.get('code', [None])[0]
 
 app()