add wordcloud support

dasmy · Jan 17, 2024 · f25a150 · f25a150
1 parent 2d6b266
commit f25a150
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -35,7 +35,8 @@ RUN pip install \
         "openpyxl>=3.1.2,<4" \
         "rdkit>=2023.3.3" \
         "scipy>=1.11.1" \
-        "scikit-learn>=1.3.0"
+        "scikit-learn>=1.3.0" \
+        "wordcloud>=1.9.3"
 
 RUN mkdir backendbuild
 WORKDIR /backendbuild

diff --git a/gpt_code_ui/kernel_program/kernel_manager.py b/gpt_code_ui/kernel_program/kernel_manager.py
@@ -168,6 +168,7 @@ def create_venv(venv_dir: pathlib.Path, install_default_packages: bool) -> pathl
                 "rdkit>=2023.3.3",
                 "scipy==1.11.1",
                 "scikit-learn==1.3.0",
+                "wordcloud>=1.9.3",
             ]
             subprocess.run([venv_python_executable, '-m', 'pip', 'install'] + default_packages)
 

diff --git a/gpt_code_ui/webapp/main.py b/gpt_code_ui/webapp/main.py
@@ -66,6 +66,7 @@ def __init__(self):
         'tabulate', # tabulate==0.9.0
         'scipy', # scipy==1.11.1
         'scikit-learn', # scikit-learn==1.3.0
+        'WordCloud', # wordcloud==1.9.3"
     For pdf extraction, you can use
         'PyPDF2', # PyPDF2==3.0.1
         'pdfminer', # pdfminer==20191125