From f25a150d3cfee3391fb7e6d9e4352a532be50146 Mon Sep 17 00:00:00 2001
From: Mathias Winkel <mathias.winkel@merckgroup.com>
Date: Wed, 17 Jan 2024 11:38:50 +0100
Subject: [PATCH] add wordcloud support

---
 Dockerfile                                   | 3 ++-
 gpt_code_ui/kernel_program/kernel_manager.py | 1 +
 gpt_code_ui/webapp/main.py                   | 1 +
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 7cb0632d..9b46a0af 100644
--- 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
index 0773550e..b90912b6 100644
--- 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
index 953bb888..7527fe8c 100644
--- 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