1 files changed, 41 insertions, 40 deletions
diff --git a/javascript/contextMenus.js b/javascript/contextMenus.js
index 4e772065..7636c4b3 100644
--- a/javascript/contextMenus.js
+++ b/javascript/contextMenus.js
@@ -123,52 +123,53 @@ contextMenuInit = function(){
   return [appendContextMenuOption, removeContextMenuOption, addContextMenuEventListener]
 }
 
-initResponse = contextMenuInit()
-appendContextMenuOption     = initResponse[0]
-removeContextMenuOption     = initResponse[1]
-addContextMenuEventListener = initResponse[2]
-
-
-//Start example Context Menu Items
-generateOnRepeat = function(genbuttonid,interruptbuttonid){
-  let genbutton = gradioApp().querySelector(genbuttonid);
-  let interruptbutton = gradioApp().querySelector(interruptbuttonid);
-  if(!interruptbutton.offsetParent){
-    genbutton.click();
-  }
-  clearInterval(window.generateOnRepeatInterval)
-  window.generateOnRepeatInterval = setInterval(function(){
+initResponse = contextMenuInit();
+appendContextMenuOption     = initResponse[0];
+removeContextMenuOption     = initResponse[1];
+addContextMenuEventListener = initResponse[2];
+
+(function(){
+  //Start example Context Menu Items
+  let generateOnRepeat = function(genbuttonid,interruptbuttonid){
+    let genbutton = gradioApp().querySelector(genbuttonid);
+    let interruptbutton = gradioApp().querySelector(interruptbuttonid);
     if(!interruptbutton.offsetParent){
       genbutton.click();
     }
-  },
-  500)
-}
-
-generateOnRepeatId = appendContextMenuOption('#txt2img_generate','Generate forever',function(){
-  generateOnRepeat('#txt2img_generate','#txt2img_interrupt');
-})
-generateOnRepeatId = appendContextMenuOption('#img2img_generate','Generate forever',function(){
-  generateOnRepeat('#img2img_generate','#img2img_interrupt');
-})
+    clearInterval(window.generateOnRepeatInterval)
+    window.generateOnRepeatInterval = setInterval(function(){
+      if(!interruptbutton.offsetParent){
+        genbutton.click();
+      }
+    },
+    500)
+  }
 
-cancelGenerateForever = function(){ 
-  clearInterval(window.generateOnRepeatInterval) 
-}
+  appendContextMenuOption('#txt2img_generate','Generate forever',function(){
+    generateOnRepeat('#txt2img_generate','#txt2img_interrupt');
+  })
+  appendContextMenuOption('#img2img_generate','Generate forever',function(){
+    generateOnRepeat('#img2img_generate','#img2img_interrupt');
+  })
 
-appendContextMenuOption('#txt2img_interrupt','Cancel generate forever',cancelGenerateForever)
-appendContextMenuOption('#txt2img_generate', 'Cancel generate forever',cancelGenerateForever)
-appendContextMenuOption('#img2img_interrupt','Cancel generate forever',cancelGenerateForever)
-appendContextMenuOption('#img2img_generate', 'Cancel generate forever',cancelGenerateForever)
-
-appendContextMenuOption('#roll','Roll three',
-  function(){ 
-    let rollbutton = get_uiCurrentTabContent().querySelector('#roll');
-    setTimeout(function(){rollbutton.click()},100)
-    setTimeout(function(){rollbutton.click()},200)
-    setTimeout(function(){rollbutton.click()},300)
+  let cancelGenerateForever = function(){ 
+    clearInterval(window.generateOnRepeatInterval) 
   }
-)
+
+  appendContextMenuOption('#txt2img_interrupt','Cancel generate forever',cancelGenerateForever)
+  appendContextMenuOption('#txt2img_generate', 'Cancel generate forever',cancelGenerateForever)
+  appendContextMenuOption('#img2img_interrupt','Cancel generate forever',cancelGenerateForever)
+  appendContextMenuOption('#img2img_generate', 'Cancel generate forever',cancelGenerateForever)
+
+  appendContextMenuOption('#roll','Roll three',
+    function(){ 
+      let rollbutton = get_uiCurrentTabContent().querySelector('#roll');
+      setTimeout(function(){rollbutton.click()},100)
+      setTimeout(function(){rollbutton.click()},200)
+      setTimeout(function(){rollbutton.click()},300)
+    }
+  )
+})();
 //End example Context Menu Items
 
 onUiUpdate(function(){