Update app.py

app.py

@@ -31,7 +31,7 @@ def complete_login(full_url):
         code = full_url.split("?code=")[1].split("&")[0]
     except IndexError:
         return (
-            gr.update(value="❌ No code found in URL. Please login again."),
+            gr.update(value="❌ OAuth code not found. Try logging in again."),
             gr.update(visible=False),
             gr.update(visible=False),
         )
@@ -48,7 +48,7 @@ def complete_login(full_url):
 
     if not devices_dict:
         return (
-            gr.update(value="⚠️ Logged in, but no active Spotify devices found. Open Spotify app."),
+            gr.update(value="⚠️ No active Spotify devices found. Open Spotify app."),
             gr.update(visible=False),
             gr.update(visible=False),
         )
@@ -80,33 +80,50 @@ with gr.Blocks() as demo:
     gr.Markdown("## 🎵 RNG Spotify Playlist Shuffler")
 
     login_url = get_auth_url()
-    gr.HTML(f'<a href="{login_url}" target="_blank"><button style="width:100%; height:40px; font-size:16px;">🔐 Login to Spotify</button></a>')
-
-    gr.Markdown("After logging in, copy the full redirected URL and paste below.")
-    redirect_url_box = gr.Textbox(label="Paste redirected URL here")
-    complete_login_btn = gr.Button("✅ Complete Login")
+    login_btn = gr.HTML(
+        f'<a href="{login_url}"><button style="width:100%; height:40px; font-size:16px;">🔐 Login to Spotify</button></a>'
+    )
 
     status = gr.Markdown()
     playlist_dd = gr.Dropdown(label="Select a Playlist", visible=False)
     device_dd = gr.Dropdown(label="Select Playback Device", visible=False)
 
-    shuffle_btn = gr.Button("🔀 Shuffle & Play", visible=True)
+    shuffle_btn = gr.Button("🔀 Shuffle & Play", visible=False)
     result = gr.Markdown()
 
+    # Hidden textbox to automatically capture redirected URL
+    redirect_url_box = gr.Textbox(visible=False)
+
+    complete_login_btn = gr.Button(visible=False)
+
     complete_login_btn.click(
         complete_login,
         inputs=[redirect_url_box],
         outputs=[status, playlist_dd, device_dd]
     )
 
+    playlist_dd.change(lambda: gr.update(visible=True), None, shuffle_btn)
+
     shuffle_btn.click(
         shuffle_and_play,
         inputs=[playlist_dd, device_dd],
         outputs=[result]
     )
 
+    # Automatic URL handling (no manual paste needed!)
+    demo.load(None, None, None, js="""
+        () => {
+            const params = new URLSearchParams(window.location.search);
+            const code = params.get('code');
+            if(code) {
+                const box = document.querySelector('#component-redirect_url_box textarea');
+                if(box){ box.value = window.location.href; }
+                const button = document.querySelector('#component-complete_login_btn');
+                if(button){ button.click(); }
+            }
+        }
+    """)
+
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", ssr_mode=False)
 
-
-