app.py

import discord
import os
import threading
from discord.ext import commands
import json
import time
import matplotlib.pyplot as plt
from io import BytesIO
import gradio_client
import gradio as gr
from gradio_client import Client


DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
intents = discord.Intents.all() 
bot = commands.Bot(command_prefix='!', intents=intents)

XP_PER_MESSAGE = 10

xp_data = {}


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
async def on_ready():
    print(f'Logged in as {bot.user.name}')


@bot.event
async def on_message(message):
    global xp_data
    
    if message.author.bot:
        return

    if message.author.id not in xp_data:
        xp_data[message.author.id] = 0
        
    old = xp_data[message.author.id]
    new = old + XP_PER_MESSAGE
    xp_data[message.author.id] = new
    level = calculate_level(new)
    
    print(f"{message.author.mention} xp: {xp_data[message.author.id]}")
    print(f"{message.author.mention} level: {level}")
    save_xp_data(xp_data)
    



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 level(ctx):
    global xp_data
    print(ctx.author.id)
    print(ctx.author.mention)
    if ctx.author.id == 811235357663297546: # lunarflu
        if ctx.author.id in xp_data:

            xp = xp_data[ctx.author.id]
            level = calculate_level(xp)

            await ctx.send(f'You are at level {level} with {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.')


    


        
""""""
def run_bot():
    bot.run(DISCORD_TOKEN)
threading.Thread(target=run_bot).start()
def greet(name):
    return "Hello " + name + "!"
demo = gr.Interface(fn=greet, inputs="text", outputs="text")
demo.launch()