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

Added ChatGPT generated piechart

Browse Source
This commit is contained in:
Stef
2025-04-18 18:48:37 +02:00
parent 2bf664e25f
commit bfe04e9c2e
2 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/process_data.py
View File

@@ -20,16 +20,15 @@ makedirs("output", exist_ok=True)
test = processRawMessages(chat)
total_frequency = 0
total_frequency_dict: dict[str, int] = Counter()
total_frequency_dict: dict[str, int] = {}
word_frequency_dict: dict[str, dict[str, int]] = {}
for author in test:
word_frequency_dict[author] = {}
messageList = test.get(author)
messageList = test[author]
if 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]
wordFreqList = Counter(processMessageList(messageList))
word_frequency_dict[author] = wordFreqList
total_frequency_dict[author] = sum(wordFreqList.values())
total_frequency = sum(total_frequency_dict.values())