insert/ai-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

lets start slashing!

Browse source
This commit is contained in:
insert username here 2022-04-23 20:25:26 -04:00 committed by GitHub
parent aed66203bb
commit df5d850e6d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 46 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
aibot.py
View file

@ -1,6 +1,6 @@
from textgenrnn import textgenrnn from textgenrnn import textgenrnn
import discord import disnake
from discord.ext import commands from disnake.ext import commands
import regex as re import regex as re
import functools import functools
from aitextgen import aitextgen from aitextgen import aitextgen
@ -17,9 +17,10 @@ bot = ChatBot(
], ],
database_uri='sqlite:///database.sqlite3' database_uri='sqlite:///database.sqlite3'
) )
client = commands.Bot(command_prefix='.') client = commands.Bot(command_prefix='.', test_guilds=[732793772697583623])
boton = True boton = True
def add_to_train(clean_content): def add_to_train(clean_content):
with open("train.txt", "a") as train: with open("train.txt", "a") as train:
train.write(f"{clean_content} \n") train.write(f"{clean_content} \n")
@ -38,6 +39,12 @@ async def on_message(message):
pass pass
elif message.content.lower().startswith("p!"): elif message.content.lower().startswith("p!"):
pass pass
elif message.content.lower().startswith("_"):
pass
elif message.channel.id == 817397092376117248:
pass
elif message.channel.id == 779060858449821716:
pass
elif message.content.lower().startswith("."): elif message.content.lower().startswith("."):
pass pass
elif message.content.lower().startswith("?"): elif message.content.lower().startswith("?"):
@ -52,10 +59,39 @@ async def on_message(message):
writing_function = functools.partial(add_to_train, message.clean_content) writing_function = functools.partial(add_to_train, message.clean_content)
await client.loop.run_in_executor(None, writing_function) await client.loop.run_in_executor(None, writing_function)
if message.channel.id == 934533774249713664: if message.channel.id == 934533774249713664:
if message.author.bot != True:
bot_response = bot.get_response(message.clean_content) bot_response = bot.get_response(message.clean_content)
await message.channel.send(bot_response) 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() @client.command()
async def aiquote(ctx): async def aiquote(ctx):
try: try:
@ -85,8 +121,6 @@ async def aiprompt(ctx):
print(finalmsg) print(finalmsg)
await ctx.send(finalmsg) await ctx.send(finalmsg)
@client.command() @client.command()
async def aidel(ctx): async def aidel(ctx):
await ctx.message.delete() await ctx.message.delete()