189 lines
8.7 KiB
TypeScript
189 lines
8.7 KiB
TypeScript
import { Fragment } from 'react'
|
|
import { ChevronUp } from 'lucide-react'
|
|
import { SettingsDropdown } from './SettingsDropdown'
|
|
import { Tooltip } from './ui/tooltip'
|
|
import type { ApiSuccessOf } from '../lib/api-client'
|
|
import type { components } from '../generated/backend-openapi'
|
|
import type { ICLoraConditioningType } from './ICLoraPanel'
|
|
import type { IcLoraAudioMode } from '../hooks/use-ic-lora'
|
|
import type { ResolutionOption } from '../lib/video-resolution'
|
|
|
|
type CatalogControl = components['schemas']['IcLoraControl']
|
|
|
|
// Short hover/focus help per IC-LoRA control. The catalog-driven controls are keyed by control id.
|
|
const CONTROL_TOOLTIPS: Record<string, string> = {
|
|
condStrength: 'How strongly the conditioning (edges/depth) constrains the result. Higher follows the control signal more closely.',
|
|
loraStrength: 'How strongly the IC-LoRA weights are applied. Lower is subtler, higher is stronger.',
|
|
duration: 'Length of the generated clip.',
|
|
}
|
|
|
|
// A catalog control's display label for a given option: explicit value_labels win, else value + unit.
|
|
function controlOptionLabel(control: CatalogControl, option: number | string): string {
|
|
return control.value_labels?.[String(option)] ?? `${option}${control.unit ?? ''}`
|
|
}
|
|
|
|
// The IC-LoRA control row in the prompt bar: conditioning-type selector + per-mode knobs
|
|
// (strength, duration, LoRA strength, and the advanced stage-2 / resolution / audio / fps).
|
|
export interface IcLoraControlsProps {
|
|
icLoraCondType?: ICLoraConditioningType
|
|
icLoraSelectorValue?: string
|
|
icLoraSelectorOptions?: { value: string; label: string }[]
|
|
onIcLoraSelectorChange?: (value: string) => void
|
|
icLoraStrength?: number
|
|
onIcLoraStrengthChange?: (strength: number) => void
|
|
availableIcLoras?: ApiSuccessOf<'listModels'>['models']
|
|
icLoraCustomRef?: string | null
|
|
onIcLoraCustomRefChange?: (ref: string | null) => void
|
|
icLoraSkipStage2?: boolean
|
|
onIcLoraSkipStage2Change?: (skip: boolean) => void
|
|
icLoraUseLoraInStage2?: boolean
|
|
onIcLoraUseLoraInStage2Change?: (use: boolean) => void
|
|
// Resolution tiers for the use_lora_in_stage_2 path (empty until source media loads).
|
|
icLoraResolutionOptions?: ResolutionOption[]
|
|
icLoraResolutionKey?: string
|
|
onIcLoraResolutionKeyChange?: (key: string) => void
|
|
icLoraResolutionFactor?: number
|
|
onIcLoraResolutionFactorChange?: (factor: number) => void
|
|
icLoraAudioMode?: IcLoraAudioMode
|
|
onIcLoraAudioModeChange?: (mode: IcLoraAudioMode) => void
|
|
icLoraLoraStrength?: number
|
|
onIcLoraLoraStrengthChange?: (strength: number) => void
|
|
icLoraFps?: number | null
|
|
onIcLoraFpsChange?: (fps: number | null) => void
|
|
isCatalogIcLora?: boolean
|
|
advancedIcLoraControls?: boolean
|
|
// Catalog-declared controls for the selected IC-LoRA, rendered generically by id + kind.
|
|
controls?: CatalogControl[]
|
|
controlValues?: Record<string, number | string>
|
|
onControlChange?: (id: string, value: number | string) => void
|
|
}
|
|
|
|
export function IcLoraSettingsControls({
|
|
icLoraCondType,
|
|
icLoraSelectorValue,
|
|
icLoraSelectorOptions,
|
|
onIcLoraSelectorChange,
|
|
icLoraStrength,
|
|
onIcLoraStrengthChange,
|
|
availableIcLoras,
|
|
icLoraCustomRef,
|
|
onIcLoraCustomRefChange,
|
|
icLoraLoraStrength,
|
|
onIcLoraLoraStrengthChange,
|
|
isCatalogIcLora,
|
|
controls,
|
|
controlValues,
|
|
onControlChange,
|
|
}: IcLoraControlsProps) {
|
|
return (
|
|
<>
|
|
{/* Unified IC-LoRA selector: canny | depth | [catalog IC-LoRAs] | custom. No tooltip:
|
|
the options are self-explanatory and the copy was wrong for catalog recipes. */}
|
|
<SettingsDropdown
|
|
title="CONDITIONING TYPE"
|
|
value={icLoraSelectorValue ?? 'canny'}
|
|
onChange={(v) => onIcLoraSelectorChange?.(v)}
|
|
options={icLoraSelectorOptions ?? []}
|
|
trigger={
|
|
<>
|
|
<span className="text-zinc-300 font-medium">{(icLoraSelectorOptions ?? []).find(o => o.value === icLoraSelectorValue)?.label ?? 'Canny Edges'}</span>
|
|
<ChevronUp className="h-3 w-3 text-zinc-500" />
|
|
</>
|
|
}
|
|
/>
|
|
{!isCatalogIcLora && icLoraCondType === 'custom' && (
|
|
<>
|
|
<div className="w-px h-4 bg-zinc-700 mx-0.5" />
|
|
<SettingsDropdown
|
|
title="IC-LORA"
|
|
value={icLoraCustomRef ?? ''}
|
|
onChange={(v) => onIcLoraCustomRefChange?.(v || null)}
|
|
options={(availableIcLoras ?? []).map(m => ({ value: m.path, label: m.name }))}
|
|
trigger={
|
|
<>
|
|
<span className="text-zinc-300 font-medium truncate max-w-[160px]">
|
|
{availableIcLoras?.find(m => m.path === icLoraCustomRef)?.name
|
|
?? (availableIcLoras?.length ? 'Select IC-LoRA' : 'No IC-LoRAs found')}
|
|
</span>
|
|
<ChevronUp className="h-3 w-3 text-zinc-500" />
|
|
</>
|
|
}
|
|
/>
|
|
</>
|
|
)}
|
|
{!isCatalogIcLora && (
|
|
<>
|
|
<div className="w-px h-4 bg-zinc-700 mx-0.5" />
|
|
<SettingsDropdown
|
|
title="STRENGTH"
|
|
tooltip={CONTROL_TOOLTIPS.condStrength}
|
|
value={String(icLoraStrength ?? 1.0)}
|
|
onChange={(v) => onIcLoraStrengthChange?.(parseFloat(v))}
|
|
options={[
|
|
{ value: '0.5', label: '0.50' },
|
|
{ value: '0.75', label: '0.75' },
|
|
{ value: '1', label: '1.00' },
|
|
{ value: '1.25', label: '1.25' },
|
|
{ value: '1.5', label: '1.50' },
|
|
{ value: '2', label: '2.00' },
|
|
]}
|
|
trigger={
|
|
<>
|
|
<span className="text-zinc-500 text-[10px]">STR</span>
|
|
<span className="text-zinc-300 font-medium">{(icLoraStrength ?? 1.0).toFixed(2)}</span>
|
|
<ChevronUp className="h-3 w-3 text-zinc-500" />
|
|
</>
|
|
}
|
|
/>
|
|
</>
|
|
)}
|
|
{/* Catalog-declared option controls (duration, …) rendered generically as dropdowns.
|
|
Structured controls (position_canvas) carry no options and render elsewhere — skip them. */}
|
|
{isCatalogIcLora && (controls ?? []).filter((c) => c.options != null).map((c) => {
|
|
const current = controlValues?.[c.id] ?? c.default ?? ''
|
|
const title = c.label.toUpperCase()
|
|
return (
|
|
<Fragment key={c.id}>
|
|
<div className="w-px h-4 bg-zinc-700 mx-0.5" />
|
|
<SettingsDropdown
|
|
title={title}
|
|
tooltip={CONTROL_TOOLTIPS[c.id]}
|
|
value={String(current)}
|
|
onChange={(v) => onControlChange?.(c.id, c.kind === 'int' ? parseInt(v, 10) : v)}
|
|
options={(c.options as (number | string)[]).map((o) => ({ value: String(o), label: controlOptionLabel(c, o) }))}
|
|
trigger={
|
|
<>
|
|
<span className="text-zinc-500 text-[10px]">{title}</span>
|
|
<span className="text-zinc-300 font-medium">{controlOptionLabel(c, current)}</span>
|
|
<ChevronUp className="h-3 w-3 text-zinc-500" />
|
|
</>
|
|
}
|
|
/>
|
|
</Fragment>
|
|
)
|
|
})}
|
|
{/* LoRA strength: a regular control for every IC-LoRA mode (canny/depth/custom/catalog).
|
|
It scales the adapter merge weight, so it always has an effect. (The advanced knobs —
|
|
S2/Res/Audio/FPS — live in IcLoraAdvancedPanel beside the prompt bar.) A continuous
|
|
slider (like the plain-LoRA strength bar) rather than a dropdown, for finer control. */}
|
|
<div className="w-px h-4 bg-zinc-700 mx-0.5" />
|
|
<Tooltip content={CONTROL_TOOLTIPS.loraStrength}>
|
|
<div className="flex items-center gap-1.5 px-2 py-1.5">
|
|
<span className="text-zinc-500 text-[10px]">LORA</span>
|
|
<input
|
|
type="range"
|
|
min={0}
|
|
max={2}
|
|
step={0.05}
|
|
value={icLoraLoraStrength ?? 1.0}
|
|
onChange={(e) => onIcLoraLoraStrengthChange?.(parseFloat(e.target.value))}
|
|
className="w-16 h-1 accent-white cursor-pointer"
|
|
aria-label="LoRA strength"
|
|
/>
|
|
<span className="text-zinc-300 font-medium w-9 text-right">{(icLoraLoraStrength ?? 1.0).toFixed(2)}</span>
|
|
</div>
|
|
</Tooltip>
|
|
</>
|
|
)
|
|
}
|