1580158
/
OneButtonPrompt
mirror of https://github.com/AIrjen/OneButtonPrompt
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

more config stuff, almost done

pull/83/head
AIrjen 2023-07-07 10:11:39 +02:00
parent 9119914537
commit 37c4e3e269
3 changed files with 159 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

150
build_dynamic_prompt.py
View File

@ -199,6 +199,37 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
humanadditionchance = 'rare'
overalladditionchance = 'extraordinary'
custominputmidrepeats = 2
custominputmidchance = 'uncommon'
minivomitmidchance = 'unique'
outfitchance = 'normal'
posechance = 'uncommon'
hairchance = 'normal'
accessorychance = 'normal'
humanoidinsideshotchance = 'legendary'
humanoidbackgroundchance = 'uncommon'
landscapeminilocationchance = 'uncommon'
generalminilocationchance = 'rare'
timperiodchance = 'normal'
focuschance = 'normal'
directionchance = 'normal'
moodchance = 'normal'
minivomitsuffixchance = 'unique'
artmovementchance = 'normal'
lightingchance = 'normal'
photoadditionchance = 'common'
lenschance = 'normal'
colorschemechance = 'normal'
vomit1chance = 'uncommon'
vomit2chance= 'uncommon'
greatworkchance = 'novel'
poemlinechance = 'novel'
songlinechance = 'novel'
quality1chance = 'uncommon'
quality2chance = 'uncommon'
for item in config:
@ -298,6 +329,66 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
if item[0] == 'overalladditionchance':
overalladditionchance = item[1]
if item[0] == 'custominputmidrepeats':
custominputmidrepeats = int(item[1])
if item[0] == 'custominputmidchance':
custominputmidchance = item[1]
if item[0] == 'minivomitmidchance':
minivomitmidchance = item[1]
if item[0] == 'outfitchance':
outfitchance = item[1]
if item[0] == 'posechance':
posechance = item[1]
if item[0] == 'hairchance':
hairchance = item[1]
if item[0] == 'accessorychance':
accessorychance = item[1]
if item[0] == 'humanoidinsideshotchance':
humanoidinsideshotchance = item[1]
if item[0] == 'humanoidbackgroundchance':
humanoidbackgroundchance = item[1]
if item[0] == 'landscapeminilocationchance':
landscapeminilocationchance = item[1]
if item[0] == 'generalminilocationchance':
generalminilocationchance = item[1]
if item[0] == 'timperiodchance':
timperiodchance = item[1]
if item[0] == 'focuschance':
focuschance = item[1]
if item[0] == 'directionchance':
directionchance = item[1]
if item[0] == 'moodchance':
moodchance = item[1]
if item[0] == 'minivomitsuffixchance':
minivomitsuffixchance = item[1]
if item[0] == 'artmovementchance':
artmovementchance = item[1]
if item[0] == 'lightingchance':
lightingchance = item[1]
if item[0] == 'photoadditionchance':
photoadditionchance = item[1]
if item[0] == 'lenschance':
lenschance = item[1]
if item[0] == 'colorschemechance':
colorschemechance = item[1]
if item[0] == 'vomit1chance':
vomit1chance = item[1]
if item[0] == 'vomit2chance':
vomit2chance = item[1]
if item[0] == 'greatworkchance':
greatworkchance = item[1]
if item[0] == 'poemlinechance':
poemlinechance = item[1]
if item[0] == 'songlinechance':
songlinechance = item[1]
if item[0] == 'quality1chance':
quality1chance = item[1]
if item[0] == 'quality2chance':
quality2chance = item[1]
generatevehicle = bool(vehiclelist) and generatevehicle
generateobject = bool(objectlist) and generateobject
@ -1295,110 +1386,109 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
# completeprompt += ", with " + random.choice(skintypelist) + " skin, "
# custom mid list
if(uncommon_dist(insanitylevel) and generatecustominputmid == True):
completeprompt += random.choice(custominputmidlist) + ", "
if(uncommon_dist(insanitylevel)):
for i in range(custominputmidrepeats):
if(chance_roll(insanitylevel, custominputmidchance) and generatecustominputmid == True):
completeprompt += random.choice(custominputmidlist) + ", "
# add in some more mini vomits
if(unique_dist(insanitylevel) and generateminivomit == True):
if(chance_roll(insanitylevel, minivomitmidchance) and generateminivomit == True):
completeprompt += " -minivomit-, "
# outfit builder
if(subjectchooser in ["animal as human","human","fictional", "non fictional", "humanoid", "manwomanrelation"] and normal_dist(insanitylevel) and generateoutfit == True and humanspecial != 1):
if(subjectchooser in ["animal as human","human","fictional", "non fictional", "humanoid", "manwomanrelation"] 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 uncommon_dist(insanitylevel) and humanspecial != 1 and generatepose == True):
if(subjectchooser in ["animal as human","human","fictional", "non fictional", "humanoid", "manwomanrelation"] 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 normal_dist(insanitylevel) and generatehairstyle == True):
if(subjectchooser in ["human","job","fictional", "non fictional", "humanoid", "manwomanrelation"] 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 normal_dist(insanitylevel) and generateaccessorie == True and generateaccessories == True):
if(subjectchooser in ["animal as human,","human","fictional", "non fictional", "humanoid", "manwomanrelation"] and chance_roll(insanitylevel, accessorychance) and generateaccessorie == True and generateaccessories == True):
completeprompt += random.choice(buildaccessorielist) + ", "
if(legendary_dist(insanitylevel) and subjectchooser not in ["landscape", "concept"] and generateinsideshot == True):
if(chance_roll(insanitylevel, humanoidinsideshotchance) and subjectchooser not in ["landscape", "concept"] and generateinsideshot == True):
insideshot = 1
completeprompt += random.choice(insideshotlist) + ", "
if(subjectchooser not in ["landscape", "concept"] and humanspecial != 1 and insideshot == 0 and uncommon_dist(insanitylevel) and generatebackground == True):
if(subjectchooser not in ["landscape", "concept"] and humanspecial != 1 and insideshot == 0 and chance_roll(insanitylevel, humanoidbackgroundchance) and generatebackground == True):
completeprompt += random.choice(backgroundtypelist) + ", "
# minilocation bit
if(subjectchooser in ["landscape"] and uncommon_dist(insanitylevel) and generateminilocationaddition == True):
if(subjectchooser in ["landscape"] and chance_roll(insanitylevel, landscapeminilocationchance) and generateminilocationaddition == True):
completeprompt += " -minilocationaddition-, "
if(rare_dist(insanitylevel) and generateminilocationaddition == True):
if(chance_roll(insanitylevel, generalminilocationchance) and generateminilocationaddition == True):
completeprompt += " -minilocationaddition-, "
# landscapes it is nice to always have a time period
if(normal_dist(insanitylevel) or subjectchooser=="landscape"):
if(chance_roll(insanitylevel, timperiodchance) or subjectchooser=="landscape"):
if(generatetimeperiod == True):
completeprompt += "-timeperiod-, "
if(mainchooser not in ["landscape"] and rare_dist(insanitylevel) and generatefocus == True):
if(mainchooser not in ["landscape"] and chance_roll(insanitylevel, focuschance) and generatefocus == True):
completeprompt += "-focus-, "
# others
if(normal_dist(insanitylevel) and generatedirection == True):
if(chance_roll(insanitylevel, directionchance) and generatedirection == True):
completeprompt += "-direction-, "
if(normal_dist(insanitylevel) and generatemood == True):
if(chance_roll(insanitylevel, moodchance) and generatemood == True):
completeprompt += "-mood-, "
# add in some more mini vomits
if(unique_dist(insanitylevel) and generateminivomit == True):
if(chance_roll(insanitylevel, minivomitsuffixchance) and generateminivomit == True):
completeprompt += " -minivomit-, "
if(normal_dist(insanitylevel) and generateartmovement == True):
if(chance_roll(insanitylevel, artmovementchance) and generateartmovement == True):
completeprompt += "-artmovement-, "
if(normal_dist(insanitylevel) and generatelighting == True):
if(chance_roll(insanitylevel, lightingchance) and generatelighting == True):
completeprompt += "-lighting-, "
# determine wether we have a photo or not
if("photo" in completeprompt.lower()):
isphoto = 1
if(common_dist(insanitylevel) and isphoto == 1 and generatephotoaddition == True):
if(chance_roll(insanitylevel, photoadditionchance) and isphoto == 1 and generatephotoaddition == True):
completeprompt += random.choice(photoadditionlist) + ", "
if(isphoto == 1 and generatecamera == True):
completeprompt += "-camera-, "
if(normal_dist(insanitylevel) or isphoto == 1):
if(chance_roll(insanitylevel, lenschance) or isphoto == 1):
if(generatelens == True):
completeprompt += "-lens-, "
if(normal_dist(insanitylevel) and generatecolorscheme == True):
completeprompt += random.choice(colorschemelist) + ", "
if(chance_roll(insanitylevel, colorschemechance) and generatecolorscheme == True):
completeprompt += "-colorscheme-, "
# vomit some cool/wierd things into the prompt
if(uncommon_dist(insanitylevel) and generatevomit == True):
if(chance_roll(insanitylevel, vomit1chance) and generatevomit == True):
completeprompt += "-vomit-, "
if(uncommon_dist(insanitylevel)):
if(chance_roll(insanitylevel, vomit2chance)):
completeprompt += "-vomit-, "
#adding a great work of art, like starry night has cool effects. But this should happen only very rarely.
if(novel_dist(insanitylevel) and generategreatwork == True):
if(chance_roll(insanitylevel, greatworkchance) and generategreatwork == True):
completeprompt += " in the style of -greatwork-, "
#adding a poemline. But this should happen only very rarely.
if(novel_dist(insanitylevel) and generatepoemline == True):
if(chance_roll(insanitylevel, poemlinechance) and generatepoemline == True):
completeprompt += " \"-poemline-\", "
#adding a songline. But this should happen only very rarely.
if(novel_dist(insanitylevel) and generatesongline == True):
if(chance_roll(insanitylevel, songlinechance) and generatesongline == True):
completeprompt += " \"-songline-\", "
# everyone loves the adding quality. The better models don't need this, but lets add it anyway
if(uncommon_dist(insanitylevel) and generatequality == True):
if(chance_roll(insanitylevel, quality1chance) and generatequality == True):
completeprompt += "-quality-, "
if(uncommon_dist(insanitylevel)):
if(chance_roll(insanitylevel, quality2chance)):
completeprompt += "-quality-, "

