nodes/AIrjen-OneButtonPrompt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added logic for running on firstnames

Browse Source 
added giant first name list
This commit is contained in:
AIrjen
2023-07-08 15:52:55 +02:00
parent b2f29a32ed
commit e5083da95a
8 changed files with 1724 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
build_dynamic_prompt.py
+40 -14
View File
@@ -88,6 +88,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
manwomanrelationlist = csv_to_list(csvfilename="manwomanrelations",antilist=antilist,skipheader=True,gender=gender)
waterlocationlist = csv_to_list("waterlocations", antilist)
containerlist = csv_to_list("containers", antilist)
firstnamelist = csv_to_list(csvfilename="firstnames",antilist=antilist,skipheader=True,gender=gender)
humanlist = fictionallist + nonfictionallist + humanoidlist
objecttotallist = objectlist + buildinglist + vehiclelist + foodlist + spacelist
@@ -163,6 +164,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
generatenonfictionalcharacter = True
generatehumanoids = True
generatejob = True
generatefirstnames = True
generatelandscape = True
generateevent = True
generateconcepts = True
@@ -270,6 +272,8 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
generatehumanoids = False
if item[0] == 'subject_job' and item[1] != 'on':
generatejob = False
if item[0] == 'subject_firstnames' and item[1] != 'on':
generatefirstnames = False
# landscape
if item[0] == 'subject_landscape' and item[1] != 'on':
generatelandscape = False
@@ -459,7 +463,8 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
generatemanwoman = bool(manwomanlist) and generatemanwoman
generatemanwomanrelation = bool(manwomanrelationlist) and generatemanwomanrelation
generatejob = bool(joblist) and generatejob
generatehumanoid = generatefictionalcharacter or generatenonfictionalcharacter or generatehumanoids or generatemanwoman or generatejob or generatemanwomanrelation
generatefirstnames = bool(firstnamelist) and generatefirstnames
generatehumanoid = generatefictionalcharacter or generatenonfictionalcharacter or generatehumanoids or generatemanwoman or generatejob or generatemanwomanrelation or generatefirstnames
if(generatefictionalcharacter):
@@ -488,6 +493,9 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
if(generatehumanoid):
mainchooserlist.append("humanoid")
if(generatefirstnames):
humanoidsubjectchooserlist.append("firstname")
generateanimal = bool(animallist) and generateanimal
@@ -870,7 +878,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
else:
subjectchooser = "animal as human"
if(mainchooser == "humanoid"):
#humanoidsubjectchooserlist = ["human", "job", "fictional", "non fictional", "humanoid", "manwomanrelation"]
#humanoidsubjectchooserlist = ["human", "job", "fictional", "non fictional", "humanoid", "manwomanrelation", "firstname"]
subjectchooser = random.choice(humanoidsubjectchooserlist)
if(subtypehumanoid != "all"):
if(subtypehumanoid == "generic humans"):
@@ -885,6 +893,8 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
subjectchooser = "humanoid"
if(subtypehumanoid == "based on job or title"):
subjectchooser = "job"
if(subtypehumanoid == "based on first name"):
subjectchooser = "firstname"
if(mainchooser == "landscape"):
subjectchooser = "landscape"
if(mainchooser == "concept"):
@@ -1204,10 +1214,10 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
if(chance_roll(insanitylevel, subjectdescriptor2chance) and generatedescriptors == True):
completeprompt += "-descriptor- "
if(subjectchooser in ["animal as human,","human", "job", "fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, subjectbodytypechance) and generatebodytype == True):
if(subjectchooser in ["animal as human,","human", "job", "fictional", "non fictional", "humanoid", "manwomanrelation","firstname"] and chance_roll(insanitylevel, subjectbodytypechance) and generatebodytype == True):
completeprompt += "-bodytype- "
if(subjectchooser in ["object","animal as human,","human", "job", "fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, subjectculturechance) and generatedescriptors == True):
if(subjectchooser in ["object","animal as human,","human", "job", "fictional", "non fictional", "humanoid", "manwomanrelation","firstname"] and chance_roll(insanitylevel, subjectculturechance) and generatedescriptors == True):
completeprompt += "-culture- "
if(mainchooser == "object"):
@@ -1344,6 +1354,21 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
completeprompt += ":" + random.choice(hybridhumanlist) + ":" + str(random.randint(1,5)) + "] "
hybridorswap = ""
if(subjectchooser == "firstname"):
if rare_dist(insanitylevel):
hybridorswaplist = ["hybrid", "swap"]
hybridorswap = random.choice(hybridorswaplist)
completeprompt += "["
completeprompt += "-firstname- "
if(hybridorswap == "hybrid"):
completeprompt += "|" + "-firstname-" + "] "
if(hybridorswap == "swap"):
completeprompt += ":" + "-firstname-" + ":" + str(random.randint(1,5)) + "] "
hybridorswap = ""
else:
completeprompt += " " + givensubject + " "
@@ -1412,7 +1437,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
# riding an animal, holding an object or driving a vehicle, rare
if(subjectchooser in ["animal as human,","human","fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, humanadditionchance) and generatehumanaddition == True):
if(subjectchooser in ["animal as human,","human","fictional", "non fictional", "humanoid", "manwomanrelation","firstname"] and chance_roll(insanitylevel, humanadditionchance) and generatehumanaddition == True):
humanspecial = 1
completeprompt += "-humanaddition- "
@@ -1428,7 +1453,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
# SD understands emoji's. Can be used to manipulate facial expressions.
# emoji, legendary
if(subjectchooser in ["animal as human,","human","fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, emojichance) and generateemoji== True):
if(subjectchooser in ["animal as human,","human","fictional", "non fictional", "humanoid", "manwomanrelation","firstname"] and chance_roll(insanitylevel, emojichance) and generateemoji== True):
completeprompt += "-emoji-, "
@@ -1439,7 +1464,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
# Job
# either go job or activity, not both
if(subjectchooser in ["animal as human","human","fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, joboractivitychance) and humanspecial != 1 and generatesubject == True):
if(subjectchooser in ["animal as human","human","fictional", "non fictional", "humanoid", "manwomanrelation","firstname"] and chance_roll(insanitylevel, joboractivitychance) and humanspecial != 1 and generatesubject == True):
joboractivitylist = [joblist,humanactivitylist]
completeprompt += random.choice(random.choice(joboractivitylist)) + ", "
@@ -1458,16 +1483,16 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
completeprompt += " -minivomit-, "
# outfit builder
if(subjectchooser in ["animal as human","human","fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, outfitchance) and generateoutfit == True and humanspecial != 1):
if(subjectchooser in ["animal as human","human","fictional", "non fictional", "humanoid", "manwomanrelation", "firstname"] and chance_roll(insanitylevel, outfitchance) and generateoutfit == True and humanspecial != 1):
completeprompt += ", wearing " + random.choice(buildoutfitlist) + ", "
if(subjectchooser in ["animal as human","human","fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, posechance) and humanspecial != 1 and generatepose == True):
if(subjectchooser in ["animal as human","human","fictional", "non fictional", "humanoid", "manwomanrelation", "firstname"] and chance_roll(insanitylevel, posechance) and humanspecial != 1 and generatepose == True):
completeprompt += random.choice(poselist) + ", "
if(subjectchooser in ["human","job","fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, hairchance) and generatehairstyle == True):
if(subjectchooser in ["human","job","fictional", "non fictional", "humanoid", "manwomanrelation", "firstname"] and chance_roll(insanitylevel, hairchance) and generatehairstyle == True):
completeprompt += random.choice(buildhairlist) + ", "
if(subjectchooser in ["animal as human,","human","fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, accessorychance) and generateaccessorie == True and generateaccessories == True):
if(subjectchooser in ["animal as human,","human","fictional", "non fictional", "humanoid", "manwomanrelation", "firstname"] and chance_roll(insanitylevel, accessorychance) and generateaccessorie == True and generateaccessories == True):
completeprompt += random.choice(buildaccessorielist) + ", "
if(chance_roll(insanitylevel, humanoidinsideshotchance) and subjectchooser not in ["landscape", "concept"] and generateinsideshot == True):
@@ -1871,9 +1896,10 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
"-styletilora-" in completeprompt or
"-manwomanrelation-" in completeprompt or
"-waterlocation-" in completeprompt or
"-container-" in completeprompt):
allwildcardslistnohybrid = [ "-color-","-object-", "-animal-", "-fictional-","-nonfictional-","-building-","-vehicle-","-location-","-conceptprefix-","-food-","-haircolor-","-hairstyle-","-job-", "-accessory-", "-humanoid-", "-manwoman-", "-human-", "-colorscheme-", "-mood-", "-genderdescription-", "-artmovement-", "-malefemale-", "-bodytype-", "-minilocation-", "-minilocationaddition-", "-pose-", "-season-", "-minioutfit-", "-elaborateoutfit-", "-minivomit-", "-vomit-", "-rpgclass-", "-subjectfromfile-", "-brand-", "-space-", "-artist-", "-imagetype-", "-othertype-", "-quality-", "-lighting-", "-camera-", "-lens-","-imagetypequality-", "-poemline-", "-songline-", "-greatwork-", "-artistfantasy-", "-artistpopular-", "-artistromanticism-", "-artistphotography-", "-emoji-", "-timeperiod-", "-shotsize-", "-musicgenre-", "-animaladdition-", "-addontolocationinside-", "-addontolocation-", "-objectaddition-", "-humanaddition-", "-overalladdition-", "-focus-", "-direction-", "-styletilora-", "-manwomanrelation-", "-waterlocation-", "-container-"]
allwildcardslistnohybridlists = [colorlist, objectlist, animallist, fictionallist, nonfictionallist, buildinglist, vehiclelist, locationlist,conceptprefixlist,foodlist,haircolorlist, hairstylelist,joblist, accessorielist, humanoidlist, manwomanlist, humanlist, colorschemelist, moodlist, genderdescriptionlist, artmovementlist, malefemalelist, bodytypelist, minilocationlist, minilocationadditionslist, poselist, seasonlist, minioutfitlist, elaborateoutfitlist, minivomitlist, vomitlist, rpgclasslist, customsubjectslist, brandlist, spacelist, artistlist, imagetypelist, othertypelist, qualitylist, lightinglist, cameralist, lenslist, imagetypequalitylist, poemlinelist, songlinelist, greatworklist, fantasyartistlist, popularartistlist, romanticismartistlist, photographyartistlist, emojilist, timeperiodlist, shotsizelist, musicgenrelist, animaladditionlist, addontolocationinsidelist, addontolocationlist, objectadditionslist, humanadditionlist, overalladditionlist, focuslist, directionlist, stylestiloralist, manwomanrelationlist, waterlocationlist, containerlist]
"-container-" in completeprompt or
"-firstname-" in completeprompt):
allwildcardslistnohybrid = [ "-color-","-object-", "-animal-", "-fictional-","-nonfictional-","-building-","-vehicle-","-location-","-conceptprefix-","-food-","-haircolor-","-hairstyle-","-job-", "-accessory-", "-humanoid-", "-manwoman-", "-human-", "-colorscheme-", "-mood-", "-genderdescription-", "-artmovement-", "-malefemale-", "-bodytype-", "-minilocation-", "-minilocationaddition-", "-pose-", "-season-", "-minioutfit-", "-elaborateoutfit-", "-minivomit-", "-vomit-", "-rpgclass-", "-subjectfromfile-", "-brand-", "-space-", "-artist-", "-imagetype-", "-othertype-", "-quality-", "-lighting-", "-camera-", "-lens-","-imagetypequality-", "-poemline-", "-songline-", "-greatwork-", "-artistfantasy-", "-artistpopular-", "-artistromanticism-", "-artistphotography-", "-emoji-", "-timeperiod-", "-shotsize-", "-musicgenre-", "-animaladdition-", "-addontolocationinside-", "-addontolocation-", "-objectaddition-", "-humanaddition-", "-overalladdition-", "-focus-", "-direction-", "-styletilora-", "-manwomanrelation-", "-waterlocation-", "-container-", "-firstname-"]
allwildcardslistnohybridlists = [colorlist, objectlist, animallist, fictionallist, nonfictionallist, buildinglist, vehiclelist, locationlist,conceptprefixlist,foodlist,haircolorlist, hairstylelist,joblist, accessorielist, humanoidlist, manwomanlist, humanlist, colorschemelist, moodlist, genderdescriptionlist, artmovementlist, malefemalelist, bodytypelist, minilocationlist, minilocationadditionslist, poselist, seasonlist, minioutfitlist, elaborateoutfitlist, minivomitlist, vomitlist, rpgclasslist, customsubjectslist, brandlist, spacelist, artistlist, imagetypelist, othertypelist, qualitylist, lightinglist, cameralist, lenslist, imagetypequalitylist, poemlinelist, songlinelist, greatworklist, fantasyartistlist, popularartistlist, romanticismartistlist, photographyartistlist, emojilist, timeperiodlist, shotsizelist, musicgenrelist, animaladditionlist, addontolocationinsidelist, addontolocationlist, objectadditionslist, humanadditionlist, overalladditionlist, focuslist, directionlist, stylestiloralist, manwomanrelationlist, waterlocationlist, containerlist, firstnamelist]
allwildcardslistwithhybrid = ["-material-", "-descriptor-", "-outfit-", "-conceptsuffix-","-culture-", "-objecttotal-"]
allwildcardslistwithhybridlists = [materiallist, descriptorlist,outfitlist,conceptsuffixlist,culturelist, objecttotallist]
csvfiles/config/default_config.csv
+1
View File
@@ -19,6 +19,7 @@ subject_fictional;on
subject_nonfictional;on
subject_humanoid;on
subject_job;on
subject_firstnames;on
# landscape;
subject_landscape;on
# concept;
1 #this file controls the generation process
19 subject_nonfictional on
20 subject_humanoid on
21 subject_job on
22 subject_firstnames on
23 # landscape
24 subject_landscape on
25 # concept
csvfiles/cultures.csv
-1
View File
@@ -199,7 +199,6 @@ Papuan
Mennonite
Basque
Hmong
Basque
Catalan
Bavarian
Swiss
1 Japanese
199 Mennonite
200 Basque
201 Hmong
Basque
202 Catalan
203 Bavarian
204 Swiss
csvfiles/firstnames.csv
+1596
View File
File diff suppressed because it is too large Load Diff
csvfiles/nature.csv
+81
View File
@@ -0,0 +1,81 @@
Tree
Bush
Flower
Oak tree
Maple tree
Pine tree
Apple tree
Cherry blossom tree
Redwood tree
Birch tree
Willow tree
Palm tree
Magnolia tree
Roses
Sunflowers
Daisies
Tulips
Orchids
Lilies
Poppies
Hydrangeas
Irises
Marigolds
Leaf
Twig
Pinecone
Acorn
Moss
Seedling
Petal
Fern
Mushroom
Mossy rock
Beehive
Vine
Willow basket
Coral reef
Honeycomb
Wildflower bouquet
Cactus spines
Pebble
Bamboo shoot
Sunflower field
Pine needle carpet
River stone
Seashell
Autumn leaf pile
Seaweed
Pine bark
Butterfly cocoon
Dandelion fluff
Bird nest
Water lily pad
Herb garden
Wild berry bush
Tree stump
Coral reef
Wild mushroom patch
Tulip bulb
Oak tree branch
Hummingbird feeder
Pond water lilies
Straw bale
Bouquet
Forest
Meadow
Garden
-waterlocation-
Rainforest
Waterfall
Rock formation
Foliage
Wilderness
Biodiversity
Ecosystem
Flora
Fauna
Woodland
Arboretum
Orchard
Herb garden
1 Tree
2 Bush
3 Flower
4 Oak tree
5 Maple tree
6 Pine tree
7 Apple tree
8 Cherry blossom tree
9 Redwood tree
10 Birch tree
11 Willow tree
12 Palm tree
13 Magnolia tree
14 Roses
15 Sunflowers
16 Daisies
17 Tulips
18 Orchids
19 Lilies
20 Poppies
21 Hydrangeas
22 Irises
23 Marigolds
24 Leaf
25 Twig
26 Pinecone
27 Acorn
28 Moss
29 Seedling
30 Petal
31 Fern
32 Mushroom
33 Mossy rock
34 Beehive
35 Vine
36 Willow basket
37 Coral reef
38 Honeycomb
39 Wildflower bouquet
40 Cactus spines
41 Pebble
42 Bamboo shoot
43 Sunflower field
44 Pine needle carpet
45 River stone
46 Seashell
47 Autumn leaf pile
48 Seaweed
49 Pine bark
50 Butterfly cocoon
51 Dandelion fluff
52 Bird nest
53 Water lily pad
54 Herb garden
55 Wild berry bush
56 Tree stump
57 Coral reef
58 Wild mushroom patch
59 Tulip bulb
60 Oak tree branch
61 Hummingbird feeder
62 Pond water lilies
63 Straw bale
64 Bouquet
65 Forest
66 Meadow
67 Garden
68 -waterlocation-
69 Rainforest
70 Waterfall
71 Rock formation
72 Foliage
73 Wilderness
74 Biodiversity
75 Ecosystem
76 Flora
77 Fauna
78 Woodland
79 Arboretum
80 Orchard
81 Herb garden
csvfiles/objects.csv
-20
View File
@@ -281,26 +281,6 @@ Fractal
Pentagram
-conceptsuffix-potion
Roast beef
Oak tree
Maple tree
Pine tree
Apple tree
Cherry blossom tree
Redwood tree
Birch tree
Willow tree
Palm tree
Magnolia tree
Roses
Sunflowers
Daisies
Tulips
Orchids
Lilies
Poppies
Hydrangeas
Irises
Marigolds
-object- in a bottle
Faberge egg
Quilt
1 Pen
281 Pentagram
282 -conceptsuffix-potion
283 Roast beef
Oak tree
Maple tree
Pine tree
Apple tree
Cherry blossom tree
Redwood tree
Birch tree
Willow tree
Palm tree
Magnolia tree
Roses
Sunflowers
Daisies
Tulips
Orchids
Lilies
Poppies
Hydrangeas
Irises
Marigolds
284 -object- in a bottle
285 Faberge egg
286 Quilt
prompttester.py
+1 -1
View File
@@ -46,4 +46,4 @@ def generateprompts(amount = 1,insanitylevel="5",subject="all", artist="all", im
print("")
print("All done!")
generateprompts(10,5,"humanoid","all","all",False,"","","PREFIXPROMPT" ,"SUFFIXPROMPT","",1,"","",True,"",100, "female", "all", "fictional characters", "all")
generateprompts(10,5,"humanoid","all","all",False,"","","PREFIXPROMPT" ,"SUFFIXPROMPT","",1,"","",True,"",100, "female", "all", "all", "all")
scripts/onebuttonprompt.py
+5
View File
@@ -87,6 +87,7 @@ generatefictionalcharacter = True
generatenonfictionalcharacter = True
generatehumanoids = True
generatejob = True
generatefirstnames = True
generatelandscape = True
generateevent = True
generateconcepts = True
@@ -122,6 +123,8 @@ for item in config:
generatehumanoids = False
if item[0] == 'subject_job' and item[1] != 'on':
generatejob = False
if item[0] == 'subject_firstnames' and item[1] != 'on':
generatefirstnames = False
# landscape
if item[0] == 'subject_landscape' and item[1] != 'on':
generatelandscape = False
@@ -178,6 +181,8 @@ if(generatehumanoids):
subjectsubtypeshumanoid.append("humanoids")
if(generatejob):
subjectsubtypeshumanoid.append("based on job or title")
if(generatefirstnames):
subjectsubtypeshumanoid.append("based on first name")
# concepts
if(generateevent):