Upload 55 files

app.py

@@ -521,10 +521,11 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
 
     with gr.Tab("FLUX Prompt Generator"):
         from prompt import (PromptGenerator, HuggingFaceInferenceNode, florence_caption,
-            ARTFORM, PHOTO_TYPE, BODY_TYPES, DEFAULT_TAGS, ROLES, HAIRSTYLES, ADDITIONAL_DETAILS,
+            ARTFORM, PHOTO_TYPE, ROLES, HAIRSTYLES, LIGHTING, COMPOSITION, POSE, BACKGROUND,
             PHOTOGRAPHY_STYLES, DEVICE, PHOTOGRAPHER, ARTIST, DIGITAL_ARTFORM, PLACE,
-            LIGHTING, CLOTHING, COMPOSITION, POSE, BACKGROUND, pg_title)
-        
+            FEMALE_DEFAULT_TAGS, MALE_DEFAULT_TAGS, FEMALE_BODY_TYPES, MALE_BODY_TYPES,
+            FEMALE_CLOTHING, MALE_CLOTHING, FEMALE_ADDITIONAL_DETAILS, MALE_ADDITIONAL_DETAILS, pg_title)
+
         prompt_generator = PromptGenerator()
         huggingface_node = HuggingFaceInferenceNode()
 
@@ -533,9 +534,9 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
         with gr.Row():
             with gr.Column(scale=2):
                 with gr.Accordion("Basic Settings"):
-                    pg_seed = gr.Slider(0, 30000, label='Seed', step=1, value=random.randint(0,30000))
                     pg_custom = gr.Textbox(label="Custom Input Prompt (optional)")
                     pg_subject = gr.Textbox(label="Subject (optional)")
+                    pg_gender = gr.Radio(["female", "male"], label="Gender", value="female")
                     
                     # Add the radio button for global option selection
                     pg_global_option = gr.Radio(
@@ -549,11 +550,11 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
                     pg_photo_type = gr.Dropdown(["disabled", "random"] + PHOTO_TYPE, label="Photo Type", value="disabled")
             
                 with gr.Accordion("Character Details", open=False):
-                    pg_body_types = gr.Dropdown(["disabled", "random"] + BODY_TYPES, label="Body Types", value="disabled")
-                    pg_default_tags = gr.Dropdown(["disabled", "random"] + DEFAULT_TAGS, label="Default Tags", value="disabled")
+                    pg_body_types = gr.Dropdown(["disabled", "random"] + FEMALE_BODY_TYPES + MALE_BODY_TYPES, label="Body Types", value="disabled")
+                    pg_default_tags = gr.Dropdown(["disabled", "random"] + FEMALE_DEFAULT_TAGS + MALE_DEFAULT_TAGS, label="Default Tags", value="disabled")
                     pg_roles = gr.Dropdown(["disabled", "random"] + ROLES, label="Roles", value="disabled")
                     pg_hairstyles = gr.Dropdown(["disabled", "random"] + HAIRSTYLES, label="Hairstyles", value="disabled")
-                    pg_clothing = gr.Dropdown(["disabled", "random"] + CLOTHING, label="Clothing", value="disabled")
+                    pg_clothing = gr.Dropdown(["disabled", "random"] + FEMALE_CLOTHING + MALE_CLOTHING, label="Clothing", value="disabled")
             
                 with gr.Accordion("Scene Details", open=False):
                     pg_place = gr.Dropdown(["disabled", "random"] + PLACE, label="Place", value="disabled")
@@ -563,7 +564,7 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
                     pg_background = gr.Dropdown(["disabled", "random"] + BACKGROUND, label="Background", value="disabled")
             
                 with gr.Accordion("Style and Artist", open=False):
-                    pg_additional_details = gr.Dropdown(["disabled", "random"] + ADDITIONAL_DETAILS, label="Additional Details", value="disabled")
+                    pg_additional_details = gr.Dropdown(["disabled", "random"] + FEMALE_ADDITIONAL_DETAILS + MALE_ADDITIONAL_DETAILS, label="Additional Details", value="disabled")
                     pg_photography_styles = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHY_STYLES, label="Photography Styles", value="disabled")
                     pg_device = gr.Dropdown(["disabled", "random"] + DEVICE, label="Device", value="disabled")
                     pg_photographer = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHER, label="Photographer", value="disabled")
@@ -587,94 +588,87 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', fill_width=True, css=css) as app:
             
             with gr.Column(scale=2):
                 with gr.Accordion("Prompt Generation with LLM", open=False):
-                    pg_model = gr.Dropdown(["Mixtral", "Mistral", "Llama 3", "Mistral-Nemo"], label="Model", value="Llama 3")
                     pg_happy_talk = gr.Checkbox(label="Happy Talk", value=True)
                     pg_compress = gr.Checkbox(label="Compress", value=True)
                     pg_compression_level = gr.Radio(["soft", "medium", "hard"], label="Compression Level", value="hard")
                     pg_poster = gr.Checkbox(label="Poster", value=False)
                     pg_custom_base_prompt = gr.Textbox(label="Custom Base Prompt", lines=5)
-                pg_generate_text_button = gr.Button("Generate Prompt with LLM")
+                pg_generate_text_button = gr.Button("Generate Prompt with LLM (Llama 3.1 70B)")
                 pg_text_output = gr.Textbox(label="Generated Text", lines=10)
 
-    description_ui()
-
-    def create_caption(image):
-        if image is not None:
-            return florence_caption(image)
-        return ""
+        def create_caption(image):
+            if image is not None:
+                return florence_caption(image)
+            return ""
 
-    pg_create_caption_button.click(
-        create_caption,
-        inputs=[pg_input_image],
-        outputs=[pg_caption_output],
-        show_api=False,
-    )
+        pg_create_caption_button.click(
+            create_caption,
+            inputs=[pg_input_image],
+            outputs=[pg_caption_output]
+        )
 
-    pg_generate_button.click(
-        prompt_generator.generate_prompt,
-        inputs=[pg_seed, pg_custom, pg_subject, pg_artform, pg_photo_type, pg_body_types,
-                pg_default_tags, pg_roles, pg_hairstyles,
-                pg_additional_details, pg_photography_styles, pg_device, pg_photographer,
-                pg_artist, pg_digital_artform,
-                pg_place, pg_lighting, pg_clothing, pg_composition, pg_pose, pg_background],
-        outputs=[pg_output, gr.Number(visible=False), pg_t5xxl_output, pg_clip_l_output, pg_clip_g_output],
-        show_api=False,
-    )
+        def generate_prompt_with_dynamic_seed(*args):
+            # Generate a new random seed
+            dynamic_seed = random.randint(0, 1000000)
+            
+            # Call the generate_prompt function with the dynamic seed
+            result = prompt_generator.generate_prompt(dynamic_seed, *args)
+            
+            # Return the result along with the used seed
+            return [dynamic_seed] + list(result)
+
+        pg_generate_button.click(
+            generate_prompt_with_dynamic_seed,
+            inputs=[pg_custom, pg_subject, pg_gender, pg_artform, pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles,
+                    pg_additional_details, pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform,
+                    pg_place, pg_lighting, pg_clothing, pg_composition, pg_pose, pg_background, pg_input_image],
+            outputs=[gr.Number(label="Used Seed", visible=True), pg_output, gr.Number(visible=False), pg_t5xxl_output, pg_clip_l_output, pg_clip_g_output]
+        )
 
-    pg_add_caption_button.click(
-        prompt_generator.add_caption_to_prompt,
-        inputs=[pg_output, pg_caption_output],
-        outputs=[pg_output],
-        show_api=False,
-    )
+        pg_add_caption_button.click(
+            prompt_generator.add_caption_to_prompt,
+            inputs=[pg_output, pg_caption_output],
+            outputs=[pg_output]
+        )
 
-    pg_generate_text_button.click(
-        huggingface_node.generate,
-        inputs=[pg_model, pg_output, pg_happy_talk, pg_compress, pg_compression_level,
-                pg_poster, pg_custom_base_prompt],
-        outputs=pg_text_output,
-        show_api=False,
-    )
+        pg_generate_text_button.click(
+            huggingface_node.generate,
+            inputs=[pg_output, pg_happy_talk, pg_compress, pg_compression_level, pg_poster, pg_custom_base_prompt],
+            outputs=pg_text_output
+        )
 
-    def update_all_options(choice):
-        updates = {}
-        if choice == "Disabled":
-            for dropdown in [
-                pg_artform, pg_photo_type, pg_body_types, pg_default_tags,
-                pg_roles, pg_hairstyles, pg_clothing,
-                pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
-                pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
-            ]:
-                updates[dropdown] = gr.update(value="disabled")
-        elif choice == "Random":
-            for dropdown in [
-                pg_artform, pg_photo_type, pg_body_types, pg_default_tags,
-                pg_roles, pg_hairstyles, pg_clothing,
+        def update_all_options(choice):
+            updates = {}
+            if choice == "Disabled":
+                for dropdown in [
+                    pg_artform, pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles, pg_clothing,
+                    pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
+                    pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
+                ]:
+                    updates[dropdown] = gr.update(value="disabled")
+            elif choice == "Random":
+                for dropdown in [
+                    pg_artform, pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles, pg_clothing,
+                    pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
+                    pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
+                ]:
+                    updates[dropdown] = gr.update(value="random")
+            else:  # No Figure Random
+                for dropdown in [pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles, pg_clothing, pg_pose, pg_additional_details]:
+                    updates[dropdown] = gr.update(value="disabled")
+                for dropdown in [pg_artform, pg_place, pg_lighting, pg_composition, pg_background, pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform]:
+                    updates[dropdown] = gr.update(value="random")
+            return updates
+        
+        pg_global_option.change(
+            update_all_options,
+            inputs=[pg_global_option],
+            outputs=[
+                pg_artform, pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles, pg_clothing,
                 pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
                 pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
-            ]:
-                updates[dropdown] = gr.update(value="random")
-        else:  # No Figure Random
-            for dropdown in [pg_photo_type, pg_body_types, pg_default_tags,
-                                pg_roles, pg_hairstyles, pg_clothing, pg_pose, pg_additional_details]:
-                updates[dropdown] = gr.update(value="disabled")
-            for dropdown in [pg_artform, pg_place, pg_lighting, pg_composition,
-                                pg_background, pg_photography_styles, pg_device, pg_photographer,
-                                pg_artist, pg_digital_artform]:
-                updates[dropdown] = gr.update(value="random")
-        return updates
-    
-    pg_global_option.change(
-        update_all_options,
-        inputs=[pg_global_option],
-        outputs=[
-            pg_artform, pg_photo_type, pg_body_types, pg_default_tags,
-            pg_roles, pg_hairstyles, pg_clothing,
-            pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
-            pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
-        ],
-        show_api=False,
-    )
+            ]
+        )
 
 app.queue()
 app.launch()

data/female_additional_details.json

