Adding tv show episode titles

2023-10-30 07:55:47 +01:00
parent ce6a956cde
commit 3a96cf2d3f
6 changed files with 6138 additions and 10 deletions
@@ -65,6 +65,7 @@ generateconcepts = True
 generatepoemline = True
 generatesongline = True
 generatecardname = True
+generateepisodetitle = True


 for item in config:
@@ -115,6 +116,8 @@ for item in config:
            generatesongline = False
        if item[0] == 'subject_cardname' and item[1] != 'on':
            generatecardname = False
+        if item[0] == 'subject_episodetitle' and item[1] != 'on':
+            generateepisodetitle = False

 # build up all subjects we can choose based on the loaded config file
 if(generatevehicle or generateobject or generatefood or generatebuilding or generatespace):
@@ -125,7 +128,7 @@ if(generatemanwoman or generatemanwomanrelation or generatefictionalcharacter or
     subjects.append("humanoid")
 if(generatelandscape):
     subjects.append("landscape")
-if(generateevent or generateconcepts or generatepoemline or generatesongline or generatecardname):
+if(generateevent or generateconcepts or generatepoemline or generatesongline or generatecardname or generateepisodetitle):
     subjects.append("concept")


@@ -177,6 +180,9 @@ if(generatesongline):
     subjectsubtypesconcept.append("lines from songs")
 if(generatecardname):
     subjectsubtypesconcept.append("names from card based games")
+if(generateepisodetitle):
+     subjectsubtypesconcept.append("episode titles from tv shows")
+     

 class OneButtonPrompt:

@@ -141,6 +141,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
    settinglist = csv_to_list("settings", antilist)
    charactertypelist = csv_to_list("charactertypes", antilist)
    objectstoholdlist = csv_to_list("objectstohold", antilist)
+    episodetitlelist = csv_to_list(csvfilename="episodetitles",antilist=antilist,skipheader=True)
    

    # additional descriptor lists
@@ -312,6 +313,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
    generatepoemline = True
    generatesongline = True
    generatecardname = True
+    generateepisodetitle = True

    custominputprefixrepeats = 2
    custominputprefixchance = 'uncommon'
@@ -437,6 +439,8 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
            generatesongline = False
        if item[0] == 'subject_cardname' and item[1] != 'on':
            generatecardname = False
+        if item[0] == 'subject_episodetitle' and item[1] != 'on':
+            generateepisodetitle = False
        
        # main list stuff
        if item[0] == 'custominputprefixrepeats':
@@ -685,6 +689,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
    generatepoemline = bool(poemlinelist) and generatepoemline 
    generatesongline = bool(songlinelist) and generatesongline
    generatecardname = bool(cardnamelist) and generatecardname
+    generateepisodetitle = bool(episodetitlelist) and generateepisodetitle
    


@@ -705,6 +710,9 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
    if(generatecardname):
        eventsubjectchooserlist.append("cardname")

+    if(generateepisodetitle):
+        eventsubjectchooserlist.append("episodetitle")
+
    if(generateconcept):
        mainchooserlist.append("concept")

@@ -817,6 +825,7 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
    generatepoemline = bool(poemlinelist) and not specialmode
    generatesongline = bool(songlinelist) and not specialmode
    generatecardname = bool(cardnamelist) and not specialmode
+    generateepisodetitle = bool(episodetitlelist) and not specialmode
    
    generateminilocationaddition = bool(minilocationadditionslist) and not specialmode
    generateminivomit = bool(minivomitlist) and not specialmode
@@ -1177,6 +1186,8 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
                    subjectchooser = "songline"
                if(subtypeconcept == "names from card based games"):
                    subjectchooser = "cardname"
+                if(subtypeconcept == "episode titles from tv shows"):
+                    subjectchooser = "episodetitle"

        # After we chose the subject, lets set all things ready for He/She/It etc
        
@@ -1908,6 +1919,11 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all

                    if(subjectchooser == "cardname"):
                        completeprompt += " \" -cardname- \" "
+
+                    if(subjectchooser == "episodetitle"):
+                        completeprompt += " \" -episodetitle- \" "
+
+                    
                # making subject override work with X and Y concepts, much fun!
                elif(givensubject != "" and subjectchooser == "concept"):
                        if(random.randint(0,3) == 0):
@@ -2779,11 +2795,12 @@ def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all
    "-element-" in completeprompt or
    "-setting-" in completeprompt or
    "-charactertype-" in completeprompt or
-    "-objectstohold-" in completeprompt):
+    "-objectstohold-" in completeprompt or
+    "-episodetitle-" 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-","-subjectfromfile-", "-brand-", "-space-", "-artist-", "-imagetype-", "-othertype-", "-quality-", "-lighting-", "-camera-", "-lens-","-imagetypequality-", "-poemline-", "-songline-", "-greatwork-", "-fantasyartist-", "-popularartist-", "-romanticismartist-", "-photographyartist-", "-emoji-", "-timeperiod-", "-shotsize-", "-musicgenre-", "-animaladdition-", "-addontolocationinside-", "-addontolocation-", "-objectaddition-", "-humanaddition-", "-overalladdition-", "-focus-", "-direction-", "-styletilora-", "-manwomanrelation-", "-waterlocation-", "-container-", "-firstname-", "-flora-", "-print-", "-miniactivity-", "-pattern-", "-animalsuffixaddition-", "-chair-", "-cardname-", "-covering-", "-heshe-", "-hisher-", "-himher-", "-outfitdescriptor-", "-hairdescriptor-", "-hairvomit-", "-humandescriptor-", "-manwomanmultiple-", "-facepart-", "-buildfacepart-", "-outfitvomit-", "-locationdescriptor-", "-basicbitchdescriptor-", "-animaldescriptor-", "-humanexpression-", "-humanvomit-", "-eyecolor-", "-fashiondesigner-", "-colorcombination-", "-materialcombination-", "-oppositefictional-", "-oppositenonfictional-", "-photoaddition-", "-age-", "-agecalculator-", "-gregmode-"
-                                    ,"-portraitartist-", "-characterartist-" , "-landscapeartist-", "-scifiartist-", "-graphicdesignartist-", "-digitalartist-", "-architectartist-", "-cinemaartist-", "-setting-", "-charactertype-", "-objectstohold-"]
+                                    ,"-portraitartist-", "-characterartist-" , "-landscapeartist-", "-scifiartist-", "-graphicdesignartist-", "-digitalartist-", "-architectartist-", "-cinemaartist-", "-setting-", "-charactertype-", "-objectstohold-", "-episodetitle-"]
        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, customoutfitslist, 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, floralist, printlist, miniactivitylist, patternlist, animalsuffixadditionlist, chairlist, cardnamelist, coveringlist, heshelist, hisherlist, himherlist, outfitdescriptorlist, hairdescriptorlist, hairvomitlist, humandescriptorlist, manwomanmultiplelist, facepartlist, buildfacepartlist, outfitvomitlist, locationdescriptorlist, basicbitchdescriptorlist, animaldescriptorlist, humanexpressionlist, humanvomitlist, eyecolorlist, fashiondesignerlist, colorcombinationlist, materialcombinationlist, oppositefictionallist, oppositenonfictionallist, photoadditionlist, agelist, agecalculatorlist, gregmodelist
