Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import requests
 import pandas as pd
 import gradio as gr
+from tempfile import NamedTemporaryFile
 
 # —————————————
 # Config Spotify (Client Credentials)
@@ -30,10 +31,7 @@ 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 = []
+    genres = resp.json().get('genres', []) if resp.status_code == 200 else []
     _artist_cache[aid] = genres
     return genres
 
@@ -62,7 +60,8 @@ def fetch_playlist_table(url):
         r.raise_for_status()
         for item in r.json().get('items', []):
             tr = item.get('track', {})
-            if not tr: continue
+            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({
@@ -79,16 +78,23 @@ def fetch_playlist_table(url):
     return df
 
 # Interfaz Gradio
+
 def main(url):
     df = fetch_playlist_table(url)
-    return df
+    # Guarda CSV temporal
+    tmp = NamedTemporaryFile(delete=False, suffix='.csv')
+    df.to_csv(tmp.name, index=False)
+    return df, tmp.name
 
 iface = gr.Interface(
     fn=main,
     inputs=gr.Textbox(label="URL de playlist"),
-    outputs=gr.Dataframe(headers=['Artist','Title','ISRC','URL','URI','Genres']),
+    outputs=[
+        gr.Dataframe(headers=['Artist','Title','ISRC','URL','URI','Genres']),
+        gr.File(label="Descargar CSV")
+    ],
     title="🎵 Tabla de Playlist Spotify",
-    description="Trae Artist, Title, ISRC, URL, URI y Géneros de cada track"
+    description="Trae Artist, Title, ISRC, URL, URI y Géneros de cada track y permite descargar CSV"
 )
 
 if __name__ == '__main__':