@@ -0,0 +1,183 @@
+[
+	"a purple iridescent suit",
+	"wearing a (necklace)",
+	"wearing ((earrings))",
+	"wearing a (bracelet)",
+	"wearing one or multiple (rings)",
+	"wearing a (brooch)",
+	"wearing (eyeglasses)",
+	"wearing (sunglasses)",
+	"wearing a (hat)",
+	"wearing a (scarf)",
+	"wearing a (headband)",
+	"wearing a (nose ring)",
+	"wearing a (lip ring)",
+	"wearing a (tongue ring)",
+	"wearing an (eyebrow ring)",
+	"wearing (face tattoos)",
+	"wearing a (wreath)",
+	"wearing a (crown)",
+	"wearing a (tiara)",
+	"wearing a (crown of thorns)",
+	"wearing a (crown of jewels)",
+	"wearing (bohemian clothes)",
+	"wearing (chic clothes)",
+	"wearing (glamorous clothes)",
+	"wearing (grunge clothes)",
+	"wearing (preppy clothes)",
+	"wearing (punk clothes)",
+	"wearing (retro clothes)",
+	"wearing (rockabilly clothes)",
+	"wearing (romantic clothes)",
+	"wearing (tomboy clothes)",
+	"wearing (urban clothes)",
+	"wearing (camo clothes)",
+	"wearing (robes)",
+	"wearing (excessive amount of jewellery)",
+	"wearing (vintage clothes)",
+	"wearing (western clothes)",
+	"wearing (minimalist clothes)",
+	"wearing (sportswear clothes)",
+	"wearing (flapper clothes)",
+	"wearing (pin-up clothes)",
+	"wearing (mid-century modern clothes)",
+	"wearing (art deco clothes)",
+	"wearing (victorian clothes)",
+	"wearing (edwardian clothes)",
+	"wearing (elizabethan clothes)",
+	"wearing (retro 70s clothes)",
+	"wearing (retro 80s clothes)",
+	"wearing (retro 90s clothes)",
+	"wearing (retro 00s clothes)",
+	"wearing (musical equipment)",
+	"wearing (leather)",
+	"wearing (bdsm leather)",
+	"wearing (shiny latex)",
+	"wearing (shiny latex suit)",
+	"wearing (silk)",
+	"wearing (full tweed set)",
+	"wearing (clothes made entirely of feathers)",
+	"wearing (clothes made entirely of fur)",
+	"wearing (clothes made entirely of leather)",
+	"wearing (clothes made entirely of metal)",
+	"wearing (clothes made entirely of plastic)",
+	"wearing (clothes adorned with shimmering jewels or crystals)",
+	"waring (clothes adorned with sequins)",
+	"wearing (clothes with exaggerated or extreme silhouettes)",
+	"wearing (clothes with exaggerated or extreme footwear)",
+	"wearing (clothes with exaggerated or extreme headwear)",
+	"wearing (clothes with exaggerated or extreme facial or body piercings or tattoos)",
+	"wearing (clothes with multiple layers or tiers)",
+	"wearing (clothes with exaggerated or extreme colors)",
+	"wearing (clothes with exaggerated or extreme patterns)",
+	"wearing (cloak)",
+	"wearing an astronaut armor",
+	"wearing a bio mechanical suit",
+	"wearing a bio hazard suit",
+	"(( working with laptop))",
+	"with Heat deformation",
+	"(((future soldier, full body armor, futuristic football, shoulder pads, guns, grenades, weapons, bullet proof vest, high tech, straps, belts, camouflage)))",
+	"((full body, zoomed out)) long slender legs 80mm",
+	"(((sci-fi, future war, cyberpunk, cyborg, future fashion, beautiful face, glowing tattoos)))",
+	"((angry expression, pretty face))",
+	"(((full body, athletic body, action pose, detailed black soldier outfit, slender long legs)))",
+	"playing epic electric guitar solo in front of a huge crowd",
+	"singing epic solo into a microphone in front of a huge crowd",
+	"as a ((gelatinous horror dripping alien creature))",
+	"in a tie or bowtie, lending a touch of formal elegance or quirky charm, knotted around the collar to elevate the outfit",
+	"with anklets, delicate chains or beads that gracefully encircle the ankle, adding a touch of femininity with every step",
+	"donning a belt, functional yet fashionable, cinching the waist or sitting low on the hips, often with a statement buckle",
+	"wearing gloves, either elegant satin for formal events or rugged leather for a tougher look, complementing the attire and mood",
+	"with a choker, snugly encircling the neck, often made of lace, velvet, or leather, exuding a mix of elegance and edge",
+	"in stockings or tights, sheer or opaque, enhancing the legs while adding a touch of sophistication or playful patterns",
+	"with a satchel or bag, a functional accessory that speaks volumes about personal style, be it a minimalist tote or an embellished clutch",
+	"wearing cufflinks, subtle symbols of elegance, adorning the sleeves of a formal shirt, showcasing attention to detail",
+	"with a pendant, a piece of jewelry that dangles gracefully from a necklace, often holding sentimental or symbolic value",
+	"in layered necklaces, a blend of chains of varying lengths, creating depth and showcasing multiple pendants or charms",
+	"sporting a watch, a timeless accessory that blends functionality with style, either minimalist or grand, reflecting personal tastes",
+	"wearing a veil, a delicate piece of fabric that adds mystery and allure, often seen in bridal or ceremonial attire",
+	"donning a cape or cloak, adding drama to the ensemble, flowing gracefully with every movement, evoking a sense of fantasy or regality",
+	"with a tiara or diadem, a jeweled headpiece that signifies royalty or celebration, resting gracefully atop the head",
+	"Adorned with a crown, symbolizing royalty and authority with gemstones and metals",
+	"With a sparkling tiara, reminiscent of princesses or beauty queens",
+	"With a poignant crown of thorns, symbolizing sacrifice and resilience",
+	"With a jewel-encrusted crown, reflecting affluence and grandeur",
+	"In bohemian attire, embodying the free spirits with patterns and fringes",
+	"Dressed in chic fashion, blending comfort with high style",
+	"In glamorous attire, shiny or sequined, perfect for red-carpet events",
+	"Donning grunge wear, with flannels and combat boots, reflecting a rebellious spirit",
+	"In preppy clothes, with clean lines and classic patterns",
+	"Sporting punk fashion, with leather, studs, and bold hairstyles",
+	"In retro outfits, channeling specific decades with authentic pieces",
+	"Wearing rockabilly style, with pin-up influences and rock 'n' roll flair",
+	"In romantic attire, featuring soft fabrics and feminine silhouettes",
+	"Dressed in tomboy fashion, with boyish charm and comfort-focused pieces",
+	"In urban streetwear, blending comfort with trendy, city-inspired looks",
+	"Wearing camouflage, either for function or as a bold fashion statement",
+	"Draped in robes, from casual bathrobes to formal ceremonial garments",
+	"Adorned with excessive jewelry, creating a bold and opulent look",
+	"In vintage clothes, authentically representing past eras and styles",
+	"Sporting western wear, with cowboy hats, boots, and denim",
+	"In minimalist attire, focusing on clean lines and a less-is-more approach",
+	"Wearing sportswear, blending function and style for athletic pursuits",
+	"In flapper style, with dropped waists and ornate beading from the 1920s",
+	"Dressed as a pin-up, channeling retro glamour with curve-hugging styles",
+	"In mid-century modern fashion, reflecting the sleek designs of the 1950s and 60s",
+	"Wearing art deco inspired clothes, with geometric patterns and luxurious fabrics",
+	"In Victorian-style attire, with high necklines, corsets, and full skirts",
+	"Dressed in Edwardian fashion, featuring S-shaped silhouettes and large hats",
+	"In Elizabethan costume, with ruffs, farthingales, and ornate embroidery",
+	"Sporting 70s retro looks, with bell-bottoms, platforms, and psychedelic prints",
+	"In 80s inspired outfits, featuring bold colors, shoulder pads, and leg warmers",
+	"Wearing 90s fashion, with crop tops, high-waisted jeans, and chunky shoes",
+	"In early 2000s style, featuring low-rise jeans, crop tops, and platform sandals",
+	"With musical equipment, like guitars, drumsticks, or headphones as accessories",
+	"In leather attire, from sleek jackets to edgy pants or skirts",
+	"Wearing BDSM-inspired leather, with harnesses, collars, or studded accessories",
+	"In shiny latex, hugging every curve with a glossy, futuristic appeal",
+	"Sporting a full latex suit, covering from neck to toe in sleek, shiny material",
+	"Draped in silk, with flowing fabrics that catch the light beautifully",
+	"In a full tweed set, channeling British countryside chic",
+	"Wearing an outfit made entirely of feathers, creating a bold, avian-inspired look",
+	"In a fur ensemble, either faux or real, exuding luxury and warmth",
+	"Sporting an all-leather outfit, from jacket to pants and accessories",
+	"In a metallic outfit, with clothes seemingly forged from shining metals",
+	"Wearing an ensemble made entirely of plastic, futuristic and unconventional",
+	"Adorned in clothes shimmering with jewels or crystals, catching light at every angle",
+	"In a sequin-covered outfit, sparkling and eye-catching from every view",
+	"Wearing clothes with exaggerated silhouettes, pushing the boundaries of shape and form",
+	"In footwear with extreme designs, from towering platforms to avant-garde shapes",
+	"Sporting headwear with exaggerated proportions or unconventional materials",
+	"With extreme facial piercings or full-body tattoos, turning the skin into a canvas",
+	"In an outfit with multiple layers or tiers, creating depth and visual interest",
+	"Wearing clothes in extremely bold or neon colors, making a vibrant statement",
+	"In attire with exaggerated or surreal patterns, creating visual intrigue",
+	"Draped in a mysterious cloak, adding an air of intrigue or fantasy",
+	"In astronaut armor, representing space exploration and futuristic concepts",
+	"Wearing a bio-mechanical suit, blending organic forms with mechanical elements",
+	"In a biohazard suit, fully enclosed for protection or as a fashion statement",
+	"Engaged with a laptop, representing technology integration in fashion",
+	"Surrounded by heat deformation effects, as if the very air is bending around them",
+	"In futuristic soldier gear, with high-tech armor, weapons, and camouflage",
+	"Posed to accentuate long, slender legs, often with 80mm or higher heels",
+	"In sci-fi inspired wear, with cyberpunk elements and glowing tattoos",
+	"With an angry expression juxtaposed against delicate features",
+	"In a detailed black tactical outfit, emphasizing an athletic build and long legs",
+	"Mid-performance with an electric guitar, captivating an imaginary audience",
+	"Belting out a solo performance, microphone in hand, before a cheering crowd",
+	"Transformed into a gelatinous alien creature, dripping and otherworldly",
+	"With a tie or bowtie, adding a formal or quirky touch to the ensemble",
+	"Wearing delicate anklets, adding a feminine charm to the overall look",
+	"With a statement belt, either cinching the waist or riding low on the hips",
+	"In elegant or rugged gloves, complementing the outfit's overall mood",
+	"Sporting a choker, from delicate lace to edgy leather designs",
+	"In stockings or tights, either sheer for elegance or patterned for fun",
+	"Carrying a stylish bag, from minimalist totes to ornate clutches",
+	"With cufflinks, adding a touch of sophistication to formal wear",
+	"Wearing a meaningful pendant, dangling from a necklace",
+	"In layered necklaces, creating depth with varying lengths and styles",
+	"Sporting a watch, balancing functionality with personal style",
+	"Wearing a veil, adding mystery and allure to the overall look",
+	"Draped in a cape or cloak, bringing drama and movement to the ensemble",
+	"Crowned with a tiara or diadem, adding a regal touch to the outfit"
+]

data/female_body_types.json

@@ -0,0 +1,44 @@
+[
+    "pretty",
+    "chubby",
+    "midweight",
+    "overweight",
+    "fat",
+    "flabby",
+    "buxom",
+    "voluptuous",
+    "hefty",
+    "pudgy",
+    "plump",
+    "obese",
+    "morbidly obese",
+    "stout",
+    "rotund",
+    "thick-bodied",
+    "thicc",
+    "thick",
+    "beefy",
+    "portly",
+    "tubby",
+    "overweight",
+    "(slightly overweight)",
+    "buff",
+    "burly",
+    "fit",
+    "well-built",
+    "well-endowed",
+    "muscular",
+    "stocky",
+    "big-boned",
+    "curvy",
+    "flabby",
+    "flyweight",
+    "skinny",
+    "too skinny",
+    "anorexic",
+    "not skinny",
+    "slender",
+    "lanky",
+    "slim",
+    "slight"
+]

data/female_clothing.json

