Update charts.py

charts.py

@@ -1,10 +1,23 @@
 import plotly.graph_objects as go
 
-def spider_chart(classifier, text):
+def add_emoji(emotions_list):
+    for s in range(len(emotions_list)):
+      if emotions_list[s]=="surprise": emotions_list[s]="surprise 😲"
+      elif emotions_list[s]=="joy": emotions_list[s]="joy 😀"
+      elif emotions_list[s]=="anger": emotions_list[s]="anger 🤬"
+      elif emotions_list[s]=="neutral": emotions_list[s]="neutral 😐"
+      elif emotions_list[s]=="disgust": emotions_list[s]="disgust 🤢"
+      elif emotions_list[s]=="fear": emotions_list[s]="fear 😨"
+      elif emotions_list[s]=="sadness": emotions_list[s]="sadness 😭"
+      else: print(s)
+    
+    return emotions_list
+
+def spider_chart(dictionary):
 
   fig = go.Figure(data=go.Scatterpolar(
-    r=[round(s['score']*100,2) for s in classifier.predict(text)[0]],
-    theta= [l['label'] for l in classifier.predict(text)[0]],
+    r=[round(s*100,2) for s in list(dictionary.values())],
+    theta= add_emoji(list(dictionary.keys())),
     fill='toself'))
 
   fig.update_layout(