context = ["Joe: I just watched "The Book of Henry" and it was really touching.", 
"Chicken: What's it about?",
"Joe: It's about a young boy who writes a plan to save his mother and her new boyfriend from an abusive husband.",
"Chicken: That sounds like a sad movie.",
"Joe: It is, but it's also hopeful. The boy's plan works and he saves his family.",
"Chicken: I'll have to check it out."
]


and cstring = """
Joe: I just watched "The Book of Henry" and it was really touching.
Chicken: What's it about?
Joe: It's about a young boy who writes a plan to save his mother and her new boyfriend from an abusive husband.
Chicken: That sounds like a sad movie.
Joe: It is, but it's also hopeful. The boy's plan works and he saves his family.
Chicken: I'll have to check it out.
"""

context = ["Joe: I just watched "The Book of Henry" and it was really touching.", 
"Chicken: What's it about?",
"Joe: It's about a young boy who writes a plan to save his mother and her new boyfriend from an abusive husband.",
"Chicken: That sounds like a sad movie.",
"Joe: It is, but it's also hopeful. The boy's plan works and he saves his family.",
"Chicken: I'll have to check it out."
]


and cstring = """
Joe: I just watched "The Book of Henry" and it was really touching.
Chicken: What's it about?
Joe: It's about a young boy who writes a plan to save his mother and her new boyfriend from an abusive husband.
Chicken: That sounds like a sad movie.
Joe: It is, but it's also hopeful. The boy's plan works and he saves his family.
Chicken: I'll have to check it out.
"""

sess_dict = sessionized.to_dict('records')
items = []
counter = 0
for row in sess_dict:
    context = []
    cstring = ''
    for i in range(10,0,-1):
        try:
            if sess_dict[counter-i]['chat_session_id'] == row['chat_session_id']:
                msg = f"{sess_dict[counter-i]['sender']}: {sess_dict[counter-i]['text']}"
                if len(context) > 0:
                    cstring += '\n'
                context.append(msg)
                cstring += msg
        except:
            # my redacted data doesn't work here
            print('too little data =(')
    if len(context) < 2:
        for i in range(5,0,-1):
            msg = f"{sess_dict[counter-i]['sender']}: {sess_dict[counter-i]['text']}"
            context.append(msg)
            cstring += '\n'
            cstring += msg
    items.append(cstring)
    counter+= 1

sess_dict = sessionized.to_dict('records')
items = []
counter = 0
for row in sess_dict:
    context = []
    cstring = ''
    for i in range(10,0,-1):
        try:
            if sess_dict[counter-i]['chat_session_id'] == row['chat_session_id']:
                msg = f"{sess_dict[counter-i]['sender']}: {sess_dict[counter-i]['text']}"
                if len(context) > 0:
                    cstring += '\n'
                context.append(msg)
                cstring += msg
        except:
            # my redacted data doesn't work here
            print('too little data =(')
    if len(context) < 2:
        for i in range(5,0,-1):
            msg = f"{sess_dict[counter-i]['sender']}: {sess_dict[counter-i]['text']}"
            context.append(msg)
            cstring += '\n'
            cstring += msg
    items.append(cstring)
    counter+= 1