Update app.py

app.py

@@ -27,46 +27,60 @@ def get_auth_url():
     return sp_oauth.get_authorize_url()
 
 def try_login(current_url):
-    global sp, user_playlists, device_map
-
-    if "?code=" not in current_url:
-        return (
-            "🔐 Please login to Spotify.",
-            gr.update(visible=False),
-            gr.update(visible=False),
-            gr.update(visible=False),
-        )
-
-    code = current_url.split("?code=")[1].split("&")[0]
-    token_info = sp_oauth.get_access_token(code, as_dict=True)
-    access_token = token_info["access_token"]
-    sp = spotipy.Spotify(auth=access_token)
-
-    playlists = sp.current_user_playlists(limit=50)["items"]
-    user_playlists.clear()
-    for p in playlists:
-        name = p["name"]
-        user_playlists[name] = p["id"]
-
-    devices = sp.devices()["devices"]
-    device_map.clear()
-    for d in devices:
-        device_map[d["name"]] = d["id"]
-
-    if not device_map:
-        return (
-            "⚠️ No active Spotify devices found.",
-            gr.update(visible=False),
-            gr.update(visible=False),
-            gr.update(visible=False),
-        )
-
-    return (
-        "✅ Logged in!",
-        gr.update(visible=True, choices=list(user_playlists.keys())),
-        gr.update(visible=True, choices=list(device_map.keys())),
-        gr.update(visible=True),
-    )
+    global sp, user_playlists, device_map
+
+    if "?code=" not in current_url:
+        return (
+            "🔐 Please login to Spotify.",
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False),
+        )
+
+    code = current_url.split("?code=")[1].split("&")[0]
+    try:
+        token_info = sp_oauth.get_access_token(code)
+        access_token = token_info["access_token"]
+    except Exception as e:
+        print("🔁 Token refresh failed:", str(e))
+        return (
+            f"❌ Failed to log in: {e}",
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False),
+        )
+
+    sp = spotipy.Spotify(auth=access_token)
+
+    try:
+        playlists = sp.current_user_playlists(limit=50)["items"]
+    except Exception as e:
+        return f"❌ Failed to get playlists: {e}", *[gr.update(visible=False)] * 3
+
+    user_playlists.clear()
+    for p in playlists:
+        user_playlists[p["name"]] = p["id"]
+
+    devices = sp.devices()["devices"]
+    device_map.clear()
+    for d in devices:
+        device_map[d["name"]] = d["id"]
+
+    if not device_map:
+        return (
+            "⚠️ No active Spotify devices found.",
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=False),
+        )
+
+    return (
+        "✅ Logged in!",
+        gr.update(visible=True, choices=list(user_playlists.keys())),
+        gr.update(visible=True, choices=list(device_map.keys())),
+        gr.update(visible=True),
+    )
+
 
 def background_queue_tracks(uris, device_id):
     for i, uri in enumerate(uris):