diff --git a/main.py b/main.py index 8430ce1..067287e 100644 --- a/main.py +++ b/main.py @@ -13,7 +13,7 @@ def form_example(): if request.method == 'POST': search = request.form.get('search') result = searchSpotify(search) - return render_template("found.html", query=search, trackName=result[0], trackArtist=result[1], trackAlbum=result[2], trackPreview=result[3], trackImage=result[4], trackID=result[5],) + return render_template("found.html", query=search, tracks=result) # otherwise handle the GET request return render_template("index.html") diff --git a/spotifySearch.py b/spotifySearch.py index b23bfd7..7b3bb69 100644 --- a/spotifySearch.py +++ b/spotifySearch.py @@ -5,13 +5,17 @@ from dotenv import load_dotenv load_dotenv() -def searchSpotify(spotifySearch, limit=1): +def searchSpotify(spotifySearch, limit=10): sp = spotipy.Spotify(auth_manager=SpotifyClientCredentials(client_id=os.getenv("client_id"), client_secret=os.getenv("client_secret"))) results = sp.search(q=spotifySearch, limit=limit) - trackName = results['tracks']['items'][0]['name'] - trackArtist = results['tracks']['items'][0]['artists'][0]['name'] - trackAlbum = results['tracks']['items'][0]['album']['name'] - trackPreview = results['tracks']['items'][0]['preview_url'] - trackImage = results['tracks']['items'][0]['album']['images'][0]['url'] - trackID = results['tracks']['items'][0]['uri'] - return (trackName, trackArtist, trackAlbum, trackPreview, trackImage, trackID) + + tracks = [] + for i in range(limit): + trackName = results['tracks']['items'][i]['name'] + trackArtist = results['tracks']['items'][i]['artists'][0]['name'] + trackAlbum = results['tracks']['items'][i]['album']['name'] + trackPreview = results['tracks']['items'][i]['preview_url'] + trackImage = results['tracks']['items'][i]['album']['images'][0]['url'] + trackID = results['tracks']['items'][i]['uri'] + tracks.append([trackName, trackArtist, trackAlbum, trackPreview, trackImage, trackID]) + return tracks diff --git a/templates/found.html b/templates/found.html index 679864f..21e0022 100644 --- a/templates/found.html +++ b/templates/found.html @@ -12,7 +12,9 @@