Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import requests
 import pandas as pd
+from io import BytesIO
 
 # —————— PARCHE Pydantic para FastAPI ——————
 import pydantic
@@ -14,104 +15,95 @@ def patched_get_type(schema):
     if not isinstance(schema, dict):
         return "Any"
     return _orig_get_type(schema)
-
 client_utils.get_type = patched_get_type
 # ————————————————————————————————
 
 import gradio as gr
-from io import BytesIO
 
-# Configuración Spotify (Client Credentials)
+# Config Spotify (Client Credentials)
 CLIENT_ID     = os.getenv("SPOTIFY_CLIENT_IDS", "").split(',')[0]
 CLIENT_SECRET = os.getenv("SPOTIFY_CLIENT_SECRETS", "").split(',')[0]
 
-# Función para obtener token
+# Obtiene token
 def get_token():
-    resp = requests.post(
+    r = requests.post(
         "https://accounts.spotify.com/api/token",
-        data={"grant_type": "client_credentials"},
+        data={"grant_type":"client_credentials"},
         auth=(CLIENT_ID, CLIENT_SECRET)
     )
-    resp.raise_for_status()
-    return resp.json()["access_token"]
+    r.raise_for_status()
+    return r.json()["access_token"]
 
-# Extrae el ID de playlist de la URL
+# Extrae ID de playlist
 def extract_pid(url):
     return url.strip().rstrip('/').split('/')[-1].split('?')[0]
 
-# Caché de géneros por artista
+# Cache géneros
 _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)
-    genres = resp.json().get('genres', []) if resp.status_code == 200 else []
+    r = requests.get(f"https://api.spotify.com/v1/artists/{aid}", headers=headers)
+    genres = r.json().get('genres', []) if r.status_code==200 else []
     _artist_cache[aid] = genres
     return genres
 
-# Fetch y devuelve DataFrame de toda la playlist
+# Trae tabla
 def fetch_playlist_table(url):
     if not CLIENT_ID or not CLIENT_SECRET:
-        return pd.DataFrame([{'Error': 'Faltan credenciales'}])
-
+        return pd.DataFrame([{'Error':'Faltan credenciales'}])
     token = get_token()
     pid = extract_pid(url)
-    headers = {"Authorization": f"Bearer {token}"}
-
-    # Metadata de playlist
+    headers={"Authorization":f"Bearer {token}"}
     meta = requests.get(f"https://api.spotify.com/v1/playlists/{pid}", headers=headers)
-    if meta.status_code != 200:
-        return pd.DataFrame([{'Error': f"Playlist error {meta.status_code}: {meta.json()}"}])
-    total = meta.json().get('tracks', {}).get('total', 0)
-
-    # Paginación de tracks
-    records = []
-    limit = 100
-    offset = 0
-    while offset < total:
-        params = {'limit': limit, 'offset': offset}
-        r = requests.get(f"https://api.spotify.com/v1/playlists/{pid}/tracks", headers=headers, params=params)
+    if meta.status_code!=200:
+        return pd.DataFrame([{'Error':f"Playlist error {meta.status_code}"}])
+    total = meta.json().get('tracks',{}).get('total',0)
+    records=[]
+    limit=100
+    offset=0
+    while offset<total:
+        r = requests.get(
+            f"https://api.spotify.com/v1/playlists/{pid}/tracks",
+            headers=headers,
+            params={'limit':limit,'offset':offset}
+        )
         r.raise_for_status()
-        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 []
+        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)
+                '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
-
-    df = pd.DataFrame(records)
-    return df
-
-# Función principal: retorna DataFrame y CSV en memoria
-def main(url):
-    df = fetch_playlist_table(url)
-    # Convertir DataFrame a CSV en memoria
-    csv_buffer = BytesIO()
-    df.to_csv(csv_buffer, index=False)
-    csv_bytes = csv_buffer.getvalue()
-    return df, ('playlist.csv', csv_bytes)
-
-# Interfaz Gradio con gr.Download
-iface = gr.Interface(
+        offset+=limit
+    return pd.DataFrame(records)
+
+# Función principal: retorna DataFrame y bytes para descargar
+ndef main(url):
+    df=fetch_playlist_table(url)
+    buf=BytesIO()
+    df.to_csv(buf,index=False)
+    buf.seek(0)
+    return df, ("playlist.csv", buf.getvalue())
+
+iface=gr.Interface(
     fn=main,
-    inputs=gr.Textbox(label="URL de playlist", placeholder="https://open.spotify.com/playlist/..."),
+    inputs=gr.Textbox(label="URL de playlist"),
     outputs=[
-        gr.Dataframe(headers=['Artist','Title','ISRC','URL','URI','Genres'], label="Tabla de Playlist Spotify"),
-        gr.Download(label="Descargar CSV")
+        gr.Dataframe(headers=['Artist','Title','ISRC','URL','URI','Genres'], label="Tabla Spotify"),
+        gr.File(label="Descargar CSV")
     ],
-    title="🎵 Tabla de Playlist Spotify",
-    description="Trae Artist, Title, ISRC, URL, URI y Géneros de cada track y permite descargar CSV sin usar disco"
+    title="🎵 Spotify Playlist Table",
+    description="Muestra y descarga CSV sin usar disco"
 )
 
-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)