Update app.py

app.py

@@ -1,10 +1,5 @@
 import requests
 import time
-import gradio as gr
-import pandas as pd
-from tempfile import NamedTemporaryFile
-from openpyxl import Workbook
-import shutil
 
 # Lista de credenciales de API de Spotify
 client_ids = [
@@ -20,6 +15,7 @@ client_secrets = [
     '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}")
@@ -34,137 +30,48 @@ 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}")
-    time.sleep(60)  # Pausa de 1 minuto para evitar el límite de tasa
-    return obtener_token(client_ids[current_api_index], client_secrets[current_api_index])
+    return client_ids[current_api_index], client_secrets[current_api_index]
 
-def manejar_limite_de_tasa(response, token):
-    if response.status_code == 429:  # Límite alcanzado
-        print("Límite de peticiones alcanzado, esperando antes de cambiar API Key...")
-        retry_after = int(response.headers.get('Retry-After', 60))  # Espera recomendada
-        time.sleep(retry_after)
-        return cambiar_api_key()
+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 buscar_playlists_spotify(token, query, limit=40):
-    print("Buscando playlists en Spotify...")
-    url = 'https://api.spotify.com/v1/search'
+def realizar_solicitud_con_nuevo_token(url, params):
+    token = obtener_nuevo_token()
     headers = {'Authorization': f'Bearer {token}'}
-    playlists = []
+    response = session.get(url, headers=headers, params=params)
+    print(f"Solicitud realizada con nuevo token. Código de estado: {response.status_code}")
+    return response
 
-    try:
-        if limit <= 40:
-            params = {'q': query, 'type': 'playlist', 'limit': limit}
-            response = requests.get(url, headers=headers, params=params)
-            token = manejar_limite_de_tasa(response, token)
-            playlists.extend(response.json().get('playlists', {}).get('items', []))
-        else:
-            offset = 0
-            while limit > 0:
-                params = {'q': query, 'type': 'playlist', 'limit': min(40, limit), 'offset': offset}
-                response = requests.get(url, headers=headers, params=params)
-                token = manejar_limite_de_tasa(response, token)
-                playlists.extend(response.json().get('playlists', {}).get('items', []))
-                limit -= min(40, limit)
-                offset += 40
-                time.sleep(3)  # Pausa de 3 segundos entre las solicitudes para evitar el límite de tasa
-    except Exception as e:
-        print(f"Error al buscar playlists: {e}")
+# Ejemplo de uso
+def main():
+    url = 'https://api.spotify.com/v1/search'
+    params = {'q': 'rancheras', 'type': 'playlist', 'limit': 40}
     
-    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)
-        token = manejar_limite_de_tasa(response, 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)
-    token = manejar_limite_de_tasa(response, 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)
-    token = manejar_limite_de_tasa(response, 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=40):
-    # 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(3)  # Pausa de 3 segundos 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=40, 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()
+    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()