Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import requests
 import pandas as pd
 from tempfile import NamedTemporaryFile
-from openpyxl import Workbook
 import shutil
 
 # Funciones para Spotify
@@ -35,7 +34,7 @@ def buscar_playlists_spotify(token, query, limit=50):
 
     return [{'playlist_id': playlist['id'], 'playlist_name': playlist['name']} for playlist in playlists]
 
-def obtener_canciones_playlist_spotify(token, playlist_id):
+def obtener_canciones_playlist_spotify(token, playlist_id, playlist_name):
     print(f"Obteniendo canciones de la playlist {playlist_id} de Spotify...")
     url = f'https://api.spotify.com/v1/playlists/{playlist_id}/tracks'
     headers = {'Authorization': f'Bearer {token}'}
@@ -46,8 +45,10 @@ def obtener_canciones_playlist_spotify(token, playlist_id):
         for item in tracks:
             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'),
@@ -60,7 +61,8 @@ def obtener_canciones_playlist_spotify(token, playlist_id):
                     'instrumentalness': audio_features.get('instrumentalness', 'No disponible'),
                     'link': track['external_urls']['spotify'],
                     'record_label': obtener_record_label_spotify(track['album']['id'], token),
-                    'source': 'Spotify'
+                    'source': 'Spotify',
+                    'genero_artista': ', '.join(artista_info.get('genres', []))
                 })
     return canciones
 
@@ -68,7 +70,11 @@ def obtener_caracteristicas_audio(token, track_id):
     url = f'https://api.spotify.com/v1/audio-features/{track_id}'
     headers = {'Authorization': f'Bearer {token}'}
     response = requests.get(url, headers=headers)
-    return response.json() if response.status_code == 200 else {}
+    if response.status_code == 200:
+        return response.json()
+    else:
+        print(f"No se pudieron obtener las características de audio para la pista {track_id}")
+        return {}
 
 def obtener_record_label_spotify(album_id, token):
     url = f'https://api.spotify.com/v1/albums/{album_id}'
@@ -77,6 +83,15 @@ def obtener_record_label_spotify(album_id, token):
     album_info = response.json() if response.status_code == 200 else {}
     return album_info.get('label', 'No disponible')
 
+def obtener_info_artista(artista_id, token):
+    url = f'https://api.spotify.com/v1/artists/{artista_id}'
+    headers = {'Authorization': f'Bearer {token}'}
+    response = requests.get(url, headers=headers)
+    artista_info = response.json() if response.status_code == 200 else {}
+    return {
+        'genres': artista_info.get('genres', [])
+    }
+
 # Función principal de la interfaz
 def interface(query, num_spotify_playlists=50, project_name="Proyecto"):
     # Obtener tokens y claves
@@ -88,7 +103,7 @@ def interface(query, num_spotify_playlists=50, project_name="Proyecto"):
     playlists_spotify = buscar_playlists_spotify(token_spotify, query, num_spotify_playlists)
     canciones_spotify = []
     for playlist in playlists_spotify:
-        songs = obtener_canciones_playlist_spotify(token_spotify, playlist['playlist_id'])
+        songs = obtener_canciones_playlist_spotify(token_spotify, playlist['playlist_id'], playlist['playlist_name'])
         canciones_spotify.extend(songs)
 
     # Crear DataFrame
@@ -114,6 +129,6 @@ iface = gr.Interface(
     ],
     outputs=[gr.Dataframe(), gr.File(label="Descargar Excel")],
     title="Busca Playlists con Keywords",
-    description="Ingresa tu busqueda para encontrar playlists en Spotify con esas palabras en sus titulos."
+    description="Ingresa tu búsqueda para encontrar playlists en Spotify con esas palabras en sus títulos."
 )
 iface.launch()