AIrjen
d5c1679be8
which you will almost never see, but they are awesome if they work. And some minor bug fixes, and improved a bit of testing.
104 lines
4.0 KiB
Python
104 lines
4.0 KiB
Python
import sys, os
|
|
import random
|
|
import uuid
|
|
import re
|
|
from datetime import datetime
|
|
sys.path.append(os.path.abspath(".."))
|
|
|
|
|
|
from build_dynamic_prompt import *
|
|
|
|
|
|
|
|
def generateprompts(amount = 1,insanitylevel="5",subject="all", artist="all", imagetype="all",onlyartists=False, workprompt="", antistring="",prefixprompt="", suffixprompt="", negativeprompt="",promptcompounderlevel = "1", seperator="comma",givensubject="",smartsubject=True,giventypeofimage="",imagemodechance=20, gender = "all", subtypeobject = "all", subtypehumanoid = "all", subtypeconcept = "all", advancedprompting = True, hardturnoffemojis=False, seed=0, overrideoutfit=""):
|
|
loops = int(amount) # amount of images to generate
|
|
steps = 0
|
|
|
|
insanitylevel = int(insanitylevel)
|
|
while steps < loops:
|
|
# build prompt
|
|
|
|
result = build_dynamic_prompt(insanitylevel,subject,artist,imagetype, onlyartists,antistring,prefixprompt,suffixprompt,promptcompounderlevel, seperator,givensubject,smartsubject,giventypeofimage,imagemodechance, gender, subtypeobject, subtypehumanoid, subtypeconcept, advancedprompting, hardturnoffemojis, seed, overrideoutfit)
|
|
|
|
print("")
|
|
print("loop " + str(steps))
|
|
print("")
|
|
if(onlyartists == True):
|
|
print(result)
|
|
print("")
|
|
|
|
if(result.count("-")>1 and imagetype == "only templates"):
|
|
print("Is there a mistake in wildcards?")
|
|
print("")
|
|
print(result)
|
|
break
|
|
|
|
if(givensubject != "" and givensubject not in result and imagetype == "only templates"):
|
|
print("No givensubject, there must be an issue:")
|
|
print("")
|
|
print(result)
|
|
break
|
|
|
|
if(" OR " in result or ";" in result):
|
|
print("There is a mistake in a OR statement")
|
|
print("")
|
|
print(result)
|
|
break
|
|
|
|
# Use regex to find words enclosed by hyphens, the wildcards1
|
|
# make some minor exceptions
|
|
resultnew = result
|
|
resultnew = resultnew.replace("-eye-", " eye ")
|
|
resultnew = resultnew.replace("-of-", " of ")
|
|
resultnew = resultnew.replace("-the-", " the ")
|
|
resultnew = resultnew.replace("-up-", " up ")
|
|
resultnew = resultnew.replace("-in-", " in ")
|
|
resultnew = resultnew.replace("-au-", " au ")
|
|
resultnew = resultnew.replace("-da-", " da ")
|
|
resultnew = resultnew.replace("-doo-", " doo ")
|
|
resultnew = resultnew.replace("-and-", " and ")
|
|
matches = re.findall(r'-\w+-', resultnew)
|
|
|
|
# Filter out matches with commas and spaces
|
|
wildcards = [match for match in matches if ',' not in match and ' ' not in match]
|
|
|
|
|
|
if(wildcards):
|
|
print("There is a wildcard still in the prompt")
|
|
print("")
|
|
print(result)
|
|
break
|
|
|
|
#if("game" in result or "movie" in result):
|
|
# print("TEST THIS")
|
|
# print("")
|
|
# print(result)
|
|
# break
|
|
|
|
|
|
steps += 1
|
|
|
|
|
|
print("")
|
|
print("All done!")
|
|
|
|
generateprompts(20000,5
|
|
,"all" # subject
|
|
,"all" # artists
|
|
,"all" # image type "only other types", "only templates mode", "art blaster mode", "quality vomit mode", "color cannon mode", "unique art mode", "massive madness mode", "photo fantasy mode", "subject only mode", "fixed styles mode"
|
|
, False # only artists
|
|
,"","","PREFIXPROMPT"
|
|
,"SUFFIXPROMPT"
|
|
,"",1,""
|
|
,"" # subject override
|
|
,True,
|
|
"",20
|
|
, "all" # gender
|
|
, "all" # object types
|
|
, "all" # humanoid types -- all,generic humans,generic human relations, multiple humans, celebrities e.a.,fictional,humanoids, based on job or title,based on first name
|
|
, "all" # concept types
|
|
, False # prompt switching
|
|
, True # Turn off emojis
|
|
, 0 # seed
|
|
, "" #outfit override
|
|
) |