generate_samples.py

def generate(prompt, filename):
    # Simulate the generation process (you can replace this with your actual function)
    print(f"Generating '{prompt}' and saving as '{filename}'")

# Options for each category
expressions = ["happy", "sad", "crying", "angry", "open mouth"]
genders = ["male", "female", "non-binary"]
ages = ["kid", "teen", "young adult", "adult", "old"]
ethnicities = ["Caucasian", "Hispanic", "Black", "Middle-Eastern", "South Asian", "East Asian"]
styles = ["realistic", "cartoon", "comics"]
frames = ["portrait", "upper body", "full body", "profile"]

# Iterate through all combinations
for expression in expressions:
    for gender in genders:
        for age in ages:
            for ethnicity in ethnicities:
                for style in styles:
                    for frame in frames:
                        prompt = f"{expression}, {gender}, {age}, {ethnicity}, {style}, {frame}"
                        filename = f"{expression}_{gender}_{age}_{ethnicity}_{style}_{frame}.jpg"
                        generate(prompt, filename)