Update app.py

app.py

@@ -243,7 +243,7 @@ def analyze_single_message(text, thresholds):
 
     return abuse_score, threshold_labels, top_patterns, result, stage, darvo_score
 
-def analyze_composite(msg1, msg2, msg3, *answers_and_none):
+def analyze_composite(msg1, date1, msg2, date2, msg3, date3, *answers_and_none):
     none_selected_checked = answers_and_none[-1]
     responses_checked = any(answers_and_none[:-1])
     none_selected = not responses_checked and none_selected_checked
@@ -260,17 +260,19 @@ def analyze_composite(msg1, msg2, msg3, *answers_and_none):
         )
 
     messages = [msg1, msg2, msg3]
-    active = [m for m in messages if m.strip()]
+    dates = [date1, date2, date3]
+    active = [(m, d) for m, d in zip(messages, dates) if m.strip()]
     if not active:
         return "Please enter at least one message."
 
-    results = [analyze_single_message(m, THRESHOLDS.copy()) for m in active]
-    abuse_scores = [r[0] for r in results]
-    top_labels = [r[2][0][0] for r in results]
-    top_scores = [r[2][0][1] for r in results]
-    sentiments = [r[3]['label'] for r in results]
-    stages = [r[4] for r in results]
-    darvo_scores = [r[5] for r in results]
+    results = [(analyze_single_message(m, THRESHOLDS.copy()), d) for m, d in active]
+    abuse_scores = [r[0][0] for r in results]
+    top_labels = [r[0][2][0][0] for r in results]
+    top_scores = [r[0][2][0][1] for r in results]
+    sentiments = [r[0][3]['label'] for r in results]
+    stages = [r[0][4] for r in results]
+    darvo_scores = [r[0][5] for r in results]
+    dates_used = [r[1] or "Undated" for r in results]  # Store dates for future mapping
 
     composite_abuse = int(round(sum(abuse_scores) / len(abuse_scores)))
     top_label = f"{top_labels[0]} – {int(round(top_scores[0] * 100))}%"
@@ -300,7 +302,8 @@ def analyze_composite(msg1, msg2, msg3, *answers_and_none):
 
     return out
     
-textbox_inputs = [gr.Textbox(label=f"Message {i+1}") for i in range(3)]
+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)]
+textbox_inputs = [item for pair in message_date_pairs for item in pair]  # Flatten for Gradio input
 quiz_boxes = [gr.Checkbox(label=q) for q, _ in ESCALATION_QUESTIONS]
 none_box = gr.Checkbox(label="None of the above")