Update app.py

app.py

@@ -3,7 +3,6 @@ import gradio as gr
 import spotipy
 from spotipy.oauth2 import SpotifyOAuth
 import random
-import time
 
 # Load environment variables securely from Hugging Face secrets
 CLIENT_ID = os.environ["SPOTIFY_CLIENT_ID"]
@@ -20,23 +19,37 @@ sp_oauth = SpotifyOAuth(
 
 auth_url = sp_oauth.get_authorize_url()
 
-sp = None  # Will be set after authorization
+sp = None
 user_playlists = {}
 selected_playlist_id = None
 
 with gr.Blocks(title="RNG Spotify Playlist Shuffler") as demo:
-    gr.Markdown("## 🎵 RNG Spotify Playlist Shuffler", elem_id="title")
+    gr.Markdown("## 🎵 RNG Spotify Playlist Shuffler")
     gr.Markdown("A clean and simple way to shuffle your Spotify playlists with one click.")
 
     with gr.Column():
-        login_link = gr.Markdown(f"### 🔐 Step 1: [Click here to login to Spotify]({auth_url})")
-        code_box = gr.Textbox(label="Step 2: Paste authorization code", placeholder="...?code=ABC123", visible=True)
-        authorize_button = gr.Button("✅ Step 3: Complete Authorization", visible=True)
+        login_link = gr.Markdown(f"[🔐 Step 1: Login to Spotify]({auth_url})")
+        code_box = gr.Textbox(label="Step 2: Paste Authorization Code", visible=True)
+        authorize_button = gr.Button("✅ Step 3: Authorize", visible=True)
         login_status = gr.Markdown(visible=False)
-        playlist_dropdown = gr.Dropdown(choices=[], label="Step 4: 🎧 Select a Playlist", visible=False)
+        playlist_dropdown = gr.Dropdown(choices=[], label="Step 4: Select Playlist", visible=False)
         shuffle_button = gr.Button("🔀 Step 5: Shuffle and Play", visible=False)
         result_box = gr.Markdown(visible=False)
 
+        # Autofill JavaScript
+        gr.HTML("""
+        <script>
+        window.onload = () => {
+            const params = new URLSearchParams(window.location.search);
+            const code = params.get("code");
+            if (code) {
+                document.querySelector("textarea").value = code;
+                document.querySelector("button").click();
+            }
+        }
+        </script>
+        """)
+
     def check_login(code):
         global sp, user_playlists
         if not code:
@@ -54,7 +67,7 @@ with gr.Blocks(title="RNG Spotify Playlist Shuffler") as demo:
             gr.update(visible=True, value="✅ Logged in! Select your playlist below."),
             gr.update(visible=True, choices=list(user_playlists.keys())),
             gr.update(visible=True),
-            gr.update(visible=True)
+            gr.update(visible=False)
         )
 
     def shuffle_and_play(playlist_name):
@@ -95,4 +108,3 @@ with gr.Blocks(title="RNG Spotify Playlist Shuffler") as demo:
 
 if __name__ == "__main__":
     demo.launch()
-