From bde16fc226a23020db72a09a254410732188107f Mon Sep 17 00:00:00 2001
From: PassiveLemon <lemonl3mn@protonmail.com>
Date: Sun, 7 May 2023 09:52:08 -0400
Subject: [PATCH] Automated install of pip packages for nodes

---
 services/comfy/entrypoint.sh | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/services/comfy/entrypoint.sh b/services/comfy/entrypoint.sh
index f5a7430..ae89a3d 100755
--- a/services/comfy/entrypoint.sh
+++ b/services/comfy/entrypoint.sh
@@ -22,4 +22,16 @@ for to_path in "${!MOUNTS[@]}"; do
   echo Mounted $(basename "${from_path}")
 done
 
+if [ "$(ls -A /stable-diffusion/custom_nodes)" ]; then
+  chmod 777 -R "/stable-diffusion/custom_nodes/"
+  apt-get install build-essential -y
+  for dir in "/stable-diffusion/custom_nodes/*/"; do
+    if [ -d $dir ]; then
+      echo $dir
+      cd $dir
+      pip install -r requirements.txt
+    fi
+  done
+fi
+
 exec "$@"