mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
bce6b0e610
- Implement LoRA locking to prevent specific LoRAs from being changed during randomization - Add visual styling for locked state with amber accents and distinct backgrounds - Introduce `roll_mode` configuration with 'backend' (execute current selection while generating new) and 'frontend' (execute newly generated selection) behaviors - Move LoraPoolNode to 'Lora Manager/randomizer' category and remove standalone class mappings - Standardize RETURN_NAMES in LoraRandomizerNode for consistency
113 lines
4.1 KiB
Vue
113 lines
4.1 KiB
Vue
<template>
|
|
<div class="lora-randomizer-widget">
|
|
<LoraRandomizerSettingsView
|
|
:count-mode="state.countMode.value"
|
|
:count-fixed="state.countFixed.value"
|
|
:count-min="state.countMin.value"
|
|
:count-max="state.countMax.value"
|
|
:model-strength-min="state.modelStrengthMin.value"
|
|
:model-strength-max="state.modelStrengthMax.value"
|
|
:use-same-clip-strength="state.useSameClipStrength.value"
|
|
:clip-strength-min="state.clipStrengthMin.value"
|
|
:clip-strength-max="state.clipStrengthMax.value"
|
|
:roll-mode="state.rollMode.value"
|
|
:is-rolling="state.isRolling.value"
|
|
:is-clip-strength-disabled="state.isClipStrengthDisabled.value"
|
|
@update:count-mode="state.countMode.value = $event"
|
|
@update:count-fixed="state.countFixed.value = $event"
|
|
@update:count-min="state.countMin.value = $event"
|
|
@update:count-max="state.countMax.value = $event"
|
|
@update:model-strength-min="state.modelStrengthMin.value = $event"
|
|
@update:model-strength-max="state.modelStrengthMax.value = $event"
|
|
@update:use-same-clip-strength="state.useSameClipStrength.value = $event"
|
|
@update:clip-strength-min="state.clipStrengthMin.value = $event"
|
|
@update:clip-strength-max="state.clipStrengthMax.value = $event"
|
|
@update:roll-mode="state.rollMode.value = $event"
|
|
@roll="handleRoll"
|
|
/>
|
|
</div>
|
|
</template>
|
|
|
|
<script setup lang="ts">
|
|
import { onMounted } from 'vue'
|
|
import LoraRandomizerSettingsView from './lora-randomizer/LoraRandomizerSettingsView.vue'
|
|
import { useLoraRandomizerState } from '../composables/useLoraRandomizerState'
|
|
import type { ComponentWidget, RandomizerConfig, LoraEntry } from '../composables/types'
|
|
|
|
// Props
|
|
const props = defineProps<{
|
|
widget: ComponentWidget
|
|
node: { id: number; inputs?: any[]; widgets?: any[]; graph?: any }
|
|
}>()
|
|
|
|
// State management
|
|
const state = useLoraRandomizerState(props.widget)
|
|
|
|
// Handle roll button click
|
|
const handleRoll = async () => {
|
|
try {
|
|
console.log('[LoraRandomizerWidget] Roll button clicked')
|
|
|
|
// Get pool config from connected pool_config input
|
|
const poolConfig = (props.node as any).getPoolConfig?.() || null
|
|
|
|
// Get locked loras from the loras widget
|
|
const lorasWidget = props.node.widgets?.find((w: any) => w.name === "loras")
|
|
const lockedLoras: LoraEntry[] = (lorasWidget?.value || []).filter((lora: LoraEntry) => lora.locked === true)
|
|
|
|
console.log('[LoraRandomizerWidget] Pool config:', poolConfig)
|
|
console.log('[LoraRandomizerWidget] Locked loras:', lockedLoras)
|
|
|
|
// Call API to get random loras
|
|
const randomLoras = await state.rollLoras(poolConfig, lockedLoras)
|
|
|
|
console.log('[LoraRandomizerWidget] Got random LoRAs:', randomLoras)
|
|
|
|
// Update the loras widget with the new selection
|
|
// This will be handled by emitting an event or directly updating the loras widget
|
|
// For now, we'll emit a custom event that the parent widget handler can catch
|
|
if (typeof (props.widget as any).onRoll === 'function') {
|
|
(props.widget as any).onRoll(randomLoras)
|
|
}
|
|
} catch (error) {
|
|
console.error('[LoraRandomizerWidget] Error rolling LoRAs:', error)
|
|
alert('Failed to roll LoRAs: ' + (error as Error).message)
|
|
}
|
|
}
|
|
|
|
// Lifecycle
|
|
onMounted(async () => {
|
|
// Setup serialization
|
|
props.widget.serializeValue = async () => {
|
|
const config = state.buildConfig()
|
|
console.log('[LoraRandomizerWidget] Serializing config:', config)
|
|
return config
|
|
}
|
|
|
|
// Handle external value updates (e.g., loading workflow, paste)
|
|
props.widget.onSetValue = (v) => {
|
|
console.log('[LoraRandomizerWidget] Restoring from config:', v)
|
|
state.restoreFromConfig(v as RandomizerConfig)
|
|
}
|
|
|
|
// Restore from saved value
|
|
if (props.widget.value) {
|
|
console.log('[LoraRandomizerWidget] Restoring from saved value:', props.widget.value)
|
|
state.restoreFromConfig(props.widget.value as RandomizerConfig)
|
|
}
|
|
})
|
|
</script>
|
|
|
|
<style scoped>
|
|
.lora-randomizer-widget {
|
|
padding: 12px;
|
|
background: rgba(40, 44, 52, 0.6);
|
|
border-radius: 4px;
|
|
height: 100%;
|
|
display: flex;
|
|
flex-direction: column;
|
|
overflow: hidden;
|
|
box-sizing: border-box;
|
|
}
|
|
</style>
|