Update app.py

app.py

@@ -15,21 +15,26 @@ def generate_abuse_score_chart(dates, scores, labels):
     import io
     from PIL import Image
     from datetime import datetime
+    import re
 
-    try:
-        parsed_dates = [datetime.strptime(d, "%Y-%m-%d") for d in dates]
-    except Exception:
-        parsed_dates = list(range(len(dates)))
+    # Determine if all entries are valid dates
+    if all(re.match(r"\d{4}-\d{2}-\d{2}", d) for d in dates):
+        parsed_x = [datetime.strptime(d, "%Y-%m-%d") for d in dates]
+        x_labels = [d.strftime("%Y-%m-%d") for d in parsed_x]
+    else:
+        parsed_x = list(range(1, len(dates) + 1))
+        x_labels = [f"Message {i+1}" for i in range(len(dates))]
 
     fig, ax = plt.subplots(figsize=(8, 3))
-    ax.plot(parsed_dates, scores, marker='o', linestyle='-', color='darkred', linewidth=2)
+    ax.plot(parsed_x, scores, marker='o', linestyle='-', color='darkred', linewidth=2)
 
-    for i, (x, y) in enumerate(zip(parsed_dates, scores)):
+    for i, (x, y) in enumerate(zip(parsed_x, 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_xticks(parsed_x)
+    ax.set_xticklabels(x_labels)
+    ax.set_xlabel("")  # No axis label
     ax.set_ylabel("Abuse Score (%)")
     ax.set_ylim(0, 105)
     ax.grid(True)