Update app.py

app.py

@@ -49,12 +49,12 @@ def heatmap(top_n):
 		api_name="/get_heatmap_pivot_table"
     )
     print(pivot_table)
-    print(type(pivot_table))
+    print(type(pivot_table)) # it's a dict
 
-    # df = pd.read_csv(StringIO(pivot_table), delim_whitespace=True, index_col=0)
+    df = pd.DataFrame.from_dict(pivot_table)
     
     plt.figure(figsize=(10, 8))
-    sns.heatmap(pivot_table,
+    sns.heatmap(df,
                 cmap='YlOrRd',
                 cbar_kws={'label': 'Weighted Frequency'},
                 square=True)