import os
import json
import random
import gradio as gr
import spotipy
from spotipy.oauth2 import SpotifyOAuth
# ——————————————————————————————————————————————
# 1) Spotify credentials (stored in HF Secrets)
CLIENT_ID = os.environ["SPOTIFY_CLIENT_ID"]
CLIENT_SECRET = os.environ["SPOTIFY_CLIENT_SECRET"]
# 2) Your Space’s URL must match this Redirect URI exactly
REDIRECT_URI = "https://jisaacso219-rng-shuffle.hf.space/"
# 3) Include `streaming` for Web Playback SDK + playback scopes
SCOPE = (
"streaming "
"user-read-playback-state "
"user-modify-playback-state "
"playlist-read-private"
)
sp_oauth = SpotifyOAuth(
client_id=CLIENT_ID,
client_secret=CLIENT_SECRET,
redirect_uri=REDIRECT_URI,
scope=SCOPE,
show_dialog=True,
)
sp = None
user_playlists = {}
def get_auth_url():
return sp_oauth.get_authorize_url()
def check_login(code: str):
"""
Called on page‐load via demo.load(). Once we have a `code`:
- exchange it for an access token
- fetch playlists
- inject the Web Playback SDK + token + debug hooks
"""
global sp, user_playlists
if not code:
return (
gr.update(visible=False), # status
gr.update(visible=False), # dropdown
gr.update(visible=False), # sdk_html
)
# 1) Exchange code for token
tokinfo = sp_oauth.get_access_token(code, as_dict=True)
access_token = tokinfo["access_token"]
sp = spotipy.Spotify(auth=access_token)
# 2) Fetch your playlists
items = sp.current_user_playlists(limit=50)["items"]
user_playlists = {p["name"]: p["id"] for p in items}
# 3) Build SDK + injection scripts
sdk_js = f"""
"""
return (
gr.update(visible=True, value="✅ Logged in! Select a playlist below."),
gr.update(visible=True, choices=list(user_playlists.keys())),
gr.update(visible=True, value=sdk_js),
)
def load_playlist_info(playlist_name: str):
"""
When a playlist is selected, show its artwork, name, and description.
"""
pid = user_playlists[playlist_name]
data = sp.playlist(pid)
img = data["images"][0]["url"] if data["images"] else ""
owner = data["owner"]["display_name"]
desc = data.get("description", "")
html = f"""
{playlist_name} by {owner}
{desc}
"""
return html, gr.update(visible=True)
def shuffle_and_play(playlist_name: str):
"""
Gather all track URIs, RNG-shuffle them, then fire off the Web Playback SDK
to play them in-browser on window._webDeviceId.
"""
pid = user_playlists.get(playlist_name)
if not pid:
return gr.update(value="⚠️ No playlist selected.", visible=True)
# fetch + shuffle
tracks, results = [], sp.playlist_tracks(pid)
tracks.extend(results["items"])
while results["next"]:
results = sp.next(results)
tracks.extend(results["items"])
uris = [t["track"]["uri"] for t in tracks if t["track"]]
random.shuffle(uris)
# in-page JS to start playback
uris_json = json.dumps(uris)
play_js = f"""