@@ -0,0 +1,344 @@
+[
+	"white crop top, denim shorts,silver bracelet,white texture,denim texture",
+	"white top, sleeveless, button-up, white bottoms, ribbed texture",
+	"white crop top, blue jeans, silver belt, textured fabric, denim texture",
+	"white tank top, black pants, pink and blue goalie pads, black belt, white skates",
+	"black long-sleeve dress, fishnet stockings",
+	"black bikini top, black bikini bottom, hoop earrings",
+	"white towel, white head wrap",
+	"gray crop top, black skirt, ribbed texture, v-neckline",
+	"black bodysuit, sheer texture, light wash jeans, golden bangles",
+	"pink dress, spaghetti straps, form-fitting, floor-length",
+	"black sports bra, black leggings, textured fabric, small white logo",
+	"black bikini, knit texture, gold accessories, aviator sunglasses",
+	"patterned crop top, green leather pants, silver necklaces, red bracelet, hoop earrings",
+	"black bikini top, striped pants, beige textures, open white shirt",
+	"pink dress, orange sash, gold heels, floral headpiece",
+	"pastel pink tank top, blue ripped jeans, black platform heels, neutral shoulder bag",
+	"white sleeveless top, ribbed texture, high-cut white bottoms",
+	"black dress, plunging neckline, satin texture, brown handbag, chain strap, diamond necklace, wristwatch, silver bracelet",
+	"brown crop top, brown skirt, smooth texture",
+	"black sports bra, black leggings, white sneakers, textured fabric",
+	"blue dress, bodycon fit, sleeveless, zipper detail, V-neckline, subtle sheen",
+	"beige tank top, black denim shorts, smooth fabric, ripped texture",
+	"black tube top, gold necklace, gold arm cuff, red headwrap",
+	"black jumpsuit, zipper detail, form-fitting, long-sleeved",
+	"white crop top, blue jeans, silver necklace",
+	"light blue crop top, ribbed texture, long sleeves, denim jeans",
+	"black swimsuit, plunging neckline, sleeveless",
+	"blue dress, strap sleeves, high slit, textured fabric, hoop earrings, wristwatch, bracelet",
+	"black sleeveless top, black leather pants, gold necklace",
+	"black tank top, chain belt, black shorts, patterned boots",
+	"pink hoodie, pink shorts, textured fabric",
+	"black bikini top, denim shorts, frayed hems",
+	"blue bikini, tie-up detail, vibrant prints, double-strap top, high-cut bottoms, textured fabric",
+	"tan crop top, tan high-waisted pants, ripped knee, checked shirt tied, white sneakers",
+	"brown tank top, plaid skirt, smooth texture, fitted top, flared skirt",
+	"white dress, black sandals, dark sunglasses, textured fabric, sheer sleeves, leather bag",
+	"black sports bra, black shorts, black arm band, dark wrist watch, clear eyeglasses",
+	"beige hoodie, matching pants, textured fabric",
+	"red satin dress, off-shoulder style, knotted front",
+	"white crop top, lace details, high-waisted skirt, pleated texture",
+	"black dress, sleeveless top, plunging neckline, white cross necklace, textured handbag, silver bracelet",
+	"pink halter-neck dress, ribbed texture, form-fitting",
+	"yellow lace-up top, white lace bralette, blue denim shorts, green handbag",
+	"white crop top, white jeans, glasses",
+	"red dress, sleeveless, v-neckline",
+	"white blouse, intricate lace, sheer sleeves",
+	"white bikini top, white mesh cover-up, white skirt, silver necklace",
+	"black oversized t-shirt, white sport socks",
+	"black strap top, lace trim, glossy texture",
+	"white crop top, ribbed texture, short sleeves, high-waisted shorts, elastic waistband",
+	"pink blazer, white top, pink trousers, silky fabric",
+	"green sports bra, green shorts, white sneakers, textured fabric",
+	"blue crop top, blue shorts, white sneakers, texture appears smooth",
+	"black tank top, denim shorts, textured fabric, light blue, frayed hems",
+	"green sports bra, green shorts, white socks, multicolored sneakers",
+	"straw hat, fringe bikini top, bikini bottom, earth tones",
+	"black dress, strap details, sheer textures",
+	"grey sweatshirt, soft texture, round neckline",
+	"white bikini top, ruffled edges, light fabric, headscarf with print, necklace",
+	"red bikini top, snakeskin pattern, textured fabric",
+	"white tank top, form-fitting, sleeveless",
+	"pink shiny dress, plunging neckline, sleeveless, textured fabric",
+	"strawberry-print top, white with red, matching shorts, silver loop earrings",
+	"pink crop-top, blue denim shorts, smooth texture, subtle sheen",
+	"textured fabric",
+	"white top, off-shoulder design, ruffled texture",
+	"beige strapless top, golden necklace",
+	"pink bikini top, shiny fabric, thin straps, pendant necklace, aviator sunglasses",
+	"white top, blue headband, black headphones",
+	"dark blue top, v-neckline, smooth texture, necklace with pendant",
+	"white top, black accents, mesh details, logo text",
+	"yellow crop top, yellow skirt, white high heels, gold necklace",
+	"white floral dress, off-shoulder design, sheer fabric",
+	"colorful bikini top, gold chain necklace, gold hoop earrings",
+	"denim jumpsuit, blue color, zipper front, sleeveless",
+	"white halter top, smooth texture",
+	"beige fedora, black bikini top, patterned cover-up, light-colored textures",
+	"green bikini top, white necklace",
+	"black halter top, glossy texture",
+	"black jacket, black crop top, black pants, various textures",
+	"black strapless top, golden necklace, golden bracelet",
+	"tie-dye bikini, blue-green-black hues, sports shorts, black with green and blue patterns",
+	"grey turtleneck, sleeveless, soft texture",
+	"beige tracksuit, zipper hoodie, matching pants, soft fabric",
+	"white tank top, blue jeans, black hair tie, gold necklace, blonde hair",
+	"satin top, silver color, strapless design, necklace, bracelet",
+	"red dress, plunging neckline, sleeveless, smooth texture, large hoop earrings",
+	"white floral dress, puff sleeves, low-cut neckline, butterfly pendant necklace",
+	"white fluffy robe, gold chain necklace, red lipstick",
+	"blue dress, sleeveless, plunging neckline, cinched waist, flowy texture",
+	"black jacket, purple shirt, denim overalls, white hair accessory, cream cardigan, simple necklace",
+	"sleeveless top, beige color, ribbed texture, high-waisted pants, black color, button details",
+	"white halter top, knotted center, sleeveless, smooth texture",
+	"cream crop top, beige striped shirt, blue ripped jeans",
+	"patterned white hat, large black glasses, denim jacket, red top, gold ring, white manicure",
+	"black rimmed glasses, beige blazer, black top, gold necklace",
+	"brown blouse, shoulder strap, gold necklace, stud earrings, eyeglasses, white manicure, gold rings",
+	"light green tank top, white drawstrings, ribbed texture, light green bottoms, fitted waistband, casual style",
+	"white strapless top, satin texture",
+	"white lace top, denim bottoms",
+	"camouflage crop top, camouflage shorts, brown boots, silver bracelets, sunglasses, hoop earrings",
+	"black top, sheer sleeves, black skirt, black boots, glasses",
+	"white ribbed top, plunging neckline, long sleeves, drawstring front, black jeans, distressed details",
+	"sleeveless top, beige color, snug fit, shorts, sitting",
+	"red dress, textured fabric, thin straps",
+	"yellow shirt, lace bralette, hoop earrings",
+	"white top, sheer sleeves, lace details, pastel pink bralette",
+	"black bodysuit, sheer gloves, patterned scarf, hoop earrings",
+	"black corset top, plaid skirt, glossy texture, lace details",
+	"beige bikini top, gold chain necklace",
+	"blue lace top, cleavage-revealing, long sleeves",
+	"black tank top, smooth texture",
+	"black halter top, white flower accessory",
+	"light pink hairband, green floral dress, gold earrings, light pink nail polish",
+	"white top, low neckline, soft texture",
+	"black tank top, black headband, headphones",
+	"black polka-dot bikini, thin necklace",
+	"black sleeveless top, textured fabric, shiny accessories",
+	"grey cardigan, white shirt, denim overalls, purple jacket, black hair ties",
+	"black leather dress, high-heel boots",
+	"blue camo sports bra, blue camo leggings",
+	"long-sleeve top, white bandeau, pink hues, sheer texture, butterfly prints",
+	"black tank top, white shirt, denim jeans",
+	"blue bikini top, blue high-waisted bottoms, large hoop earrings, white sneakers",
+	"beige hoodie, blue highlights, black sunglasses, blue sneakers, textured sole",
+	"beige hoodie, blue t-shirt, black leggings, blue sneakers, textured materials, oversized top, fitted bottoms",
+	"black crop top, black sheer-panelled bottoms, solid colors, smooth textures",
+	"straw hat, leopard print bikini, white cover-up",
+	"black sports bra, black leggings, smooth textures",
+	"black bikini top, shiny texture, gold necklace",
+	"white sports bra, white leggings, cream textured cardigan",
+	"red dress, thin straps, v-neckline, satin texture",
+	"black tank top, smooth texture",
+	"white halter top, gold necklace",
+	"black bikini, strap details",
+	"black bodysuit, white snow jacket, visible textures",
+	"no visible clothing",
+	"pink beret, light blouse, pink jacket, soft textures",
+	"fur-lined hood, beige coat, texture-soft",
+	"graphic jacket, multicolored, white hoodie, blue jeans",
+	"white tank top, spaghetti straps, lace details, natural textures",
+	"red crop top, white bikini bottoms, ribbed fabric",
+	"red strapless dress, smooth texture",
+	"sleeveless top, neutral color, soft texture",
+	"black sleeveless dress, textured fabric, sheer details",
+	"metallic green dress, crystal choker necklace",
+	"white tank top, gold necklace, natural texture",
+	"pearl headpiece, white textured dress, bejeweled adornments, sheer sleeves",
+	"white dress, deep neckline, sleeveless, lace-up sides, figure-hugging",
+	"white deep v-neck top, gold necklace, neutral makeup, straight hair",
+	"none visible",
+	"patterned blazer, patterned shorts, black bralette, warm colors, glossy texture",
+	"sleeveless top, beige color, denim shorts, light wash, leather seat texture",
+	"floral bikini top, white color, printed texture",
+	"blue dress, white trim, sparkling necklace, silver earrings, silver bracelet",
+	"silver chain necklace, pink pendant, gold bracelet",
+	"white headband, white ribbed top, black undergarment, light-colored jacket",
+	"yellow bikini top, black phone case",
+	"white crop top, light blue ripped jeans, smooth texture, silver necklace",
+	"peach sports bra, peach skirt, smooth fabric",
+	"patterned dress, black and white, deep neckline, short sleeves, textured fabric, black boots, laced footwear",
+	"black sleeveless dress, black heels, silver bracelet, silver necklace",
+	"light pink dress, sheer texture, fringe details, brown belt",
+	"red lace dress, black fur hat, red gloves",
+	"white cropped top, light blue jeans, textured fabric",
+	"pink cropped top, ribbed texture, tie-front detail, blue ripped jeans, casual style, light wash denim, black shoulder bag, brown belt",
+	"blue patterned bikini, sheer sleeves, ruffled cuffs",
+	"red dress, low neckline, sleeveless, smooth texture",
+	"pink sports bra, pink leggings, barefoot, black accents",
+	"striped one-piece swimsuit, blue and white colors, textured fabric",
+	"red ribbed sweater, black leather pants",
+	"white crop top, long sleeves, white skirt, textured fabric, green handbag, gold necklace, white wristwatch",
+	"white crop top, crisscross neckline, white pants, silver chain accents, white handbag, quilted texture",
+	"ribbed beige turtleneck dress, chest cut-out, black handbag, black and white sneakers",
+	"white tank top, black pants, smooth textures",
+	"White long-sleeve top, ribbed texture",
+	"black hoodie, pink leggings, brown shoes, white socks",
+	"Blue-purple bikini top, white lace-up detail, purple patterned skirt",
+	"strapless top, aqua color, textured fabric, high-waisted trousers, matching color, cinched ankles",
+	"pink dress, white cardigan, lacy sleeves",
+	"denim jumpsuit, light blue, sleeveless, lace-up front, fringed hem",
+	"bikini top, bikini bottom, earth tones, string ties",
+	"geometric pattern bodysuit, brown and black colors, plunging neckline, long sleeves, glossy black belt, black shoulder bag",
+	"white sleeveless top, silver zipper, textured fabric, blue denim shorts, frayed hems",
+	"red frilly dress, white heels, blue hair",
+	"white cropped shirt, white shorts, smooth fabric",
+	"yellow patterned bikini top, white bikini bottom, sunglasses, wrist accessories",
+	"grey bikini top, grey skirt, textured fabric, silver wrist accessory",
+	"strapless yellow dress, blue waist sash, beige high heels, black floral hair accessory",
+	"white crop top, black pants, pink and blue leg pads, vaughn glove, white skates",
+	"purple bikini top, lace-up detail, purple print skirt, textured fabric",
+	"pink tank top, blue ripped jeans, black strappy heels, beige shoulder bag",
+	"white tank top, blue denim shorts, textured fabric, visible brand writing",
+	"white cropped top, textured fabric, light-wash denim jeans, high-waisted, lace-up sides",
+	"white cropped cardigan, white pleated skirt, black shoulder bag, dark sunglasses, white bralette, gold necklace",
+	"pink crop top, white midriff band, pink shorts, glossy tan heels",
+	"black bikini top, black bikini bottoms, smooth fabric",
+	"sleeveless top, tan cropped pants, black sunglasses, black backpack straps, ribbed texture top, light colors",
+	"white crochet dress, brown belt, textures visible",
+	"white off-shoulder top, white shorts, textured fabric, ruffled sleeves",
+	"blue crop top, long sleeves, ribbed texture, plunging neckline, gray plaid skirt, pleated design",
+	"floral dress, white base, red and green patterns, V-neckline, long sleeves, brown belt, textured fabric",
+	"purple dress, gold heels, textured fabric, sleeveless, cut-out detailing",
+	"white hat, white dress, tiered ruffles, lace texture, small purse, silver watch, neutral heels",
+	"pink crop top, pink skirt, pink high heels, sleeveless, button details, smooth texture",
+	"black bodysuit, lace texture, sheer fabric",
+	"left: tan bodysuit, right: black dress, plunging neckline, short sleeves",
+	"sleeveless sequined dress, black and blue tones, sheer fabric, high slit, silver sequined dress, beige tones, strappy sandals, ankle strap, heeled footwear",
+	"black corset, lace texture, black gloves, chandelier earrings, silver necklace",
+	"brown polka-dot swimsuit, wide-brimmed straw hat, smooth texture",
+	"pink hearts bodysuit, white base, long sleeves",
+	"gray crop top, gold necklace, gold earrings, light makeup",
+	"Tan crochet bikini top, tan bikini bottom, gold waist chain, black halterneck bikini top, black tie-side bikini bottom, tan crochet bikini top, tan crochet skirt, gold waist chain",
+	"white hat, sunglasses, olive green cropped shirt, olive green skirt, black belt, black bikini bottom visible",
+	"beige dress, deep neckline, ribbed texture, white heels, criss-cross straps",
+	"blue tank top, textured fabric, lace-up front, white shorts, striped pattern, white sneakers",
+	"grey cardigan, white crop top, animal print shorts, blue hair",
+	"green dress, deep neckline, long sleeves, smooth texture, waist belt",
+	"pink bikini, pink heels, pink sunglasses",
+	"patterned jacket, patterned skirt, bikini top, vibrant colors, shiny texture",
+	"white cropped t-shirt, black graphic design, casual style",
+	"Red bikini, white polka dots, bow details",
+	"blue polka-dot dress, front buttoning, sleeveless design, white sunglasses, tied waist belt",
+	"white sports bra, white leggings, black sneakers, textures visible",
+	"lavender floral dress, white sunglasses, silver hoop earrings",
+	"sports bra, leggings, pastel green, form-fitting, sleeveless, high-waisted",
+	"white top, sheer overlay, light textures",
+	"white crop top, blue denim shorts, black and white sneakers, ribbed texture top",
+	"red crop top, blue jeans, light textures",
+	"blue sports bra, blue leggings, smooth texture",
+	"white crop top, high-waisted skirt, textured pink heels, black shoulder bag",
+	"striped swimsuit, blue and white, thin straps",
+	"white bikini top, white bikini bottoms, smooth texture",
+	"denim tube top, denim shorts, light blue, frayed hems, gold necklace, gold bracelet, navel piercing",
+	"black cap, black-framed glasses, red and black striped sweater, denim shorts, black belt, clear-lens glasses",
+	"crocheted bikini, white color, tassel details, tied straps",
+	"black halter dress, cut-out details, form-fitting, knee-length",
+	"white bikini, halter neck top, string bikini bottom, smooth texture",
+	"black dress, sleeveless, cut-out details, form-fitting",
+	"sports bra, leggings, pastel colors, form-fitting, athletic wear",
+	"blue and white striped top, front zipper, black shorts, denim texture",
+	"patterned dress, white with blue spots, short sleeves, v-neckline, textured fabric",
+	"blue bikini, white-orange socks, red-yellow roller skates",
+	"white long-sleeve top, distressed blue jeans, white sneakers",
+	"Black blazer, v-neckline, satin texture",
+	"white off-shoulder top, white bikini bottom, gold necklace, black sunglasses, textured fabrics",
+	"floral bikini, vibrant colors, textured fabric",
+	"green jumpsuit, cut-out torso, wide-leg pants, sleeveless top, flowy fabric",
+	"Pink dress, text patterns, sleeveless, high neckline, cut-out detail",
+	"black corset top, ripped blue jeans, white blazer, golden necklace, sunglasses on head",
+	"pink tank top, blue denim shorts, diamond choker necklace, brown patterned handbag",
+	"black corset top, patterned skirt, glossy texture",
+	"White halter top, blue denim jeans, gold bracelet, beige handbag",
+	"blue fringe haircut, pink sunglasses, pink denim jacket, white tank top, pink skirt, pink heart-shaped purse",
+	"pink crop top, pink shorts, ribbed texture, athletic wear",
+	"white off-shoulder top, wide sleeves, straw hat, textured fabric",
+	"denim jacket, black dress, white sneakers, blue backpack",
+	"white sports bra, black shorts, white sneakers, white socks",
+	"brown corset top, brown leather pants, glossy texture",
+	"brown cropped top, denim shorts, eyeglasses, wrist watch",
+	"white cropped top, blue denim jeans, smooth texture, frayed edges",
+	"light blue corset, denim jeans, silver necklace, black belt, light blue handbag",
+	"white cropped top, patterned green skirt, fabric texture visible, white undergarment",
+	"yellow crop top, blue denim shorts, white strapless top, tan belt, frayed hems",
+	"white crochet top, fringed detailing, denim jeans, tan shoulder bag",
+	"leopard print halter-top, blue ripped jeans",
+	"white long-sleeve top, plunging neckline, high-waisted jeans, distressed denim, blue color, tan ankle boots",
+	"black cut-out dress, gold bracelet, black and white heels, beige handbag",
+	"black crop top, deep neckline, long sleeves, denim shorts, frayed hem, black bralette, multiple straps, sheer sleeves, mesh pattern, choker necklace, high-waisted shorts, button closure",
+	"beige hoodie, matching sweatpants, visible drawstrings, soft texture",
+	"gray t-shirt, blue jeans, smooth texture, casual style",
+	"pink bikini top, pink skirt, sunglasses, gold watch",
+	"lace top, brown skirt, strapless design, textured fabric",
+	"strappy dress, beige color, form-fitting, textured fabric, sleeveless, knee-length",
+	"black t-shirt, rib cage print, white shorts",
+	"light blue hoodie, matching shorts, soft, cotton texture, hoodie drawstrings, exposed midriff",
+	"blue patterned top, blue patterned shorts, black glasses",
+	"black sequin dress, high neckline, sleeveless, side slit, strappy high heels, silver earrings",
+	"black sequined dress, high neckline, sleeveless, back slit, strappy high heels, metallic color",
+	"transparent patterned top, black bralette, glossy black pants, glittery sneakers",
+	"black eyeliner, silver hoop earrings, gray-scale makeup, camouflage long sleeve top, black crop top, glossy black pants, glittery silver sneakers",
+	"blue dress, short sleeves, shiny texture, watch, ring",
+	"white crop top, black headphones, black sunglasses",
+	"button-up dress, peach color, short sleeves, collar, thigh length, fabric texture",
+	"white crop top, yellow sweatpants, textured fabric",
+	"striped swimsuit, glasses",
+	"pink dress, short sleeves, crew neckline, ribbed texture",
+	"pink dress, tiered layers, spaghetti straps, v-neckline, pleated texture, silver watch",
+	"red sports bra, red leggings, smooth texture, athletic wear",
+	"light jacket, matching trousers, white sneakers, black bag",
+	"Yellow checkered bikini, white straps, smooth texture",
+	"white ribbed dress, yellow-tinted sunglasses",
+	"long-sleeve top, ribbed texture, brown color, matching skirt, high-waisted",
+	"black sports bra, striped sheer pants",
+	"golden bikini, textured fabric",
+	"white crop top, blue denim shorts, cow print boots, silver necklace",
+	"orange dress, ribbed texture, short sleeves, side cut-outs, tie-up details",
+	"red bikini top, red bikini bottom, ribbed texture, tied sides",
+	"black bikini, textured fabric",
+	"white corset top, floral pattern, denim jeans, corset laces",
+	"white tank top, grey sports bra, grey shorts, textured fabrics, visible branding",
+	"white ribbed top, front tie, long sleeves, black leather pants, high-waisted",
+	"brown hat, black tank top, white denim shorts, frayed shorts hem",
+	"black crop top, mesh sleeves, checkered mini skirt, black boots, textured fabrics",
+	"black sweatshirt, cartoon graphic, glasses, light shorts, casual style",
+	"crop top, olive green, ribbed texture, high-waisted leggings, olive green, fitted",
+	"black bikini, glossy texture, gold earrings",
+	"black v-neck top, black belt, black trousers, blue handbag, golden necklace",
+	"black crop top, long sleeves, high-waisted leggings, mesh panels, dark hues, white sneakers",
+	"white crop top, gray trousers, white sandals, black sunglasses, gold belt",
+	"gray cardigan, white crop top, leopard print shorts, blue hair",
+	"white v-neck top, knotted waist, light-colored pants, fitted texture",
+	"white crop top, grey shorts, white sneakers, text on top",
+	"black swimsuit, straw hat, textured fabric",
+	"off-shoulder top, white with stripes, casual shorts",
+	"white crop top, glitter texture, black leather jacket, black leather pants",
+	"black tank top, shoulder straps, scoop neckline",
+	"Sleeveless top, pink hue, denim shorts, ripped texture, sunglasses on head",
+	"black long-sleeve top, light blue jeans, silver belt, form-fitting, casual style",
+	"floral jumpsuit, red and beige, texture visible, high heels, tan-colored",
+	"red dress, sleeveless, V-neckline",
+	"dark sunglasses, white collared shirt, navy blue sweater, pleated white skirt, black shoulder bag with gold hardware",
+	"white tank top, textured fabric, large hoop earrings, headscarf with print",
+	"black swimsuit, red headband, sunglasses",
+	"black tank top, sheer fabric, patterned design, golden bracelet",
+	"camouflage bikini, white-brown-beige tones, high-waisted bottoms, sleeveless top, hoop earrings",
+	"patterned crop top, light colors, tie front, long sleeves, white pants, soft texture",
+	"green tank top, white shorts, black glasses",
+	"Red patterned top, deep neckline, short sleeves",
+	"white crop top, lace-up detail, textured skirt, cream color, sleeveless, midriff-baring",
+	"white bathrobe, textured fabric, glasses",
+	"white cropped top, blue trim, long sleeves, blue track pants",
+	"white crop top, pastel green leggings, textured fabric",
+	"spotted swimsuit, light-dark contrast, thin straps, skin exposure",
+	"black dress, double-breasted, v-neckline, sleeveless, silver necklace",
+	"black beanie, black cropped top, black leggings, black boots, red bag, gold watch",
+	"beige fedora, white cropped top, textured terracotta skirt",
+	"blue denim jeans, blue strapless top, texture appears soft",
+	"White lace dress, Off-shoulder sleeves, Corset-style top, Drawstring details",
+	"sleeveless top, beige color, ribbed texture, high neckline, light blue jeans, fitted style, denim material, wrist accessory, silver color, sparkling texture",
+	"white strapless top, white skirt, shiny fabric",
+	"light white dress, thin straps, sheer texture"
+]

