Update app.py

app.py

@@ -55,16 +55,16 @@ class SentimentAnalyzer:
         }
         return sentiment_scores
 
-    # def emotion_analysis_for_graph(self,text):
+    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:float(score)
-    #     }
-    #     print(score_dict)
-    #     return score_dict
+        list_of_emotion=text.split(":")
+        label=list_of_emotion[0]
+        score=list_of_emotion[1]
+        score_dict={
+          label:float(score)
+        }
+        print(score_dict)
+        return score_dict
 
 
 class Summarizer:
@@ -203,7 +203,7 @@ class LangChain_Document_QA:
         customer_emotion_fig=self._display_graph(customer_emotion_score)
         customer_emotion_fig.update_layout(title="Emotion Analysis",width=770)
 
-        return scores
+        return scores,customer_fig,customer_emotion_fig
 
 
     def clear_func(self):
@@ -265,7 +265,7 @@ class LangChain_Document_QA:
           emptyBtn.click(self.clear_func,[],[])
           emptyBtn.click(lambda: None, None, chatbot, queue=False)
 
-          Sentiment_btn.click(self._on_sentiment_btn_click,[],[txt5])
+          Sentiment_btn.click(self._on_sentiment_btn_click,[],[txt5,plot,plot_3])
 
       demo.title = "AI Mental Healthcare ChatBot"
       demo.launch()