mirror of
https://github.com/StefBuwalda/whatsapp-wordcloud.git
synced 2025-10-30 11:19:57 +00:00
29 lines
729 B
Python
29 lines
729 B
Python
from os import makedirs
|
|
from backend.functions import (
|
|
processRawMessages,
|
|
processMessageList,
|
|
)
|
|
from collections import Counter
|
|
|
|
|
|
# Open and read the chats from the '/data/_chat.txt' file exported by Whatsapp
|
|
try:
|
|
file = open("data/_chat.txt", encoding="utf8")
|
|
chat = file.read()
|
|
file.close()
|
|
except FileNotFoundError:
|
|
print("Sorry, the file /data/_chat.txt does not exist.")
|
|
exit()
|
|
|
|
makedirs("output", exist_ok=True)
|
|
|
|
test = processRawMessages(chat)
|
|
|
|
frequency_dictionary: dict[str, dict[str, int]] = {}
|
|
|
|
for author in test:
|
|
frequency_dictionary[author] = {}
|
|
messageList = test.get(author)
|
|
if messageList:
|
|
frequency_dictionary[author] = Counter(processMessageList(messageList))
|