data/female_default_tags.json

@@ -0,0 +1,6 @@
+[
+	"a woman",
+	"a young woman",
+	"a middle aged woman",
+	"an old woman"
+]

data/male_additional_details.json

@@ -0,0 +1,183 @@
+[
+	"a purple iridescent suit",
+	"wearing a (necklace)",
+	"wearing ((earrings))",
+	"wearing a (bracelet)",
+	"wearing one or multiple (rings)",
+	"wearing a (brooch)",
+	"wearing (eyeglasses)",
+	"wearing (sunglasses)",
+	"wearing a (hat)",
+	"wearing a (scarf)",
+	"wearing a (headband)",
+	"wearing a (nose ring)",
+	"wearing a (lip ring)",
+	"wearing a (tongue ring)",
+	"wearing an (eyebrow ring)",
+	"wearing (face tattoos)",
+	"wearing a (wreath)",
+	"wearing a (crown)",
+	"wearing a (tiara)",
+	"wearing a (crown of thorns)",
+	"wearing a (crown of jewels)",
+	"wearing (bohemian clothes)",
+	"wearing (chic clothes)",
+	"wearing (glamorous clothes)",
+	"wearing (grunge clothes)",
+	"wearing (preppy clothes)",
+	"wearing (punk clothes)",
+	"wearing (retro clothes)",
+	"wearing (rockabilly clothes)",
+	"wearing (romantic clothes)",
+	"wearing (tomboy clothes)",
+	"wearing (urban clothes)",
+	"wearing (camo clothes)",
+	"wearing (robes)",
+	"wearing (excessive amount of jewellery)",
+	"wearing (vintage clothes)",
+	"wearing (western clothes)",
+	"wearing (minimalist clothes)",
+	"wearing (sportswear clothes)",
+	"wearing (flapper clothes)",
+	"wearing (pin-up clothes)",
+	"wearing (mid-century modern clothes)",
+	"wearing (art deco clothes)",
+	"wearing (victorian clothes)",
+	"wearing (edwardian clothes)",
+	"wearing (elizabethan clothes)",
+	"wearing (retro 70s clothes)",
+	"wearing (retro 80s clothes)",
+	"wearing (retro 90s clothes)",
+	"wearing (retro 00s clothes)",
+	"wearing (musical equipment)",
+	"wearing (leather)",
+	"wearing (bdsm leather)",
+	"wearing (shiny latex)",
+	"wearing (shiny latex suit)",
+	"wearing (silk)",
+	"wearing (full tweed set)",
+	"wearing (clothes made entirely of feathers)",
+	"wearing (clothes made entirely of fur)",
+	"wearing (clothes made entirely of leather)",
+	"wearing (clothes made entirely of metal)",
+	"wearing (clothes made entirely of plastic)",
+	"wearing (clothes adorned with shimmering jewels or crystals)",
+	"waring (clothes adorned with sequins)",
+	"wearing (clothes with exaggerated or extreme silhouettes)",
+	"wearing (clothes with exaggerated or extreme footwear)",
+	"wearing (clothes with exaggerated or extreme headwear)",
+	"wearing (clothes with exaggerated or extreme facial or body piercings or tattoos)",
+	"wearing (clothes with multiple layers or tiers)",
+	"wearing (clothes with exaggerated or extreme colors)",
+	"wearing (clothes with exaggerated or extreme patterns)",
+	"wearing (cloak)",
+	"wearing an astronaut armor",
+	"wearing a bio mechanical suit",
+	"wearing a bio hazard suit",
+	"(( working with laptop))",
+	"with Heat deformation",
+	"(((future soldier, full body armor, futuristic football, shoulder pads, guns, grenades, weapons, bullet proof vest, high tech, straps, belts, camouflage)))",
+	"((full body, zoomed out)) long slender legs 80mm",
+	"(((sci-fi, future war, cyberpunk, cyborg, future fashion, beautiful face, glowing tattoos)))",
+	"((angry expression, pretty face))",
+	"(((full body, athletic body, action pose, detailed black soldier outfit, slender long legs)))",
+	"playing epic electric guitar solo in front of a huge crowd",
+	"singing epic solo into a microphone in front of a huge crowd",
+	"as a ((gelatinous horror dripping alien creature))",
+	"in a tie or bowtie, lending a touch of formal elegance or quirky charm, knotted around the collar to elevate the outfit",
+	"with anklets, delicate chains or beads that gracefully encircle the ankle, adding a touch of femininity with every step",
+	"donning a belt, functional yet fashionable, cinching the waist or sitting low on the hips, often with a statement buckle",
+	"wearing gloves, either elegant satin for formal events or rugged leather for a tougher look, complementing the attire and mood",
+	"with a choker, snugly encircling the neck, often made of lace, velvet, or leather, exuding a mix of elegance and edge",
+	"in stockings or tights, sheer or opaque, enhancing the legs while adding a touch of sophistication or playful patterns",
+	"with a satchel or bag, a functional accessory that speaks volumes about personal style, be it a minimalist tote or an embellished clutch",
+	"wearing cufflinks, subtle symbols of elegance, adorning the sleeves of a formal shirt, showcasing attention to detail",
+	"with a pendant, a piece of jewelry that dangles gracefully from a necklace, often holding sentimental or symbolic value",
+	"in layered necklaces, a blend of chains of varying lengths, creating depth and showcasing multiple pendants or charms",
+	"sporting a watch, a timeless accessory that blends functionality with style, either minimalist or grand, reflecting personal tastes",
+	"wearing a veil, a delicate piece of fabric that adds mystery and allure, often seen in bridal or ceremonial attire",
+	"donning a cape or cloak, adding drama to the ensemble, flowing gracefully with every movement, evoking a sense of fantasy or regality",
+	"with a tiara or diadem, a jeweled headpiece that signifies royalty or celebration, resting gracefully atop the head",
+	"Adorned with a crown, symbolizing royalty and authority with gemstones and metals",
+	"With a sparkling tiara, reminiscent of princesses or beauty queens",
+	"With a poignant crown of thorns, symbolizing sacrifice and resilience",
+	"With a jewel-encrusted crown, reflecting affluence and grandeur",
+	"In bohemian attire, embodying the free spirits with patterns and fringes",
+	"Dressed in chic fashion, blending comfort with high style",
+	"In glamorous attire, shiny or sequined, perfect for red-carpet events",
+	"Donning grunge wear, with flannels and combat boots, reflecting a rebellious spirit",
+	"In preppy clothes, with clean lines and classic patterns",
+	"Sporting punk fashion, with leather, studs, and bold hairstyles",
+	"In retro outfits, channeling specific decades with authentic pieces",
+	"Wearing rockabilly style, with pin-up influences and rock 'n' roll flair",
+	"In romantic attire, featuring soft fabrics and feminine silhouettes",
+	"Dressed in tomboy fashion, with boyish charm and comfort-focused pieces",
+	"In urban streetwear, blending comfort with trendy, city-inspired looks",
+	"Wearing camouflage, either for function or as a bold fashion statement",
+	"Draped in robes, from casual bathrobes to formal ceremonial garments",
+	"Adorned with excessive jewelry, creating a bold and opulent look",
+	"In vintage clothes, authentically representing past eras and styles",
+	"Sporting western wear, with cowboy hats, boots, and denim",
+	"In minimalist attire, focusing on clean lines and a less-is-more approach",
+	"Wearing sportswear, blending function and style for athletic pursuits",
+	"In flapper style, with dropped waists and ornate beading from the 1920s",
+	"Dressed as a pin-up, channeling retro glamour with curve-hugging styles",
+	"In mid-century modern fashion, reflecting the sleek designs of the 1950s and 60s",
+	"Wearing art deco inspired clothes, with geometric patterns and luxurious fabrics",
+	"In Victorian-style attire, with high necklines, corsets, and full skirts",
+	"Dressed in Edwardian fashion, featuring S-shaped silhouettes and large hats",
+	"In Elizabethan costume, with ruffs, farthingales, and ornate embroidery",
+	"Sporting 70s retro looks, with bell-bottoms, platforms, and psychedelic prints",
+	"In 80s inspired outfits, featuring bold colors, shoulder pads, and leg warmers",
+	"Wearing 90s fashion, with crop tops, high-waisted jeans, and chunky shoes",
+	"In early 2000s style, featuring low-rise jeans, crop tops, and platform sandals",
+	"With musical equipment, like guitars, drumsticks, or headphones as accessories",
+	"In leather attire, from sleek jackets to edgy pants or skirts",
+	"Wearing BDSM-inspired leather, with harnesses, collars, or studded accessories",
+	"In shiny latex, hugging every curve with a glossy, futuristic appeal",
+	"Sporting a full latex suit, covering from neck to toe in sleek, shiny material",
+	"Draped in silk, with flowing fabrics that catch the light beautifully",
+	"In a full tweed set, channeling British countryside chic",
+	"Wearing an outfit made entirely of feathers, creating a bold, avian-inspired look",
+	"In a fur ensemble, either faux or real, exuding luxury and warmth",
+	"Sporting an all-leather outfit, from jacket to pants and accessories",
+	"In a metallic outfit, with clothes seemingly forged from shining metals",
+	"Wearing an ensemble made entirely of plastic, futuristic and unconventional",
+	"Adorned in clothes shimmering with jewels or crystals, catching light at every angle",
+	"In a sequin-covered outfit, sparkling and eye-catching from every view",
+	"Wearing clothes with exaggerated silhouettes, pushing the boundaries of shape and form",
+	"In footwear with extreme designs, from towering platforms to avant-garde shapes",
+	"Sporting headwear with exaggerated proportions or unconventional materials",
+	"With extreme facial piercings or full-body tattoos, turning the skin into a canvas",
+	"In an outfit with multiple layers or tiers, creating depth and visual interest",
+	"Wearing clothes in extremely bold or neon colors, making a vibrant statement",
+	"In attire with exaggerated or surreal patterns, creating visual intrigue",
+	"Draped in a mysterious cloak, adding an air of intrigue or fantasy",
+	"In astronaut armor, representing space exploration and futuristic concepts",
+	"Wearing a bio-mechanical suit, blending organic forms with mechanical elements",
+	"In a biohazard suit, fully enclosed for protection or as a fashion statement",
+	"Engaged with a laptop, representing technology integration in fashion",
+	"Surrounded by heat deformation effects, as if the very air is bending around them",
+	"In futuristic soldier gear, with high-tech armor, weapons, and camouflage",
+	"Posed to accentuate long, slender legs, often with 80mm or higher heels",
+	"In sci-fi inspired wear, with cyberpunk elements and glowing tattoos",
+	"With an angry expression juxtaposed against delicate features",
+	"In a detailed black tactical outfit, emphasizing an athletic build and long legs",
+	"Mid-performance with an electric guitar, captivating an imaginary audience",
+	"Belting out a solo performance, microphone in hand, before a cheering crowd",
+	"Transformed into a gelatinous alien creature, dripping and otherworldly",
+	"With a tie or bowtie, adding a formal or quirky touch to the ensemble",
+	"Wearing delicate anklets, adding a subtle charm to the overall look",
+	"With a statement belt, either cinching the waist or riding low on the hips",
+	"In elegant or rugged gloves, complementing the outfit's overall mood",
+	"Sporting a choker, from delicate designs to edgy leather styles",
+	"In dress socks or patterned socks, adding a pop of color or sophistication",
+	"Carrying a stylish bag, from minimalist briefcases to trendy messenger bags",
+	"With cufflinks, adding a touch of sophistication to formal wear",
+	"Wearing a meaningful pendant, dangling from a necklace",
+	"In layered necklaces, creating depth with varying lengths and styles",
+	"Sporting a watch, balancing functionality with personal style",
+	"Wearing a ceremonial sash, adding formality and significance to the attire",
+	"Draped in a cape or cloak, bringing drama and movement to the ensemble",
+	"Crowned with a laurel wreath, symbolizing victory or achievement"
+]

