Update app.py

app.py

@@ -146,18 +146,17 @@ def analyze_single_message(text, thresholds):
     # Compute weighted average across all patterns (not just top 2)
     weighted_total = 0.0
     weight_sum = 0.0
-for label, score in zip(LABELS, scores):
-    weight = PATTERN_WEIGHTS.get(label, 1.0)
-    weighted_total += score * weight
-    weight_sum += weight
+    for label, score in zip(LABELS, scores):
+        weight = PATTERN_WEIGHTS.get(label, 1.0)
+        weighted_total += score * weight
+        weight_sum += weight
 
-abuse_score_raw = (weighted_total / weight_sum) * 100
+    abuse_score_raw = (weighted_total / weight_sum) * 100
 
-# Cap abuse intensity at 100 only if high-threat patterns are present
-if "threat" in threshold_labels or "control" in threshold_labels or "insults" in threshold_labels:
-    abuse_score = min(abuse_score_raw, 100)
-else:
-    abuse_score = min(abuse_score_raw, 95)
+    if "threat" in threshold_labels or "control" in threshold_labels or "insults" in threshold_labels:
+        abuse_score = min(abuse_score_raw, 100)
+    else:
+        abuse_score = min(abuse_score_raw, 95)
 
     stage = get_risk_stage(threshold_labels, sentiment)
 
@@ -171,6 +170,7 @@ else:
     print(f"Motifs: {motifs}")
     print(f"Contradiction: {contradiction_flag}")
     print("------------------\n")
+
     return abuse_score, threshold_labels, top_patterns, result, stage
 
 def analyze_composite(msg1, msg2, msg3, *answers_and_none):
@@ -200,6 +200,7 @@ def analyze_composite(msg1, msg2, msg3, *answers_and_none):
     out += generate_risk_snippet(composite_abuse, top_label, escalation_score)
     out += f"\n\n{stage_text}"
     return out
+    
 textbox_inputs = [gr.Textbox(label=f"Message {i+1}") for i in range(3)]
 quiz_boxes = [gr.Checkbox(label=q) for q, _ in ESCALATION_QUESTIONS]
 none_box = gr.Checkbox(label="None of the above")