First Commit

simulate_xp.py

@@ -0,0 +1,76 @@
+import random
+import math
+import matplotlib.pyplot as plt
+
+
+def calculate_xp_needed_for_next_level(level: int) -> int:
+    return int(
+        35 * (level**2) + 35 * level
+    )
+
+def get_level_from_xp(total_xp: int) -> int:
+    level = 0
+    while True:
+        xp_needed = calculate_xp_needed_for_next_level(level)
+        if total_xp < xp_needed:
+            break
+        level += 1
+    return level
+
+
+def simulate_xp(num_messages=1000):
+    total_xp = 0
+    xp_per_message = []
+    xp_progression = []
+    level_ups = []
+    current_level = 0
+
+    for i in range(num_messages):
+        # Simulate a message length between 5 and 120 characters
+        msg_length = random.randint(5, 120)
+        base = random.randint(2, 5)
+        length_bonus = min(msg_length, 100) // 20  # +1 XP per 20 chars, max +5
+        extra_xp = base + length_bonus
+        extra_xp = min(extra_xp, 8)  # cap at 8 XP per message
+        xp_per_message.append(extra_xp)
+        total_xp += extra_xp
+        xp_progression.append(total_xp)
+
+        # Check for level up
+        next_level_xp = calculate_xp_needed_for_next_level(current_level)
+        if total_xp >= next_level_xp:
+            current_level += 1
+            level_ups.append(i + 1)  # Message number where level up occurred
+
+    avg_xp = total_xp / num_messages
+    level = get_level_from_xp(total_xp)
+
+    print(f"Simulated {num_messages} messages.")
+    print(f"Total XP: {total_xp}")
+    print(f"Average XP per message: {avg_xp:.2f}")
+    print(f"Min XP in a message: {min(xp_per_message)}")
+    print(f"Max XP in a message: {max(xp_per_message)}")
+    print(f"Level reached: {level}")
+
+    # Plot XP progression and level-ups
+    plt.figure(figsize=(10, 6))
+    plt.plot(range(1, num_messages + 1), xp_progression, label="Total XP")
+    if level_ups:
+        plt.scatter(
+            level_ups,
+            [xp_progression[i - 1] for i in level_ups],
+            color="red",
+            label="Level Up",
+            zorder=5,
+        )
+    plt.xlabel("Message Number")
+    plt.ylabel("Cumulative XP")
+    plt.title("XP Progression Over Messages")
+    plt.grid(True)
+    plt.legend()
+    plt.tight_layout()
+    plt.show()
+
+
+if __name__ == "__main__":
+    simulate_xp(1000000)