data/male_body_types.json

@@ -0,0 +1,40 @@
+[
+    "chubby",
+    "midweight",
+    "overweight",
+    "fat",
+    "flabby",
+    "hefty",
+    "pudgy",
+    "plump",
+    "obese",
+    "morbidly obese",
+    "stout",
+    "rotund",
+    "thick-bodied",
+    "thicc",
+    "thick",
+    "beefy",
+    "portly",
+    "tubby",
+    "overweight",
+    "(slightly overweight)",
+    "buff",
+    "burly",
+    "fit",
+    "well-built",
+    "well-endowed",
+    "muscular",
+    "stocky",
+    "big-boned",
+    "flabby",
+    "flyweight",
+    "skinny",
+    "too skinny",
+    "anorexic",
+    "not skinny",
+    "slender",
+    "lanky",
+    "slim",
+    "slight"
+]

data/male_clothing.json

@@ -0,0 +1,196 @@
+[
+	"white t-shirt, blue jeans, brown leather belt, white sneakers",
+	"black suit, white dress shirt, red tie, polished black shoes",
+	"gray hoodie, black sweatpants, white athletic shoes",
+	"blue denim jacket, white t-shirt, black jeans, brown boots",
+	"khaki chinos, light blue button-up shirt, brown loafers",
+	"navy blue blazer, white dress shirt, gray slacks, black dress shoes",
+	"red and black plaid flannel shirt, dark wash jeans, work boots",
+	"black leather jacket, white t-shirt, dark jeans, black motorcycle boots",
+	"green cargo pants, black t-shirt, hiking boots",
+	"white polo shirt, khaki shorts, boat shoes",
+	"black tuxedo, white dress shirt, black bow tie, patent leather shoes",
+	"blue swim trunks, white tank top, flip flops",
+	"gray sweatshirt, black joggers, running shoes",
+	"brown tweed jacket, cream sweater, dark brown trousers, oxford shoes",
+	"black turtleneck, dark gray slacks, Chelsea boots",
+	"white linen shirt, light blue shorts, sandals",
+	"red basketball jersey, black shorts, high-top sneakers",
+	"navy blue pea coat, gray scarf, dark jeans, leather gloves",
+	"olive green military jacket, white t-shirt, camouflage pants, combat boots",
+	"purple v-neck sweater, light gray chinos, brown brogues",
+	"black wetsuit, surfboard shorts",
+	"white chef's jacket, black and white checkered pants, non-slip shoes",
+	"orange high-visibility vest, blue work shirt, denim jeans, steel-toe boots",
+	"light blue scrubs, white lab coat, comfortable shoes",
+	"black clergy shirt with white collar, black trousers",
+	"tan safari jacket, khaki shorts, hiking boots, wide-brimmed hat",
+	"red and white striped rugby shirt, white shorts, cleats",
+	"gray pinstripe suit, light blue dress shirt, patterned tie, oxford shoes",
+	"black leather vest, white t-shirt, ripped jeans, cowboy boots",
+	"yellow raincoat, waterproof pants, rubber boots",
+	"white martial arts gi, black belt",
+	"blue and white striped sailor shirt, white pants, boat shoes",
+	"green scrubs, surgical cap, face mask",
+	"black wetsuit, diving boots, gloves",
+	"red and black checkered lumberjack shirt, jeans, work boots",
+	"white cricket uniform, protective pads, spiked shoes",
+	"blue denim overalls, red plaid shirt, work boots",
+	"black judge's robe, white collar",
+	"tan fishing vest, khaki shorts, waterproof boots",
+	"white chef's hat, double-breasted chef's jacket, black pants",
+	"orange prison jumpsuit",
+	"blue letter carrier uniform, shorts, comfortable walking shoes",
+	"green surgical gown, cap, mask, gloves",
+	"black cassock, white collar",
+	"red and yellow firefighter turnout gear, helmet",
+	"blue police uniform, badge, utility belt",
+	"white karate uniform, colored belt",
+	"gray sweatshirt, matching sweatpants, running shoes",
+	"black leather punk jacket with studs, ripped jeans, combat boots",
+	"colorful Hawaiian shirt, khaki shorts, sandals",
+	"white dress shirt, suspenders, bow tie, black trousers",
+	"blue coveralls, work boots, hard hat",
+	"camouflage hunting jacket and pants, boots",
+	"black graduation gown, cap with tassel",
+	"red and white Santa Claus suit, black boots",
+	"green elf costume, pointy shoes, hat with bell",
+	"white space suit, helmet",
+	"blue jeans, flannel shirt, cowboy hat, boots",
+	"black ninja outfit, face mask",
+	"brown monk's robe, rope belt, sandals",
+	"colorful clown costume, oversized shoes, red nose",
+	"gray business suit, white shirt, tie, dress shoes",
+	"white sailor's uniform, hat",
+	"green army uniform, combat boots, beret",
+	"blue airline pilot uniform, hat",
+	"orange astronaut suit, white helmet",
+	"red and white striped Where's Waldo shirt, blue jeans, glasses",
+	"black and white referee uniform, whistle",
+	"blue mailman uniform, hat, mail bag",
+	"white lab coat, stethoscope, scrubs",
+	"brown UPS uniform, shorts, brown shoes",
+	"gray janitor uniform, name tag",
+	"black and white striped prisoner uniform",
+	"blue denim jacket, white t-shirt, ripped jeans, sneakers",
+	"green camouflage military uniform, boots, hat",
+	"red lifeguard swimsuit, whistle, sunglasses",
+	"white cricket uniform, protective gear",
+	"blue mechanic's coveralls, work boots, tool belt",
+	"black tuxedo t-shirt, jeans, sneakers",
+	"tan safari outfit, pith helmet",
+	"red and white candy striper uniform",
+	"blue air force uniform, hat",
+	"green park ranger uniform, hat",
+	"white naval officer uniform, hat",
+	"black and white waiter uniform, bow tie",
+	"orange construction worker vest, hard hat, jeans",
+	"purple wizard robe, pointed hat",
+	"brown Franciscan monk habit",
+	"red and gold matador outfit",
+	"blue train conductor uniform, hat",
+	"green Robin Hood costume, feathered cap",
+	"white fencing uniform, mask",
+	"black gothic outfit, leather pants, boots",
+	"red and white baseball uniform, cap",
+	"blue Olympic gymnast uniform",
+	"green Boy Scout uniform, badges",
+	"white martial arts gi, black belt",
+	"tan zookeeper uniform, khaki shorts",
+	"blue NASCAR driver uniform, helmet",
+	"red Canadian Mountie uniform, hat",
+	"black and white mime outfit, face paint",
+	"green leprechaun costume, top hat",
+	"white ice cream man uniform, hat",
+	"blue and white Greek evzone uniform, pom-pom shoes",
+	"red British Royal Guard uniform, bearskin hat",
+	"black ninja outfit, mask",
+	"white chef uniform, tall hat",
+	"brown cowboy outfit, hat, boots",
+	"blue and white sailor uniform, hat",
+	"green army camouflage, helmet",
+	"red firefighter uniform, helmet",
+	"black SWAT team uniform, bulletproof vest",
+	"white astronaut spacesuit, helmet",
+	"blue airline pilot uniform, cap",
+	"tan safari guide outfit, hat",
+	"purple and gold king costume, crown",
+	"gray business suit, red tie",
+	"blue denim overalls, plaid shirt",
+	"black leather biker jacket, jeans",
+	"white doctor's coat, stethoscope",
+	"green surgical scrubs, mask",
+	"red and white Santa Claus outfit",
+	"brown UPS delivery uniform",
+	"orange prisoner jumpsuit",
+	"blue postal worker uniform",
+	"black and white referee shirt, whistle",
+	"tan park ranger uniform, hat",
+	"white naval officer uniform, hat",
+	"green army dress uniform, medals",
+	"blue air force flight suit",
+	"red and white lifeguard outfit",
+	"black tuxedo, bow tie",
+	"white chef's jacket, checkered pants",
+	"blue mechanic's coveralls",
+	"green landscaper uniform, work boots",
+	"tan archaeologist outfit, hat",
+	"purple wizard costume, star-covered robe",
+	"red and gold circus ringmaster outfit",
+	"blue train conductor uniform, hat",
+	"white cricket player uniform",
+	"black gothic outfit, leather jacket",
+	"red and white racing driver suit",
+	"blue Olympic swimmer outfit",
+	"green environmental activist t-shirt, cargo pants",
+	"white karate uniform, black belt",
+	"tan wildlife photographer vest, khaki pants",
+	"blue NASCAR pit crew uniform",
+	"red Canadian Mountie uniform, hat",
+	"black and white mime costume",
+	"green St. Patrick's Day outfit, shamrock accessories",
+	"white ice cream vendor uniform",
+	"blue Greek fisherman outfit",
+	"red British Royal Guard uniform",
+	"black martial arts uniform",
+	"white sushi chef outfit",
+	"brown cowboy costume, hat and boots",
+	"blue sailor uniform, white hat",
+	"green camouflage hunting outfit",
+	"red firefighter turnout gear",
+	"black tactical police uniform",
+	"white space shuttle astronaut suit",
+	"blue commercial pilot uniform",
+	"tan desert explorer outfit",
+	"purple medieval jester costume",
+	"gray Wall Street businessman suit",
+	"blue denim farmer overalls",
+	"black rock band leather outfit",
+	"white lab researcher coat",
+	"green hospital scrubs",
+	"red holiday elf costume",
+	"brown delivery driver uniform",
+	"orange highway worker safety vest",
+	"blue meter reader uniform",
+	"black and white soccer referee jersey",
+	"tan zookeeper uniform",
+	"white cruise ship officer uniform",
+	"green military cadet uniform",
+	"blue coast guard uniform",
+	"red ski patrol jacket",
+	"black formal waiter uniform",
+	"white professional tennis outfit",
+	"blue auto mechanic uniform",
+	"green waste management worker uniform",
+	"tan construction site manager outfit",
+	"purple comic book superhero costume",
+	"red circus acrobat leotard",
+	"blue ship captain uniform",
+	"white cricket umpire outfit",
+	"black nightclub bouncer attire",
+	"red and white sports mascot costume",
+	"blue competitive swimmer outfit",
+	"green eco-tour guide uniform",
+	"white Taekwondo uniform",
+	"tan nature documentary filmmaker outfit"
+]

