dream

1
import statistics
2
import logging
3

4
logging.basicConfig(format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.DEBUG)
5
logger = logging.getLogger(__name__)
6

7

8
def is_introvert(dialog):
9
    is_extravert_across_five_turns = []
10

11
    # we need to get five first turns
12
    human_utterances = dialog["human_utterances"]
13

14
    first_5 = human_utterances[:5]
15

16
    for human_utterance in first_5:
17
        user_uttr_annotations = human_utterance["annotations"]
18

19
        # Extraversion (outgoing/energetic vs. solitary/reserved)
20
        # is_extravert = -1
21
        # Neuroticism (sensitive/nervous vs. secure/confident)
22
        # is_neu = -1
23
        # Extraversion (outgoing/energetic vs. solitary/reserved)
24
        # is_agr = -1
25
        # Conscientiousness (efficient/organized vs. easy-going/careless)
26
        # is_con = -1
27
        # Openness to experience (inventive/curious vs. consistent/cautious)
28
        # is_opn = -1
29

30
        personality_detection = user_uttr_annotations.get("personality_detection", {})
31

32
        logger.info(f"personality_detection: {personality_detection}")
33

34
        if len(personality_detection) == 5:
35
            is_extravert = personality_detection[0]
36

37
            is_extravert_across_five_turns.append(is_extravert)
38

39
    is_extravert_across_five_turns_median = statistics.median(is_extravert_across_five_turns)
40

41
    logger.info(f"is_extravert (across first five turns): {is_extravert_across_five_turns_median}")
42

43
    if is_extravert_across_five_turns_median > 0.8:
44
        return True
45

46
    return False
47