Update app.py

app.py

@@ -42,7 +42,9 @@ def extract_track_id_from_url(url):
 
 
 def app():
-
+    if st.session_state.get('code') is not None:
+        if st.button("Go Back"):
+            st.session_state['code'] = None
     # Step 1: User Authorization
     if st.session_state.get('code') is None:
         auth_params = {
@@ -70,6 +72,8 @@ def app():
             st.session_state['access_token'] = token_response.get('access_token')
 
         if 'access_token' in st.session_state:
+            if st.button("Go Back"):
+                
             st.write("Enter a Spotify Track URL or URI to get its audio features.")
             track_input = st.text_input("Spotify Track URL/URI", "")
             traindata = []