36
csvfiles/config/default_config.csv
View File

@ -58,4 +58,38 @@ objectadditionsrepeats;2
objectadditionschance;uncommon
humanadditionchance;rare
overalladditionchance;extraordinary
# humanoid other stuff;
emojichance;legendary
joboractivitychance;normal
# things in the middle of the prompt;
custominputmidrepeats;2
custominputmidchance;uncommon
minivomitmidchance;unique
# humanoid main stuff;
outfitchance;normal
posechance;uncommon
hairchance;normal
accessorychance;normal
humanoidinsideshotchance;legendary
humanoidbackgroundchance;uncommon
# minilocation stuff;
landscapeminilocationchance;uncommon
generalminilocationchance;rare
# general stuff add the end of the prompt;
timperiodchance;normal
focuschance;normal
directionchance;normal
moodchance;normal
minivomitsuffixchance;unique
artmovementchance;normal
lightingchance;normal
photoadditionchance;common
lenschance;normal
colorschemechance;normal
vomit1chance;uncommon
vomit2chance;uncommon
greatworkchance;novel
poemlinechance;novel
songlinechance;novel
quality1chance;uncommon
quality2chance;uncommon
1 #this file controls the generation process
58 objectadditionschance uncommon
59 humanadditionchance rare
60 overalladditionchance extraordinary
61 # humanoid other stuff
62 emojichance legendary
63 joboractivitychance normal
64 # things in the middle of the prompt
65 custominputmidrepeats 2
66 custominputmidchance uncommon
67 minivomitmidchance unique
68 # humanoid main stuff
69 outfitchance normal
70 posechance uncommon
71 hairchance normal
72 accessorychance normal
73 humanoidinsideshotchance legendary
74 humanoidbackgroundchance uncommon
75 # minilocation stuff
76 landscapeminilocationchance uncommon
77 generalminilocationchance rare
78 # general stuff add the end of the prompt
79 timperiodchance normal
80 focuschance normal
81 directionchance normal
82 moodchance normal
83 minivomitsuffixchance unique
84 artmovementchance normal
85 lightingchance normal
86 photoadditionchance common
87 lenschance normal
88 colorschemechance normal
89 vomit1chance uncommon
90 vomit2chance uncommon
91 greatworkchance novel
92 poemlinechance novel
93 songlinechance novel
94 quality1chance uncommon
95 quality2chance uncommon

6
random_functions.py
View File

@ -58,12 +58,14 @@ def chance_roll(insanitylevel, chance):
properties = chance_mapping[chance]
set_number = properties['set_number']
message = properties['message']
if(message != ""):
print(message)
# if we have insanity level of 10, then every under rare is alwas true
if (set_number <= 35 and insanitylevel >= 10):
if(message != ""):
print(message)
return True
roll = random.randint(1, set_number) < insanitylevel
if(message != "" and roll == True):
print(message)
return roll
else:
raise ValueError(f"Invalid chance value: {chance}")