Update app.py

app.py

@@ -56,20 +56,12 @@ class SentimentAnalyzer:
         return sentiment_scores
 
     def emotion_analysis_for_graph(self,text):
-
-        # list_of_emotion=text.split(":")
-        # label=list_of_emotion[0]
-        # score=list_of_emotion[1]
-        # score_dict={
-        #   label:score
-        # }
-        # print(score_dict)
-        emotions_match = re.search(r'\[(.*?)\]', text)
-        emotions_str = emotions_match.group(1)
-        emotions = emotions_str.split(', ')
-        scores_match = re.search(r'\[(.*?)\]', text, re.DOTALL)
-        scores_str = scores_match.group(1)
-        scores = list(map(float, scores_str.split(', ')))
+        start_index = text.find("[")
+        end_index = text.find("]")
+        list1_text = text[start_index + 1: end_index]
+        list2_text = text[end_index + 2:-1]
+        emotions = list(map(str.strip, list1_text.split(",")))
+        scores = list(map(float, list2_text.split(",")))
         score_dict={"Emotion": emotions, "Score": scores}
         print(score_dict)
         return score_dict