added custom stopwords in prompt

config.json

@@ -28,7 +28,8 @@
   },
   "wordcloud": {
     "font": "RobotoSlab-Thin",
-    "colormap": "prism"
+    "colormap": "prism",
+    "custom_stopwords": "file, use"
   },
   "node": {},
   "conversation_set": {}

controllers/data_analysis.py

@@ -49,9 +49,18 @@ def wordcloud_from_text(
     **kwargs: Any,
 ) -> WordCloud:
     """Creates a wordcloud from the given text. Returns a WordCloud object."""
-    custom_stopwords: list[str] = kwargs.get("stopwords", [])
     default_stopwords: set[str] = load_nltk_stopwords()
-    stop_words: set[str] = default_stopwords.union(set(custom_stopwords))
+
+    custom_stopwords: str = kwargs.get("custom_stopwords", "")
+    custom_stopwords_list: list[str] = (
+        custom_stopwords.split(sep=",") if custom_stopwords else []
+    )
+    custom_stopwords_list = [
+        word.strip().lower() for word in custom_stopwords_list if word.strip()
+    ]
+
+    stop_words: set[str] = default_stopwords.union(set(custom_stopwords_list))
+
     background_color = kwargs.get("background_color", None)
     if background_color is None:
         mode = kwargs.get("mode", "RGBA")

main.py

@@ -102,13 +102,15 @@ def main() -> None:
     wordcloud_folder.mkdir(parents=True, exist_ok=True)
 
     font_path: str = f"assets/fonts/{configs_dict['wordcloud']['font']}.ttf"
-    colormap = configs_dict["wordcloud"]["colormap"]
+    colormap: str = configs_dict["wordcloud"]["colormap"]
+    custom_stopwords: str = configs_dict["wordcloud"]["custom_stopwords"]
 
     generate_all_wordclouds(
         conv_set=all_conversations_set,
         dir_path=wordcloud_folder,
         font_path=font_path,
         colormap=colormap,
+        custom_stopwords=custom_stopwords,
     )
 
     print(f"\nDone ✅ ! Check the output 🔡☁️ here : {wordcloud_folder.as_uri()} 🔗\n")

views/prompt_user.py

@@ -10,7 +10,7 @@
 def validate_header(string: str) -> bool:
     """Returns True if the given text is a valid markdown header."""
     return (
-        1 <= string.count(x="#") <= 6
+        1 <= string.count("#") <= 6
         and string.startswith("#")
         and string[len(string.split()[0])] == " "
     )
@@ -125,6 +125,12 @@ def prompt_user(default_configs: dict[str, Any]) -> dict[str, Any]:
         style=custom_style,
     ).ask()
 
+    user_configs["wordcloud"]["custom_stopwords"] = text(
+        message="Enter custom stopwords (separated by commas) :",
+        default=default_configs["wordcloud"]["custom_stopwords"],
+        style=custom_style,
+    ).ask()
+
     # ------------------ node ---------------------
 
     user_configs["node"] = {}