data/male_default_tags.json

@@ -0,0 +1,6 @@
+[
+	"a man",
+	"a young man",
+	"a middle aged man",
+	"an old man"
+]

prompt.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import random
 import json
@@ -10,14 +11,13 @@ import torch
 from PIL import Image
 from transformers import AutoProcessor, AutoModelForCausalLM
 import random
-import spaces
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 # Initialize Florence model
 device = "cuda" if torch.cuda.is_available() else "cpu"
-florence_model = AutoModelForCausalLM.from_pretrained('gokaygokay/Florence-2-Flux', trust_remote_code=True).to("cpu").eval()
-florence_processor = AutoProcessor.from_pretrained('gokaygokay/Florence-2-Flux', trust_remote_code=True)
+florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True).to("cpu").eval()
+florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', trust_remote_code=True)
 
 # Florence caption function
 @spaces.GPU(duration=30)
@@ -50,24 +50,31 @@ def load_json_file(file_name):
     with open(file_path, "r") as file:
         return json.load(file)
 
+# Load gender-specific JSON files
+FEMALE_DEFAULT_TAGS = load_json_file("female_default_tags.json")
+MALE_DEFAULT_TAGS = load_json_file("male_default_tags.json")
+FEMALE_BODY_TYPES = load_json_file("female_body_types.json")
+MALE_BODY_TYPES = load_json_file("male_body_types.json")
+FEMALE_CLOTHING = load_json_file("female_clothing.json")
+MALE_CLOTHING = load_json_file("male_clothing.json")
+FEMALE_ADDITIONAL_DETAILS = load_json_file("female_additional_details.json")
+MALE_ADDITIONAL_DETAILS = load_json_file("male_additional_details.json")
+
+# Load non-gender-specific JSON files
 ARTFORM = load_json_file("artform.json")
 PHOTO_TYPE = load_json_file("photo_type.json")
