Update app.py

app.py

@@ -1,21 +1,31 @@
+import gradio as gr
 import requests
+import pandas as pd
 import time
+from tempfile import NamedTemporaryFile
+from openpyxl import Workbook
+import shutil
 
 # Lista de credenciales de API de Spotify
 client_ids = [
-    '807008cb2ce041178c1871973fc81716', '107286416613436ab5c63b56d3564f9e', 
-    '528d1d6733ba46eb8e9532e56ced61eb', 'aed6435c51484dc18b8c18aa74dd51f8', 
-    '9df51caba5d247dc921b21de35a47c44', '191227c66e0d4be692bc8ee73ea6eb3d', 
+    '807008cb2ce041178c1871973fc81716',
+    '107286416613436ab5c63b56d3564f9e',
+    '528d1d6733ba46eb8e9532e56ced61eb',
+    'aed6435c51484dc18b8c18aa74dd51f8',
+    '9df51caba5d247dc921b21de35a47c44',
+    '191227c66e0d4be692bc8ee73ea6eb3d',
     'e272c0705c7c4fd68937c58adaa446ed'
 ]
 client_secrets = [
-    'b6c3fbe2304145e4b268f05eefd6ab2a', '4de4c14ed109420c9517c083b8018f8c', 
-    '3dbdcf9a12634cfca9e6aca8d3093020', '8662b33d594f4d198ea025d4aa9f0b98', 
-    '0e39502ec7e74fe99bb74245678d5f0d', '2d2a895d85874c088897dd9894dc64ad', 
+    'b6c3fbe2304145e4b268f05eefd6ab2a',
+    '4de4c14ed109420c9517c083b8018f8c',
+    '3dbdcf9a12634cfca9e6aca8d3093020',
+    '8662b33d594f4d198ea025d4aa9f0b98',
+    '0e39502ec7e74fe99bb74245678d5f0d',
+    '2d2a895d85874c088897dd9894dc64ad',
     '9fdfa58ea0a94ce7a0cb34fa19fb7d74'
 ]
 current_api_index = 0
-session = requests.Session()
 
 def obtener_token(client_id, client_secret):
     print(f"Obteniendo token de Spotify con client_id: {client_id}")
@@ -30,48 +40,151 @@ def obtener_token(client_id, client_secret):
         print(f"Error al obtener token para client_id: {client_id}, status_code: {response.status_code}, response: {response.text}")
         return None
 
-def cerrar_y_reiniciar_sesion():
-    global session
-    if session:
-        session.close()
-        print("Sesión cerrada.")
-    session = requests.Session()
-    print("Nueva sesión creada.")
-
 def cambiar_api_key():
     global current_api_index
     current_api_index = (current_api_index + 1) % len(client_ids)
     print(f"Cambiando a la siguiente API Key, índice actual: {current_api_index}")
-    return client_ids[current_api_index], client_secrets[current_api_index]
-
-def obtener_nuevo_token():
-    cerrar_y_reiniciar_sesion()
-    client_id, client_secret = cambiar_api_key()
-    token = obtener_token(client_id, client_secret)
-    print(f"Nuevo token obtenido: {token}")
-    return token
+    return obtener_token(client_ids[current_api_index], client_secrets[current_api_index])
 
-def realizar_solicitud_con_nuevo_token(url, params):
-    token = obtener_nuevo_token()
+def buscar_playlists_spotify(token, query, limit=50):
+    print("Buscando playlists en Spotify...")
+    url = 'https://api.spotify.com/v1/search'
     headers = {'Authorization': f'Bearer {token}'}
-    response = session.get(url, headers=headers, params=params)
-    print(f"Solicitud realizada con nuevo token. Código de estado: {response.status_code}")
-    return response
+    playlists = []
 
-# Ejemplo de uso
-def main():
-    url = 'https://api.spotify.com/v1/search'
-    params = {'q': 'rancheras', 'type': 'playlist', 'limit': 40}
+    try:
+        if limit <= 50:
+            params = {'q': query, 'type': 'playlist', 'limit': limit}
+            response = requests.get(url, headers=headers, params=params)
+            if response.status_code == 429:  # Límite alcanzado
+                print("Límite de peticiones alcanzado, cambiando API Key...")
+                token = cambiar_api_key()
+                time.sleep(1)  # Pausa de 1 segundo para evitar el límite de tasa
+                response = requests.get(url, headers={'Authorization': f'Bearer {token}'}, params=params)
+            playlists.extend(response.json().get('playlists', {}).get('items', []))
+        else:
+            offset = 0
+            while limit > 0:
+                params = {'q': query, 'type': 'playlist', 'limit': min(50, limit), 'offset': offset}
+                response = requests.get(url, headers=headers, params=params)
+                if response.status_code == 429:  # Límite alcanzado
+                    print("Límite de peticiones alcanzado, cambiando API Key...")
+                    token = cambiar_api_key()
+                    time.sleep(1)  # Pausa de 1 segundo para evitar el límite de tasa
+                    response = requests.get(url, headers={'Authorization': f'Bearer {token}'}, params=params)
+                playlists.extend(response.json().get('playlists', {}).get('items', []))
+                limit -= min(50, limit)
+                offset += 50
+                time.sleep(5)  # Pausa de 5 segundos entre las solicitudes para evitar el límite de tasa
+    except Exception as e:
+        print(f"Error al buscar playlists: {e}")
     
