Update app.py

app.py

@@ -49,9 +49,21 @@ def heatmap(top_n):
 		api_name="/get_heatmap_pivot_table"
     )
     print(pivot_table)
-    print(type(pivot_table)) # it's a dict
+    print(type(pivot_table)) 
+    """
+    pivot_table is a dict like:
+    {'headers': ['Index', 'economy', 'human rights', 'immigrant', 'politics'], 
+    'data': [['anger', 55880.0, 557679.0, 147766.0, 180094.0], 
+             ['disgust', 26911.0, 123112.0, 64567.0, 46460.0], 
+             ['fear', 51466.0, 188898.0, 113174.0, 150578.0], 
+             ['neutral', 77005.0, 192945.0, 20549.0, 190793.0]], 
+    'metadata': None}
+    """
+    
 
-    df = pd.DataFrame.from_dict(pivot_table)
+    # transfere dictionary to df
+    df = pd.DataFrame(diction['data'], columns=diction['headers'])
+    df.set_index('Index', inplace=True)
     
     plt.figure(figsize=(10, 8))
     sns.heatmap(df,