Overhaul UI: new color palette, spacing, and visual hierarchy
Replace red accent with amber, add Inter font, introduce 4-level depth palette via CSS variables, expand padding/gaps, wrap sidebar and content sections in cards, add section/subsection header typography classes, and style scrollbars for dark theme. Pure visual changes — no functional or data-flow modifications. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
387
main.py
387
main.py
@@ -20,72 +20,129 @@ from tab_comfy_ng import render_comfy_monitor
|
|||||||
def index():
|
def index():
|
||||||
# -- Streamlit dark theme --
|
# -- Streamlit dark theme --
|
||||||
ui.dark_mode(True)
|
ui.dark_mode(True)
|
||||||
ui.colors(primary='#FF4B4B')
|
ui.colors(primary='#F59E0B')
|
||||||
|
ui.add_head_html(
|
||||||
|
'<link rel="stylesheet" href="https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&display=swap">'
|
||||||
|
)
|
||||||
ui.add_css('''
|
ui.add_css('''
|
||||||
/* === Streamlit Dark Theme === */
|
/* === Dark Theme with Depth Palette === */
|
||||||
|
:root {
|
||||||
|
--bg-page: #0B0E14;
|
||||||
|
--bg-surface-1: #13161E;
|
||||||
|
--bg-surface-2: #1A1E2A;
|
||||||
|
--bg-surface-3: #242836;
|
||||||
|
--border: rgba(255,255,255,0.08);
|
||||||
|
--text-primary: #EAECF0;
|
||||||
|
--text-secondary: rgba(234,236,240,0.55);
|
||||||
|
--accent: #F59E0B;
|
||||||
|
--accent-subtle: rgba(245,158,11,0.12);
|
||||||
|
--negative: #EF4444;
|
||||||
|
}
|
||||||
|
|
||||||
/* Backgrounds */
|
/* Backgrounds */
|
||||||
body.body--dark,
|
body.body--dark,
|
||||||
.q-page.body--dark,
|
.q-page.body--dark,
|
||||||
.body--dark .q-page { background: #0E1117 !important; }
|
.body--dark .q-page { background: var(--bg-page) !important; }
|
||||||
.body--dark .q-drawer { background: #262730 !important; }
|
.body--dark .q-drawer { background: var(--bg-surface-1) !important; }
|
||||||
.body--dark .q-card { background: #262730 !important; border-radius: 0.5rem; }
|
.body--dark .q-card {
|
||||||
|
background: var(--bg-surface-2) !important;
|
||||||
|
border: 1px solid var(--border);
|
||||||
|
border-radius: 0.75rem;
|
||||||
|
}
|
||||||
.body--dark .q-tab-panels { background: transparent !important; }
|
.body--dark .q-tab-panels { background: transparent !important; }
|
||||||
.body--dark .q-tab-panel { background: transparent !important; }
|
.body--dark .q-tab-panel { background: transparent !important; }
|
||||||
.body--dark .q-expansion-item { background: transparent !important; }
|
.body--dark .q-expansion-item { background: transparent !important; }
|
||||||
|
|
||||||
/* Text */
|
/* Text */
|
||||||
.body--dark { color: #FAFAFA !important; }
|
.body--dark { color: var(--text-primary) !important; }
|
||||||
.body--dark .q-field__label { color: rgba(250,250,250,0.6) !important; }
|
.body--dark .q-field__label { color: var(--text-secondary) !important; }
|
||||||
.body--dark .text-caption { color: rgba(250,250,250,0.6) !important; }
|
.body--dark .text-caption { color: var(--text-secondary) !important; }
|
||||||
.body--dark .text-subtitle1,
|
.body--dark .text-subtitle1,
|
||||||
.body--dark .text-subtitle2 { color: #FAFAFA !important; }
|
.body--dark .text-subtitle2 { color: var(--text-primary) !important; }
|
||||||
|
|
||||||
/* Inputs & textareas */
|
/* Inputs & textareas */
|
||||||
.body--dark .q-field--outlined .q-field__control {
|
.body--dark .q-field--outlined .q-field__control {
|
||||||
background: #262730 !important;
|
background: var(--bg-surface-3) !important;
|
||||||
border-radius: 0.5rem !important;
|
border-radius: 0.5rem !important;
|
||||||
}
|
}
|
||||||
.body--dark .q-field--outlined .q-field__control:before {
|
.body--dark .q-field--outlined .q-field__control:before {
|
||||||
border-color: rgba(250,250,250,0.2) !important;
|
border-color: var(--border) !important;
|
||||||
border-radius: 0.5rem !important;
|
border-radius: 0.5rem !important;
|
||||||
}
|
}
|
||||||
.body--dark .q-field--outlined.q-field--focused .q-field__control:after {
|
.body--dark .q-field--outlined.q-field--focused .q-field__control:after {
|
||||||
border-color: #FF4B4B !important;
|
border-color: var(--accent) !important;
|
||||||
}
|
}
|
||||||
.body--dark .q-field__native,
|
.body--dark .q-field__native,
|
||||||
.body--dark .q-field__input { color: #FAFAFA !important; }
|
.body--dark .q-field__input { color: var(--text-primary) !important; }
|
||||||
|
|
||||||
/* Sidebar inputs get main bg */
|
/* Sidebar inputs get page bg */
|
||||||
.body--dark .q-drawer .q-field--outlined .q-field__control {
|
.body--dark .q-drawer .q-field--outlined .q-field__control {
|
||||||
background: #0E1117 !important;
|
background: var(--bg-page) !important;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Buttons */
|
/* Buttons */
|
||||||
.body--dark .q-btn--standard { border-radius: 0.5rem !important; }
|
.body--dark .q-btn--standard { border-radius: 0.5rem !important; }
|
||||||
|
.body--dark .q-btn--outline {
|
||||||
|
transition: background 0.15s ease;
|
||||||
|
}
|
||||||
|
.body--dark .q-btn--outline:hover {
|
||||||
|
background: var(--accent-subtle) !important;
|
||||||
|
}
|
||||||
|
|
||||||
/* Tabs */
|
/* Tabs */
|
||||||
.body--dark .q-tab--active { color: #FF4B4B !important; }
|
.body--dark .q-tab--active { color: var(--accent) !important; }
|
||||||
.body--dark .q-tab__indicator { background: #FF4B4B !important; }
|
.body--dark .q-tab__indicator { background: var(--accent) !important; }
|
||||||
|
|
||||||
/* Separators */
|
/* Separators */
|
||||||
.body--dark .q-separator { background: rgba(250,250,250,0.2) !important; }
|
.body--dark .q-separator { background: var(--border) !important; }
|
||||||
|
|
||||||
/* Expansion items */
|
/* Expansion items */
|
||||||
.body--dark .q-expansion-item__content { padding: 4px 0; }
|
.body--dark .q-expansion-item__content { padding: 12px 16px; }
|
||||||
.body--dark .q-item { border-radius: 0.5rem; }
|
.body--dark .q-item { border-radius: 0.5rem; }
|
||||||
|
|
||||||
/* Splitter */
|
/* Splitter */
|
||||||
.body--dark .q-splitter__separator { background: rgba(250,250,250,0.2) !important; }
|
.body--dark .q-splitter__separator { background: var(--border) !important; }
|
||||||
|
.body--dark .q-splitter__before,
|
||||||
|
.body--dark .q-splitter__after { padding: 0 8px; }
|
||||||
|
|
||||||
/* Action row wrap */
|
/* Action row wrap */
|
||||||
.action-row { flex-wrap: wrap !important; gap: 4px !important; }
|
.action-row { flex-wrap: wrap !important; gap: 8px !important; }
|
||||||
|
|
||||||
/* Notifications */
|
/* Notifications */
|
||||||
.body--dark .q-notification { border-radius: 0.5rem; }
|
.body--dark .q-notification { border-radius: 0.5rem; }
|
||||||
|
|
||||||
/* Font */
|
/* Font */
|
||||||
body { font-family: "Source Sans Pro", "Source Sans 3", sans-serif !important; }
|
body { font-family: "Inter", "Source Sans Pro", "Source Sans 3", sans-serif !important; }
|
||||||
|
|
||||||
|
/* Surface utility classes (need .body--dark to beat .body--dark .q-card specificity) */
|
||||||
|
.body--dark .surface-1 { background: var(--bg-surface-1) !important; }
|
||||||
|
.body--dark .surface-2 { background: var(--bg-surface-2) !important; }
|
||||||
|
.body--dark .surface-3 { background: var(--bg-surface-3) !important; }
|
||||||
|
|
||||||
|
/* Typography utility classes */
|
||||||
|
.section-header {
|
||||||
|
font-size: 0.8rem;
|
||||||
|
font-weight: 600;
|
||||||
|
text-transform: uppercase;
|
||||||
|
letter-spacing: 0.05em;
|
||||||
|
color: var(--text-secondary) !important;
|
||||||
|
}
|
||||||
|
.subsection-header {
|
||||||
|
font-size: 0.85rem;
|
||||||
|
font-weight: 500;
|
||||||
|
color: var(--text-primary) !important;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Scrollbar */
|
||||||
|
::-webkit-scrollbar { width: 6px; height: 6px; }
|
||||||
|
::-webkit-scrollbar-track { background: transparent; }
|
||||||
|
::-webkit-scrollbar-thumb {
|
||||||
|
background: rgba(255,255,255,0.12);
|
||||||
|
border-radius: 3px;
|
||||||
|
}
|
||||||
|
::-webkit-scrollbar-thumb:hover {
|
||||||
|
background: rgba(255,255,255,0.2);
|
||||||
|
}
|
||||||
''')
|
''')
|
||||||
|
|
||||||
config = load_config()
|
config = load_config()
|
||||||
@@ -106,9 +163,9 @@ def index():
|
|||||||
'text-subtitle1 q-pa-lg')
|
'text-subtitle1 q-pa-lg')
|
||||||
return
|
return
|
||||||
|
|
||||||
ui.label(f'Editing: {state.file_path.name}').classes('text-h5 q-mb-md')
|
ui.label(f'Editing: {state.file_path.name}').classes('text-h5 q-mb-lg').style('font-weight: 600')
|
||||||
|
|
||||||
with ui.tabs().classes('w-full') as tabs:
|
with ui.tabs().classes('w-full').style('border-bottom: 1px solid var(--border)') as tabs:
|
||||||
ui.tab('batch', label='Batch Processor')
|
ui.tab('batch', label='Batch Processor')
|
||||||
ui.tab('timeline', label='Timeline')
|
ui.tab('timeline', label='Timeline')
|
||||||
ui.tab('raw', label='Raw Editor')
|
ui.tab('raw', label='Raw Editor')
|
||||||
@@ -148,7 +205,7 @@ def index():
|
|||||||
# ------------------------------------------------------------------
|
# ------------------------------------------------------------------
|
||||||
# Sidebar (rendered AFTER helpers are attached)
|
# Sidebar (rendered AFTER helpers are attached)
|
||||||
# ------------------------------------------------------------------
|
# ------------------------------------------------------------------
|
||||||
with ui.left_drawer(value=True).classes('q-pa-md').style('width: 300px'):
|
with ui.left_drawer(value=True).classes('q-pa-md').style('width: 320px'):
|
||||||
render_sidebar(state)
|
render_sidebar(state)
|
||||||
|
|
||||||
# ------------------------------------------------------------------
|
# ------------------------------------------------------------------
|
||||||
@@ -165,172 +222,172 @@ def index():
|
|||||||
def render_sidebar(state: AppState):
|
def render_sidebar(state: AppState):
|
||||||
ui.label('Navigator').classes('text-h6')
|
ui.label('Navigator').classes('text-h6')
|
||||||
|
|
||||||
# --- Path input ---
|
# --- Path input + Pin ---
|
||||||
path_input = ui.input(
|
with ui.card().classes('w-full q-pa-md q-mb-md'):
|
||||||
'Current Path',
|
path_input = ui.input(
|
||||||
value=str(state.current_dir),
|
'Current Path',
|
||||||
).classes('w-full')
|
value=str(state.current_dir),
|
||||||
|
).classes('w-full')
|
||||||
|
|
||||||
def on_path_enter():
|
def on_path_enter():
|
||||||
p = resolve_path_case_insensitive(path_input.value)
|
p = resolve_path_case_insensitive(path_input.value)
|
||||||
if p is not None and p.is_dir():
|
if p is not None and p.is_dir():
|
||||||
state.current_dir = p
|
state.current_dir = p
|
||||||
state.config['last_dir'] = str(p)
|
state.config['last_dir'] = str(p)
|
||||||
|
save_config(state.current_dir, state.config['favorites'], state.config)
|
||||||
|
state.loaded_file = None
|
||||||
|
state.file_path = None
|
||||||
|
path_input.set_value(str(p))
|
||||||
|
render_file_list.refresh()
|
||||||
|
# Auto-load inside render_file_list already refreshed main content
|
||||||
|
# if files exist; only refresh here for the empty-directory case.
|
||||||
|
if not state.loaded_file:
|
||||||
|
state._render_main.refresh()
|
||||||
|
|
||||||
|
path_input.on('keydown.enter', lambda _: on_path_enter())
|
||||||
|
|
||||||
|
def pin_folder():
|
||||||
|
d = str(state.current_dir)
|
||||||
|
if d not in state.config['favorites']:
|
||||||
|
state.config['favorites'].append(d)
|
||||||
|
save_config(state.current_dir, state.config['favorites'], state.config)
|
||||||
|
render_favorites.refresh()
|
||||||
|
|
||||||
|
ui.button('Pin Folder', icon='push_pin', on_click=pin_folder).classes('w-full')
|
||||||
|
|
||||||
|
# --- Favorites ---
|
||||||
|
with ui.card().classes('w-full q-pa-md q-mb-md'):
|
||||||
|
ui.label('Favorites').classes('section-header')
|
||||||
|
|
||||||
|
@ui.refreshable
|
||||||
|
def render_favorites():
|
||||||
|
for fav in list(state.config['favorites']):
|
||||||
|
with ui.row().classes('w-full items-center'):
|
||||||
|
ui.button(
|
||||||
|
fav,
|
||||||
|
on_click=lambda f=fav: _jump_to(f),
|
||||||
|
).props('flat dense').classes('col')
|
||||||
|
ui.button(
|
||||||
|
icon='close',
|
||||||
|
on_click=lambda f=fav: _unpin(f),
|
||||||
|
).props('flat dense color=negative')
|
||||||
|
|
||||||
|
def _jump_to(fav: str):
|
||||||
|
state.current_dir = Path(fav)
|
||||||
|
state.config['last_dir'] = fav
|
||||||
save_config(state.current_dir, state.config['favorites'], state.config)
|
save_config(state.current_dir, state.config['favorites'], state.config)
|
||||||
state.loaded_file = None
|
state.loaded_file = None
|
||||||
state.file_path = None
|
state.file_path = None
|
||||||
path_input.set_value(str(p))
|
path_input.set_value(fav)
|
||||||
render_file_list.refresh()
|
render_file_list.refresh()
|
||||||
# Auto-load inside render_file_list already refreshed main content
|
|
||||||
# if files exist; only refresh here for the empty-directory case.
|
|
||||||
if not state.loaded_file:
|
if not state.loaded_file:
|
||||||
state._render_main.refresh()
|
state._render_main.refresh()
|
||||||
|
|
||||||
path_input.on('keydown.enter', lambda _: on_path_enter())
|
def _unpin(fav: str):
|
||||||
|
if fav in state.config['favorites']:
|
||||||
|
state.config['favorites'].remove(fav)
|
||||||
|
save_config(state.current_dir, state.config['favorites'], state.config)
|
||||||
|
render_favorites.refresh()
|
||||||
|
|
||||||
# --- Pin / Unpin ---
|
render_favorites()
|
||||||
def pin_folder():
|
|
||||||
d = str(state.current_dir)
|
|
||||||
if d not in state.config['favorites']:
|
|
||||||
state.config['favorites'].append(d)
|
|
||||||
save_config(state.current_dir, state.config['favorites'], state.config)
|
|
||||||
render_favorites.refresh()
|
|
||||||
|
|
||||||
ui.button('Pin Folder', icon='push_pin', on_click=pin_folder).classes('w-full')
|
|
||||||
|
|
||||||
@ui.refreshable
|
|
||||||
def render_favorites():
|
|
||||||
for fav in list(state.config['favorites']):
|
|
||||||
with ui.row().classes('w-full items-center'):
|
|
||||||
ui.button(
|
|
||||||
fav,
|
|
||||||
on_click=lambda f=fav: _jump_to(f),
|
|
||||||
).props('flat dense').classes('col')
|
|
||||||
ui.button(
|
|
||||||
icon='close',
|
|
||||||
on_click=lambda f=fav: _unpin(f),
|
|
||||||
).props('flat dense color=negative')
|
|
||||||
|
|
||||||
def _jump_to(fav: str):
|
|
||||||
state.current_dir = Path(fav)
|
|
||||||
state.config['last_dir'] = fav
|
|
||||||
save_config(state.current_dir, state.config['favorites'], state.config)
|
|
||||||
state.loaded_file = None
|
|
||||||
state.file_path = None
|
|
||||||
path_input.set_value(fav)
|
|
||||||
render_file_list.refresh()
|
|
||||||
if not state.loaded_file:
|
|
||||||
state._render_main.refresh()
|
|
||||||
|
|
||||||
def _unpin(fav: str):
|
|
||||||
if fav in state.config['favorites']:
|
|
||||||
state.config['favorites'].remove(fav)
|
|
||||||
save_config(state.current_dir, state.config['favorites'], state.config)
|
|
||||||
render_favorites.refresh()
|
|
||||||
|
|
||||||
render_favorites()
|
|
||||||
|
|
||||||
ui.separator()
|
|
||||||
|
|
||||||
# --- Snippet Library ---
|
# --- Snippet Library ---
|
||||||
ui.label('Snippet Library').classes('text-subtitle1 q-mt-md')
|
with ui.card().classes('w-full q-pa-md q-mb-md'):
|
||||||
|
ui.label('Snippet Library').classes('section-header')
|
||||||
|
|
||||||
with ui.expansion('Add New Snippet'):
|
with ui.expansion('Add New Snippet'):
|
||||||
snip_name_input = ui.input('Name', placeholder='e.g. Cinematic').classes('w-full')
|
snip_name_input = ui.input('Name', placeholder='e.g. Cinematic').classes('w-full')
|
||||||
snip_content_input = ui.textarea('Content', placeholder='4k, high quality...').classes('w-full')
|
snip_content_input = ui.textarea('Content', placeholder='4k, high quality...').classes('w-full')
|
||||||
|
|
||||||
def save_snippet():
|
def save_snippet():
|
||||||
name = snip_name_input.value
|
name = snip_name_input.value
|
||||||
content = snip_content_input.value
|
content = snip_content_input.value
|
||||||
if name and content:
|
if name and content:
|
||||||
state.snippets[name] = content
|
state.snippets[name] = content
|
||||||
|
save_snippets(state.snippets)
|
||||||
|
snip_name_input.set_value('')
|
||||||
|
snip_content_input.set_value('')
|
||||||
|
ui.notify(f"Saved '{name}'")
|
||||||
|
render_snippet_list.refresh()
|
||||||
|
|
||||||
|
ui.button('Save Snippet', on_click=save_snippet).classes('w-full')
|
||||||
|
|
||||||
|
@ui.refreshable
|
||||||
|
def render_snippet_list():
|
||||||
|
if not state.snippets:
|
||||||
|
return
|
||||||
|
ui.label('Click to copy snippet text:').classes('text-caption')
|
||||||
|
for name, content in list(state.snippets.items()):
|
||||||
|
with ui.row().classes('w-full items-center'):
|
||||||
|
async def copy_snippet(c=content):
|
||||||
|
await ui.run_javascript(
|
||||||
|
f'navigator.clipboard.writeText({json.dumps(c)})', timeout=3.0)
|
||||||
|
ui.notify('Copied to clipboard')
|
||||||
|
|
||||||
|
ui.button(
|
||||||
|
f'{name}',
|
||||||
|
on_click=copy_snippet,
|
||||||
|
).props('flat dense').classes('col')
|
||||||
|
ui.button(
|
||||||
|
icon='delete',
|
||||||
|
on_click=lambda n=name: _del_snippet(n),
|
||||||
|
).props('flat dense color=negative')
|
||||||
|
|
||||||
|
def _del_snippet(name: str):
|
||||||
|
if name in state.snippets:
|
||||||
|
del state.snippets[name]
|
||||||
save_snippets(state.snippets)
|
save_snippets(state.snippets)
|
||||||
snip_name_input.set_value('')
|
|
||||||
snip_content_input.set_value('')
|
|
||||||
ui.notify(f"Saved '{name}'")
|
|
||||||
render_snippet_list.refresh()
|
render_snippet_list.refresh()
|
||||||
|
|
||||||
ui.button('Save Snippet', on_click=save_snippet).classes('w-full')
|
render_snippet_list()
|
||||||
|
|
||||||
@ui.refreshable
|
|
||||||
def render_snippet_list():
|
|
||||||
if not state.snippets:
|
|
||||||
return
|
|
||||||
ui.label('Click to copy snippet text:').classes('text-caption')
|
|
||||||
for name, content in list(state.snippets.items()):
|
|
||||||
with ui.row().classes('w-full items-center'):
|
|
||||||
async def copy_snippet(c=content):
|
|
||||||
await ui.run_javascript(
|
|
||||||
f'navigator.clipboard.writeText({json.dumps(c)})', timeout=3.0)
|
|
||||||
ui.notify('Copied to clipboard')
|
|
||||||
|
|
||||||
ui.button(
|
|
||||||
f'{name}',
|
|
||||||
on_click=copy_snippet,
|
|
||||||
).props('flat dense').classes('col')
|
|
||||||
ui.button(
|
|
||||||
icon='delete',
|
|
||||||
on_click=lambda n=name: _del_snippet(n),
|
|
||||||
).props('flat dense color=negative')
|
|
||||||
|
|
||||||
def _del_snippet(name: str):
|
|
||||||
if name in state.snippets:
|
|
||||||
del state.snippets[name]
|
|
||||||
save_snippets(state.snippets)
|
|
||||||
render_snippet_list.refresh()
|
|
||||||
|
|
||||||
render_snippet_list()
|
|
||||||
|
|
||||||
ui.separator()
|
|
||||||
|
|
||||||
# --- File List ---
|
# --- File List ---
|
||||||
@ui.refreshable
|
with ui.card().classes('w-full q-pa-md q-mb-md'):
|
||||||
def render_file_list():
|
@ui.refreshable
|
||||||
json_files = sorted(state.current_dir.glob('*.json'))
|
def render_file_list():
|
||||||
json_files = [f for f in json_files if f.name not in ('.editor_config.json', '.editor_snippets.json')]
|
json_files = sorted(state.current_dir.glob('*.json'))
|
||||||
|
json_files = [f for f in json_files if f.name not in ('.editor_config.json', '.editor_snippets.json')]
|
||||||
|
|
||||||
if not json_files:
|
if not json_files:
|
||||||
ui.label('No JSON files in this folder.').classes('text-caption')
|
ui.label('No JSON files in this folder.').classes('text-caption')
|
||||||
ui.button('Generate Templates', on_click=lambda: _gen_templates()).classes('w-full')
|
ui.button('Generate Templates', on_click=lambda: _gen_templates()).classes('w-full')
|
||||||
return
|
return
|
||||||
|
|
||||||
with ui.expansion('Create New JSON'):
|
with ui.expansion('Create New JSON'):
|
||||||
new_fn_input = ui.input('Filename', placeholder='my_prompt_vace').classes('w-full')
|
new_fn_input = ui.input('Filename', placeholder='my_prompt_vace').classes('w-full')
|
||||||
|
|
||||||
def create_new():
|
def create_new():
|
||||||
fn = new_fn_input.value
|
fn = new_fn_input.value
|
||||||
if not fn:
|
if not fn:
|
||||||
return
|
return
|
||||||
if not fn.endswith('.json'):
|
if not fn.endswith('.json'):
|
||||||
fn += '.json'
|
fn += '.json'
|
||||||
path = state.current_dir / fn
|
path = state.current_dir / fn
|
||||||
first_item = DEFAULTS.copy()
|
first_item = DEFAULTS.copy()
|
||||||
first_item[KEY_SEQUENCE_NUMBER] = 1
|
first_item[KEY_SEQUENCE_NUMBER] = 1
|
||||||
save_json(path, {KEY_BATCH_DATA: [first_item]})
|
save_json(path, {KEY_BATCH_DATA: [first_item]})
|
||||||
new_fn_input.set_value('')
|
new_fn_input.set_value('')
|
||||||
render_file_list.refresh()
|
render_file_list.refresh()
|
||||||
|
|
||||||
ui.button('Create', on_click=create_new).classes('w-full')
|
ui.button('Create', on_click=create_new).classes('w-full')
|
||||||
|
|
||||||
ui.label('Select File').classes('text-subtitle2 q-mt-sm')
|
ui.label('Select File').classes('subsection-header q-mt-sm')
|
||||||
file_names = [f.name for f in json_files]
|
file_names = [f.name for f in json_files]
|
||||||
ui.radio(
|
ui.radio(
|
||||||
file_names,
|
file_names,
|
||||||
value=file_names[0] if file_names else None,
|
value=file_names[0] if file_names else None,
|
||||||
on_change=lambda e: state._load_file(e.value) if e.value else None,
|
on_change=lambda e: state._load_file(e.value) if e.value else None,
|
||||||
).classes('w-full')
|
).classes('w-full')
|
||||||
|
|
||||||
# Auto-load first file if nothing loaded yet
|
# Auto-load first file if nothing loaded yet
|
||||||
if file_names and not state.loaded_file:
|
if file_names and not state.loaded_file:
|
||||||
state._load_file(file_names[0])
|
state._load_file(file_names[0])
|
||||||
|
|
||||||
def _gen_templates():
|
def _gen_templates():
|
||||||
generate_templates(state.current_dir)
|
generate_templates(state.current_dir)
|
||||||
render_file_list.refresh()
|
render_file_list.refresh()
|
||||||
|
|
||||||
render_file_list()
|
render_file_list()
|
||||||
|
|
||||||
ui.separator()
|
|
||||||
|
|
||||||
# --- Comfy Monitor toggle ---
|
# --- Comfy Monitor toggle ---
|
||||||
def on_monitor_toggle(e):
|
def on_monitor_toggle(e):
|
||||||
|
|||||||
207
tab_batch_ng.py
207
tab_batch_ng.py
@@ -156,69 +156,68 @@ def render_batch_processor(state: AppState):
|
|||||||
batch_list = data.get(KEY_BATCH_DATA, [])
|
batch_list = data.get(KEY_BATCH_DATA, [])
|
||||||
|
|
||||||
# Source file data for importing
|
# Source file data for importing
|
||||||
json_files = sorted(state.current_dir.glob('*.json'))
|
with ui.card().classes('w-full q-pa-md q-mb-lg'):
|
||||||
json_files = [f for f in json_files if f.name not in (
|
json_files = sorted(state.current_dir.glob('*.json'))
|
||||||
'.editor_config.json', '.editor_snippets.json')]
|
json_files = [f for f in json_files if f.name not in (
|
||||||
file_options = {f.name: f.name for f in json_files}
|
'.editor_config.json', '.editor_snippets.json')]
|
||||||
|
file_options = {f.name: f.name for f in json_files}
|
||||||
|
|
||||||
src_file_select = ui.select(
|
src_file_select = ui.select(
|
||||||
file_options,
|
file_options,
|
||||||
value=file_path.name,
|
value=file_path.name,
|
||||||
label='Source File:',
|
label='Source File:',
|
||||||
).classes('w-64')
|
).classes('w-64')
|
||||||
|
|
||||||
src_seq_select = ui.select([], label='Source Sequence:').classes('w-64')
|
src_seq_select = ui.select([], label='Source Sequence:').classes('w-64')
|
||||||
|
|
||||||
# Track loaded source data
|
# Track loaded source data
|
||||||
_src_cache = {'data': None, 'batch': [], 'name': None}
|
_src_cache = {'data': None, 'batch': [], 'name': None}
|
||||||
|
|
||||||
def _update_src():
|
def _update_src():
|
||||||
name = src_file_select.value
|
name = src_file_select.value
|
||||||
if name and name != _src_cache['name']:
|
if name and name != _src_cache['name']:
|
||||||
src_data, _ = load_json(state.current_dir / name)
|
src_data, _ = load_json(state.current_dir / name)
|
||||||
_src_cache['data'] = src_data
|
_src_cache['data'] = src_data
|
||||||
_src_cache['batch'] = src_data.get(KEY_BATCH_DATA, [])
|
_src_cache['batch'] = src_data.get(KEY_BATCH_DATA, [])
|
||||||
_src_cache['name'] = name
|
_src_cache['name'] = name
|
||||||
if _src_cache['batch']:
|
if _src_cache['batch']:
|
||||||
opts = {i: format_seq_label(s.get(KEY_SEQUENCE_NUMBER, i+1))
|
opts = {i: format_seq_label(s.get(KEY_SEQUENCE_NUMBER, i+1))
|
||||||
for i, s in enumerate(_src_cache['batch'])}
|
for i, s in enumerate(_src_cache['batch'])}
|
||||||
src_seq_select.set_options(opts, value=0)
|
src_seq_select.set_options(opts, value=0)
|
||||||
else:
|
else:
|
||||||
src_seq_select.set_options({})
|
src_seq_select.set_options({})
|
||||||
|
|
||||||
src_file_select.on_value_change(lambda _: _update_src())
|
src_file_select.on_value_change(lambda _: _update_src())
|
||||||
_update_src()
|
_update_src()
|
||||||
|
|
||||||
# --- Add New Sequence ---
|
# --- Add New Sequence ---
|
||||||
ui.label('Add New Sequence').classes('text-subtitle1 q-mt-md')
|
ui.label('Add New Sequence').classes('section-header q-mt-md')
|
||||||
|
|
||||||
def _add_sequence(new_item):
|
def _add_sequence(new_item):
|
||||||
new_item[KEY_SEQUENCE_NUMBER] = max_main_seq_number(batch_list) + 1
|
new_item[KEY_SEQUENCE_NUMBER] = max_main_seq_number(batch_list) + 1
|
||||||
for k in [KEY_PROMPT_HISTORY, KEY_HISTORY_TREE, 'note', 'loras']:
|
for k in [KEY_PROMPT_HISTORY, KEY_HISTORY_TREE, 'note', 'loras']:
|
||||||
new_item.pop(k, None)
|
new_item.pop(k, None)
|
||||||
batch_list.append(new_item)
|
batch_list.append(new_item)
|
||||||
data[KEY_BATCH_DATA] = batch_list
|
data[KEY_BATCH_DATA] = batch_list
|
||||||
save_json(file_path, data)
|
save_json(file_path, data)
|
||||||
render_sequence_list.refresh()
|
render_sequence_list.refresh()
|
||||||
|
|
||||||
with ui.row():
|
with ui.row():
|
||||||
def add_empty():
|
def add_empty():
|
||||||
_add_sequence(DEFAULTS.copy())
|
_add_sequence(DEFAULTS.copy())
|
||||||
|
|
||||||
def add_from_source():
|
def add_from_source():
|
||||||
item = copy.deepcopy(DEFAULTS)
|
item = copy.deepcopy(DEFAULTS)
|
||||||
src_batch = _src_cache['batch']
|
src_batch = _src_cache['batch']
|
||||||
sel_idx = src_seq_select.value
|
sel_idx = src_seq_select.value
|
||||||
if src_batch and sel_idx is not None:
|
if src_batch and sel_idx is not None:
|
||||||
item.update(copy.deepcopy(src_batch[int(sel_idx)]))
|
item.update(copy.deepcopy(src_batch[int(sel_idx)]))
|
||||||
elif _src_cache['data']:
|
elif _src_cache['data']:
|
||||||
item.update(copy.deepcopy(_src_cache['data']))
|
item.update(copy.deepcopy(_src_cache['data']))
|
||||||
_add_sequence(item)
|
_add_sequence(item)
|
||||||
|
|
||||||
ui.button('Add Empty', icon='add', on_click=add_empty)
|
ui.button('Add Empty', icon='add', on_click=add_empty)
|
||||||
ui.button('From Source', icon='file_download', on_click=add_from_source)
|
ui.button('From Source', icon='file_download', on_click=add_from_source)
|
||||||
|
|
||||||
ui.separator()
|
|
||||||
|
|
||||||
# --- Standard / LoRA / VACE key sets ---
|
# --- Standard / LoRA / VACE key sets ---
|
||||||
lora_keys = ['lora 1 high', 'lora 1 low', 'lora 2 high', 'lora 2 low',
|
lora_keys = ['lora 1 high', 'lora 1 low', 'lora 2 high', 'lora 2 low',
|
||||||
@@ -250,36 +249,36 @@ def render_batch_processor(state: AppState):
|
|||||||
ui.button('Sort by Number', icon='sort', on_click=sort_by_number).props('flat')
|
ui.button('Sort by Number', icon='sort', on_click=sort_by_number).props('flat')
|
||||||
|
|
||||||
for i, seq in enumerate(batch_list):
|
for i, seq in enumerate(batch_list):
|
||||||
_render_sequence_card(
|
with ui.card().classes('w-full q-mb-sm'):
|
||||||
i, seq, batch_list, data, file_path, state,
|
_render_sequence_card(
|
||||||
_src_cache, src_seq_select,
|
i, seq, batch_list, data, file_path, state,
|
||||||
standard_keys, render_sequence_list,
|
_src_cache, src_seq_select,
|
||||||
)
|
standard_keys, render_sequence_list,
|
||||||
|
)
|
||||||
|
|
||||||
render_sequence_list()
|
render_sequence_list()
|
||||||
|
|
||||||
ui.separator()
|
|
||||||
|
|
||||||
# --- Save & Snap ---
|
# --- Save & Snap ---
|
||||||
with ui.row().classes('w-full items-end q-gutter-md'):
|
with ui.card().classes('w-full q-pa-md q-mt-lg'):
|
||||||
commit_input = ui.input('Change Note (Optional)',
|
with ui.row().classes('w-full items-end q-gutter-md'):
|
||||||
placeholder='e.g. Added sequence 3').classes('col')
|
commit_input = ui.input('Change Note (Optional)',
|
||||||
|
placeholder='e.g. Added sequence 3').classes('col')
|
||||||
|
|
||||||
def save_and_snap():
|
def save_and_snap():
|
||||||
data[KEY_BATCH_DATA] = batch_list
|
data[KEY_BATCH_DATA] = batch_list
|
||||||
tree_data = data.get(KEY_HISTORY_TREE, {})
|
tree_data = data.get(KEY_HISTORY_TREE, {})
|
||||||
htree = HistoryTree(tree_data)
|
htree = HistoryTree(tree_data)
|
||||||
snapshot_payload = copy.deepcopy(data)
|
snapshot_payload = copy.deepcopy(data)
|
||||||
snapshot_payload.pop(KEY_HISTORY_TREE, None)
|
snapshot_payload.pop(KEY_HISTORY_TREE, None)
|
||||||
note = commit_input.value if commit_input.value else 'Batch Update'
|
note = commit_input.value if commit_input.value else 'Batch Update'
|
||||||
htree.commit(snapshot_payload, note=note)
|
htree.commit(snapshot_payload, note=note)
|
||||||
data[KEY_HISTORY_TREE] = htree.to_dict()
|
data[KEY_HISTORY_TREE] = htree.to_dict()
|
||||||
save_json(file_path, data)
|
save_json(file_path, data)
|
||||||
state.restored_indicator = None
|
state.restored_indicator = None
|
||||||
commit_input.set_value('')
|
commit_input.set_value('')
|
||||||
ui.notify('Batch Saved & Snapshot Created!', type='positive')
|
ui.notify('Batch Saved & Snapshot Created!', type='positive')
|
||||||
|
|
||||||
ui.button('Save & Snap', icon='save', on_click=save_and_snap).props('color=primary')
|
ui.button('Save & Snap', icon='save', on_click=save_and_snap).props('color=primary')
|
||||||
|
|
||||||
|
|
||||||
# ======================================================================
|
# ======================================================================
|
||||||
@@ -321,7 +320,7 @@ def _render_sequence_card(i, seq, batch_list, data, file_path, state,
|
|||||||
batch_list[idx] = item
|
batch_list[idx] = item
|
||||||
commit('Copied!')
|
commit('Copied!')
|
||||||
|
|
||||||
ui.button('Copy Src', icon='file_download', on_click=copy_source).props('dense')
|
ui.button('Copy Src', icon='file_download', on_click=copy_source).props('outline')
|
||||||
|
|
||||||
# Clone Next
|
# Clone Next
|
||||||
def clone_next(idx=i, sn=seq_num, s=seq):
|
def clone_next(idx=i, sn=seq_num, s=seq):
|
||||||
@@ -334,7 +333,7 @@ def _render_sequence_card(i, seq, batch_list, data, file_path, state,
|
|||||||
batch_list.insert(pos, new_seq)
|
batch_list.insert(pos, new_seq)
|
||||||
commit('Cloned to Next!')
|
commit('Cloned to Next!')
|
||||||
|
|
||||||
ui.button('Clone Next', icon='content_copy', on_click=clone_next).props('dense')
|
ui.button('Clone Next', icon='content_copy', on_click=clone_next).props('outline')
|
||||||
|
|
||||||
# Clone End
|
# Clone End
|
||||||
def clone_end(s=seq):
|
def clone_end(s=seq):
|
||||||
@@ -343,7 +342,7 @@ def _render_sequence_card(i, seq, batch_list, data, file_path, state,
|
|||||||
batch_list.append(new_seq)
|
batch_list.append(new_seq)
|
||||||
commit('Cloned to End!')
|
commit('Cloned to End!')
|
||||||
|
|
||||||
ui.button('Clone End', icon='vertical_align_bottom', on_click=clone_end).props('dense')
|
ui.button('Clone End', icon='vertical_align_bottom', on_click=clone_end).props('outline')
|
||||||
|
|
||||||
# Clone Sub
|
# Clone Sub
|
||||||
def clone_sub(idx=i, sn=seq_num, s=seq):
|
def clone_sub(idx=i, sn=seq_num, s=seq):
|
||||||
@@ -360,7 +359,10 @@ def _render_sequence_card(i, seq, batch_list, data, file_path, state,
|
|||||||
batch_list.insert(pos, new_seq)
|
batch_list.insert(pos, new_seq)
|
||||||
commit(f'Created {format_seq_label(new_seq[KEY_SEQUENCE_NUMBER])}!')
|
commit(f'Created {format_seq_label(new_seq[KEY_SEQUENCE_NUMBER])}!')
|
||||||
|
|
||||||
ui.button('Clone Sub', icon='link', on_click=clone_sub).props('dense')
|
ui.button('Clone Sub', icon='link', on_click=clone_sub).props('outline')
|
||||||
|
|
||||||
|
# Spacer before Promote
|
||||||
|
ui.element('div').classes('col')
|
||||||
|
|
||||||
# Promote
|
# Promote
|
||||||
def promote(idx=i, s=seq):
|
def promote(idx=i, s=seq):
|
||||||
@@ -375,14 +377,17 @@ def _render_sequence_card(i, seq, batch_list, data, file_path, state,
|
|||||||
# and sees the file is now single (no KEY_BATCH_DATA)
|
# and sees the file is now single (no KEY_BATCH_DATA)
|
||||||
state._render_main.refresh()
|
state._render_main.refresh()
|
||||||
|
|
||||||
ui.button('Promote', icon='north_west', on_click=promote).props('dense')
|
ui.button('Promote', icon='north_west', on_click=promote)
|
||||||
|
|
||||||
|
# Spacer before Delete
|
||||||
|
ui.element('div').classes('col')
|
||||||
|
|
||||||
# Delete
|
# Delete
|
||||||
def delete(idx=i):
|
def delete(idx=i):
|
||||||
batch_list.pop(idx)
|
batch_list.pop(idx)
|
||||||
commit()
|
commit()
|
||||||
|
|
||||||
ui.button(icon='delete', on_click=delete).props('dense color=negative')
|
ui.button(icon='delete', on_click=delete).props('color=negative')
|
||||||
|
|
||||||
ui.separator()
|
ui.separator()
|
||||||
|
|
||||||
@@ -390,13 +395,13 @@ def _render_sequence_card(i, seq, batch_list, data, file_path, state,
|
|||||||
with ui.splitter(value=66).classes('w-full') as splitter:
|
with ui.splitter(value=66).classes('w-full') as splitter:
|
||||||
with splitter.before:
|
with splitter.before:
|
||||||
dict_textarea('General Prompt', seq, 'general_prompt').classes(
|
dict_textarea('General Prompt', seq, 'general_prompt').classes(
|
||||||
'w-full').props('outlined rows=2')
|
'w-full q-mt-sm').props('outlined rows=2')
|
||||||
dict_textarea('General Negative', seq, 'general_negative').classes(
|
dict_textarea('General Negative', seq, 'general_negative').classes(
|
||||||
'w-full').props('outlined rows=2')
|
'w-full q-mt-sm').props('outlined rows=2')
|
||||||
dict_textarea('Specific Prompt', seq, 'current_prompt').classes(
|
dict_textarea('Specific Prompt', seq, 'current_prompt').classes(
|
||||||
'w-full').props('outlined rows=10')
|
'w-full q-mt-sm').props('outlined rows=10')
|
||||||
dict_textarea('Specific Negative', seq, 'negative').classes(
|
dict_textarea('Specific Negative', seq, 'negative').classes(
|
||||||
'w-full').props('outlined rows=2')
|
'w-full q-mt-sm').props('outlined rows=2')
|
||||||
|
|
||||||
with splitter.after:
|
with splitter.after:
|
||||||
# Sequence number
|
# Sequence number
|
||||||
@@ -452,7 +457,7 @@ def _render_sequence_card(i, seq, batch_list, data, file_path, state,
|
|||||||
with ui.expansion('LoRA Settings', icon='style').classes('w-full'):
|
with ui.expansion('LoRA Settings', icon='style').classes('w-full'):
|
||||||
with ui.row().classes('w-full q-gutter-md'):
|
with ui.row().classes('w-full q-gutter-md'):
|
||||||
for lora_idx in range(1, 4):
|
for lora_idx in range(1, 4):
|
||||||
with ui.column().classes('col'):
|
with ui.card().classes('col q-pa-sm surface-3'):
|
||||||
ui.label(f'LoRA {lora_idx}').classes('text-subtitle2')
|
ui.label(f'LoRA {lora_idx}').classes('text-subtitle2')
|
||||||
for tier, tier_label in [('high', 'High'), ('low', 'Low')]:
|
for tier, tier_label in [('high', 'High'), ('low', 'Low')]:
|
||||||
k = f'lora {lora_idx} {tier}'
|
k = f'lora {lora_idx} {tier}'
|
||||||
@@ -474,8 +479,7 @@ def _render_sequence_card(i, seq, batch_list, data, file_path, state,
|
|||||||
lora_input.on('blur', on_lora_blur)
|
lora_input.on('blur', on_lora_blur)
|
||||||
|
|
||||||
# --- Custom Parameters ---
|
# --- Custom Parameters ---
|
||||||
ui.separator()
|
ui.label('Custom Parameters').classes('section-header q-mt-md')
|
||||||
ui.label('Custom Parameters').classes('text-caption')
|
|
||||||
|
|
||||||
custom_keys = [k for k in seq.keys() if k not in standard_keys]
|
custom_keys = [k for k in seq.keys() if k not in standard_keys]
|
||||||
if custom_keys:
|
if custom_keys:
|
||||||
@@ -537,11 +541,11 @@ def _render_vace_settings(i, seq, batch_list, data, file_path, refresh_list):
|
|||||||
|
|
||||||
ui.button('Shift', icon='arrow_downward', on_click=shift_fts).props('dense')
|
ui.button('Shift', icon='arrow_downward', on_click=shift_fts).props('dense')
|
||||||
|
|
||||||
dict_input(ui.input, 'Transition', seq, 'transition').props('outlined')
|
dict_input(ui.input, 'Transition', seq, 'transition').props('outlined').classes('q-mt-sm')
|
||||||
|
|
||||||
# VACE Schedule
|
# VACE Schedule
|
||||||
sched_val = max(0, min(int(seq.get('vace schedule', 1)), len(VACE_MODES) - 1))
|
sched_val = max(0, min(int(seq.get('vace schedule', 1)), len(VACE_MODES) - 1))
|
||||||
with ui.row().classes('w-full items-center'):
|
with ui.row().classes('w-full items-center q-mt-sm'):
|
||||||
vs_input = dict_number('VACE Schedule', seq, 'vace schedule', default=1,
|
vs_input = dict_number('VACE Schedule', seq, 'vace schedule', default=1,
|
||||||
min=0, max=len(VACE_MODES) - 1).classes('col').props('outlined')
|
min=0, max=len(VACE_MODES) - 1).classes('col').props('outlined')
|
||||||
mode_label = ui.label(VACE_MODES[sched_val]).classes('text-caption')
|
mode_label = ui.label(VACE_MODES[sched_val]).classes('text-caption')
|
||||||
@@ -566,8 +570,8 @@ def _render_vace_settings(i, seq, batch_list, data, file_path, refresh_list):
|
|||||||
ui.button('Mode Reference', icon='help', on_click=ref_dlg.open).props('flat dense')
|
ui.button('Mode Reference', icon='help', on_click=ref_dlg.open).props('flat dense')
|
||||||
|
|
||||||
# Input A / B frames
|
# Input A / B frames
|
||||||
ia_input = dict_number('Input A Frames', seq, 'input_a_frames').props('outlined')
|
ia_input = dict_number('Input A Frames', seq, 'input_a_frames').props('outlined').classes('q-mt-sm')
|
||||||
ib_input = dict_number('Input B Frames', seq, 'input_b_frames').props('outlined')
|
ib_input = dict_number('Input B Frames', seq, 'input_b_frames').props('outlined').classes('q-mt-sm')
|
||||||
|
|
||||||
# VACE Length + output calculation
|
# VACE Length + output calculation
|
||||||
input_a = int(seq.get('input_a_frames', 16))
|
input_a = int(seq.get('input_a_frames', 16))
|
||||||
@@ -582,7 +586,7 @@ def _render_vace_settings(i, seq, batch_list, data, file_path, refresh_list):
|
|||||||
else:
|
else:
|
||||||
base_length = max(stored_total - input_a - input_b, 1)
|
base_length = max(stored_total - input_a - input_b, 1)
|
||||||
|
|
||||||
with ui.row().classes('w-full items-center'):
|
with ui.row().classes('w-full items-center q-mt-sm'):
|
||||||
vl_input = ui.number('VACE Length', value=base_length, min=1).classes('col').props(
|
vl_input = ui.number('VACE Length', value=base_length, min=1).classes('col').props(
|
||||||
'outlined')
|
'outlined')
|
||||||
output_label = ui.label(f'Output: {stored_total}').classes('text-bold')
|
output_label = ui.label(f'Output: {stored_total}').classes('text-bold')
|
||||||
@@ -607,7 +611,7 @@ def _render_vace_settings(i, seq, batch_list, data, file_path, refresh_list):
|
|||||||
for inp in (vs_input, ia_input, ib_input, vl_input):
|
for inp in (vs_input, ia_input, ib_input, vl_input):
|
||||||
inp.on('update:model-value', recalc_vace)
|
inp.on('update:model-value', recalc_vace)
|
||||||
|
|
||||||
dict_number('Reference Switch', seq, 'reference switch').props('outlined')
|
dict_number('Reference Switch', seq, 'reference switch').props('outlined').classes('q-mt-sm')
|
||||||
|
|
||||||
|
|
||||||
# ======================================================================
|
# ======================================================================
|
||||||
@@ -638,13 +642,14 @@ def _render_mass_update(batch_list, data, file_path, state: AppState, refresh_li
|
|||||||
source_select.on_value_change(update_fields)
|
source_select.on_value_change(update_fields)
|
||||||
update_fields()
|
update_fields()
|
||||||
|
|
||||||
ui.label('Apply to:').classes('text-subtitle2 q-mt-md')
|
ui.label('Apply to:').classes('subsection-header q-mt-md')
|
||||||
select_all_cb = ui.checkbox('Select All')
|
select_all_cb = ui.checkbox('Select All')
|
||||||
target_checks = {}
|
target_checks = {}
|
||||||
for idx, s in enumerate(batch_list):
|
with ui.scroll_area().style('max-height: 250px'):
|
||||||
sn = s.get(KEY_SEQUENCE_NUMBER, idx + 1)
|
for idx, s in enumerate(batch_list):
|
||||||
cb = ui.checkbox(format_seq_label(sn))
|
sn = s.get(KEY_SEQUENCE_NUMBER, idx + 1)
|
||||||
target_checks[idx] = cb
|
cb = ui.checkbox(format_seq_label(sn))
|
||||||
|
target_checks[idx] = cb
|
||||||
|
|
||||||
def on_select_all(e):
|
def on_select_all(e):
|
||||||
for cb in target_checks.values():
|
for cb in target_checks.values():
|
||||||
|
|||||||
186
tab_comfy_ng.py
186
tab_comfy_ng.py
@@ -55,7 +55,7 @@ def render_comfy_monitor(state: AppState):
|
|||||||
|
|
||||||
# Add server section
|
# Add server section
|
||||||
ui.separator()
|
ui.separator()
|
||||||
ui.label('Add New Server').classes('text-subtitle1')
|
ui.label('Add New Server').classes('section-header')
|
||||||
with ui.row().classes('w-full items-end'):
|
with ui.row().classes('w-full items-end'):
|
||||||
new_name = ui.input('Server Name', placeholder='e.g. Render Node 2').classes('col')
|
new_name = ui.input('Server Name', placeholder='e.g. Render Node 2').classes('col')
|
||||||
new_url = ui.input('URL', placeholder='http://192.168.1.50:8188').classes('col')
|
new_url = ui.input('URL', placeholder='http://192.168.1.50:8188').classes('col')
|
||||||
@@ -152,18 +152,18 @@ def _render_single_instance(state: AppState, instance_config: dict, index: int,
|
|||||||
running_cnt = len(queue_data.get('queue_running', []))
|
running_cnt = len(queue_data.get('queue_running', []))
|
||||||
pending_cnt = len(queue_data.get('queue_pending', []))
|
pending_cnt = len(queue_data.get('queue_pending', []))
|
||||||
|
|
||||||
with ui.card().classes('q-pa-sm'):
|
with ui.card().classes('q-pa-md text-center').style('min-width: 100px'):
|
||||||
ui.label('Status')
|
ui.label('Status')
|
||||||
ui.label('Online' if running_cnt > 0 else 'Idle').classes(
|
ui.label('Online' if running_cnt > 0 else 'Idle').classes(
|
||||||
'text-positive' if running_cnt > 0 else 'text-grey')
|
'text-positive' if running_cnt > 0 else 'text-grey')
|
||||||
with ui.card().classes('q-pa-sm'):
|
with ui.card().classes('q-pa-md text-center').style('min-width: 100px'):
|
||||||
ui.label('Pending')
|
ui.label('Pending')
|
||||||
ui.label(str(pending_cnt))
|
ui.label(str(pending_cnt))
|
||||||
with ui.card().classes('q-pa-sm'):
|
with ui.card().classes('q-pa-md text-center').style('min-width: 100px'):
|
||||||
ui.label('Running')
|
ui.label('Running')
|
||||||
ui.label(str(running_cnt))
|
ui.label(str(running_cnt))
|
||||||
else:
|
else:
|
||||||
with ui.card().classes('q-pa-sm'):
|
with ui.card().classes('q-pa-md text-center').style('min-width: 100px'):
|
||||||
ui.label('Status')
|
ui.label('Status')
|
||||||
ui.label('Offline').classes('text-negative')
|
ui.label('Offline').classes('text-negative')
|
||||||
ui.label(f'Could not connect to {comfy_url}').classes('text-negative')
|
ui.label(f'Could not connect to {comfy_url}').classes('text-negative')
|
||||||
@@ -173,104 +173,106 @@ def _render_single_instance(state: AppState, instance_config: dict, index: int,
|
|||||||
ui.button('Refresh Status', icon='refresh', on_click=refresh_status).props('flat dense')
|
ui.button('Refresh Status', icon='refresh', on_click=refresh_status).props('flat dense')
|
||||||
|
|
||||||
# --- Live View ---
|
# --- Live View ---
|
||||||
ui.label('Live View').classes('text-subtitle1 q-mt-md')
|
with ui.card().classes('w-full q-pa-md q-mt-md'):
|
||||||
toggle_key = f'live_toggle_{index}'
|
ui.label('Live View').classes('section-header')
|
||||||
|
toggle_key = f'live_toggle_{index}'
|
||||||
|
|
||||||
live_checkbox = ui.checkbox('Enable Live Preview', value=False)
|
live_checkbox = ui.checkbox('Enable Live Preview', value=False)
|
||||||
# Store reference so poll_all timer can disable it on timeout
|
# Store reference so poll_all timer can disable it on timeout
|
||||||
state._live_checkboxes[toggle_key] = live_checkbox
|
state._live_checkboxes[toggle_key] = live_checkbox
|
||||||
|
|
||||||
@ui.refreshable
|
@ui.refreshable
|
||||||
def render_live_view():
|
def render_live_view():
|
||||||
if not live_checkbox.value:
|
if not live_checkbox.value:
|
||||||
ui.label('Live Preview is disabled.').classes('text-caption')
|
ui.label('Live Preview is disabled.').classes('text-caption')
|
||||||
return
|
|
||||||
|
|
||||||
# Record start time
|
|
||||||
if toggle_key not in state.live_toggles or state.live_toggles.get(toggle_key) is None:
|
|
||||||
state.live_toggles[toggle_key] = time.time()
|
|
||||||
|
|
||||||
timeout_val = config.get('monitor_timeout', 0)
|
|
||||||
if timeout_val > 0:
|
|
||||||
start = state.live_toggles.get(toggle_key, time.time())
|
|
||||||
remaining = (timeout_val * 60) - (time.time() - start)
|
|
||||||
if remaining <= 0:
|
|
||||||
live_checkbox.set_value(False)
|
|
||||||
state.live_toggles[toggle_key] = None
|
|
||||||
ui.label('Preview timed out.').classes('text-caption')
|
|
||||||
return
|
return
|
||||||
ui.label(f'Auto-off in: {int(remaining)}s').classes('text-caption')
|
|
||||||
|
|
||||||
iframe_h = ui.slider(min=600, max=2500, step=50, value=1000).classes('w-full')
|
# Record start time
|
||||||
ui.label().bind_text_from(iframe_h, 'value', backward=lambda v: f'Height: {v}px')
|
if toggle_key not in state.live_toggles or state.live_toggles.get(toggle_key) is None:
|
||||||
|
state.live_toggles[toggle_key] = time.time()
|
||||||
|
|
||||||
viewer_base = config.get('viewer_url', '').strip()
|
timeout_val = config.get('monitor_timeout', 0)
|
||||||
parsed = urllib.parse.urlparse(viewer_base)
|
if timeout_val > 0:
|
||||||
if viewer_base and parsed.scheme in ('http', 'https'):
|
start = state.live_toggles.get(toggle_key, time.time())
|
||||||
safe_src = html.escape(viewer_base, quote=True)
|
remaining = (timeout_val * 60) - (time.time() - start)
|
||||||
ui.label(f'Viewing: {viewer_base}').classes('text-caption')
|
if remaining <= 0:
|
||||||
|
live_checkbox.set_value(False)
|
||||||
|
state.live_toggles[toggle_key] = None
|
||||||
|
ui.label('Preview timed out.').classes('text-caption')
|
||||||
|
return
|
||||||
|
ui.label(f'Auto-off in: {int(remaining)}s').classes('text-caption')
|
||||||
|
|
||||||
iframe_container = ui.column().classes('w-full')
|
iframe_h = ui.slider(min=600, max=2500, step=50, value=1000).classes('w-full')
|
||||||
|
ui.label().bind_text_from(iframe_h, 'value', backward=lambda v: f'Height: {v}px')
|
||||||
|
|
||||||
def update_iframe():
|
viewer_base = config.get('viewer_url', '').strip()
|
||||||
iframe_container.clear()
|
parsed = urllib.parse.urlparse(viewer_base)
|
||||||
with iframe_container:
|
if viewer_base and parsed.scheme in ('http', 'https'):
|
||||||
ui.html(
|
safe_src = html.escape(viewer_base, quote=True)
|
||||||
f'<iframe src="{safe_src}" width="100%" height="{int(iframe_h.value)}px"'
|
ui.label(f'Viewing: {viewer_base}').classes('text-caption')
|
||||||
f' style="border: 2px solid #666; border-radius: 8px;"></iframe>'
|
|
||||||
)
|
|
||||||
|
|
||||||
iframe_h.on_value_change(lambda _: update_iframe())
|
iframe_container = ui.column().classes('w-full')
|
||||||
update_iframe()
|
|
||||||
else:
|
|
||||||
ui.label('No valid viewer URL configured.').classes('text-warning')
|
|
||||||
|
|
||||||
state._live_refreshables[toggle_key] = render_live_view
|
def update_iframe():
|
||||||
live_checkbox.on_value_change(lambda _: render_live_view.refresh())
|
iframe_container.clear()
|
||||||
render_live_view()
|
with iframe_container:
|
||||||
|
ui.html(
|
||||||
|
f'<iframe src="{safe_src}" width="100%" height="{int(iframe_h.value)}px"'
|
||||||
|
f' style="border: 2px solid #666; border-radius: 8px;"></iframe>'
|
||||||
|
)
|
||||||
|
|
||||||
|
iframe_h.on_value_change(lambda _: update_iframe())
|
||||||
|
update_iframe()
|
||||||
|
else:
|
||||||
|
ui.label('No valid viewer URL configured.').classes('text-warning')
|
||||||
|
|
||||||
|
state._live_refreshables[toggle_key] = render_live_view
|
||||||
|
live_checkbox.on_value_change(lambda _: render_live_view.refresh())
|
||||||
|
render_live_view()
|
||||||
|
|
||||||
# --- Latest Output ---
|
# --- Latest Output ---
|
||||||
ui.label('Latest Output').classes('text-subtitle1 q-mt-md')
|
with ui.card().classes('w-full q-pa-md q-mt-md'):
|
||||||
img_container = ui.column().classes('w-full')
|
ui.label('Latest Output').classes('section-header')
|
||||||
|
img_container = ui.column().classes('w-full')
|
||||||
|
|
||||||
async def check_image():
|
async def check_image():
|
||||||
img_container.clear()
|
img_container.clear()
|
||||||
loop = asyncio.get_event_loop()
|
loop = asyncio.get_event_loop()
|
||||||
res, err = await loop.run_in_executor(
|
res, err = await loop.run_in_executor(
|
||||||
None, lambda: _fetch_blocking(f'{comfy_url}/history', timeout=2))
|
None, lambda: _fetch_blocking(f'{comfy_url}/history', timeout=2))
|
||||||
with img_container:
|
with img_container:
|
||||||
if err is not None:
|
if err is not None:
|
||||||
ui.label(f'Error fetching image: {err}').classes('text-negative')
|
ui.label(f'Error fetching image: {err}').classes('text-negative')
|
||||||
return
|
return
|
||||||
try:
|
try:
|
||||||
history = res.json()
|
history = res.json()
|
||||||
except (ValueError, Exception):
|
except (ValueError, Exception):
|
||||||
ui.label('Invalid response from server').classes('text-negative')
|
ui.label('Invalid response from server').classes('text-negative')
|
||||||
return
|
return
|
||||||
if not history:
|
if not history:
|
||||||
ui.label('No history found.').classes('text-caption')
|
ui.label('No history found.').classes('text-caption')
|
||||||
return
|
return
|
||||||
last_prompt_id = list(history.keys())[-1]
|
last_prompt_id = list(history.keys())[-1]
|
||||||
outputs = history[last_prompt_id].get('outputs', {})
|
outputs = history[last_prompt_id].get('outputs', {})
|
||||||
found_img = None
|
found_img = None
|
||||||
for node_output in outputs.values():
|
for node_output in outputs.values():
|
||||||
if 'images' in node_output:
|
if 'images' in node_output:
|
||||||
for img_info in node_output['images']:
|
for img_info in node_output['images']:
|
||||||
if img_info['type'] == 'output':
|
if img_info['type'] == 'output':
|
||||||
found_img = img_info
|
found_img = img_info
|
||||||
break
|
break
|
||||||
|
if found_img:
|
||||||
|
break
|
||||||
if found_img:
|
if found_img:
|
||||||
break
|
params = urllib.parse.urlencode({
|
||||||
if found_img:
|
'filename': found_img['filename'],
|
||||||
params = urllib.parse.urlencode({
|
'subfolder': found_img['subfolder'],
|
||||||
'filename': found_img['filename'],
|
'type': found_img['type'],
|
||||||
'subfolder': found_img['subfolder'],
|
})
|
||||||
'type': found_img['type'],
|
img_url = f'{comfy_url}/view?{params}'
|
||||||
})
|
ui.image(img_url).classes('w-full').style('max-width: 600px')
|
||||||
img_url = f'{comfy_url}/view?{params}'
|
ui.label(f'Last Output: {found_img["filename"]}').classes('text-caption')
|
||||||
ui.image(img_url).classes('w-full').style('max-width: 600px')
|
else:
|
||||||
ui.label(f'Last Output: {found_img["filename"]}').classes('text-caption')
|
ui.label('Last run had no image output.').classes('text-caption')
|
||||||
else:
|
|
||||||
ui.label('Last run had no image output.').classes('text-caption')
|
|
||||||
|
|
||||||
ui.button('Check Latest Image', icon='image', on_click=check_image).props('flat')
|
ui.button('Check Latest Image', icon='image', on_click=check_image).props('flat')
|
||||||
|
|||||||
107
tab_raw_ng.py
107
tab_raw_ng.py
@@ -11,64 +11,63 @@ def render_raw_editor(state: AppState):
|
|||||||
data = state.data_cache
|
data = state.data_cache
|
||||||
file_path = state.file_path
|
file_path = state.file_path
|
||||||
|
|
||||||
ui.label(f'Raw Editor: {file_path.name}').classes('text-h6')
|
with ui.card().classes('w-full q-pa-md'):
|
||||||
|
ui.label(f'Raw Editor: {file_path.name}').classes('text-h6 q-mb-md')
|
||||||
|
|
||||||
hide_history = ui.checkbox(
|
hide_history = ui.checkbox(
|
||||||
'Hide History (Safe Mode)',
|
'Hide History (Safe Mode)',
|
||||||
value=True,
|
value=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
@ui.refreshable
|
@ui.refreshable
|
||||||
def render_editor():
|
def render_editor():
|
||||||
# Prepare display data
|
# Prepare display data
|
||||||
if hide_history.value:
|
if hide_history.value:
|
||||||
display_data = copy.deepcopy(data)
|
display_data = copy.deepcopy(data)
|
||||||
display_data.pop(KEY_HISTORY_TREE, None)
|
display_data.pop(KEY_HISTORY_TREE, None)
|
||||||
display_data.pop(KEY_PROMPT_HISTORY, None)
|
display_data.pop(KEY_PROMPT_HISTORY, None)
|
||||||
else:
|
else:
|
||||||
display_data = data
|
display_data = data
|
||||||
|
|
||||||
try:
|
|
||||||
json_str = json.dumps(display_data, indent=4, ensure_ascii=False)
|
|
||||||
except Exception as e:
|
|
||||||
ui.notify(f'Error serializing JSON: {e}', type='negative')
|
|
||||||
json_str = '{}'
|
|
||||||
|
|
||||||
text_area = ui.textarea(
|
|
||||||
'JSON Content',
|
|
||||||
value=json_str,
|
|
||||||
).classes('w-full font-mono').props('outlined rows=30')
|
|
||||||
|
|
||||||
ui.separator()
|
|
||||||
|
|
||||||
def do_save():
|
|
||||||
try:
|
try:
|
||||||
input_data = json.loads(text_area.value)
|
json_str = json.dumps(display_data, indent=4, ensure_ascii=False)
|
||||||
|
|
||||||
# Merge hidden history back in if safe mode
|
|
||||||
if hide_history.value:
|
|
||||||
if KEY_HISTORY_TREE in data:
|
|
||||||
input_data[KEY_HISTORY_TREE] = data[KEY_HISTORY_TREE]
|
|
||||||
if KEY_PROMPT_HISTORY in data:
|
|
||||||
input_data[KEY_PROMPT_HISTORY] = data[KEY_PROMPT_HISTORY]
|
|
||||||
|
|
||||||
save_json(file_path, input_data)
|
|
||||||
|
|
||||||
data.clear()
|
|
||||||
data.update(input_data)
|
|
||||||
state.last_mtime = get_file_mtime(file_path)
|
|
||||||
|
|
||||||
ui.notify('Raw JSON Saved Successfully!', type='positive')
|
|
||||||
render_editor.refresh()
|
|
||||||
|
|
||||||
except json.JSONDecodeError as e:
|
|
||||||
ui.notify(f'Invalid JSON Syntax: {e}', type='negative')
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
ui.notify(f'Unexpected Error: {e}', type='negative')
|
ui.notify(f'Error serializing JSON: {e}', type='negative')
|
||||||
|
json_str = '{}'
|
||||||
|
|
||||||
ui.button('Save Raw Changes', icon='save', on_click=do_save).props(
|
text_area = ui.textarea(
|
||||||
'color=primary'
|
'JSON Content',
|
||||||
).classes('w-full')
|
value=json_str,
|
||||||
|
).classes('w-full font-mono').props('outlined rows=30')
|
||||||
|
|
||||||
hide_history.on_value_change(lambda _: render_editor.refresh())
|
def do_save():
|
||||||
render_editor()
|
try:
|
||||||
|
input_data = json.loads(text_area.value)
|
||||||
|
|
||||||
|
# Merge hidden history back in if safe mode
|
||||||
|
if hide_history.value:
|
||||||
|
if KEY_HISTORY_TREE in data:
|
||||||
|
input_data[KEY_HISTORY_TREE] = data[KEY_HISTORY_TREE]
|
||||||
|
if KEY_PROMPT_HISTORY in data:
|
||||||
|
input_data[KEY_PROMPT_HISTORY] = data[KEY_PROMPT_HISTORY]
|
||||||
|
|
||||||
|
save_json(file_path, input_data)
|
||||||
|
|
||||||
|
data.clear()
|
||||||
|
data.update(input_data)
|
||||||
|
state.last_mtime = get_file_mtime(file_path)
|
||||||
|
|
||||||
|
ui.notify('Raw JSON Saved Successfully!', type='positive')
|
||||||
|
render_editor.refresh()
|
||||||
|
|
||||||
|
except json.JSONDecodeError as e:
|
||||||
|
ui.notify(f'Invalid JSON Syntax: {e}', type='negative')
|
||||||
|
except Exception as e:
|
||||||
|
ui.notify(f'Unexpected Error: {e}', type='negative')
|
||||||
|
|
||||||
|
ui.button('Save Raw Changes', icon='save', on_click=do_save).props(
|
||||||
|
'color=primary'
|
||||||
|
).classes('w-full q-mt-md')
|
||||||
|
|
||||||
|
hide_history.on_value_change(lambda _: render_editor.refresh())
|
||||||
|
render_editor()
|
||||||
|
|||||||
@@ -68,11 +68,12 @@ def _render_graph_or_log(mode, all_nodes, htree, selected_nodes,
|
|||||||
"""Render graph visualization or linear log view."""
|
"""Render graph visualization or linear log view."""
|
||||||
if mode in ('Horizontal', 'Vertical'):
|
if mode in ('Horizontal', 'Vertical'):
|
||||||
direction = 'LR' if mode == 'Horizontal' else 'TB'
|
direction = 'LR' if mode == 'Horizontal' else 'TB'
|
||||||
try:
|
with ui.card().classes('w-full q-pa-md'):
|
||||||
graph_dot = htree.generate_graph(direction=direction)
|
try:
|
||||||
_render_graphviz(graph_dot)
|
graph_dot = htree.generate_graph(direction=direction)
|
||||||
except Exception as e:
|
_render_graphviz(graph_dot)
|
||||||
ui.label(f'Graph Error: {e}').classes('text-negative')
|
except Exception as e:
|
||||||
|
ui.label(f'Graph Error: {e}').classes('text-negative')
|
||||||
|
|
||||||
elif mode == 'Linear Log':
|
elif mode == 'Linear Log':
|
||||||
ui.label('Chronological list of all snapshots.').classes('text-caption')
|
ui.label('Chronological list of all snapshots.').classes('text-caption')
|
||||||
@@ -82,9 +83,9 @@ def _render_graph_or_log(mode, all_nodes, htree, selected_nodes,
|
|||||||
|
|
||||||
card_style = ''
|
card_style = ''
|
||||||
if is_selected:
|
if is_selected:
|
||||||
card_style = 'background: #3d1f1f !important;'
|
card_style = 'background: rgba(239, 68, 68, 0.1) !important; border-left: 3px solid var(--negative);'
|
||||||
elif is_head:
|
elif is_head:
|
||||||
card_style = 'background: #1a2332 !important;'
|
card_style = 'background: var(--accent-subtle) !important; border-left: 3px solid var(--accent);'
|
||||||
with ui.card().classes('w-full q-mb-sm').style(card_style):
|
with ui.card().classes('w-full q-mb-sm').style(card_style):
|
||||||
with ui.row().classes('w-full items-center'):
|
with ui.row().classes('w-full items-center'):
|
||||||
if selection_mode_on:
|
if selection_mode_on:
|
||||||
@@ -145,7 +146,7 @@ def _render_batch_delete(htree, data, file_path, state, refresh_fn):
|
|||||||
|
|
||||||
def _render_node_manager(all_nodes, htree, data, file_path, restore_fn, refresh_fn):
|
def _render_node_manager(all_nodes, htree, data, file_path, restore_fn, refresh_fn):
|
||||||
"""Render node selector with restore, rename, delete, and preview."""
|
"""Render node selector with restore, rename, delete, and preview."""
|
||||||
ui.label('Manage Version').classes('text-subtitle1 q-mt-md')
|
ui.label('Manage Version').classes('section-header')
|
||||||
|
|
||||||
def fmt_node(n):
|
def fmt_node(n):
|
||||||
ts = time.strftime('%b %d %H:%M', time.localtime(n['timestamp']))
|
ts = time.strftime('%b %d %H:%M', time.localtime(n['timestamp']))
|
||||||
@@ -186,7 +187,7 @@ def _render_node_manager(all_nodes, htree, data, file_path, restore_fn, refresh_
|
|||||||
ui.button('Update Label', on_click=rename_node).props('flat')
|
ui.button('Update Label', on_click=rename_node).props('flat')
|
||||||
|
|
||||||
# Danger zone
|
# Danger zone
|
||||||
with ui.expansion('Danger Zone (Delete)', icon='warning').classes('w-full q-mt-md'):
|
with ui.expansion('Danger Zone (Delete)', icon='warning').classes('w-full q-mt-md').style('border-left: 3px solid var(--negative)'):
|
||||||
ui.label('Deleting a node cannot be undone.').classes('text-warning')
|
ui.label('Deleting a node cannot be undone.').classes('text-warning')
|
||||||
|
|
||||||
def delete_selected():
|
def delete_selected():
|
||||||
@@ -226,7 +227,7 @@ def render_timeline_tab(state: AppState):
|
|||||||
'text-info q-pa-sm')
|
'text-info q-pa-sm')
|
||||||
|
|
||||||
# --- View mode + Selection toggle ---
|
# --- View mode + Selection toggle ---
|
||||||
with ui.row().classes('w-full items-center q-gutter-md'):
|
with ui.row().classes('w-full items-center q-gutter-md q-mb-md'):
|
||||||
ui.label('Version History').classes('text-h6 col')
|
ui.label('Version History').classes('text-h6 col')
|
||||||
view_mode = ui.toggle(
|
view_mode = ui.toggle(
|
||||||
['Horizontal', 'Vertical', 'Linear Log'],
|
['Horizontal', 'Vertical', 'Linear Log'],
|
||||||
@@ -249,11 +250,10 @@ def render_timeline_tab(state: AppState):
|
|||||||
if selection_mode.value and state.timeline_selected_nodes:
|
if selection_mode.value and state.timeline_selected_nodes:
|
||||||
_render_batch_delete(htree, data, file_path, state, render_timeline.refresh)
|
_render_batch_delete(htree, data, file_path, state, render_timeline.refresh)
|
||||||
|
|
||||||
ui.separator()
|
with ui.card().classes('w-full q-pa-md q-mt-md'):
|
||||||
|
_render_node_manager(
|
||||||
_render_node_manager(
|
all_nodes, htree, data, file_path,
|
||||||
all_nodes, htree, data, file_path,
|
_restore_and_refresh, render_timeline.refresh)
|
||||||
_restore_and_refresh, render_timeline.refresh)
|
|
||||||
|
|
||||||
def _toggle_select(nid, checked):
|
def _toggle_select(nid, checked):
|
||||||
if checked:
|
if checked:
|
||||||
|
|||||||
Reference in New Issue
Block a user