Create app.py

app.py

@@ -0,0 +1,115 @@
+import gradio as gr
+import requests
+import pandas as pd
+from tempfile import NamedTemporaryFile
+from openpyxl import Workbook
+
+# Funciones para Spotify
+def obtener_token(client_id, client_secret):
+    print("Obteniendo token de Spotify...")
+    url = 'https://accounts.spotify.com/api/token'
+    headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+    payload = {'grant_type': 'client_credentials'}
+    response = requests.post(url, headers=headers, data=payload, auth=(client_id, client_secret))
+    return response.json().get('access_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}'}
+    playlists = []
+
+    if limit <= 50:
+        params = {'q': query, 'type': 'playlist', 'limit': limit}
+        response = requests.get(url, headers=headers, 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)
+            playlists.extend(response.json().get('playlists', {}).get('items', []))
+            limit -= min(50, limit)
+            offset += 50
+
+    return [{'playlist_id': playlist['id'], 'playlist_name': playlist['name']} for playlist in playlists]
+
+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}'}
+    response = requests.get(url, headers=headers)
+    canciones = []
+    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({
+                    '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'),
+                    'link': track['external_urls']['spotify'],
+                    'record_label': obtener_record_label_spotify(track['album']['id'], token),
+                    'source': 'Spotify'
+                })
+    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)
+    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)
+    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(query, num_spotify_playlists=50):
+    # Obtener tokens y claves
+    client_id = 'b4a2add66ffb4f1198b94b087b365c65'
+    client_secret = '8045eacf956a477299d2bc41752f1f73'
+
+    # Spotify
+    token_spotify = obtener_token(client_id, client_secret)
+    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'])
+        canciones_spotify.extend(songs)
+
+    # 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)
+
+    with NamedTemporaryFile(delete=False, suffix='.xlsx') as tmpfile:
+        df.to_excel(tmpfile.name, index=False)
+        return df, tmpfile.name  # Devuelve el DataFrame y el enlace al archivo Excel
+
+# Configuración de Gradio
+iface = gr.Interface(
+    fn=interface,
+    inputs=[
+        gr.Textbox(label="Search Query"),
+        gr.Number(label="Number of Spotify Playlists", 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()