moved many functions out of sparkytron to phixxycom
This commit is contained in:
phixxy
parent
9a1097fc0a
commit
6e2fb0ae0d
2 changed files with 285 additions and 92 deletions
282
extensions/phixxycom.py
Normal file
282
extensions/phixxycom.py
Normal file
|
|
@ -0,0 +1,282 @@
|
|||
import os
|
||||
import io
|
||||
import base64
|
||||
import time
|
||||
import asyncssh
|
||||
from PIL import Image, PngImagePlugin
|
||||
from discord.ext import commands, tasks
|
||||
|
||||
class PhixxyCom(commands.Cog):
|
||||
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
self.SERVER = os.getenv('ftp_server')
|
||||
self.USERNAME = os.getenv('ftp_username')
|
||||
self.PASSWORD = os.getenv('ftp_password')
|
||||
self.working_dir = "tmp/phixxy.com/"
|
||||
self.data_dir = "data/phixxy.com/"
|
||||
self.folder_setup()
|
||||
self.phixxy_loop.start()
|
||||
|
||||
def folder_setup(self):
|
||||
try:
|
||||
if not os.path.exists(self.working_dir):
|
||||
os.mkdir(self.working_dir)
|
||||
if not os.path.exists(self.data_dir):
|
||||
os.mkdir(self.data_dir)
|
||||
except:
|
||||
print("PhixxyCom failed to make directories")
|
||||
|
||||
async def upload_sftp(self, local_filename, server_folder, server_filename):
|
||||
remotepath = server_folder + server_filename
|
||||
async with asyncssh.connect(self.SERVER, username=self.USERNAME, password=self.PASSWORD) as conn:
|
||||
async with conn.start_sftp_client() as sftp:
|
||||
await sftp.put(local_filename, remotepath=remotepath)
|
||||
|
||||
async def delete_local_pngs(self, local_folder):
|
||||
for filename in os.listdir(local_folder):
|
||||
if ".png" in filename:
|
||||
os.remove(local_folder + filename)
|
||||
|
||||
async def delete_ftp_pngs(self,server_folder):
|
||||
async with asyncssh.connect(os.getenv('ftp_server'), username=os.getenv('ftp_username'), password=os.getenv('ftp_password')) as conn:
|
||||
async with conn.start_sftp_client() as sftp:
|
||||
for filename in (await sftp.listdir(server_folder)):
|
||||
if '.png' in filename:
|
||||
try:
|
||||
print("Deleting", filename)
|
||||
await sftp.remove(server_folder+filename)
|
||||
except:
|
||||
print("Couldn't delete", filename)
|
||||
|
||||
async def extract_image_tags(self,code):
|
||||
count = code.count("<img")
|
||||
tags = []
|
||||
for x in range(0,count):
|
||||
index1 = code.find("<img")
|
||||
index2 = code[index1:].find(">") + index1 + 1
|
||||
img_tag = code[index1:index2]
|
||||
tags.append(img_tag)
|
||||
code = code[index2:]
|
||||
return tags
|
||||
|
||||
async def extract_image_alt_text(self,tags):
|
||||
alt_texts = []
|
||||
for tag in tags:
|
||||
index1 = tag.find("alt") + 5
|
||||
index2 = tag[index1:].find("\"") + index1
|
||||
alt_text = tag[index1:index2]
|
||||
alt_texts.append(alt_text)
|
||||
return alt_texts
|
||||
|
||||
async def generate_images(self, local_folder, image_list):
|
||||
url = os.getenv('stablediffusion_url')
|
||||
if url == "disabled":
|
||||
return
|
||||
file_list = []
|
||||
for image in image_list:
|
||||
filename = image.replace(" ", "").lower() + ".png"
|
||||
payload = {"prompt": image, "steps": 25}
|
||||
response = await self.bot.http_session.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
|
||||
r = await response.json()
|
||||
for i in r['images']:
|
||||
image = Image.open(io.BytesIO(base64.b64decode(i.split(",", 1)[0])))
|
||||
png_payload = {"image": "data:image/png;base64," + i}
|
||||
response2 = await self.bot.http_session.post(url=f'{url}/sdapi/v1/png-info', json=png_payload)
|
||||
pnginfo = PngImagePlugin.PngInfo()
|
||||
json_response = await response2.json()
|
||||
pnginfo.add_text("parameters", json_response.get("info"))
|
||||
image.save(local_folder + filename, pnginfo=pnginfo)
|
||||
file_list.append(filename)
|
||||
return file_list
|
||||
|
||||
async def add_image_filenames(self, code, file_list):
|
||||
for filename in file_list:
|
||||
code = code.replace("src=\"\"", "src=\""+ filename + "\"", 1)
|
||||
return code
|
||||
|
||||
async def upload_html_and_imgs(self, local_folder):
|
||||
|
||||
for filename in os.listdir(local_folder):
|
||||
if ".png" in filename:
|
||||
await self.upload_sftp(local_folder + filename, (os.getenv('ftp_public_html') + 'ai-webpage/'), filename)
|
||||
#explicitly upload html files last!
|
||||
for filename in os.listdir(local_folder):
|
||||
if ".html" in filename:
|
||||
await self.upload_sftp(local_folder + filename, (os.getenv('ftp_public_html') + 'ai-webpage/'), filename)
|
||||
|
||||
async def delete_derp_files(self, server_folder):
|
||||
async with asyncssh.connect(self.SERVER, username=self.USERNAME, password=self.PASSWORD) as conn:
|
||||
async with conn.start_sftp_client() as sftp:
|
||||
for filename in (await sftp.listdir(server_folder)):
|
||||
if filename == '.' or filename == '..' or filename == 'style.css' or filename == 'myScript.js':
|
||||
pass
|
||||
else:
|
||||
try:
|
||||
print("Deleting", filename)
|
||||
await sftp.remove(server_folder+filename)
|
||||
except:
|
||||
print("Couldn't delete", filename)
|
||||
|
||||
async def meme_handler(self, folder):
|
||||
for file in os.listdir(folder):
|
||||
filepath = folder + file
|
||||
await self.update_meme_webpage(filepath)
|
||||
|
||||
async def update_meme_webpage(self, filename):
|
||||
server_folder = (os.getenv('ftp_public_html') + 'ai-memes/')
|
||||
new_file_name = str(time.time_ns()) + ".png"
|
||||
await self.upload_sftp(filename, server_folder, new_file_name)
|
||||
print("Uploaded", new_file_name)
|
||||
with open("phixxy.com/ai-memes/index.html", 'r') as f:
|
||||
html_data = f.read()
|
||||
html_insert = '<!--ADD IMG HERE-->\n <img src="' + new_file_name + '" loading="lazy">'
|
||||
html_data = html_data.replace('<!--ADD IMG HERE-->',html_insert)
|
||||
with open("phixxy.com/ai-memes/index.html", "w") as f:
|
||||
f.writelines(html_data)
|
||||
await self.upload_sftp("phixxy.com/ai-memes/index.html", server_folder, "index.html")
|
||||
os.rename(filename, 'tmp/' + new_file_name)
|
||||
|
||||
async def upload_ftp_ai_images(self, folder):
|
||||
for filename in os.listdir(folder):
|
||||
if filename[-4:] == '.png':
|
||||
filepath = folder + filename
|
||||
prompt = "Unknown Prompt" # Will have to update this later
|
||||
|
||||
html_file = "phixxy.com/ai-images/index.html"
|
||||
html_insert = '''<!--REPLACE THIS COMMENT-->
|
||||
<div>
|
||||
<img src="<!--filename-->" loading="lazy">
|
||||
<p class="image-description"><!--description--></p>
|
||||
</div>'''
|
||||
server_folder = (os.getenv('ftp_public_html') + 'ai-images/')
|
||||
new_filename = str(time.time_ns()) + ".png"
|
||||
await self.upload_sftp(filepath, server_folder, new_filename)
|
||||
print("Uploaded", new_filename)
|
||||
with open(html_file, 'r') as f:
|
||||
html_data = f.read()
|
||||
html_insert = html_insert.replace("<!--filename-->", new_filename)
|
||||
html_insert = html_insert.replace("<!--description-->", prompt)
|
||||
html_data = html_data.replace("<!--REPLACE THIS COMMENT-->", html_insert)
|
||||
with open(html_file, "w") as f:
|
||||
f.writelines(html_data)
|
||||
await self.upload_sftp(html_file, server_folder, "index.html")
|
||||
os.rename(filepath, f"tmp/{new_filename}")
|
||||
|
||||
@commands.command()
|
||||
async def generate_blog(self, ctx):
|
||||
start_time = time.time()
|
||||
topic = ''
|
||||
filename = f"{self.data_dir}ai-blog/index.html"
|
||||
with open(filename, 'r', encoding="utf-8") as f:
|
||||
html_data = f.read()
|
||||
current_time = time.time()
|
||||
current_struct_time = time.localtime(current_time)
|
||||
date = time.strftime("%B %d, %Y", current_struct_time)
|
||||
if date in html_data:
|
||||
print("I already wrote a blog post today!")
|
||||
return
|
||||
blogpost_file = f"{self.data_dir}blog_topics.txt"
|
||||
if os.path.isfile(blogpost_file):
|
||||
with open(blogpost_file, 'r') as f:
|
||||
blogpost_topics = f.read()
|
||||
f.seek(0)
|
||||
topic = f.readline()
|
||||
blogpost_topics = blogpost_topics.replace(topic, '')
|
||||
with open(blogpost_file, 'w') as f:
|
||||
f.write(blogpost_topics)
|
||||
if topic != '':
|
||||
print("Writing blogpost")
|
||||
else:
|
||||
print("No topic given for blogpost, generating one.")
|
||||
topic = await self.answer_question("Give me one topic for an absurd blogpost.")
|
||||
|
||||
|
||||
post_div = '''<!--replace this with a post-->
|
||||
<div class="post">
|
||||
<h2 class="post-title"><!--POST_TITLE--></h2>
|
||||
<p class="post-date"><!--POST_DATE--></p>
|
||||
<div class="post-content">
|
||||
<!--POST_CONTENT-->
|
||||
</div>
|
||||
</div>'''
|
||||
title_prompt = 'generate an absurd essay title about ' + topic
|
||||
title = await self.answer_question(title_prompt, model="gpt-3.5-turbo")
|
||||
prompt = 'Write a satirical essay with a serious tone titled: "' + title + '". Do not label parts of the essay.'
|
||||
content = await self.answer_question(prompt, model="gpt-4")
|
||||
if title in content[:len(title)]:
|
||||
content = content.replace(title, '', 1)
|
||||
content = f"<p>{content}</p>"
|
||||
content = content.replace('\n\n', "</p><p>")
|
||||
content = content.replace("<p></p>", '')
|
||||
|
||||
post_div = post_div.replace("<!--POST_TITLE-->", title)
|
||||
post_div = post_div.replace("<!--POST_DATE-->", date)
|
||||
post_div = post_div.replace("<!--POST_CONTENT-->", content)
|
||||
|
||||
html_data = html_data.replace("<!--replace this with a post-->", post_div)
|
||||
with open(filename, 'w', encoding="utf-8") as f:
|
||||
f.write(html_data)
|
||||
await self.upload_sftp(filename, (os.getenv('ftp_public_html') + 'ai-blog/'), "index.html")
|
||||
run_time = time.time() - start_time
|
||||
print("It took " + str(run_time) + " seconds to generate the blog post!")
|
||||
output = "Blog Updated! (" + str(run_time) + " seconds) https://ai.phixxy.com/ai-blog"
|
||||
print(output)
|
||||
|
||||
@commands.command(
|
||||
description="Website",
|
||||
help="Generates a website using gpt 3.5. Usage: !website (topic)",
|
||||
brief="Generate a website"
|
||||
)
|
||||
async def website(self, ctx):
|
||||
server_folder = os.getenv('ftp_public_html') + 'ai-webpage/'
|
||||
local_folder = f"{self.working_dir}/webpage/"
|
||||
working_file = local_folder + "index.html"
|
||||
if not os.path.exists(local_folder):
|
||||
os.mkdir(local_folder)
|
||||
try:
|
||||
await ctx.send("Please wait, this will take a long time! You will be able to view the website here: https://ai.phixxy.com/ai-webpage/")
|
||||
with open(working_file, "w") as f:
|
||||
f.write("<!DOCTYPE html><html><head><script>setTimeout(function(){location.reload();}, 10000);</script><title>Generating Website</title><style>body {font-size: 24px;text-align: center;margin-top: 100px;}</style></head><body><p>This webpage is currently being generated. The page will refresh once it is complete. Please be patient.</p></body></html>")
|
||||
await self.upload_sftp(working_file, server_folder, "index.html")
|
||||
topic = ctx.message.content.split(" ", maxsplit=1)[1]
|
||||
prompt = "Generate a webpage using html and inline css. The webpage topic should be " + topic + ". Feel free to add image tags with alt text. Leave the image source blank. The images will be added later."
|
||||
code = await self.answer_question(prompt)
|
||||
|
||||
|
||||
await self.delete_local_pngs(local_folder)
|
||||
await self.delete_ftp_pngs(server_folder)
|
||||
|
||||
tags = await self.extract_image_tags(code)
|
||||
alt_texts = await self.extract_image_alt_text(tags)
|
||||
file_list = await self.generate_images(local_folder, alt_texts)
|
||||
code = await self.add_image_filenames(code, file_list)
|
||||
|
||||
with open(working_file, 'w') as f:
|
||||
f.write(code)
|
||||
f.close()
|
||||
|
||||
await self.upload_html_and_imgs(local_folder)
|
||||
|
||||
await ctx.send("Finished https://ai.phixxy.com/ai-webpage/")
|
||||
except Exception as error:
|
||||
await self.handle_error(error)
|
||||
await ctx.send("Failed, Try again.")
|
||||
|
||||
@tasks.loop(seconds=1)
|
||||
async def phixxy_loop(self):
|
||||
current_time = time.localtime()
|
||||
#Run every minute
|
||||
if current_time.tm_sec == 0:
|
||||
await self.meme_handler('tmp/meme/')
|
||||
await self.upload_ftp_ai_images('tmp/sfw/')
|
||||
if current_time.tm_hour == 17 and current_time.tm_min == 0 and current_time.tm_sec == 0:
|
||||
bot_stuff_channel = self.bot.get_channel(544408659174883328)
|
||||
await bot_stuff_channel.send("Running Daily Tasks")
|
||||
try:
|
||||
await self.generate_blog()
|
||||
except Exception as error:
|
||||
await self.handle_error(error)
|
||||
|
||||
async def setup(bot):
|
||||
await bot.add_cog(PhixxyCom(bot))
|
||||
|
|
@ -36,46 +36,16 @@ intents.message_content = True
|
|||
bot = commands.Bot(command_prefix='!', intents=intents)
|
||||
#discord setup END
|
||||
|
||||
async def upload_sftp(local_filename, server_folder, server_filename):
|
||||
remotepath = server_folder + server_filename
|
||||
async with asyncssh.connect(ftp_server, username=ftp_username, password=ftp_password) as conn:
|
||||
async with conn.start_sftp_client() as sftp:
|
||||
await sftp.put(local_filename, remotepath=remotepath)
|
||||
|
||||
|
||||
async def handle_error(error):
|
||||
print(error)
|
||||
current_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
||||
log_line = current_time + ': ' + str(error) + '\n'
|
||||
with open("databases/error_log.txt", 'a') as f:
|
||||
with open("data/error_log.txt", 'a') as f:
|
||||
f.write(log_line)
|
||||
return error
|
||||
|
||||
async def upload_ftp_ai_images(folder):
|
||||
for filename in os.listdir(folder):
|
||||
if filename[-4:] == '.png':
|
||||
filepath = folder + filename
|
||||
prompt = "Unknown Prompt" # Will have to update this later
|
||||
|
||||
html_file = "phixxy.com/ai-images/index.html"
|
||||
html_insert = '''<!--REPLACE THIS COMMENT-->
|
||||
<div>
|
||||
<img src="<!--filename-->" loading="lazy">
|
||||
<p class="image-description"><!--description--></p>
|
||||
</div>'''
|
||||
server_folder = (os.getenv('ftp_public_html') + 'ai-images/')
|
||||
new_filename = str(time.time_ns()) + ".png"
|
||||
await upload_sftp(filepath, server_folder, new_filename)
|
||||
print("Uploaded", new_filename)
|
||||
with open(html_file, 'r') as f:
|
||||
html_data = f.read()
|
||||
html_insert = html_insert.replace("<!--filename-->", new_filename)
|
||||
html_insert = html_insert.replace("<!--description-->", prompt)
|
||||
html_data = html_data.replace("<!--REPLACE THIS COMMENT-->", html_insert)
|
||||
with open(html_file, "w") as f:
|
||||
f.writelines(html_data)
|
||||
await upload_sftp(html_file, server_folder, "index.html")
|
||||
os.rename(filepath, f"tmp/{new_filename}")
|
||||
|
||||
def create_channel_config(filepath):
|
||||
config_dict = {
|
||||
"personality":"average",
|
||||
|
|
@ -199,56 +169,16 @@ async def delete_all_files(path, safe_folders):
|
|||
elif os.path.isfile(path+filename):
|
||||
os.remove(path+filename)
|
||||
|
||||
async def delete_derp_files(server_folder):
|
||||
async with asyncssh.connect(ftp_server, username=ftp_username, password=ftp_password) as conn:
|
||||
async with conn.start_sftp_client() as sftp:
|
||||
for filename in (await sftp.listdir(server_folder)):
|
||||
if filename == '.' or filename == '..' or filename == 'style.css' or filename == 'myScript.js':
|
||||
pass
|
||||
else:
|
||||
try:
|
||||
print("Deleting", filename)
|
||||
await sftp.remove(server_folder+filename)
|
||||
except:
|
||||
print("Couldn't delete", filename)
|
||||
|
||||
async def meme_handler(folder):
|
||||
for file in os.listdir(folder):
|
||||
filepath = folder + file
|
||||
await update_meme_webpage(filepath)
|
||||
|
||||
|
||||
@tasks.loop(seconds=1) # Run the task every second
|
||||
async def task_loop():
|
||||
current_time = time.localtime()
|
||||
#Run every minute
|
||||
if current_time.tm_sec == 0:
|
||||
await meme_handler('tmp/meme/')
|
||||
await upload_ftp_ai_images('tmp/sfw/')
|
||||
|
||||
#Run daily tasks
|
||||
if current_time.tm_hour == 17 and current_time.tm_min == 0 and current_time.tm_sec == 0:
|
||||
bot_stuff = bot.get_channel(544408659174883328)
|
||||
output = 'The following tasks failed:\n```'
|
||||
failed_tasks = []
|
||||
await bot_stuff.send("<@242018983241318410> The current time is 5pm. Running daily tasks!")
|
||||
if current_time.tm_hour == 0 and current_time.tm_min == 0 and current_time.tm_sec == 0:
|
||||
try:
|
||||
await delete_all_files("tmp/")
|
||||
except Exception as error:
|
||||
await handle_error(error)
|
||||
failed_tasks.append("Delete tmp/")
|
||||
try:
|
||||
await delete_derp_files("/home/debian/websites/derp.phixxy.com/files/")
|
||||
except Exception as error:
|
||||
await handle_error(error)
|
||||
failed_tasks.append("Delete derp files")
|
||||
if failed_tasks != []:
|
||||
for failed_task in failed_tasks:
|
||||
output += failed_task + '\n'
|
||||
output += '```'
|
||||
await bot_stuff.send(output)
|
||||
else:
|
||||
await bot_stuff.send("All daily tasks successfully ran!")
|
||||
|
||||
async def create_session():
|
||||
return aiohttp.ClientSession()
|
||||
|
|
@ -272,32 +202,13 @@ async def on_disconnect():
|
|||
async def on_ready():
|
||||
folders_made = await folder_setup()
|
||||
await delete_all_files("tmp/", folders_made)
|
||||
|
||||
# Import plugins from extensions folder
|
||||
for plugin_file in os.listdir('extensions/'):
|
||||
if plugin_file != '__init__.py' and plugin_file[-3:] == '.py':
|
||||
await bot.load_extension(f'extensions.{plugin_file[:-3]}')
|
||||
|
||||
print('We have logged in as {0.user}'.format(bot))
|
||||
|
||||
task_loop.start()
|
||||
|
||||
|
||||
async def update_meme_webpage(filename):
|
||||
server_folder = (os.getenv('ftp_public_html') + 'ai-memes/')
|
||||
new_file_name = str(time.time_ns()) + ".png"
|
||||
await upload_sftp(filename, server_folder, new_file_name)
|
||||
print("Uploaded", new_file_name)
|
||||
with open("phixxy.com/ai-memes/index.html", 'r') as f:
|
||||
html_data = f.read()
|
||||
html_insert = '<!--ADD IMG HERE-->\n <img src="' + new_file_name + '" loading="lazy">'
|
||||
html_data = html_data.replace('<!--ADD IMG HERE-->',html_insert)
|
||||
with open("phixxy.com/ai-memes/index.html", "w") as f:
|
||||
f.writelines(html_data)
|
||||
await upload_sftp("phixxy.com/ai-memes/index.html", server_folder, "index.html")
|
||||
os.rename(filename, 'tmp/' + new_file_name)
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_reaction_add(reaction, user):
|
||||
if not random.randint(0,9):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue