diff --git a/extensions/phixxycom.py b/extensions/phixxycom.py index 2128434..bc610ec 100644 --- a/extensions/phixxycom.py +++ b/extensions/phixxycom.py @@ -11,7 +11,6 @@ class PhixxyCom(commands.Cog): def __init__(self, bot): self.bot = bot - self.upload_enabled = os.getenv('upload_phixxy') self.SERVER = os.getenv('ftp_server') self.USERNAME = os.getenv('ftp_username') self.PASSWORD = os.getenv('ftp_password') @@ -46,11 +45,10 @@ class PhixxyCom(commands.Cog): return "Unknown Prompt" async def upload_sftp(self, local_filename, server_folder, server_filename): - if self.upload_enabled.lower() == "true": - 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) + 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): @@ -158,33 +156,32 @@ class PhixxyCom(commands.Cog): async def upload_ftp_ai_images(self, ai_dict): try: - if self.upload_enabled.lower() == "true": - for folder in ai_dict: - for filename in os.listdir(folder): - if filename[-4:] == '.png': - filepath = folder + filename - self.bot.logger.info(f"Found file = {filename}") - prompt = self.find_prompt_from_filename(ai_dict[folder], filename) - self.bot.logger.info(f"Found prompt = {prompt}") - html_file = f"{self.data_dir}ai-images/index.html" - html_insert = ''' -
{content}
" - content = content.replace('\n\n', "") - content = content.replace("
", '') + 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: + 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 != '': + self.bot.logger.info("Writing blogpost") + else: + messages = self.get_last_5_messages() + question = f"you have a blog and you are inspired based on this short text chat interaction:\n{messages}\nwhat will the topic of your next blog be? just tell me the topic and a one sentence description" + self.bot.logger.info("No topic given for blogpost, generating one.") + topic = await self.answer_question(question) + post_div = ''' +{content}
" + content = content.replace('\n\n', "") + content = content.replace("
", '') - post_div = post_div.replace("", title) - post_div = post_div.replace("", date) - post_div = post_div.replace("", content) - - html_data = html_data.replace("", 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 - self.bot.logger.debug("It took " + str(run_time) + " seconds to generate the blog post!") - output = "Blog Updated! (" + str(run_time) + " seconds) https://ai.phixxy.com/ai-blog" - return output + post_div = post_div.replace("", title) + post_div = post_div.replace("", date) + post_div = post_div.replace("", content) + + html_data = html_data.replace("", 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 + self.bot.logger.debug("It took " + str(run_time) + " seconds to generate the blog post!") + output = "Blog Updated! (" + str(run_time) + " seconds) https://ai.phixxy.com/ai-blog" + return output @commands.command() async def force_blog(self, ctx): @@ -371,6 +367,9 @@ class PhixxyCom(commands.Cog): await ctx.send(output) async def setup(bot): - asyncssh.set_log_level(30) - asyncssh.set_sftp_log_level(30) - await bot.add_cog(PhixxyCom(bot)) \ No newline at end of file + if os.getenv("upload_phixxy") == "true": + asyncssh.set_log_level(30) + asyncssh.set_sftp_log_level(30) + await bot.add_cog(PhixxyCom(bot)) + else: + pass \ No newline at end of file