stef_buwalda/whatsapp-wordcloud
1
0
Fork 0
mirror of https://github.com/StefBuwalda/whatsapp-wordcloud.git synced 2025-10-30 11:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

setting up the data to generate a piechart

Browse Source
This commit is contained in:
Stef
2025-04-18 18:21:50 +02:00
parent e7cfc81146
commit 2bf664e25f
4 changed files with 27 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/process_data.py
View File

@@ -19,10 +19,17 @@ makedirs("output", exist_ok=True)
test = processRawMessages(chat)
frequency_dictionary: dict[str, dict[str, int]] = {}
total_frequency = 0
total_frequency_dict: dict[str, int] = Counter()
word_frequency_dict: dict[str, dict[str, int]] = {}
for author in test:
frequency_dictionary[author] = {}
word_frequency_dict[author] = {}
messageList = test.get(author)
if messageList:
frequency_dictionary[author] = Counter(processMessageList(messageList))
wordList = Counter(processMessageList(messageList))
word_frequency_dict[author] = wordList
for count in wordList.items():
total_frequency_dict.update({author: count[1]})
total_frequency += count[1]