Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import requests
+import pandas as pd
 import gradio as gr
 
 # —————————————
@@ -8,70 +9,87 @@ import gradio as gr
 CLIENT_ID     = os.getenv("SPOTIFY_CLIENT_IDS", "").split(',')[0]
 CLIENT_SECRET = os.getenv("SPOTIFY_CLIENT_SECRETS", "").split(',')[0]
 
+# Función para obtener token
 def get_token():
     resp = requests.post(
         "https://accounts.spotify.com/api/token",
-        data={"grant_type":"client_credentials"},
-        auth=(CLIENT_ID, CLIENT_SECRET),
+        data={"grant_type": "client_credentials"},
+        auth=(CLIENT_ID, CLIENT_SECRET)
     )
     resp.raise_for_status()
     return resp.json()["access_token"]
 
+# Extrae el ID de playlist de la URL
 def extract_pid(url):
     return url.strip().rstrip('/').split('/')[-1].split('?')[0]
 
-# —————————————
-# Función full-fetch
-# —————————————
-def fetch_full_playlist(url):
+# Caché de géneros por artista
+_artist_cache = {}
+
+def get_artist_genres(aid, headers):
+    if aid in _artist_cache:
+        return _artist_cache[aid]
+    resp = requests.get(f"https://api.spotify.com/v1/artists/{aid}", headers=headers)
+    if resp.status_code == 200:
+        genres = resp.json().get('genres', [])
+    else:
+        genres = []
+    _artist_cache[aid] = genres
+    return genres
+
+# Fetch y devuelve DataFrame
+def fetch_playlist_table(url):
     if not CLIENT_ID or not CLIENT_SECRET:
-        return "❌ Faltan credenciales SPOTIFY_CLIENT_IDS o SPOTIFY_CLIENT_SECRETS"
+        return pd.DataFrame([{'Error': 'Faltan credenciales'}])
+
     token = get_token()
     pid = extract_pid(url)
-    if len(pid) != 22:
-        return f"❌ ID inválido: '{pid}'"
     headers = {"Authorization": f"Bearer {token}"}
 
-    # 1) Metadata básica
+    # Metadata de playlist
     meta = requests.get(f"https://api.spotify.com/v1/playlists/{pid}", headers=headers)
     if meta.status_code != 200:
-        return f"❌ Error {meta.status_code}: no es pública o no existe"
-    info  = meta.json()
-    name  = info.get("name", "–")
-    total = info.get("tracks", {}).get("total", 0)
+        return pd.DataFrame([{'Error': f"Playlist error {meta.status_code}: {meta.json()}"}])
+    total = meta.json().get('tracks', {}).get('total', 0)
 
-    # 2) Traer TODOS los tracks en batches de 100
-    nombres = []
-    limit  = 100
+    # Paginación de tracks
+    records = []
+    limit = 100
     offset = 0
     while offset < total:
-        ruta = f"https://api.spotify.com/v1/playlists/{pid}/tracks"
-        params = {"limit": limit, "offset": offset}
-        r = requests.get(ruta, headers=headers, params=params)
+        params = {'limit': limit, 'offset': offset}
+        r = requests.get(f"https://api.spotify.com/v1/playlists/{pid}/tracks", headers=headers, params=params)
         r.raise_for_status()
-        items = r.json().get("items", [])
-        for it in items:
-            track = it.get("track")
-            if track and track.get("name"):
-                nombres.append(track["name"])
+        for item in r.json().get('items', []):
+            tr = item.get('track', {})
+            if not tr: continue
+            aid = tr['artists'][0]['id'] if tr.get('artists') else None
+            genres = get_artist_genres(aid, headers) if aid else []
+            records.append({
+                'Artist': tr['artists'][0]['name'] if tr.get('artists') else '',
+                'Title':  tr.get('name', ''),
+                'ISRC': tr.get('external_ids', {}).get('isrc', ''),
+                'URL':  tr.get('external_urls', {}).get('spotify', ''),
+                'URI':  tr.get('uri', ''),
+                'Genres': ', '.join(genres)
+            })
         offset += limit
 
-    # 3) Formatear salida
-    # Si la lista es gigante, tal vez quieras cambiar salto de línea por comas, o paginarla.
-    salida = [f"✅ “{name}”: {total} tracks"]
-    salida += [f"{i+1}. {t}" for i, t in enumerate(nombres)]
-    return "\n".join(salida)
+    df = pd.DataFrame(records)
+    return df
 
-# —————————————
 # Interfaz Gradio
-# —————————————
+def main(url):
+    df = fetch_playlist_table(url)
+    return df
+
 iface = gr.Interface(
-    fn=fetch_full_playlist,
+    fn=main,
     inputs=gr.Textbox(label="URL de playlist"),
-    outputs="text",
-    title="🔽 Baja toda la playlist",
-    description="Descarga el nombre y ¡TODOS! los temas de una playlist pública"
+    outputs=gr.Dataframe(headers=['Artist','Title','ISRC','URL','URI','Genres']),
+    title="🎵 Tabla de Playlist Spotify",
+    description="Trae Artist, Title, ISRC, URL, URI y Géneros de cada track"
 )
 
-if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+if __name__ == '__main__':
+    iface.launch(server_name='0.0.0.0', server_port=7860)