mirror of
https://github.com/StefBuwalda/whatsapp-wordcloud.git
synced 2025-11-01 12:19:57 +00:00
32 lines
893 B
Python
32 lines
893 B
Python
from config import wordcloud # type: ignore
|
|
from os import makedirs
|
|
from functions import (
|
|
processRawMessages,
|
|
processRawMessages2,
|
|
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 = processRawMessages2(chat)
|
|
|
|
for author in test.keys():
|
|
messageList = test.get(author)
|
|
if messageList:
|
|
wordList = processMessageList(messageList)
|
|
freq_dict = Counter(wordList)
|
|
image = wordcloud.generate_from_frequencies(freq_dict) # type: ignore
|
|
image.to_file(f"output/{author}.png") # type: ignore
|
|
messages = processRawMessages(chat)
|