refactoring

word_cloud.py

@@ -0,0 +1,23 @@
+from wordcloud import WordCloud  # type: ignore
+from backend.process_data import frequency_dictionary
+
+
+wordcloud = WordCloud(
+    width=800,
+    height=400,
+    background_color="black",  # or 'black', or any HTML color
+    colormap="viridis",  # matplotlib colormap ('plasma', 'cool', 'inferno')
+    # font_path="path/to/font.ttf",  # Use a custom font
+    max_words=100,  # Max number of words to include
+    min_font_size=10,
+    max_font_size=100,
+    prefer_horizontal=1,  # Between 0 (all vertical) and 1 (all horizontal)
+    scale=4,  # Higher = better resolution
+    contour_color="steelblue",  # Outline color (when using contour_width)
+    contour_width=1,  # For consistent layout between runs
+)
+
+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