-BODY_TYPES = load_json_file("body_types.json")
-DEFAULT_TAGS = load_json_file("default_tags.json")
 ROLES = load_json_file("roles.json")
 HAIRSTYLES = load_json_file("hairstyles.json")
-ADDITIONAL_DETAILS = load_json_file("additional_details.json")
-PHOTOGRAPHY_STYLES = load_json_file("photography_styles.json")
-DEVICE = load_json_file("device.json")
-PHOTOGRAPHER = load_json_file("photographer.json")
-ARTIST = load_json_file("artist.json")
-DIGITAL_ARTFORM = load_json_file("digital_artform.json")
 PLACE = load_json_file("place.json")
 LIGHTING = load_json_file("lighting.json")
-CLOTHING = load_json_file("clothing.json")
 COMPOSITION = load_json_file("composition.json")
 POSE = load_json_file("pose.json")
 BACKGROUND = load_json_file("background.json")
+PHOTOGRAPHY_STYLES = load_json_file("photography_styles.json")
+DEVICE = load_json_file("device.json")
+PHOTOGRAPHER = load_json_file("photographer.json")
+ARTIST = load_json_file("artist.json")
+DIGITAL_ARTFORM = load_json_file("digital_artform.json")
 
 class PromptGenerator:
     def __init__(self, seed=None):
@@ -88,12 +95,12 @@ class PromptGenerator:
             return input_str
 
     def clean_consecutive_commas(self, input_string):
-        cleaned_string = re.sub(r',\s*,', ',', input_string)
+        cleaned_string = re.sub(r',\s*,', ', ', input_string)
         return cleaned_string
 
     def process_string(self, replaced, seed):
-        replaced = re.sub(r'\s*,\s*', ',', replaced)
-        replaced = re.sub(r',+', ',', replaced)
+        replaced = re.sub(r'\s*,\s*', ', ', replaced)
+        replaced = re.sub(r',+', ', ', replaced)
         original = replaced
         
         first_break_clipl_index = replaced.find("BREAK_CLIPL")
@@ -119,24 +126,31 @@ class PromptGenerator:
         t5xxl = replaced
         
         original = original.replace("BREAK_CLIPL", "").replace("BREAK_CLIPG", "")
-        original = re.sub(r'\s*,\s*', ',', original)
-        original = re.sub(r',+', ',', original)
-        clip_l = re.sub(r'\s*,\s*', ',', clip_l)
-        clip_l = re.sub(r',+', ',', clip_l)
-        clip_g = re.sub(r'\s*,\s*', ',', clip_g)
-        clip_g = re.sub(r',+', ',', clip_g)
-        if clip_l.startswith(","):
-            clip_l = clip_l[1:]
-        if clip_g.startswith(","):
-            clip_g = clip_g[1:]
-        if original.startswith(","):
-            original = original[1:]
-        if t5xxl.startswith(","):
-            t5xxl = t5xxl[1:]
+        original = re.sub(r'\s*,\s*', ', ', original)
+        original = re.sub(r',+', ', ', original)
+        clip_l = re.sub(r'\s*,\s*', ', ', clip_l)
+        clip_l = re.sub(r',+', ', ', clip_l)
+        clip_g = re.sub(r'\s*,\s*', ', ', clip_g)
+        clip_g = re.sub(r',+', ', ', clip_g)
+        if clip_l.startswith(", "):
+            clip_l = clip_l[2:]
+        if clip_g.startswith(", "):
+            clip_g = clip_g[2:]
+        if original.startswith(", "):
+            original = original[2:]
+        if t5xxl.startswith(", "):
+            t5xxl = t5xxl[2:]
+
+        # Add spaces after commas
+        replaced = re.sub(r',(?!\s)', ', ', replaced)
+        original = re.sub(r',(?!\s)', ', ', original)
+        clip_l = re.sub(r',(?!\s)', ', ', clip_l)
+        clip_g = re.sub(r',(?!\s)', ', ', clip_g)
+        t5xxl = re.sub(r',(?!\s)', ', ', t5xxl)
 
         return original, seed, t5xxl, clip_l, clip_g
 
-    def generate_prompt(self, seed, custom, subject, artform, photo_type, body_types, default_tags, roles, hairstyles,
+    def generate_prompt(self, seed, custom, subject, gender, artform, photo_type, body_types, default_tags, roles, hairstyles,
                         additional_details, photography_styles, device, photographer, artist, digital_artform,
                         place, lighting, clothing, composition, pose, background, input_image):
         kwargs = locals()
@@ -155,6 +169,7 @@ class PromptGenerator:
         )
 
         subject = kwargs.get("subject", "")
+        gender = kwargs.get("gender", "female")
 
         if is_photographer:
             selected_photo_style = self.get_choice(kwargs.get("photography_styles", ""), PHOTOGRAPHY_STYLES)
@@ -169,18 +184,18 @@ class PromptGenerator:
         if not subject:
             if default_tags == "random":
                 if body_type != "disabled" and body_type != "random":
-                    selected_subject = self.get_choice(kwargs.get("default_tags", ""), DEFAULT_TAGS).replace("a ", "").replace("an ", "")
+                    selected_subject = self.get_choice(kwargs.get("default_tags", ""), FEMALE_DEFAULT_TAGS if gender == "female" else MALE_DEFAULT_TAGS).replace("a ", "").replace("an ", "")
                     components.append("a ")
                     components.append(body_type)
                     components.append(selected_subject)
                 elif body_type == "disabled":
-                    selected_subject = self.get_choice(kwargs.get("default_tags", ""), DEFAULT_TAGS)
+                    selected_subject = self.get_choice(kwargs.get("default_tags", ""), FEMALE_DEFAULT_TAGS if gender == "female" else MALE_DEFAULT_TAGS)
                     components.append(selected_subject)
                 else:
-                    body_type = self.get_choice(body_type, BODY_TYPES)
+                    body_type = self.get_choice(body_type, FEMALE_BODY_TYPES if gender == "female" else MALE_BODY_TYPES)
                     components.append("a ")
                     components.append(body_type)
-                    selected_subject = self.get_choice(kwargs.get("default_tags", ""), DEFAULT_TAGS).replace("a ", "").replace("an ", "")
+                    selected_subject = self.get_choice(kwargs.get("default_tags", ""), FEMALE_DEFAULT_TAGS if gender == "female" else MALE_DEFAULT_TAGS).replace("a ", "").replace("an ", "")
                     components.append(selected_subject)
             elif default_tags == "disabled":
                 pass
@@ -193,7 +208,7 @@ class PromptGenerator:
             elif body_type == "disabled":
                 pass
             else:
-                body_type = self.get_choice(body_type, BODY_TYPES)
+                body_type = self.get_choice(body_type, FEMALE_BODY_TYPES if gender == "female" else MALE_BODY_TYPES)
                 components.append("a ")
                 components.append(body_type)
             components.append(subject)
@@ -201,13 +216,13 @@ class PromptGenerator:
         params = [
             ("roles", ROLES),
             ("hairstyles", HAIRSTYLES),
-            ("additional_details", ADDITIONAL_DETAILS),
+            ("additional_details", FEMALE_ADDITIONAL_DETAILS if gender == "female" else MALE_ADDITIONAL_DETAILS),
         ]
         for param in params:
             components.append(self.get_choice(kwargs.get(param[0], ""), param[1]))
         for i in reversed(range(len(components))):
             if components[i] in PLACE:
-                components[i] += ","
+                components[i] += ", "
                 break
         if kwargs.get("clothing", "") != "disabled" and kwargs.get("clothing", "") != "random":
             components.append(", dressed in ")
@@ -215,49 +230,49 @@ class PromptGenerator:
             components.append(clothing)
         elif kwargs.get("clothing", "") == "random":
             components.append(", dressed in ")
-            clothing = self.get_choice(kwargs.get("clothing", ""), CLOTHING)
+            clothing = self.get_choice(kwargs.get("clothing", ""), FEMALE_CLOTHING if gender == "female" else MALE_CLOTHING)
             components.append(clothing)
 
         if kwargs.get("composition", "") != "disabled" and kwargs.get("composition", "") != "random":
-            components.append(",")
+            components.append(", ")
             composition = kwargs.get("composition", "")
             components.append(composition)
         elif kwargs.get("composition", "") == "random": 
-            components.append(",")
+            components.append(", ")
             composition = self.get_choice(kwargs.get("composition", ""), COMPOSITION)
             components.append(composition)
         
         if kwargs.get("pose", "") != "disabled" and kwargs.get("pose", "") != "random":
-            components.append(",")
+            components.append(", ")
             pose = kwargs.get("pose", "")
             components.append(pose)
         elif kwargs.get("pose", "") == "random":
-            components.append(",")
+            components.append(", ")
             pose = self.get_choice(kwargs.get("pose", ""), POSE)
             components.append(pose)
         components.append("BREAK_CLIPG")
         if kwargs.get("background", "") != "disabled" and kwargs.get("background", "") != "random":
-            components.append(",")
+            components.append(", ")
             background = kwargs.get("background", "")
             components.append(background)
         elif kwargs.get("background", "") == "random": 
-            components.append(",")
+            components.append(", ")
             background = self.get_choice(kwargs.get("background", ""), BACKGROUND)
             components.append(background)
 
         if kwargs.get("place", "") != "disabled" and kwargs.get("place", "") != "random":
-            components.append(",")
+            components.append(", ")
             place = kwargs.get("place", "")
             components.append(place)
         elif kwargs.get("place", "") == "random": 
-            components.append(",")
+            components.append(", ")
             place = self.get_choice(kwargs.get("place", ""), PLACE)
-            components.append(place + ",")
+            components.append(place + ", ")
 
         lighting = kwargs.get("lighting", "").lower()
         if lighting == "random":
             selected_lighting = ", ".join(self.rng.sample(LIGHTING, self.rng.randint(2, 5)))
-            components.append(",")
+            components.append(", ")
             components.append(selected_lighting)
         elif lighting == "disabled":
             pass
@@ -302,14 +317,15 @@ class PromptGenerator:
             return f"{prompt}, {caption}"
         return prompt
 
+import os
+from openai import OpenAI
+
 class HuggingFaceInferenceNode:
     def __init__(self):
-        self.clients = {
-            "Mixtral": InferenceClient("NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO"),
-            "Mistral": InferenceClient("mistralai/Mistral-7B-Instruct-v0.3"),
-            "Llama 3": InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct"),
-            "Mistral-Nemo": InferenceClient("mistralai/Mistral-Nemo-Instruct-2407")
-        }
+        self.client = OpenAI(
+            base_url="https://api-inference.huggingface.co/v1/",
+            #api_key=huggingface_token,
+        )
         self.prompts_dir = "./prompts"
         os.makedirs(self.prompts_dir, exist_ok=True)
 
@@ -326,10 +342,8 @@ class HuggingFaceInferenceNode:
         
         print(f"Prompt saved to {filename}")
 
