diff --git a/sparkytron3000.py b/sparkytron3000.py index 3a3624c..59b4b2f 100644 --- a/sparkytron3000.py +++ b/sparkytron3000.py @@ -73,28 +73,33 @@ async def handle_error(error): with open("databases/error_log.txt", 'a') as f: f.write(log_line) return error - -async def upload_ftp_ai_images(filename, prompt): - html_file = "phixxy.com/ai-images/index.html" - html_insert = ''' -
- -

-
''' - img_list = [] - server_folder = (os.getenv('ftp_public_html') + 'ai-images/') - new_filename = str(time.time_ns()) + ".png" - await upload_sftp(filename, server_folder, new_filename) - print("Uploaded", new_filename) - with open(html_file, 'r') as f: - html_data = f.read() - html_insert = html_insert.replace("", new_filename) - html_insert = html_insert.replace("", prompt) - html_data = html_data.replace("", html_insert) - with open(html_file, "w") as f: - f.writelines(html_data) - await upload_sftp(html_file, server_folder, "index.html") + +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 = ''' +
+ +

+
''' + 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("", new_filename) + html_insert = html_insert.replace("", prompt) + html_data = html_data.replace("", 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 = { @@ -275,6 +280,7 @@ async def task_loop(): #Run every minute if current_time.tm_sec == 0: await meme_handler('tmp/meme/') + await upload_ftp_ai_images('tmp/sfw')