From c6fe3d2e0e068e11de4762aa2bbf9c9d0bb1bfdc Mon Sep 17 00:00:00 2001
From: Abdullah Alfaraj <abdullah.a.f2014@gmail.com>
Date: Mon, 13 Mar 2023 01:03:53 +0300
Subject: [PATCH] refresh controlnet tab when refreshing the ui

---
 index.js | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/index.js b/index.js
index fc4992a..dcd8e1e 100644
--- a/index.js
+++ b/index.js
@@ -417,6 +417,10 @@ async function refreshUI() {
         console.log('inpainting_mask_weight: ', inpainting_mask_weight)
         html_manip.autoFillInInpaintMaskWeight(inpainting_mask_weight)
 
+        await temp_config.getConfig()
+        //init ControlNet Tab
+        // g_hi_res_upscaler_models = temp_config.getUpscalerModels()
+        g_controlnet_max_models = temp_config.getControlNetMaxModelsNum()
         await control_net.initializeControlNetTab(g_controlnet_max_models)
     } catch (e) {
         console.warn(e)
@@ -708,7 +712,10 @@ async function initPlugin() {
     await loadPromptShortcut()
     await refreshPromptMenue()
 
+    await temp_config.getConfig()
     //init ControlNet Tab
+    // g_hi_res_upscaler_models = temp_config.getUpscalerModels()
+    g_controlnet_max_models = temp_config.getControlNetMaxModelsNum()
     await control_net.initializeControlNetTab(g_controlnet_max_models)
 }
 initPlugin()