Update app.py

app.py

@@ -7,6 +7,7 @@ import gradio as gr
 import uvicorn
 from fastapi.responses import HTMLResponse
 from fastapi.responses import RedirectResponse
+import pandas as pd
 
 import spotipy
 from spotipy import oauth2
@@ -55,6 +56,29 @@ def scatter_plot_fn(request: gr.Request):
         value=iris,
     )
 
+def get_features(spotify):
+    features = []
+    for index in range(0, 10):
+        results = spotify.current_user_saved_tracks(offset=index*50, limit=50)
+        track_ids = [item['track']['id'] for item in results['items']]
+        features.extend(spotify.audio_features(track_ids))
+
+    df = pd.DataFrame(data=features)
+    names = [
+        'danceability', 
+        'energy',
+        'loudness',
+        'speechiness',
+        'acousticness',
+        'instrumentalness',
+        'liveness',
+        'valence',
+        'tempo',
+    ]
+    features_means = df[names].mean()
+    # print (features_means.to_json())
+    return features_means
+
 
 ##########
 def get_started():