Update app.py

app.py

@@ -46,19 +46,13 @@ def obtener_canciones_playlist_spotify(token, playlist_id, playlist_name):
             track = item.get('track')
             if track:
                 artista_info = obtener_info_artista(track['artists'][0]['id'], token) if track['artists'] else {}
-                audio_features = obtener_caracteristicas_audio(token, track['id'])
                 canciones.append({
                     'playlist_name': playlist_name,
                     'artista': track['artists'][0]['name'] if track['artists'] else 'Desconocido',
                     'titulo': track['name'],
                     'isrc': track['external_ids'].get('isrc', 'No disponible'),
                     'popularity': track.get('popularity', 'No disponible'),
-                    'valence': audio_features.get('valence', 'No disponible'),
-                    'danceability': audio_features.get('danceability', 'No disponible'),
-                    'energy': audio_features.get('energy', 'No disponible'),
-                    'tempo': audio_features.get('tempo', 'No disponible'),
-                    'speechiness': audio_features.get('speechiness', 'No disponible'),
-                    'instrumentalness': audio_features.get('instrumentalness', 'No disponible'),
+                    'track_id': track['id'],
                     'link': track['external_urls']['spotify'],
                     'record_label': obtener_record_label_spotify(track['album']['id'], token),
                     'source': 'Spotify',
@@ -66,15 +60,17 @@ def obtener_canciones_playlist_spotify(token, playlist_id, playlist_name):
                 })
     return canciones
 
-def obtener_caracteristicas_audio(token, track_id):
-    url = f'https://api.spotify.com/v1/audio-features/{track_id}'
+def obtener_caracteristicas_audio(token, track_ids):
+    print("Obteniendo características de audio para las pistas...")
+    url = f'https://api.spotify.com/v1/audio-features'
     headers = {'Authorization': f'Bearer {token}'}
-    response = requests.get(url, headers=headers)
+    params = {'ids': ','.join(track_ids)}
+    response = requests.get(url, headers=headers, params=params)
     if response.status_code == 200:
-        return response.json()
+        return response.json().get('audio_features', [])
     else:
-        print(f"No se pudieron obtener las características de audio para la pista {track_id}")
-        return {}
+        print(f"No se pudieron obtener las características de audio para las pistas {track_ids}")
+        return []
 
 def obtener_record_label_spotify(album_id, token):
     url = f'https://api.spotify.com/v1/albums/{album_id}'
@@ -102,9 +98,27 @@ def interface(query, num_spotify_playlists=50, project_name="Proyecto"):
     token_spotify = obtener_token(client_id, client_secret)
     playlists_spotify = buscar_playlists_spotify(token_spotify, query, num_spotify_playlists)
     canciones_spotify = []
+    track_ids = []
     for playlist in playlists_spotify:
         songs = obtener_canciones_playlist_spotify(token_spotify, playlist['playlist_id'], playlist['playlist_name'])
         canciones_spotify.extend(songs)
+        track_ids.extend([song['track_id'] for song in songs])
+
+    # Obtener características de audio en un solo paso
+    audio_features_list = obtener_caracteristicas_audio(token_spotify, track_ids)
+    audio_features_dict = {af['id']: af for af in audio_features_list if af}
+
+    # Añadir características de audio a las canciones
+    for song in canciones_spotify:
+        audio_features = audio_features_dict.get(song['track_id'], {})
+        song.update({
+            'valence': audio_features.get('valence', 'No disponible'),
+            'danceability': audio_features.get('danceability', 'No disponible'),
+            'energy': audio_features.get('energy', 'No disponible'),
+            'tempo': audio_features.get('tempo', 'No disponible'),
+            'speechiness': audio_features.get('speechiness', 'No disponible'),
+            'instrumentalness': audio_features.get('instrumentalness', 'No disponible')
+        })
 
     # Crear DataFrame
     df = pd.DataFrame(canciones_spotify)