diff --git a/cogs/fun/logger.py b/cogs/fun/logger.py deleted file mode 100644 index a935214..0000000 --- a/cogs/fun/logger.py +++ /dev/null @@ -1,29 +0,0 @@ -import discord -from discord.ext import commands -import re -from datetime import datetime - -class Logger(commands.Cog): - def __init__(self, client): - self.client = client - self.first = True - self.last_update = datetime.now() - self.most_common = [] - - @commands.Cog.listener() - async def on_message(self, message): - if message.author.bot: - return - - if message.channel.id != 439466964625391637: - return - - if message.content.startswith(('.', '!', '-', '?', '/')): - return - - with open('data/naughty_step.txt', 'a') as file: - file.write(message.content + '\n') - - -def setup(client): - client.add_cog(Logger(client)) diff --git a/cogs/fun/responses.py b/cogs/fun/responses.py index a2dbcd1..3fbd5a4 100644 --- a/cogs/fun/responses.py +++ b/cogs/fun/responses.py @@ -1,17 +1,14 @@ import discord from discord.ext import commands -from textgenrnn import textgenrnn class Responses(commands.Cog): def __init__(self, client): self.client = client - self.model = textgenrnn("data/textgenrnn_weights.hdf5") @commands.Cog.listener() async def on_message(self, message: discord.Message): if message.author == self.client.user: - print("Cock") return # @applesauce @@ -19,12 +16,6 @@ class Responses(commands.Cog): await message.channel.send(f"Stop pinging us {message.author.mention} <:pinged:451198700832817202>") return - if "<@!657709911337074698>" in message.content: - print("In") - result = self.model.generate(temperature=0.7, return_as_list=True)[0] - await message.channel.send(result) - return - if message.author.id == 478006431589728259: if message.content == "<@&380535423233032193> NUMBERS REEEEEEEE": await message.channel.send(f"<@&380535423233032193> {message.author.mention} REEEEEEEE") diff --git a/cogs/fun/wolfram.py b/cogs/fun/wolfram.py new file mode 100644 index 0000000..b761e38 --- /dev/null +++ b/cogs/fun/wolfram.py @@ -0,0 +1,27 @@ +import discord +from discord.ext import commands +from wolframclient.evaluation import WolframLanguageSession +from wolframclient.language import wl, wlexpr +from util import embed + +class WolframAlpha(commands.Cog): + def __init__(self, client): + self.client = client + self.session = WolframLanguageSession() + + def __del__(self): + self.session.terminate() + + @commands.command(name="WolframLanguage", description="Evaluates WolframLanguage expressions (Not to be confused with WolframAlpha!)", usage="WolframLanguage ", aliases=["wl"]) + async def WolframLanguage(self, ctx, *expr): + msg = await ctx.send(embed=embed.make_embed("WolframLanguage", "Thinking...")) + expr = " ".join(expr) + evaluate = self.session.evaluate(expr) + + self.session.evaluate(wl.Export("eval.png", evaluate, "PNG")) + await msg.delete() + data = embed.make_embed_image("Result", "eval.png") + await ctx.send(embed=data[0], file=data[1]) + +def setup(client): + client.add_cog(WolframAlpha(client)) \ No newline at end of file diff --git a/data/textgenrnn_weights.hdf5 b/data/textgenrnn_weights.hdf5 deleted file mode 100644 index bbecbf9..0000000 Binary files a/data/textgenrnn_weights.hdf5 and /dev/null differ