populate audio widget from history

Comfy-Org · Jun 27, 2024 · 9dd50f5 · 9dd50f5
1 parent e9867a6
commit 9dd50f5
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/src/extensions/core/uploadAudio.ts b/src/extensions/core/uploadAudio.ts
@@ -97,14 +97,25 @@ app.registerExtension({
             const audios = message.audio;
             if (!audios) return;
             const audio = audios[0];
-            audioUIWidget.element.src= api.apiURL(getResourceURL(audio.subfolder, audio.filename, "output"));
+            audioUIWidget.element.src = api.apiURL(getResourceURL(audio.subfolder, audio.filename, "output"));
             audioUIWidget.element.classList.remove("empty-audio-widget");
           }
         }
         return { widget: audioUIWidget };
       }
     }
   },
+  onNodeOutputsUpdated(nodeOutputs: Record<number, any>) {
+    for (const [nodeId, output] of Object.entries(nodeOutputs)) {
+      const node = app.graph.getNodeById(Number.parseInt(nodeId));
+      if ("audio" in output) {
+        const audioUIWidget = node.widgets.find((w) => w.name === "audioUI") as unknown as DOMWidget<HTMLAudioElement>;
+        const audio = output.audio[0];
+        audioUIWidget.element.src = api.apiURL(getResourceURL(audio.subfolder, audio.filename, "output"));
+        audioUIWidget.element.classList.remove("empty-audio-widget");
+      }
+    }
+  },
 });
 
 app.registerExtension({

diff --git a/src/scripts/app.ts b/src/scripts/app.ts
@@ -65,7 +65,7 @@ export class ComfyApp {
 	ui: ComfyUI;
 	logging: ComfyLogging;
 	extensions: ComfyExtension[];
-	nodeOutputs: Record<string, any>;
+	_nodeOutputs: Record<string, any>;
 	nodePreviewImages: Record<string, typeof Image>;
 	shiftDown: boolean;
 	graph: LGraph;
@@ -116,6 +116,15 @@ export class ComfyApp {
 		this.shiftDown = false;
 	}
 
+	get nodeOutputs() {
+		return this._nodeOutputs;
+	}
+
+	set nodeOutputs(value) {
+		this._nodeOutputs = value;
+		this.#invokeExtensions("onNodeOutputsUpdated", value);
+	}
+
 	getPreviewFormatParam() {
 		let preview_format = this.ui.settings.getSettingValue("Comfy.PreviewFormat");
 		if(preview_format)