Update app.py

app.py

@@ -10,25 +10,28 @@ from PIL import Image
 from datetime import datetime
 
 # --- Timeline Visualization Function ---
-def generate_risk_stage_timeline(dates, stages):
-    stage_labels = {
-        1: "Tension",
-        2: "Escalation",
-        3: "Reconciliation",
-        4: "Calm"
-    }
+def generate_abuse_score_chart(dates, scores, labels):
+    import matplotlib.pyplot as plt
+    import io
+    from PIL import Image
+    from datetime import datetime
+
     try:
         parsed_dates = [datetime.strptime(d, "%Y-%m-%d") for d in dates]
     except Exception:
         parsed_dates = list(range(len(dates)))
 
-    fig, ax = plt.subplots(figsize=(6, 2.5))
-    ax.step(parsed_dates, stages, where='mid', color='purple', linewidth=2, marker='o')
-    ax.set_yticks(list(stage_labels.keys()))
-    ax.set_yticklabels([stage_labels[s] for s in stage_labels])
-    ax.set_title("Risk Stage Timeline")
+    fig, ax = plt.subplots(figsize=(8, 3))
+    ax.plot(parsed_dates, scores, marker='o', linestyle='-', color='darkred', linewidth=2)
+
+    for i, (x, y) in enumerate(zip(parsed_dates, scores)):
+        label = labels[i]
+        ax.text(x, y + 2, f"{label}\n{int(y)}%", ha='center', fontsize=8, color='black')
+
+    ax.set_title("Abuse Intensity Over Time")
     ax.set_xlabel("Date")
-    ax.set_ylabel("Risk Stage")
+    ax.set_ylabel("Abuse Score (%)")
+    ax.set_ylim(0, 105)
     ax.grid(True)
     plt.tight_layout()
 
@@ -331,7 +334,8 @@ def analyze_composite(msg1, date1, msg2, date2, msg3, date3, *answers_and_none):
     out += f"\n\n{stage_text}"
     out += darvo_blurb
 
-    timeline_image = generate_risk_stage_timeline(dates_used, stages)
+    pattern_labels = [r[0][2][0][0] for r in results]  # top label for each message
+    timeline_image = generate_abuse_score_chart(dates_used, abuse_scores, pattern_labels)
     return out, timeline_image
     
 message_date_pairs = [(gr.Textbox(label=f"Message {i+1}"), gr.Textbox(label=f"Date {i+1} (optional)", placeholder="e.g. 2025-04-22")) for i in range(3)]