SamanthaStorm commited on
Commit
0b7d696
·
verified ·
1 Parent(s): 438222a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -499,7 +499,7 @@ def analyze_single_message(text, thresholds):
499
  print(f"Contradiction: {contradiction_flag}")
500
  print("------------------\n")
501
 
502
- return abuse_score, threshold_labels, top_patterns, {"label": sentiment}, stage, darvo_score, get_emotional_tone_tag
503
 
504
  def analyze_composite(msg1, date1, msg2, date2, msg3, date3, *answers_and_none):
505
  none_selected_checked = answers_and_none[-1]
@@ -525,13 +525,14 @@ def analyze_composite(msg1, date1, msg2, date2, msg3, date3, *answers_and_none):
525
 
526
  results = [(analyze_single_message(m, THRESHOLDS.copy()), d) for m, d in active]
527
  for result, date in results:
528
- assert len(result) == 6, "Unexpected output from analyze_single_message"
529
  abuse_scores = [r[0][0] for r in results]
530
  top_labels = [r[0][1][0] if r[0][1] else r[0][2][0][0] for r in results]
531
  top_scores = [r[0][2][0][1] for r in results]
532
  sentiments = [r[0][3]['label'] for r in results]
533
  stages = [r[0][4] for r in results]
534
  darvo_scores = [r[0][5] for r in results]
 
535
  dates_used = [r[1] or "Undated" for r in results] # Store dates for future mapping
536
 
537
 
 
499
  print(f"Contradiction: {contradiction_flag}")
500
  print("------------------\n")
501
 
502
+ return abuse_score, threshold_labels, top_patterns, {"label": sentiment}, stage, darvo_score, tone_tag
503
 
504
  def analyze_composite(msg1, date1, msg2, date2, msg3, date3, *answers_and_none):
505
  none_selected_checked = answers_and_none[-1]
 
525
 
526
  results = [(analyze_single_message(m, THRESHOLDS.copy()), d) for m, d in active]
527
  for result, date in results:
528
+ assert len(result) == 7, "Unexpected output from analyze_single_message"
529
  abuse_scores = [r[0][0] for r in results]
530
  top_labels = [r[0][1][0] if r[0][1] else r[0][2][0][0] for r in results]
531
  top_scores = [r[0][2][0][1] for r in results]
532
  sentiments = [r[0][3]['label'] for r in results]
533
  stages = [r[0][4] for r in results]
534
  darvo_scores = [r[0][5] for r in results]
535
+ tone_tags= [r[0][6] for r in results]
536
  dates_used = [r[1] or "Undated" for r in results] # Store dates for future mapping
537
 
538