Update app.py

app.py

@@ -219,6 +219,7 @@ def scatter_plot_fn_liveness(request: gr.Request):
         results = sp.current_user()
         print(results)
     df = get_features(sp)
+    print(df)
     return gr.ScatterPlot(
         value=df,
         x="acousticness",
@@ -258,6 +259,30 @@ def get_features(spotify):
     return df
 
 
+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():
     # redirects to spotify and comes back
@@ -271,13 +296,13 @@ with gr.Blocks() as demo:
         get_started_btn = gr.Button("Get Started")
     with gr.Row():
         spider_plot = gr.Plot()
-    with gr.Row():
-        with gr.Column():
-            with gr.Row():
-                with gr.Column():
-                    energy_plot = gr.ScatterPlot(show_label=False).style(container=True)
-                with gr.Column():
-                    liveness_plot = gr.ScatterPlot(show_label=False).style(container=True)
+    # with gr.Row():
+    #     with gr.Column():
+    #         with gr.Row():
+    #             with gr.Column():
+    #                 energy_plot = gr.ScatterPlot(show_label=False).style(container=True)
+    #             with gr.Column():
+    #                 liveness_plot = gr.ScatterPlot(show_label=False).style(container=True)
     with gr.Row():
         gr.Markdown(" ### We have recommendations for you!")
     with gr.Row():
@@ -289,12 +314,13 @@ with gr.Blocks() as demo:
             headers=["Song", "Album", "Artist"],
             datatype=["str", "str", "str"],
             label="Reccomended Songs",
-            value=[["Fired Up", "Fired Up", "Randy Houser"], ["Something Just Like This", "Memories...Do Not Open", "The Chainsmokers"]]  # TODO: replace with actual reccomendations once get_started() is implemeted.
+            value=[["Fired Up", "Fired Up", "Randy Houser"], ["Something Just Like This", "Memories... Do Not Open", "The Chainsmokers"]]  # TODO: replace with actual reccomendations once get_started() is implemeted.
         )
-
+    
+    demo.load(fn=get_spider_plot, outputs = spider_plot)
     demo.load(fn=heatmap_plot_fn, outputs = heatmap_plot)
-    demo.load(fn=scatter_plot_fn_energy, outputs = energy_plot)
-    demo.load(fn=scatter_plot_fn_liveness, outputs = liveness_plot)
+    # demo.load(fn=scatter_plot_fn_energy, outputs = energy_plot)
+    # demo.load(fn=scatter_plot_fn_liveness, outputs = liveness_plot)
 
 
 gradio_app = gr.mount_gradio_app(app, demo, "/gradio")