import json

path = ''

chat_history = json.load(open(path + 'chat_history.json'))

received_saved = chat_history['Received Saved Chat History']

sent_saved = chat_history['Sent Saved Chat History']

conversations = {}

def clean_message_in_chat(chat):
    message = chat['Text']
    cleaned = message.replace("'", "'")
    chat['Text'] = cleaned
    return chat

for received in received_saved:
    if (received['Media Type'] != 'MEDIA'):
        chat = received['From']
        cleaned_chat = clean_message_in_chat(received)
        if chat in conversations:
            conversations[chat]['messages'] += [cleaned_chat]
        else:
            conversations[chat] = {
                'messages': [cleaned_chat]
            }

for sent in sent_saved:
    if (sent['Media Type'] != 'MEDIA'):
        chat = sent['To']
        cleaned_chat = clean_message_in_chat(sent)
        if chat in conversations:
            conversations[chat]['messages'] += [cleaned_chat]
        else:
            conversations[chat] = {
                'messages': [cleaned_chat]
            }

for chat, content in conversations.items():
    messages = content['messages']
    msgs_sorted_by_date = sorted(messages, key = lambda x:x['Created'])
    content['messages'] = msgs_sorted_by_date

for idx, chat in enumerate(conversations):
    print(f"{idx}: {chat} (Messages: {len(conversations[chat]['messages'])})")

selected_chat_idx = input("Select a chat: ")
selected_chat = list(conversations.values())[int(selected_chat_idx)]

messages = selected_chat['messages']
for message in messages:
    print("\n")
    message_content = message['Text']
    time = message['Created']
    if 'From' in message:
        recipient = message['From']
        print(message_content)
        print(f"From: {recipient}")
        print(time)
    else:
        recipient = message['To']
        if (len(message_content) >= 150):
            for submessage in list(message_content[0+i:150+i] for i in range(0, len(message_content), 150)):
                print(submessage.rjust(150))
        else:
            print(message_content.rjust(150))
        print(f"To: {recipient}".rjust(150))
        print(time.rjust(150))