From dfa91a317789d487bc20ba2c7711c02ee9a617f7 Mon Sep 17 00:00:00 2001
From: Mathias Winkel <mathias.winkel@merckgroup.com>
Date: Thu, 18 Jan 2024 10:55:53 +0100
Subject: [PATCH] add XlsxWriter package just in case GPT would like to use it
 for writing excel files

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

diff --git a/Dockerfile b/Dockerfile
index 9b46a0af..1b51d9e6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -36,7 +36,8 @@ RUN pip install \
         "rdkit>=2023.3.3" \
         "scipy>=1.11.1" \
         "scikit-learn>=1.3.0" \
-        "wordcloud>=1.9.3"
+        "wordcloud>=1.9.3" \
+        "XlsxWriter>=3.1.9"
 
 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 b90912b6..4ac34735 100644
--- a/gpt_code_ui/kernel_program/kernel_manager.py
+++ b/gpt_code_ui/kernel_program/kernel_manager.py
@@ -169,6 +169,7 @@ def create_venv(venv_dir: pathlib.Path, install_default_packages: bool) -> pathl
                 "scipy==1.11.1",
                 "scikit-learn==1.3.0",
                 "wordcloud>=1.9.3",
+                "XlsxWriter>=3.1.9",
             ]
             subprocess.run([venv_python_executable, '-m', 'pip', 'install'] + default_packages)