add handling to some places in javascript that can potentially cause issues #6898

1 files changed, 6 insertions, 4 deletions

diff --git a/extensions-builtin/prompt-bracket-checker/javascript/prompt-bracket-checker.js b/extensions-builtin/prompt-bracket-checker/javascript/prompt-bracket-checker.js
index eccfb0f9..251a1f57 100644
--- a/extensions-builtin/prompt-bracket-checker/javascript/prompt-bracket-checker.js
+++ b/extensions-builtin/prompt-bracket-checker/javascript/prompt-bracket-checker.js
@@ -93,10 +93,12 @@ function checkBrackets(evt) {
 }
 
 var shadowRootLoaded = setInterval(function() {
-  var shadowTextArea = document.querySelector('gradio-app').shadowRoot.querySelectorAll('#txt2img_prompt > label > textarea');
-  if(shadowTextArea.length < 1) {
-      return false;
-  }
+  var sahdowRoot = document.querySelector('gradio-app').shadowRoot;
+  if(! sahdowRoot)  return false;
+
+  var shadowTextArea = sahdowRoot.querySelectorAll('#txt2img_prompt > label > textarea');
+  if(shadowTextArea.length < 1)  return false;
+
 
   clearInterval(shadowRootLoaded);