# This file is part of sygil-webui (https://github.com/Sygil-Dev/sygil-webui/). # Copyright 2022 Sygil-Dev team. # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . # base webui import and utils. from sd_utils import * # home plugin import os from PIL import Image # from bs4 import BeautifulSoup from streamlit.runtime.in_memory_file_manager import in_memory_file_manager from streamlit.elements import image as STImage # Temp imports # end of imports # --------------------------------------------------------------------------------------------------------------- try: # this silences the annoying "Some weights of the model checkpoint were not used when initializing..." message at start. from transformers import logging logging.set_verbosity_error() except: pass class plugin_info: plugname = "imglab" description = "Image Lab" isTab = True displayPriority = 3 def getLatestGeneratedImagesFromPath(): # get the latest images from the generated images folder # get the path to the generated images folder generatedImagesPath = os.path.join(os.getcwd(), "outputs") # get all the files from the folders and subfolders files = [] # get the laest 10 images from the output folder without walking the subfolders for r, d, f in os.walk(generatedImagesPath): for file in f: if ".png" in file: files.append(os.path.join(r, file)) # sort the files by date files.sort(key=os.path.getmtime) # reverse the list so the latest images are first for f in files: img = Image.open(f) files[files.index(f)] = img # get the latest 10 files # get all the files with the .png or .jpg extension # sort files by date # get the latest 10 files latestFiles = files[-10:] # reverse the list latestFiles.reverse() return latestFiles def getImagesFromLexica(): # scrape images from lexica.art # get the html from the page # get the html with cookies and javascript apiEndpoint = r"https://lexica.art/api/trpc/prompts.infinitePrompts?batch=1&input=%7B%220%22%3A%7B%22json%22%3A%7B%22limit%22%3A10%2C%22text%22%3A%22%22%2C%22cursor%22%3A10%7D%7D%7D" # REST API call # from requests_html import HTMLSession session = HTMLSession() response = session.get(apiEndpoint) # req = requests.Session() # req.headers['user-agent'] = 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.45 Safari/537.36' # response = req.get(apiEndpoint) print(response.status_code) print(response.text) # get the json from the response # json = response.json() # get the prompts from the json print(response) # session = requests.Session() # parseEndpointJson = session.get(apiEndpoint,headers=headers,verify=False) # print(parseEndpointJson) # print('test2') # page = requests.get("https://lexica.art/", headers={'User-Agent': 'Mozilla/5.0'}) # parse the html # soup = BeautifulSoup(page.content, 'html.parser') # find all the images # print(soup) # images = soup.find_all('alt-image') # create a list to store the image urls image_urls = [] # loop through the images for image in images: # get the url image_url = image["src"] # add it to the list image_urls.append("http://www.lexica.art/" + image_url) # return the list print(image_urls) return image_urls def changeImage(): # change the image in the image holder # check if the file is not empty if len(st.session_state["uploaded_file"]) > 0: # read the file print("test2") uploaded = st.session_state["uploaded_file"][0].read() # show the image in the image holder st.session_state["previewImg"].empty() st.session_state["previewImg"].image(uploaded, use_column_width=True) def createHTMLGallery(images): html3 = """
""" mkdwn_array = [] for i in images: bImg = i.read() i = Image.save(bImg, "PNG") width, height = i.size # get random number for the id image_id = "%s" % (str(images.index(i))) (data, mimetype) = STImage._normalize_to_bytes(bImg.getvalue(), width, "auto") this_file = in_memory_file_manager.add(data, mimetype, image_id) img_str = this_file.url # img_str = 'data:image/png;base64,' + b64encode(image_io.getvalue()).decode('ascii') # get image size # make sure the image is not bigger then 150px but keep the aspect ratio if width > 150: height = int(height * (150 / width)) width = 150 if height > 150: width = int(width * (150 / height)) height = 150 # mkdwn = f"""Image""" mkdwn = f"""
Image
""" mkdwn_array.append(mkdwn) html3 += "".join(mkdwn_array) html3 += "
" return html3 def layout(): col1, col2 = st.columns(2) with col1: st.session_state["uploaded_file"] = st.file_uploader( "Choose an image or images", type=["png", "jpg", "jpeg", "webp"], accept_multiple_files=True, on_change=changeImage, ) if "previewImg" not in st.session_state: st.session_state["previewImg"] = st.empty() else: if len(st.session_state["uploaded_file"]) > 0: st.session_state["previewImg"].empty() st.session_state["previewImg"].image( st.session_state["uploaded_file"][0], use_column_width=True ) else: st.session_state["previewImg"] = st.empty()