revert to levelbot

app.py

@@ -15,6 +15,20 @@ DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
 intents = discord.Intents.all() 
 bot = commands.Bot(command_prefix='!', intents=intents)
 
+XP_PER_MESSAGE = 10
+
+
+def load_xp_data():
+    try:
+        with open('xp_data.json', 'r') as f:
+            return json.load(f)
+    except FileNotFoundError:
+        return {}
+
+
+def save_xp_data(xp_data):
+    with open('xp_data.json', 'w') as f:
+        json.dump(xp_data, f)
 
 
 @bot.event
@@ -22,62 +36,86 @@ async def on_ready():
     print(f'Logged in as {bot.user.name}')
 
 
-@bot.command()
-async def save_messages(ctx, channel_id: int):
-    if ctx.author.id == 811235357663297546:
-        channel = bot.get_channel(channel_id)
-        if not channel:
-            await ctx.send("Channel not found.")
-            return
+@bot.event
+async def on_message(message):
+    if message.author.bot:
+        return
+
+    author_id = str(message.author.id)
     
-        messages = []
-        async for message in channel.history(limit=None):
-            messages.append(message)
+    xp_data.setdefault(author_id, [])
+    timestamp = int(time.time())
+    xp_data[author_id].append((timestamp, XP_PER_MESSAGE))
     
-        with open("gradio-questions.json", 'w', encoding='utf-8') as file:
-            for message in messages:
-                file.write(f"{message.content}\n")
+    save_xp_data(xp_data)
     
-        await ctx.send(f"Messages from {channel.name} saved to gradio-questions.json")
-        
-        with open("gradio-questions.json", 'rb') as file:
-            discord_file = discord.File(file)
-            await ctx.send(file=discord_file)
+    await bot.process_commands(message)
+
+
+def calculate_level(xp):
+    return int(xp ** (1.0 / 3.0)) # 100k messages = lvl 100, good for super long term plan
 
 
 @bot.command()
-async def save_forum(ctx, channel_id: int):
-    if ctx.author.id == 811235357663297546:
-        channel = bot.get_channel(channel_id)
-        threads = channel.threads
-        messages = []
-        for thread in threads:
-            #messages.append(thread.name)
-            async for message in thread.history(limit=None):
-                messages.append(message)
-
-        messages = messages.reverse()
-        print(messages) # debug
-    
-        with open("gradio-questions.json", 'w', encoding='utf-8') as file:
-            for message in messages:
-                file.write(f"{message.content}\n")
-    
-        await ctx.send(f"Messages from {channel.name} saved to gradio-questions.json")
-        
-        with open("gradio-questions.json", 'rb') as file:
-            discord_file = discord.File(file)
-            await ctx.send(file=discord_file)
+async def level(ctx):
+    if ctx.author.id == 811235357663297546: # lunarflu
+        author_id = str(ctx.author.id)
+        if author_id in xp_data:
+            total_xp = sum(xp for _, xp in xp_data[author_id])
+            level = calculate_level(total_xp)
+            await ctx.send(f'You are at level {level} with {total_xp} total XP.')
+            # progress bar with ascii? could be nice, easy
+        else:
+            await ctx.send('You have not earned any XP yet.')
 
 
+@bot.command()
+async def plot_xp(ctx):
+    if ctx.author.id == 811235357663297546:
+        author_id = str(ctx.author.id)
+        if author_id in xp_data:
+            timestamps, xp_values = zip(*xp_data[author_id])
+            plt.plot(timestamps, xp_values)
+            plt.xlabel('Timestamp')
+            plt.ylabel('XP')
+            plt.title('XP Over Time')
+            
+            image_stream = BytesIO()
+            plt.savefig(image_stream, format='png')
+            plt.close()
+            
+            image_stream.seek(0)
+            await ctx.send(file=discord.File(fp=image_stream, filename='xp_graph.png'))
+        else:
+            await ctx.send('You have not earned any XP yet.')
 
 
+@bot.command()
+async def show_xp_data(ctx):
+    if ctx.author.id == 811235357663297546:
+        author_id = str(ctx.author.id)
+        if author_id in xp_data:
+            xp = xp_data[author_id]
+            await ctx.send(f'Your XP data: {xp}')
+        else:
+            await ctx.send('You have not earned any XP yet.')
 
 
+@bot.command()
+async def load_xp(ctx):
+    if ctx.author.id == 811235357663297546:
+        try:
+            xp_data.clear()  # Clear current XP data
+            with open('xp_data.json', 'r') as f:
+                loaded_data = json.load(f)
+                xp_data.update(loaded_data)  # Update with loaded data
+            await ctx.send('XP data has been loaded from the file.')
+        except FileNotFoundError:
+            await ctx.send('No XP data file found.')
 
 
+    
 
-        
 
         
 """"""