Add Lora syntax send to comfyui functionality: implement API endpoint and frontend integration for sending and updating LoRA codes in ComfyUI nodes.

static/js/components/LoraCard.js

@@ -1,4 +1,4 @@
-import { showToast, openCivitai, copyToClipboard } from '../utils/uiHelpers.js';
+import { showToast, openCivitai, copyToClipboard, sendLoraToWorkflow } from '../utils/uiHelpers.js';
 import { state } from '../state/index.js';
 import { showLoraModal } from './loraModal/index.js';
 import { bulkManager } from '../managers/BulkManager.js';
@@ -51,9 +51,14 @@ function handleLoraCardEvent(event) {
         return;
     }
     
-    if (event.target.closest('.fa-copy')) {
+    if (event.target.closest('.fa-copy') || event.target.closest('.fa-paper-plane')) {
         event.stopPropagation();
-        copyLoraCode(card);
+        const useSendButton = state.settings.useSendButton || true;
+        if (useSendButton) {
+            sendLoraToComfyUI(card, event.shiftKey);
+        } else {
+            copyLoraCode(card);
+        }
         return;
     }
     
@@ -181,6 +186,15 @@ async function copyLoraCode(card) {
     await copyToClipboard(loraSyntax, 'LoRA syntax copied');
 }
 
+// New function to send LoRA to ComfyUI workflow
+async function sendLoraToComfyUI(card, replaceMode) {
+    const usageTips = JSON.parse(card.dataset.usage_tips || '{}');
+    const strength = usageTips.strength || 1;
+    const loraSyntax = `<lora:${card.dataset.file_name}:${strength}>`;
+    
+    sendLoraToWorkflow(loraSyntax, replaceMode);
+}
+
 export function createLoraCard(lora) {
     const card = document.createElement('div');
     card.className = 'lora-card';
@@ -244,6 +258,8 @@ export function createLoraCard(lora) {
 
     // Get favorite status from the lora data
     const isFavorite = lora.favorite === true;
+    // Check if we're using send button instead of copy button
+    const useSendButton = state.settings.useSendButton || true;
 
     card.innerHTML = `
         <div class="card-preview ${shouldBlur ? 'blurred' : ''}">
@@ -269,8 +285,8 @@ export function createLoraCard(lora) {
                        title="${lora.from_civitai ? 'View on Civitai' : 'Not available from Civitai'}"
                        ${!lora.from_civitai ? 'style="opacity: 0.5; cursor: not-allowed"' : ''}>
                     </i>
-                    <i class="fas fa-copy" 
-                       title="Copy LoRA Syntax">
+                    <i class="${useSendButton ? 'fas fa-paper-plane' : 'fas fa-copy'}" 
+                       title="${useSendButton ? 'Send to ComfyUI (Click: Append, Shift+Click: Replace)' : 'Copy LoRA Syntax'}">
                     </i>
                     <i class="fas fa-trash" 
                        title="Delete Model">

static/js/utils/uiHelpers.js

@@ -351,4 +351,66 @@ export function getNSFWLevelName(level) {
     if (level >= 2) return 'PG13';
     if (level >= 1) return 'PG';
     return 'Unknown';
+}
+
+/**
+ * Sends LoRA syntax to the active ComfyUI workflow
+ * @param {string} loraSyntax - The LoRA syntax to send
+ * @param {boolean} replaceMode - Whether to replace existing LoRAs (true) or append (false)
+ * @returns {Promise<boolean>} - Whether the operation was successful
+ */
+export async function sendLoraToWorkflow(loraSyntax, replaceMode = false) {
+  try {
+    // Get the current workflow from localStorage
+    const workflowData = localStorage.getItem('workflow');
+    if (!workflowData) {
+      showToast('No active workflow found', 'error');
+      return false;
+    }
+    
+    // Parse the workflow JSON
+    const workflow = JSON.parse(workflowData);
+    
+    // Find all Lora Loader (LoraManager) nodes
+    const loraNodes = [];
+    if (workflow.nodes && Array.isArray(workflow.nodes)) {
+      for (const node of workflow.nodes) {
+        if (node.type === "Lora Loader (LoraManager)") {
+          loraNodes.push(node.id);
+        }
+      }
+    }
+    
+    if (loraNodes.length === 0) {
+      showToast('No Lora Loader nodes found in the workflow', 'warning');
+      return false;
+    }
+    
+    // Call the backend API to update the lora code
+    const response = await fetch('/api/update-lora-code', {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json'
+      },
+      body: JSON.stringify({
+        node_ids: loraNodes,
+        lora_code: loraSyntax,
+        mode: replaceMode ? 'replace' : 'append'
+      })
+    });
+    
+    const result = await response.json();
+    
+    if (result.success) {
+      showToast(`LoRA ${replaceMode ? 'replaced' : 'added'} to workflow`, 'success');
+      return true;
+    } else {
+      showToast(result.error || 'Failed to send LoRA to workflow', 'error');
+      return false;
+    }
+  } catch (error) {
+    console.error('Failed to send LoRA to workflow:', error);
+    showToast('Failed to send LoRA to workflow', 'error');
+    return false;
+  }
 }