-    def generate(self, model, input_text, happy_talk, compress, compression_level, poster, custom_base_prompt=""):
+    def generate(self, input_text, happy_talk, compress, compression_level, poster, custom_base_prompt=""):
         try:
-            client = self.clients[model]
-
             default_happy_prompt = """Create a detailed visually descriptive caption of this description, which will be used as a prompt for a text to image AI system (caption only, no instructions like "create an image").Remove any mention of digital artwork or artwork style. Give detailed visual descriptions of the character(s), including ethnicity, skin tone, expression etc. Imagine using keywords for a still for someone who has aphantasia. Describe the image style, e.g. any photographic or art styles / techniques utilized. Make sure to fully describe all aspects of the cinematography, with abundant technical details and visual descriptions. If there is more than one image, combine the elements and characters from all of the images creatively into a single cohesive composition with a single background, inventing an interaction between the characters. Be creative in combining the characters into a single cohesive scene. Focus on two primary characters (or one) and describe an interesting interaction between them, such as a hug, a kiss, a fight, giving an object, an emotional reaction / interaction. If there is more than one background in the images, pick the most appropriate one. Your output is only the caption itself, no comments or extra formatting. The caption is in a single long paragraph. If you feel the images are inappropriate, invent a new scene / characters inspired by these. Additionally, incorporate a specific movie director's visual style and describe the lighting setup in detail, including the type, color, and placement of light sources to create the desired mood and atmosphere. Always frame the scene, including details about the film grain, color grading, and any artifacts or characteristics specific."""
 
             default_simple_prompt = """Create a brief, straightforward caption for this description, suitable for a text-to-image AI system. Focus on the main elements, key characters, and overall scene without elaborate details. Provide a clear and concise description in one or two sentences."""
@@ -360,23 +374,24 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
                 char_limit = compression_chars[compression_level]
                 base_prompt += f" Compress the output to be concise while retaining key visual details. MAX OUTPUT SIZE no more than {char_limit} characters."
 
-            messages = f"<|im_start|>system\nYou are a helpful assistant. Try your best to give best response possible to user.<|im_end|>"
-            messages += f"\n<|im_start|>user\n{base_prompt}\nDescription: {input_text}<|im_end|>\n<|im_start|>assistant\n"
-
-            stream = client.text_generation(messages, max_new_tokens=4000, do_sample=True, stream=True, details=True, return_full_text=False)
-            output = ""
-            for response in stream:
-                if not response.token.text == "<|im_end|>":
-                    output += response.token.text
-
-            # Remove specific tokens based on the model
-            if model == "Llama 3":
-                output = output.rstrip("<|eot_id|>")
-            elif model == "Mistral":
-                output = output.rstrip("</s>")
-            elif model == "Mistral-Nemo":
-                output = output.rstrip("<|im_end|></s>")
-            
+            system_message = "You are a helpful assistant. Try your best to give the best response possible to the user."
+            user_message = f"{base_prompt}\nDescription: {input_text}"
+
+            messages = [
+                {"role": "system", "content": system_message},
+                {"role": "user", "content": user_message}
+            ]
+
+            response = self.client.chat.completions.create(
+                model="meta-llama/Meta-Llama-3.1-70B-Instruct",
+                max_tokens=1024,
+                temperature=0.7,
+                top_p=0.95,
+                messages=messages,
+            )
+
+            output = response.choices[0].message.content.strip()
+
             # Clean up the output
             if ": " in output:
                 output = output.split(": ", 1)[1].strip()
@@ -549,6 +564,160 @@ def create_interface():
 
     return demo
 
+if __name__ == "__main__":
+    demo = create_interface()
+    demo.launch()
+
+
+def create_interface():
+    prompt_generator = PromptGenerator()
+    huggingface_node = HuggingFaceInferenceNode()
+
+    with gr.Blocks(theme='bethecloud/storj_theme') as demo:
+        
+        gr.HTML(pg_title)
+
+        with gr.Row():
+            with gr.Column(scale=2):
+                with gr.Accordion("Basic Settings"):
+                    pg_custom = gr.Textbox(label="Custom Input Prompt (optional)")
+                    pg_subject = gr.Textbox(label="Subject (optional)")
+                    pg_gender = gr.Radio(["female", "male"], label="Gender", value="female")
+                    
+                    # Add the radio button for global option selection
+                    pg_global_option = gr.Radio(
+                        ["Disabled", "Random", "No Figure Rand"],
+                        label="Set all options to:",
+                        value="Disabled"
+                    )
+                
+                with gr.Accordion("Artform and Photo Type", open=False):
+                    pg_artform = gr.Dropdown(["disabled", "random"] + ARTFORM, label="Artform", value="disabled")
+                    pg_photo_type = gr.Dropdown(["disabled", "random"] + PHOTO_TYPE, label="Photo Type", value="disabled")
+            
+                with gr.Accordion("Character Details", open=False):
+                    pg_body_types = gr.Dropdown(["disabled", "random"] + FEMALE_BODY_TYPES + MALE_BODY_TYPES, label="Body Types", value="disabled")
+                    pg_default_tags = gr.Dropdown(["disabled", "random"] + FEMALE_DEFAULT_TAGS + MALE_DEFAULT_TAGS, label="Default Tags", value="disabled")
+                    pg_roles = gr.Dropdown(["disabled", "random"] + ROLES, label="Roles", value="disabled")
+                    pg_hairstyles = gr.Dropdown(["disabled", "random"] + HAIRSTYLES, label="Hairstyles", value="disabled")
+                    pg_clothing = gr.Dropdown(["disabled", "random"] + FEMALE_CLOTHING + MALE_CLOTHING, label="Clothing", value="disabled")
+            
+                with gr.Accordion("Scene Details", open=False):
+                    pg_place = gr.Dropdown(["disabled", "random"] + PLACE, label="Place", value="disabled")
+                    pg_lighting = gr.Dropdown(["disabled", "random"] + LIGHTING, label="Lighting", value="disabled")
+                    pg_composition = gr.Dropdown(["disabled", "random"] + COMPOSITION, label="Composition", value="disabled")
+                    pg_pose = gr.Dropdown(["disabled", "random"] + POSE, label="Pose", value="disabled")
+                    pg_background = gr.Dropdown(["disabled", "random"] + BACKGROUND, label="Background", value="disabled")
+            
+                with gr.Accordion("Style and Artist", open=False):
+                    pg_additional_details = gr.Dropdown(["disabled", "random"] + FEMALE_ADDITIONAL_DETAILS + MALE_ADDITIONAL_DETAILS, label="Additional Details", value="disabled")
+                    pg_photography_styles = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHY_STYLES, label="Photography Styles", value="disabled")
+                    pg_device = gr.Dropdown(["disabled", "random"] + DEVICE, label="Device", value="disabled")
+                    pg_photographer = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHER, label="Photographer", value="disabled")
+                    pg_artist = gr.Dropdown(["disabled", "random"] + ARTIST, label="Artist", value="disabled")
+                    pg_digital_artform = gr.Dropdown(["disabled", "random"] + DIGITAL_ARTFORM, label="Digital Artform", value="disabled")
+                
+                pg_generate_button = gr.Button("Generate Prompt")
+
+            with gr.Column(scale=2):
+                with gr.Accordion("Image and Caption", open=False):
+                    pg_input_image = gr.Image(label="Input Image (optional)")
+                    pg_caption_output = gr.Textbox(label="Generated Caption", lines=3)
+                    pg_create_caption_button = gr.Button("Create Caption")
+                    pg_add_caption_button = gr.Button("Add Caption to Prompt")
+
+                with gr.Accordion("Prompt Generation", open=True):
+                    pg_output = gr.Textbox(label="Generated Prompt / Input Text", lines=4)
+                    pg_t5xxl_output = gr.Textbox(label="T5XXL Output", visible=True)
+                    pg_clip_l_output = gr.Textbox(label="CLIP L Output", visible=True)
+                    pg_clip_g_output = gr.Textbox(label="CLIP G Output", visible=True)
+            
+            with gr.Column(scale=2):
+                with gr.Accordion("Prompt Generation with LLM", open=False):
+                    pg_happy_talk = gr.Checkbox(label="Happy Talk", value=True)
+                    pg_compress = gr.Checkbox(label="Compress", value=True)
+                    pg_compression_level = gr.Radio(["soft", "medium", "hard"], label="Compression Level", value="hard")
+                    pg_poster = gr.Checkbox(label="Poster", value=False)
+                    pg_custom_base_prompt = gr.Textbox(label="Custom Base Prompt", lines=5)
+                pg_generate_text_button = gr.Button("Generate Prompt with LLM (Llama 3.1 70B)")
+                pg_text_output = gr.Textbox(label="Generated Text", lines=10)
+
+        def create_caption(image):
+            if image is not None:
+                return florence_caption(image)
+            return ""
+
+        pg_create_caption_button.click(
+            create_caption,
+            inputs=[pg_input_image],
+            outputs=[pg_caption_output]
+        )
+
+        def generate_prompt_with_dynamic_seed(*args):
+            # Generate a new random seed
+            dynamic_seed = random.randint(0, 1000000)
+            
+            # Call the generate_prompt function with the dynamic seed
+            result = prompt_generator.generate_prompt(dynamic_seed, *args)
+            
+            # Return the result along with the used seed
+            return [dynamic_seed] + list(result)
+
+        pg_generate_button.click(
+            generate_prompt_with_dynamic_seed,
+            inputs=[pg_custom, pg_subject, pg_gender, pg_artform, pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles,
+                    pg_additional_details, pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform,
+                    pg_place, pg_lighting, pg_clothing, pg_composition, pg_pose, pg_background, pg_input_image],
+            outputs=[gr.Number(label="Used Seed", visible=True), pg_output, gr.Number(visible=False), pg_t5xxl_output, pg_clip_l_output, pg_clip_g_output]
+        )
+
+        pg_add_caption_button.click(
+            prompt_generator.add_caption_to_prompt,
+            inputs=[pg_output, pg_caption_output],
+            outputs=[pg_output]
+        )
+
+        pg_generate_text_button.click(
+            huggingface_node.generate,
+            inputs=[pg_output, pg_happy_talk, pg_compress, pg_ompression_level, pg_poster, pg_custom_base_prompt],
+            outputs=pg_text_output
+        )
+
+        def update_all_options(choice):
+            updates = {}
+            if choice == "Disabled":
+                for dropdown in [
+                    pg_artform, pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles, pg_clothing,
+                    pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
+                    pg_photography_styles, device, pg_photographer, pg_artist, pg_digital_artform
+                ]:
+                    updates[dropdown] = gr.update(value="disabled")
+            elif choice == "Random":
+                for dropdown in [
+                    pg_artform, pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles, pg_clothing,
+                    pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
+                    pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
+                ]:
+                    updates[dropdown] = gr.update(value="random")
+            else:  # No Figure Random
+                for dropdown in [pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles, pg_clothing, pg_pose, pg_additional_details]:
+                    updates[dropdown] = gr.update(value="disabled")
+                for dropdown in [pg_artform, pg_place, pg_lighting, pg_composition, pg_background, pg_photography_styles, device, pg_photographer, pg_artist, pg_digital_artform]:
+                    updates[dropdown] = gr.update(value="random")
+            return updates
+        
+        pg_global_option.change(
+            update_all_options,
+            inputs=[pg_global_option],
+            outputs=[
+                pg_artform, pg_photo_type, pg_body_types, pg_default_tags, pg_roles, pg_hairstyles, pg_clothing,
+                pg_place, pg_lighting, pg_composition, pg_pose, pg_background, pg_additional_details,
+                pg_photography_styles, pg_device, pg_photographer, pg_artist, pg_digital_artform
+            ]
+        )
+
+    return demo
+
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch()

requirements.txt

@@ -14,3 +14,4 @@ kornia
 numpy
 opencv-python
 deepspeed
+openai==1.37.0