Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import pandas as pd
 from tempfile import NamedTemporaryFile
+from openpyxl import Workbook
 import shutil
 
 # Funciones para Spotify
@@ -34,7 +35,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, playlist_name):
+def obtener_canciones_playlist_spotify(token, playlist_id):
     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}'}
@@ -45,10 +46,8 @@ def obtener_canciones_playlist_spotify(token, playlist_id, playlist_name):
         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'),
@@ -61,8 +60,7 @@ def obtener_canciones_playlist_spotify(token, playlist_id, playlist_name):
                     'instrumentalness': audio_features.get('instrumentalness', 'No disponible'),
                     'link': track['external_urls']['spotify'],
                     'record_label': obtener_record_label_spotify(track['album']['id'], token),
-                    'source': 'Spotify',
-                    'genero_artista': ', '.join(artista_info.get('genres', []))
+                    'source': 'Spotify'
                 })
     return canciones
 
@@ -70,11 +68,7 @@ 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)
-    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 {}
+    return response.json() if response.status_code == 200 else {}
 
 def obtener_record_label_spotify(album_id, token):
     url = f'https://api.spotify.com/v1/albums/{album_id}'
@@ -83,15 +77,6 @@ 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
@@ -103,7 +88,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'], playlist['playlist_name'])
+        songs = obtener_canciones_playlist_spotify(token_spotify, playlist['playlist_id'])
         canciones_spotify.extend(songs)
 
     # Crear DataFrame
@@ -129,6 +114,6 @@ iface = gr.Interface(
     ],
     outputs=[gr.Dataframe(), gr.File(label="Descargar Excel")],
     title="Busca Playlists con Keywords",
-    description="Ingresa tu búsqueda para encontrar playlists en Spotify con esas palabras en sus títulos."
+    description="Ingresa tu busqueda para encontrar playlists en Spotify con esas palabras en sus titulos."
 )
 iface.launch()