-                                         , portraitartistlist, characterartistlist, landscapeartistlist, scifiartistlist, graphicdesignartistlist, digitalartistlist, architectartistlist, cinemaartistlist, settinglist, charactertypelist, objectstoholdlist]
+                                         , portraitartistlist, characterartistlist, landscapeartistlist, scifiartistlist, graphicdesignartistlist, digitalartistlist, architectartistlist, cinemaartistlist, settinglist, charactertypelist, objectstoholdlist, episodetitlelist]
        
        allwildcardslistwithhybrid = ["-material-", "-descriptor-", "-outfit-", "-conceptsuffix-","-culture-", "-objecttotal-", "-outfitprinttotal-", "-element-"]
        allwildcardslistwithhybridlists = [materiallist, descriptorlist,outfitlist,conceptsuffixlist,culturelist, objecttotallist, outfitprinttotallist, elementlist]
@@ -2950,6 +2967,7 @@ def createpromptvariant(prompt = "", insanitylevel = 5, antivalues = "" , gender
    settinglist = csv_to_list("settings", antilist)
    charactertypelist = csv_to_list("charactertypes", antilist)
    objectstoholdlist = csv_to_list("objectstohold", antilist)
+    episodetitlelist = csv_to_list(csvfilename="episodetitles",antilist=antilist,skipheader=True)

    outfitdescriptorlist = csv_to_list("outfitdescriptors",antilist)
    hairdescriptorlist = csv_to_list("hairdescriptors",antilist)
@@ -3158,6 +3176,8 @@ def createpromptvariant(prompt = "", insanitylevel = 5, antivalues = "" , gender
                    prompt = prompt.replace(combination," -charactertype- ")
                if lowercase_combination in [x.lower() for x in objectstoholdlist] and chance_roll(insanitylevel, "uncommon"):
                    prompt = prompt.replace(combination," -objectstohold- ")
+                if lowercase_combination in [x.lower() for x in episodetitlelist] and chance_roll(insanitylevel, "uncommon"):
+                    prompt = prompt.replace(combination," -episodetitle- ")

                
                #if lowercase_combination in [x.lower() for x in conceptprefixlist] and chance_roll(insanitylevel, "uncommon"):
@@ -3448,12 +3468,13 @@ def createpromptvariant(prompt = "", insanitylevel = 5, antivalues = "" , gender
        "-element-" in completeprompt or
        "-setting-" in completeprompt or
        "-charactertype-" in completeprompt or
-        "-objectstohold-" in completeprompt
+        "-objectstohold-" in completeprompt or
+        "-episodetitle-" 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-", "-subjectfromfile-", "-brand-", "-space-", "-artist-", "-imagetype-", "-othertype-", "-quality-", "-lighting-", "-camera-", "-lens-","-imagetypequality-", "-poemline-", "-songline-", "-greatwork-", "-fantasyartist-", "-popularartist-", "-romanticismartist-", "-photographyartist-", "-emoji-", "-timeperiod-", "-shotsize-", "-musicgenre-", "-animaladdition-", "-objectaddition-", "-humanaddition-", "-overalladdition-", "-focus-", "-direction-", "-styletilora-", "-manwomanrelation-", "-waterlocation-", "-container-", "-firstname-", "-flora-", "-print-", "-miniactivity-", "-pattern-", "-chair-", "-cardname-", "-covering-", "-outfitdescriptor-", "-hairdescriptor-", "-hairvomit-", "-humandescriptor-", "-manwomanmultiple-", "-facepart-", "-locationdescriptor-", "-basicbitchdescriptor-", "-animaldescriptor-", "-humanexpression-", "-humanvomit-", "-eyecolor-", "-fashiondesigner-", "-colorcombination-", "-materialcombination-", "-photoaddition-", "-age-", "agecalculator-", "-gregmode-"
-                                        ,"-portraitartist-", "-characterartist-" , "-landscapeartist-", "-scifiartist-", "-graphicdesignartist-", "-digitalartist-", "-architectartist-", "-cinemaartist-" "-setting-", "-charactertype-", "-objectstohold-"]
+                                        ,"-portraitartist-", "-characterartist-" , "-landscapeartist-", "-scifiartist-", "-graphicdesignartist-", "-digitalartist-", "-architectartist-", "-cinemaartist-" "-setting-", "-charactertype-", "-objectstohold-", "-episodetitle-"]
            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, customoutfitslist, brandlist, spacelist, artistlist, imagetypelist, othertypelist, qualitylist, lightinglist, cameralist, lenslist, imagetypequalitylist, poemlinelist, songlinelist, greatworklist, fantasyartistlist, popularartistlist, romanticismartistlist, photographyartistlist, emojilist, timeperiodlist, shotsizelist, musicgenrelist, animaladditionlist, objectadditionslist, humanadditionlist, overalladditionlist, focuslist, directionlist, stylestiloralist, manwomanrelationlist, waterlocationlist, containerlist, firstnamelist, floralist, printlist, miniactivitylist, patternlist, chairlist, cardnamelist, coveringlist, outfitdescriptorlist, hairdescriptorlist, hairvomitlist, humandescriptorlist, manwomanmultiplelist, facepartlist, locationdescriptorlist, basicbitchdescriptorlist, animaldescriptorlist, humanexpressionlist, humanvomitlist, eyecolorlist, fashiondesignerlist, colorcombinationlist, materialcombinationlist, photoadditionlist, agelist, agecalculatorlist, gregmodelist
-                                             , portraitartistlist, characterartistlist, landscapeartistlist, scifiartistlist, graphicdesignartistlist, digitalartistlist, architectartistlist, cinemaartistlist, settinglist, charactertypelist, objectstoholdlist]
+                                             , portraitartistlist, characterartistlist, landscapeartistlist, scifiartistlist, graphicdesignartistlist, digitalartistlist, architectartistlist, cinemaartistlist, settinglist, charactertypelist, objectstoholdlist, episodetitlelist]
            
            allwildcardslistwithhybrid = ["-material-", "-descriptor-", "-outfit-", "-conceptsuffix-","-culture-", "-objecttotal-", "-outfitprinttotal-", "-element-"]
            allwildcardslistwithhybridlists = [materiallist, descriptorlist,outfitlist,conceptsuffixlist,culturelist, objecttotallist, outfitprinttotallist, elementlist]
