From 71adea96879090ea16798ca5573c74377b561955 Mon Sep 17 00:00:00 2001
From: Stef <stbuwalda@gmail.com>
Date: Fri, 18 Apr 2025 18:01:40 +0200
Subject: [PATCH] refactoring

---
 backend/process_data.py                    | 2 +-
 applications/wordcloud.py => word_cloud.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)
 rename applications/wordcloud.py => word_cloud.py (95%)

diff --git a/backend/process_data.py b/backend/process_data.py
index d6ab161..6adaaba 100644
--- a/backend/process_data.py
+++ b/backend/process_data.py
@@ -21,7 +21,7 @@ test = processRawMessages(chat)
 
 frequency_dictionary: dict[str, dict[str, int]] = {}
 
-for author in test.keys():
+for author in test:
     frequency_dictionary[author] = {}
     messageList = test.get(author)
     if messageList:
diff --git a/applications/wordcloud.py b/word_cloud.py
similarity index 95%
rename from applications/wordcloud.py
rename to word_cloud.py
index 0c8cc2b..42c62ca 100644
--- a/applications/wordcloud.py
+++ b/word_cloud.py
@@ -17,7 +17,7 @@ wordcloud = WordCloud(
     contour_width=1,  # For consistent layout between runs
 )
 
-for author in frequency_dictionary.keys():
+for author in frequency_dictionary:
     freq_dict = frequency_dictionary.get(author)
     image = wordcloud.generate_from_frequencies(freq_dict)  # type: ignore
     image.to_file(f"output/{author}.png")  # type: ignore