Update app.py

app.py

@@ -177,10 +177,10 @@ class LangChain_Document_QA:
 
 
 
- def _text_box(self,customer_emotion,customer_sentiment_score):
- #agent_score = ", ".join([f"{key}: {value:.2f}" for key, value in agent_sentiment_score.items()])
+ def _text_box(self,customer_emotion,agent_emotion,agent_sentiment_score,customer_sentiment_score):
+ agent_score = ", ".join([f"{key}: {value:.2f}" for key, value in agent_sentiment_score.items()])
  customer_score = ", ".join([f"{key}: {value:.2f}" for key, value in customer_sentiment_score.items()])
- return f"customer_emotion:{customer_emotion}\nCustomer_sentiment_score:{customer_score}"
+ return f"customer_emotion:{customer_emotion}\nagent_emotion:{agent_emotion}\nAgent_Sentiment_score:{agent_score}\nCustomer_sentiment_score:{customer_score}"
 
  def _on_sentiment_btn_click(self):
  client,agent=self._history_of_chat()
@@ -188,28 +188,28 @@ class LangChain_Document_QA:
  customer_emotion=sentiment.emotion_analysis(client)
  customer_sentiment_score = sentiment.analyze_sentiment_for_graph(client)
 
- # agent_emotion=sentiment.emotion_analysis(agent)
- # agent_sentiment_score = sentiment.analyze_sentiment_for_graph(agent)
+ agent_emotion=sentiment.emotion_analysis(agent)
+ agent_sentiment_score = sentiment.analyze_sentiment_for_graph(agent)
 
- scores=self._text_box(customer_emotion,customer_sentiment_score)
+ scores=self._text_box(customer_emotion,agent_emotion,agent_sentiment_score,customer_sentiment_score)
 
  customer_fig=self._display_graph(customer_sentiment_score)
- customer_fig.update_layout(title="Sentiment Analysis",width=775)
+ customer_fig.update_layout(title="Sentiment Analysis",width=770)
 
- # agent_fig=self._display_graph(agent_sentiment_score)
- # agent_fig.update_layout(title="Sentiment Analysis",width=775)
+ agent_fig=self._display_graph(agent_sentiment_score)
+ agent_fig.update_layout(title="Sentiment Analysis",width=770)
 
- #agent_emotion_score = sentiment.emotion_analysis_for_graph(agent_emotion)
+ agent_emotion_score = sentiment.emotion_analysis_for_graph(agent_emotion)
 
- # agent_emotion_fig=self._display_graph(agent_emotion_score)
- # agent_emotion_fig.update_layout(title="Emotion Analysis",width=775)
+ agent_emotion_fig=self._display_graph(agent_emotion_score)
+ agent_emotion_fig.update_layout(title="Emotion Analysis",width=770)
 
  customer_emotion_score = sentiment.emotion_analysis_for_graph(customer_emotion)
 
  customer_emotion_fig=self._display_graph(customer_emotion_score)
- customer_emotion_fig.update_layout(title="Emotion Analysis",width=775)
+ customer_emotion_fig.update_layout(title="Emotion Analysis",width=770)
 
- return scores,customer_fig,customer_emotion_fig
+ return scores,customer_fig,agent_fig,customer_emotion_fig,agent_emotion_fig
 
 
  def clear_func(self):
@@ -255,8 +255,7 @@ class LangChain_Document_QA:
  lines=4,
  placeholder="Sentiment",
  ).style(container=False)
- with gr.Row():
- with gr.Column(scale=0.50, min_width=0):
+ 
  end_btn=gr.Button(
  value="End",
  )
@@ -271,7 +270,11 @@ class LangChain_Document_QA:
  plot =gr.Plot(label="Patient", size=(500, 600))
  with gr.Column(scale=0.50, min_width=0):
  plot_3 =gr.Plot(label="Patient_Emotion", size=(500, 600))
-
+ with gr.Row():
+ with gr.Column(scale=0.50, min_width=0):
+ plot_2 =gr.Plot(label="Psychotherapist", size=(500, 600))
+ with gr.Column(scale=0.50, min_width=0):
+ plot_4 =gr.Plot(label="Psychotherapist_Emotion", size=(500, 600))
 
 
  txt_msg = txt.submit(self._add_text, [chatbot, txt], [chatbot, txt])
@@ -282,7 +285,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,plot,plot_3])
+ Sentiment_btn.click(self._on_sentiment_btn_click,[],[txt5,plot,plot_2,plot_3,plot_4])
 
  demo.title = "AI Mental Healthcare ChatBot"
  demo.launch()