-    print("Realizando primera solicitud...")
-    response = realizar_solicitud_con_nuevo_token(url, params)
-    print("Respuesta de la primera solicitud:", response.json())
-
-    # Simular cambio de API key después de recibir un código 429
-    if response.status_code == 429:
-        print("Límite de peticiones alcanzado, cambiando API key y realizando una nueva solicitud...")
-        response = realizar_solicitud_con_nuevo_token(url, params)
-        print("Respuesta de la segunda solicitud:", response.json())
-
-if __name__ == "__main__":
-    main()
+    return [{'playlist_id': playlist['id'], 'playlist_name': playlist['name']} for playlist in playlists]
+
+def obtener_canciones_playlist_spotify(token, playlist_id, playlist_name):
+    print(f"Obteniendo canciones de la playlist {playlist_id} ({playlist_name}) de Spotify...")
+    url = f'https://api.spotify.com/v1/playlists/{playlist_id}/tracks'
+    headers = {'Authorization': f'Bearer {token}'}
+    canciones = []
+
+    try:
+        response = requests.get(url, headers=headers)
+        if response.status_code == 429:  # Límite alcanzado
+            print("Límite de peticiones alcanzado, cambiando API Key...")
+            token = cambiar_api_key()
+            time.sleep(1)  # Pausa de 1 segundo para evitar el límite de tasa
+            response = requests.get(url, headers={'Authorization': f'Bearer {token}'})
+        if response.status_code == 200:
+            tracks = response.json().get('items')
+            for item in tracks:
+                track = item.get('track')
+                if track:
+                    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'),
+                        'duration': track.get('duration_ms', 'No disponible'),
+                        'release_year': track.get('album', {}).get('release_date', 'No disponible').split('-')[0] if track.get('album', {}).get('release_date') else 'No disponible',
+                        'record_label': obtener_record_label_spotify(track['album']['id'], token)
+                    })
+    except Exception as e:
+        print(f"Error al obtener canciones de la playlist: {e}")
+
+    return canciones
+
+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 == 429:  # Límite alcanzado
+        print("Límite de peticiones alcanzado al obtener características de audio, cambiando API Key...")
+        token = cambiar_api_key()
+        time.sleep(1)  # Pausa de 1 segundo para evitar el límite de tasa
+        response = requests.get(url, headers={'Authorization': f'Bearer {token}'})
+    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}'
+    headers = {'Authorization': f'Bearer {token}'}
+    response = requests.get(url, headers=headers)
+    if response.status_code == 429:  # Límite alcanzado
+        print("Límite de peticiones alcanzado al obtener la discográfica, cambiando API Key...")
+        token = cambiar_api_key()
+        time.sleep(1)  # Pausa de 1 segundo para evitar el límite de tasa
+        response = requests.get(url, headers={'Authorization': f'Bearer {token}'})
+    album_info = response.json() if response.status_code == 200 else {}
+    return album_info.get('label', 'No disponible')
+
+def interface(project_name, query, num_spotify_playlists=50):
+    # Obtener tokens y claves
+    token_spotify = obtener_token(client_ids[current_api_index], client_secrets[current_api_index])
+    if not token_spotify:
+        return pd.DataFrame(), "Error obteniendo el token de Spotify. Intenta de nuevo."
+
+    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'])
+        canciones_spotify.extend(songs)
+        time.sleep(1)  # Pausa de 1 segundo entre la obtención de canciones para evitar el límite de tasa
+
+    # Crear DataFrame
+    df = pd.DataFrame(canciones_spotify)
+    df.rename(columns={'isrc': 'ISRCs'}, inplace=True)
+
+    # Ordenar por popularidad
+    df.sort_values(by=['popularity'], ascending=False, inplace=True)
+
+    # Guardar DataFrame en un archivo Excel
+    tmpfile = NamedTemporaryFile(delete=False, suffix='.xlsx')
+    df.to_excel(tmpfile.name, index=False)
+
+    # Renombrar el archivo con el nombre del proyecto
+    project_file_name = f"{project_name}.xlsx"
+    shutil.move(tmpfile.name, project_file_name)
+
+    return df, project_file_name  # Devuelve el DataFrame y el enlace al archivo Excel
+
+# Configuración de Gradio
+iface = gr.Interface(
+    fn=interface,
+    inputs=[
+        gr.Textbox(label="Nombre del Proyecto"),
+        gr.Textbox(label="Keywords - Palabras Clave para tu búsqueda"),
+        gr.Number(label="Numero de Playlists que vamos a buscar con estas Keywords", value=50, minimum=1, maximum=1000)
+    ],
+    outputs=[gr.Dataframe(), gr.File(label="Download Excel")],
+    title="Spotify Playlist Fetcher",
+    description="Introduce una consulta de búsqueda para obtener playlists y sus canciones de Spotify. Las credenciales del cliente están preconfiguradas."
+)
+
+iface.launch()