0.57

web/js/ollama_talk.js

@@ -0,0 +1,197 @@
+import { app } from "../../../scripts/app.js";
+import { api } from "../../../scripts/api.js";
+
+// Node-specific logic
+app.registerExtension({
+  name: "Bjornulf.OllamaTalk",
+  async nodeCreated(node) {
+    if (node.comfyClass === "Bjornulf_OllamaTalk") {
+      // Set seed widget to hidden input
+      const seedWidget = node.widgets.find((w) => w.name === "seed");
+      if (seedWidget) {
+        seedWidget.type = "HIDDEN";
+      }
+
+      // Function to update the Reset Button text
+      const updateResetButtonTextNode = () => {
+        fetch("/get_current_context_size", {
+          method: "POST",
+        })
+          .then((response) => response.json())
+          .then((data) => {
+            if (data.success) {
+              if (data.value === 0) {
+                resetButton.name = "Save/Reset Context File (Empty)";
+              } else {
+                resetButton.name = `Save/Reset Context File (${data.value} lines)`;
+              }
+            } else {
+              console.error("Error in context size:", data.error);
+              resetButton.name = "Save/Reset Context File (Error)";
+            }
+          })
+          .catch((error) => {
+            console.error("Error fetching context size:", error);
+            resetButton.name = "Save/Reset Context File (Error)";
+          });
+      };
+
+      // Add reset button
+      const resetButton = node.addWidget(
+        "button",
+        "Save/Reset Context File",
+        null,
+        () => {
+          fetch("/reset_lines_context", {
+            method: "POST",
+          })
+            .then((response) => response.json())
+            .then((data) => {
+              if (data.success) {
+                // updateLineNumber();
+                updateResetButtonTextNode();
+                app.ui.toast("Counter reset successfully!", { duration: 5000 });
+              } else {
+                app.ui.toast(
+                  `Failed to reset counter: ${data.error || "Unknown error"}`,
+                  { type: "error", duration: 5000 }
+                );
+              }
+            })
+            .catch((error) => {
+              console.error("Error:", error);
+              app.ui.toast("An error occurred while resetting the counter.", {
+                type: "error",
+                duration: 5000,
+              });
+            });
+        }
+      );
+
+      // Add resume button
+      const resumeButton = node.addWidget("button", "Resume", "Resume", () => {
+        const workflow = app.graph.serialize();
+        const nodeData = workflow.nodes.find((n) => n.id === node.id);
+        const userPromptValue =
+          nodeData.widgets_values?.[
+            node.widgets.findIndex((w) => w.name === "user_prompt")
+          ];
+
+        fetch("/bjornulf_ollama_send_prompt", {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json",
+          },
+          body: JSON.stringify({
+            node_id: node.id,
+            user_prompt: userPromptValue,
+          }),
+        })
+          .then((response) => response.text())
+          .then((data) => {
+            console.log("Resume response:", data);
+          })
+          .catch((error) => console.error("Error:", error));
+      });
+
+      // Function to update button visibility based on widget values
+      const updateButtonVisibility = () => {
+        // Check context file widget
+        const contextWidget = node.widgets.find(
+          (w) => w.name === "use_context_file"
+        );
+        const isContextFileEnabled = contextWidget
+          ? contextWidget.value
+          : false;
+        resetButton.type = isContextFileEnabled ? "button" : "HIDDEN";
+
+        // Check waiting for prompt widget
+        const waitingWidget = node.widgets.find(
+          (w) => w.name === "waiting_for_prompt"
+        );
+        const isWaitingForPrompt = waitingWidget ? waitingWidget.value : false;
+        resumeButton.type = isWaitingForPrompt ? "button" : "HIDDEN";
+
+        //ALSO update reset button text node
+        updateResetButtonTextNode(); // Will trigger when... toggle / refresh page
+
+        // Force canvas redraw to update UI
+        node.setDirtyCanvas(true);
+      };
+
+      // Add a handler for the use_context_file widget
+      const contextWidget = node.widgets.find(
+        (w) => w.name === "use_context_file"
+      );
+      if (contextWidget) {
+        const originalOnChange = contextWidget.callback;
+        contextWidget.callback = function (v) {
+          if (originalOnChange) {
+            originalOnChange.call(this, v);
+          }
+          updateButtonVisibility();
+        };
+      }
+
+      // Add a handler for the waiting_for_prompt widget
+      const waitingWidget = node.widgets.find(
+        (w) => w.name === "waiting_for_prompt"
+      );
+      if (waitingWidget) {
+        const originalOnChange = waitingWidget.callback;
+        waitingWidget.callback = function (v) {
+          if (originalOnChange) {
+            originalOnChange.call(this, v);
+          }
+          updateButtonVisibility();
+        };
+      }
+
+      // Initial update of button visibility
+      setTimeout(updateButtonVisibility, 0);
+
+      // Listen for node execution events
+      api.addEventListener("executed", async () => {
+        updateResetButtonTextNode();
+      });
+
+      //If workflow is stopped during pause, cancel the run
+      const original_api_interrupt = api.interrupt;
+      api.interrupt = function () {
+          api.fetchApi('/bjornulf_ollama_interrupt', {
+              method: 'POST'
+          });
+          original_api_interrupt.apply(this, arguments);
+      }
+    }
+  },
+});
+
+// // Add listener for workflow execution
+// app.addEventListener("workflowExecuted", () => {
+//     if (node.graph.isPlaying) {
+//         updateContextSize();
+//     }
+// });
+// app.registerExtension({
+//     name: "Bjornulf.OllamaContextChat",
+//     async nodeCreated(node) {
+//         if (node.comfyClass === "Bjornulf_OllamaContextChat") {
+//             const resumeButton = node.addWidget("button", "Resume", "Resume", () => {
+//                 fetch('/bjornulf_ollama_send_prompt', { method: 'GET' })
+//                     .then(response => response.text())
+//                     .then(data => {
+//                         console.log('Resume response:', data);
+//                         // You can update the UI here if needed
+//                     })
+//                     .catch(error => console.error('Error:', error));
+//             });
+//         }
+//     }
+// });
+
+// Set seed widget to hidden input
+// const seedWidget = node.widgets.find((w) => w.name === "seed");
+// if (seedWidget) {
+//   seedWidget.type = "HIDDEN";
+// }