Component additions & lock graph

src/lib/components/ComfyApp.svelte

@@ -49,14 +49,15 @@
      resizeTimeout = setTimeout(app.resizeCanvas.bind(app), 250);
  }
 
- function queuePrompt() {
-     console.log("Queuing!");
-     const workflow = $layoutState.attrs.defaultSubgraph;
-     app.queuePrompt(0, 1, workflow);
- }
+ $: if (app?.lCanvas) {
+     app.lCanvas.allow_dragnodes = $uiState.uiUnlocked;
+     app.lCanvas.allow_interaction = $uiState.uiUnlocked;
 
- $: if (app?.lCanvas) app.lCanvas.allow_dragnodes = !$uiState.nodesLocked;
- $: if (app?.lCanvas) app.lCanvas.allow_interaction = !$uiState.graphLocked;
+     if (!$uiState.uiUnlocked) {
+         app.lCanvas.deselectAllNodes();
+         $layoutState.currentSelectionNodes = []
+     }
+ }
 
  $: if ($uiState.uiEditMode)
      $layoutState.currentSelection = []
@@ -240,9 +241,6 @@
     </div>
     <div id="bottombar">
         <div class="left">
-            <Button variant="primary" on:click={queuePrompt}>
-                Queue Prompt
-            </Button>
             <Button variant="secondary" on:click={toggleGraph}>
                 Toggle Graph
             </Button>

src/lib/components/ComfyApp.ts

@@ -131,6 +131,10 @@ export default class ComfyApp {
         LiteGraph.release_link_on_empty_shows_menu = true;
         LiteGraph.alt_drag_do_clone_nodes = true;
 
+        const uiUnlocked = get(uiState).uiUnlocked;
+        this.lCanvas.allow_dragnodes = uiUnlocked;
+        this.lCanvas.allow_interaction = uiUnlocked;
+
         (window as any).LiteGraph = LiteGraph;
 
         // await this.#invokeExtensionsAsync("init");
@@ -542,6 +546,11 @@ export default class ComfyApp {
                     // nodes have conditional logic that determines which link
                     // to follow backwards.
                     while (isFrontendParent(parent)) {
+                        if (!("getUpstreamLink" in parent)) {
+                            console.warn("[graphToPrompt] Node does not support getUpstreamLink", parent.type)
+                            break;
+                        }
+
                         const nextLink = parent.getUpstreamLink()
                         if (nextLink == null) {
                             console.warn("[graphToPrompt] No upstream link found in frontend node", parent)

src/lib/components/gradio/gallery/Gallery.svelte

@@ -48,7 +48,7 @@
 			  );
 
 	let prevValue: string[] | FileData[] | null = value;
-	let selected_image: number | null = null;
+	export let selected_image: number | null = null;
 	let old_selected_image: number | null = null;
 
 	$: if (prevValue !== value) {