components/induce_personality.py

import itertools

personality_types = [["extroverted", "introverted"], ["agreeable", "antagonistic"], ["conscientious", "unconscientious"], ["neurotic", "emotionally stable"], ["open to experience", "closed to experience"]]


def construct_big_five_words(persona_type: list):
    """Construct the list of personality traits

    e.g., introverted + antagonistic + conscientious + emotionally stable + open to experience
    """
    options = list(persona_type)
    assert options[0] in ["extroverted", "introverted"], "Invalid personality type"
    assert options[1] in ["agreeable", "antagonistic"], "Invalid personality type"
    assert options[2] in ["conscientious", "unconscientious"], "Invalid personality type"
    assert options[3] in ["neurotic", "emotionally stable"], "Invalid personality type"
    assert options[4] in ["open to experience", "closed to experience"], "Invalid personality type"
    last_item = "and " + options[-1]
    options[-1] = last_item
    return ", ".join(options)

def build_personality_prompt(persona_type: list):
    return "You are a character who is {}.".format(construct_big_five_words(persona_type))



if __name__ == "__main__":
    count = 0
    for persona_type in itertools.product(*personality_types):
        system_prompt = "You are a character who is {}.".format(construct_big_five_words(persona_type))
        print(system_prompt)
        print("\n")
        count += 1
        if count == 5:
            break