Sygil-Dev/sygil-webui
1
1
Fork 0
mirror of https://github.com/Sygil-Dev/sygil-webui.git synced 2024-12-15 22:42:14 +03:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1247 from ZeroCool940711/dev

Browse Source 
Added condition to check if the concepts library folder exist and if it doesn't then tell the user where to download the concepts from.
This commit is contained in:
ZeroCool 2022-09-20 22:16:51 -07:00 committed by GitHub
commit dfa02ac139
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
scripts/sd_concept_library.py
View File

@ -43,14 +43,17 @@ def getConceptsFromPath(page, conceptPerPage, searchText= ""):
path = os.path.join(os.getcwd(), st.session_state['defaults'].general.sd_concepts_library_folder) path = os.path.join(os.getcwd(), st.session_state['defaults'].general.sd_concepts_library_folder)
acceptedExtensions = ('jpeg', 'jpg', "png") acceptedExtensions = ('jpeg', 'jpg', "png")
concepts = [] concepts = []
# List all folders (concepts) in the path if os.path.exists(path):
folders = [f for f in os.listdir(path) if os.path.isdir(os.path.join(path, f))] # List all folders (concepts) in the path
filteredFolders = folders folders = [f for f in os.listdir(path) if os.path.isdir(os.path.join(path, f))]
filteredFolders = folders
# Filter the folders by the search text
if searchText != "": # Filter the folders by the search text
filteredFolders = [f for f in folders if searchText.lower() in f.lower()] if searchText != "":
filteredFolders = [f for f in folders if searchText.lower() in f.lower()]
else:
filteredFolders = []
conceptIndex = 1 conceptIndex = 1
for folder in filteredFolders: for folder in filteredFolders:
@ -118,14 +121,17 @@ def getTotalNumberOfConcepts(searchText= ""):
# get the path where the concepts are stored # get the path where the concepts are stored
path = os.path.join(os.getcwd(), st.session_state['defaults'].general.sd_concepts_library_folder) path = os.path.join(os.getcwd(), st.session_state['defaults'].general.sd_concepts_library_folder)
concepts = [] concepts = []
# List all folders (concepts) in the path if os.path.exists(path):
folders = [f for f in os.listdir(path) if os.path.isdir(os.path.join(path, f))] # List all folders (concepts) in the path
filteredFolders = folders folders = [f for f in os.listdir(path) if os.path.isdir(os.path.join(path, f))]
filteredFolders = folders
# Filter the folders by the search text
if searchText != "": # Filter the folders by the search text
filteredFolders = [f for f in folders if searchText.lower() in f.lower()] if searchText != "":
filteredFolders = [f for f in folders if searchText.lower() in f.lower()]
else:
filteredFolders = []
return len(filteredFolders) return len(filteredFolders)
def layout(): def layout():
@ -159,7 +165,7 @@ def layout():
st.write("You don't have any concepts in your library ") st.write("You don't have any concepts in your library ")
# Propose the user to go to "https://github.com/sd-webui/sd-concepts-library" # Propose the user to go to "https://github.com/sd-webui/sd-concepts-library"
st.markdown("To add concepts to your library, download some from the [sd-concepts-library](https://github.com/sd-webui/sd-concepts-library) \ st.markdown("To add concepts to your library, download some from the [sd-concepts-library](https://github.com/sd-webui/sd-concepts-library) \
repository and save the content of `sd-concepts-library` into ```./models/custom/sd-concepts-library``` or just create your own concept.", unsafe_allow_html=False) repository and save the content of `sd-concepts-library` into ```./models/custom/sd-concepts-library``` or just create your own concepts :wink:.", unsafe_allow_html=False)
else: else:
st.write("No concepts found in the library matching your search: " + searchInput) st.write("No concepts found in the library matching your search: " + searchInput)