Added closing of model info view after downloading model info.

- Forces model info to be read from server.
This commit is contained in:
Christian Bastian
2024-09-23 01:47:14 -04:00
parent 3026717fd7
commit 91e5c882fc

View File

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