From 91e5c882fc6bdfbdc832e0733be365772b992582 Mon Sep 17 00:00:00 2001
From: Christian Bastian <80225746+cdb-boop@users.noreply.github.com>
Date: Mon, 23 Sep 2024 01:47:14 -0400
Subject: [PATCH] Added closing of model info view after downloading model
 info. - Forces model info to be read from server.

---
 web/model-manager.js | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/web/model-manager.js b/web/model-manager.js
index 3666fba..8f427e9 100644
--- a/web/model-manager.js
+++ b/web/model-manager.js
@@ -2411,13 +2411,18 @@ class ModelInfo {
   /** @type {[HTMLElement][]} */
   #settingsElements = null;
 
+  /** @type {() -> Promise<void>} */
+  #tryHideModelInfo = () => {};
+
   /**
    * @param {ModelData} modelData
    * @param {() => Promise<void>} updateModels
    * @param {any} settingsElements
+   * @param {() => Promise<void>} tryHideModelInfo
    */
-  constructor(modelData, updateModels, settingsElements) {
+  constructor(modelData, updateModels, settingsElements, tryHideModelInfo) {
     this.#settingsElements = settingsElements;
+    this.#tryHideModelInfo = tryHideModelInfo;
     const moveDestinationInput = $el('input.search-text-area', {
       name: 'move directory',
       autocomplete: 'off',
@@ -2936,6 +2941,9 @@ class ModelInfo {
               }).catch((err) => {
                   return false;
               });
+              if (success) {
+                this.#tryHideModelInfo();
+              }
               comfyButtonAlert(e.target, success, 'mdi-check-bold', 'mdi-close-thick');
               button.disabled = false;
             },
@@ -5254,6 +5262,7 @@ class ModelManager extends ComfyDialog {
       this.#modelData,
       this.#refreshModels,
       this.#settingsView.elements.settings,
+      () => this.#tryHideModelInfo(),
     );
 
     this.#browseView = new BrowseView(