@@ -30,6 +30,7 @@ subject_concept;on
 subject_poemline;on
 subject_songline;on
 subject_cardname;on
+subject_episodetitle;on
 #----;
 # other main lists, in order of appearance;
 # setting the random chance, for normal generation;
@@ -90,13 +90,13 @@ generateprompts(10,5
                ,"","","PREFIXPROMPT" 
                ,"SUFFIXPROMPT"
                ,"",1,""
-                ,"TEST" # subject override
+                ,"" # 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
-                , "the X of Y concepts" # concept types
+                , "episode titles from tv shows" # concept types
                , False  # prompt switching
                , True  # Turn off emojis
                , 0  # seed
@@ -100,6 +100,7 @@ generateconcepts = True
 generatepoemline = True
 generatesongline = True
 generatecardname = True
+generateepisodetitle = True


 for item in config:
@@ -150,6 +151,8 @@ for item in config:
            generatesongline = False
        if item[0] == 'subject_cardname' and item[1] != 'on':
            generatecardname = False
+        if item[0] == 'subject_episodetitle' and item[1] != 'on':
+            generateepisodetitle = False

 # build up all subjects we can choose based on the loaded config file
 if(generatevehicle or generateobject or generatefood or generatebuilding or generatespace):
@@ -160,7 +163,7 @@ if(generatemanwoman or generatemanwomanrelation or generatefictionalcharacter or
     subjects.append("humanoid")
 if(generatelandscape):
     subjects.append("landscape")
-if(generateevent or generateconcepts or generatepoemline or generatesongline or generatecardname):
+if(generateevent or generateconcepts or generatepoemline or generatesongline or generatecardname or generateepisodetitle):
     subjects.append("concept")


@@ -213,6 +216,8 @@ if(generatesongline):
     subjectsubtypesconcept.append("lines from songs")
 if(generatecardname):
     subjectsubtypesconcept.append("names from card based games")
+if(generateepisodetitle):
+     subjectsubtypesconcept.append("episode titles from tv shows")