Update app.py

app.py

@@ -1,8 +1,11 @@
 
+import gradio as gr
 import requests
+import pandas as pd
 import time
-from collections import deque
-from datetime import datetime, timedelta
+from tempfile import NamedTemporaryFile
+from openpyxl import Workbook
+import shutil
 
 # Lista de credenciales de API de Spotify
 client_ids =[
@@ -30,11 +33,8 @@ client_secrets = [
     '9fdfa58ea0a94ce7a0cb34fa19fb7d74'
 ]
 current_api_index = 0
-session = requests.Session()
-
-# Registro de peticiones en los últimos 60 minutos
-request_log = deque()
 
+# Funciones para Spotify
 def obtener_token(client_id, client_secret):
     print(f"Obteniendo token de Spotify con client_id: {client_id}")
     url = 'https://accounts.spotify.com/api/token'
@@ -48,71 +48,146 @@ 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
-
-def registrar_peticion():
-    ahora = datetime.now()
-    request_log.append(ahora)
-    # Remover peticiones que tienen más de 60 minutos
-    while request_log and (ahora - request_log[0]) > timedelta(minutes=60):
-        request_log.popleft()
-
-def realizar_solicitud(url, params, token):
-    headers = {'Authorization': f'Bearer {token}'}
-    response = session.get(url, headers=headers, params=params)
-    registrar_peticion()
-    return response
-
-def realizar_solicitud_con_reintento(url, params):
-    token = obtener_nuevo_token()
-    espera = 1
-
-    while True:
-        response = realizar_solicitud(url, params, token)
-        print(f"Solicitud realizada. Código de estado: {response.status_code}")
-
-        if response.status_code == 429:
-            retry_after = int(response.headers.get('Retry-After', espera))
-            espera *= 2  # Incremento exponencial
-            print(f"Límite de peticiones alcanzado, esperando {retry_after} segundos antes de reintentar...")
-            time.sleep(retry_after)
-            token = obtener_nuevo_token()  # Obtener un nuevo token después de esperar
-        else:
-            return response
+    return obtener_token(client_ids[current_api_index], client_secrets[current_api_index])
+
+def manejar_errores(response):
+    if response.status_code == 429:
+        retry_after = int(response.headers.get('Retry-After', 1))
+        print(f"Límite de peticiones alcanzado. Reintentando en {retry_after} segundos.")
+        time.sleep(retry_after)
+        return True
+    return False
 
-# Ejemplo de uso
-def main():
+def buscar_playlists_spotify(token, query, limit=50):
+    print("Buscando playlists en Spotify...")
     url = 'https://api.spotify.com/v1/search'
-    params = {'q': 'rancheras', 'type': 'playlist', 'limit': 40}
+    headers = {'Authorization': f'Bearer {token}'}
+    playlists = []
+
+    try:
+        if limit <= 50:
+            params = {'q': query, 'type': 'playlist', 'limit': limit}
+            response = requests.get(url, headers=headers, params=params)
+            if manejar_errores(response):
+                token = cambiar_api_key()
+                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 manejar_errores(response):
+                    token = cambiar_api_key()
+                    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_reintento(url, params)
-    print("Respuesta de la primera solicitud:", response.json())
+    return [{'playlist_id': playlist['id'], 'playlist_name': playlist['name']} for playlist in playlists]
 
-    # 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_reintento(url, params)
-        print("Respuesta de la segunda solicitud:", response.json())
+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 manejar_errores(response):
+            token = cambiar_api_key()
+            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 manejar_errores(response):
+        token = cambiar_api_key()
+        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 manejar_errores(response):
+        token = cambiar_api_key()
+        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')
+
+# Función principal de la interfaz
+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])
+    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
 
-if __name__ == "__main__":
-    main()
+# 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="Enter a search query to fetch playlists and their songs from Spotify. Client credentials are pre-configured."
+)
+iface.launch()