From 2aa37755291d02c38d981b5478b7fe5b64346085 Mon Sep 17 00:00:00 2001
From: insert username here <montyd18@gmail.com>
Date: Sat, 22 Jan 2022 16:48:27 -0500
Subject: [PATCH] Update aibot.py

updates that were wayyyy behinf
---
 aibot.py | 83 +++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 82 insertions(+), 1 deletion(-)

diff --git a/aibot.py b/aibot.py
index 4129e5d..f893324 100644
--- a/aibot.py
+++ b/aibot.py
@@ -2,10 +2,59 @@ from textgenrnn import textgenrnn
 import discord
 from discord.ext import commands
 import regex as re
+import functools
+from aitextgen import aitextgen
+import subprocess
+from chatterbot import ChatBot
 
 textgen = textgenrnn('insert3_weights.hdf5')
+ai = aitextgen()
+bot = ChatBot(
+    'insert3',
+    storage_adapter='chatterbot.storage.SQLStorageAdapter',
+    logic_adapters=[
+        'chatterbot.logic.BestMatch'
+    ],
+    database_uri='sqlite:///database.sqlite3'
+)
 client = commands.Bot(command_prefix='.')
+boton = True
 
+def add_to_train(clean_content):
+    with open("train.txt", "a") as train:
+        train.write(f"{clean_content} \n")
+
+@client.event
+async def on_message(message):
+    await client.process_commands(message)
+    urls = re.findall('http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*(),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+',message.content.lower())
+    if not message.attachments:
+        if boton:
+            if message.content.lower().startswith("$"):
+                pass
+            elif urls:
+                pass
+            elif message.author.bot:
+                pass
+            elif message.content.lower().startswith("p!"):
+                pass
+            elif message.content.lower().startswith("."):
+                pass
+            elif message.content.lower().startswith("?"):
+                pass
+            elif message.content.lower().startswith("!"):
+                pass
+            elif message.content.lower().startswith("—"):
+                pass
+            elif message.content.lower().startswith("--"):
+                pass
+            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)
 
 @client.command()
 async def aiquote(ctx):
@@ -20,5 +69,37 @@ async def aiquote(ctx):
             quote = f.read()
         await ctx.send(quote)
 
+@client.command()
+async def aiprompt(ctx):
+    prompt = ctx.message.content[10:]
+    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 ctx.send(finalmsg)
+    
+    
 
-client.run('BOTTOKEN')
+@client.command()
+async def aidel(ctx):
+    await ctx.message.delete()
+    try:
+        textgen.generate_to_file('insert3.txt', n=1)
+        with open("insert3.txt") as f:
+            quote = f.read()
+        await ctx.send(quote)
+    except:
+        textgen.generate_to_file('insert3.txt', n=1)
+        with open("insert3.txt") as f:
+            quote = f.read()
+        await ctx.send(quote)
+
+
+client.run('bot token')