From dcb950f18690cbcb4309f65d8054aa6a002e72e2 Mon Sep 17 00:00:00 2001 From: phixxy Date: Sun, 9 Jul 2023 13:24:30 -0700 Subject: [PATCH] Removed reliance on this file, it is no longer needed --- website_image_gen.py | 220 ------------------------------------------- 1 file changed, 220 deletions(-) delete mode 100644 website_image_gen.py diff --git a/website_image_gen.py b/website_image_gen.py deleted file mode 100644 index b3868ae..0000000 --- a/website_image_gen.py +++ /dev/null @@ -1,220 +0,0 @@ -import requests -from PIL import Image, PngImagePlugin -import io -import base64 -from ftplib import FTP -import os - -ftp_server = "themis.feralhosting.com" -ftp_username = "bottlecap" -ftp_password = "UjSKCglyT5YlHMml" -ftp_ai_webpage = "/media/sdq1/bottlecap/www/phixxy.com/public_html/ai-webpage/" -ftp_ai_webpage_archive = "/media/sdq1/bottlecap/www/phixxy.com/public_html/webpage-archive/" - -def upload_html_and_imgs(): - with FTP(ftp_server) as ftp: - ftp.login(ftp_username, ftp_password) - ftp.cwd(ftp_ai_webpage) - for filename in os.listdir("webpage/"): - if ".png" in filename: - ftp.storbinary("STOR " + filename, open("webpage/" + filename, "rb")) - #explicitly upload html files last! - for filename in os.listdir("webpage/"): - if ".html" in filename: - ftp.storbinary("STOR " + filename, open("webpage/" + filename, "rb")) - -def upload_websites(): - with FTP(ftp_server) as ftp: - ftp.login(ftp_username, ftp_password) - for foldername in os.listdir("websites/"): - ftp.cwd(ftp_ai_webpage_archive) - if "archived.txt" not in os.listdir("websites/" +foldername): - ftp.mkd(foldername) - ftp.cwd(ftp_ai_webpage_archive + foldername) - for filename in os.listdir("websites/" +foldername): - if ".png" in filename: - ftp.storbinary("STOR " + filename, open("websites/" + foldername + "/" + filename, "rb")) - print("Uploaded websites/" + foldername + "/" + filename) - #explicitly upload html files last! - for filename in os.listdir("websites/" +foldername): - if ".html" in filename: - ftp.storbinary("STOR " + filename, open("websites/" + foldername + "/" + filename, "rb")) - print("Uploaded websites/" + foldername + "/" + filename) - with open("websites/" + foldername + "/" + "archived.txt","w") as f: - f.write("True") - f.close() - - - -def generate_website_archive_html(): - html1 = ''' - - - - webpage-archive - - - - -

webpage-archive

-

This is a list of saved websites that AI generated.

- - - -''' - with FTP(ftp_server) as ftp: - ftp.login(ftp_username, ftp_password) - ftp.cwd(ftp_ai_webpage_archive) - file_list = ftp.nlst() - dir_list = "" - for folder in file_list: - if folder != "index.html": - dir_list += "
  • " + folder + "
    • \n" - html = html1 + dir_list + html2 - with open("index.html","w") as f: - f.write(html) - f.close() - ftp.storbinary("STOR " + "index.html", open("index.html", "rb")) - - -def combine_dicts(dict1, dict2): #prioritizes dict2 args - combined_dict = {} - for key in dict1: - combined_dict[key] = dict1[key] - for key in dict2: - combined_dict[key] = dict2[key] - return combined_dict - -def extract_image_tags(code): - count = code.count("") + index1 + 1 - img_tag = code[index1:index2] - tags.append(img_tag) - code = code[index2:] - return tags - -def extract_image_alt_text(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 - - -def generate_images(image_list): - file_list = [] - for image in image_list: - filename = image.replace(" ", "").lower() + ".png" - url = "http://127.0.0.1:7861" - payload = {"prompt": image,"steps": 25} - response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload) - r = 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 = requests.post(url=f'{url}/sdapi/v1/png-info', json=png_payload) - pnginfo = PngImagePlugin.PngInfo() - pnginfo.add_text("parameters", response2.json().get("info")) - image.save("webpage/" + filename, pnginfo=pnginfo) - file_list.append(filename) - return file_list - -def add_image_filenames(code, file_list): - for filename in file_list: - code = code.replace("src=\"\"", "src=\""+ filename + "\"", 1) - return code - -def delete_local_pngs(): - for filename in os.listdir("webpage/"): - if ".png" in filename: - os.remove("webpage/" + filename) - -def delete_ftp_pngs(): - with FTP(ftp_server) as ftp: - ftp.login(ftp_username, ftp_password) - ftp.cwd(ftp_ai_webpage) - file_list = ftp.nlst() - for filename in file_list: - if ".png" in filename: - print("Deleting", filename) - ftp.delete(filename) - -delete_local_pngs() -delete_ftp_pngs() - - -with open("webpage/index.html", 'r') as f: - code = f.read() - f.close() - -tags = extract_image_tags(code) -alt_texts = extract_image_alt_text(tags) -file_list = generate_images(alt_texts) -code = add_image_filenames(code, file_list) -with open("webpage/index.html", 'w') as f: - f.write(code) - f.close() -upload_html_and_imgs() -upload_websites() -generate_website_archive_html() -print("Success") -