From df5d850e6d08bf708a214d49abb2b3e4793fe13b Mon Sep 17 00:00:00 2001 From: insert username here Date: Sat, 23 Apr 2022 20:25:26 -0400 Subject: [PATCH] lets start slashing! --- aibot.py | 58 ++++++++++++++++++++++++++++++++++++++++++++------------ 1 file changed, 46 insertions(+), 12 deletions(-) diff --git a/aibot.py b/aibot.py index f893324..10de3a6 100644 --- a/aibot.py +++ b/aibot.py @@ -1,10 +1,10 @@ from textgenrnn import textgenrnn -import discord -from discord.ext import commands +import disnake +from disnake.ext import commands import regex as re import functools from aitextgen import aitextgen -import subprocess +import subprocess from chatterbot import ChatBot textgen = textgenrnn('insert3_weights.hdf5') @@ -17,9 +17,10 @@ bot = ChatBot( ], database_uri='sqlite:///database.sqlite3' ) -client = commands.Bot(command_prefix='.') +client = commands.Bot(command_prefix='.', test_guilds=[732793772697583623]) boton = True + def add_to_train(clean_content): with open("train.txt", "a") as train: train.write(f"{clean_content} \n") @@ -38,6 +39,12 @@ async def on_message(message): pass elif message.content.lower().startswith("p!"): pass + elif message.content.lower().startswith("_"): + pass + elif message.channel.id == 817397092376117248: + pass + elif message.channel.id == 779060858449821716: + pass elif message.content.lower().startswith("."): pass elif message.content.lower().startswith("?"): @@ -51,11 +58,40 @@ async def on_message(message): else: writing_function = functools.partial(add_to_train, message.clean_content) await client.loop.run_in_executor(None, writing_function) - if message.channel.id == 934533774249713664: - if message.author.bot != True: - bot_response = bot.get_response(message.clean_content) - await message.channel.send(bot_response) + if message.channel.id == 934533774249713664: + bot_response = bot.get_response(message.clean_content) + await message.channel.send(bot_response) +@client.slash_command(description="get a quote from the ai") +async def aiquote(inter): + await inter.response.defer() + try: + textgen.generate_to_file('insert3.txt', n=1) + with open("insert3.txt") as f: + quote = f.read() + await inter.response.send_message(content=f"{quote}") + except: + textgen.generate_to_file('insert3.txt', n=1) + with open("insert3.txt") as f: + quote = f.read() + await inter.edit_original_message(content=f"{quote}") + +@client.slash_command(description="give the ai a prompt to finish off") +async def aiprompt(inter, prompt: str): + await inter.response.defer() + endmsg = subprocess.check_output(['python3','promptai.py',f'{prompt}']) + non = endmsg.replace("%1B[1m".encode(), "**".encode()) + nona = non.replace("%1B[0m".encode(), "**".encode()) + nor = nona.replace("".encode(), "".encode()) + nora = nor.replace("".encode(), "".encode()) + norx = nora.replace("\x1b[1m".encode(), "".encode()) + nore = norx.replace("\\x1b[1m".encode(), "".encode()) + nor1 = nore.replace("\x1b[0m".encode(), "".encode()) + nor2 = nor1.replace("\\x1b[0m".encode(), "".encode()) + finalmsg = nor2.decode("utf-8") + print(finalmsg) + await inter.edit_original_message(content=f"{finalmsg}") + @client.command() async def aiquote(ctx): try: @@ -67,7 +103,7 @@ async def aiquote(ctx): textgen.generate_to_file('insert3.txt', n=1) with open("insert3.txt") as f: quote = f.read() - await ctx.send(quote) + await ctx.send(quote) @client.command() async def aiprompt(ctx): @@ -84,8 +120,6 @@ async def aiprompt(ctx): finalmsg = nor2.decode("utf-8") print(finalmsg) await ctx.send(finalmsg) - - @client.command() async def aidel(ctx): @@ -102,4 +136,4 @@ async def aidel(ctx): await ctx.send(quote) -client.run('bot token') +client.run('bottoken')