Ethanfel
b02bf124fb
- save_json() now writes to a temp file then uses os.replace() for atomic writes - Replace hardcoded "batch_data", "history_tree", "prompt_history", "sequence_number" strings with constants (KEY_BATCH_DATA, etc.) across all modules - Add 29 unit tests for history_tree, utils, and json_loader - Add type hints to public functions in utils.py, json_loader.py, history_tree.py - Remove ALLOWED_BASE_DIR restriction that blocked navigating outside app CWD - Fix path text input not updating on navigation by using session state key - Add unpin button (❌) for removing pinned folders Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
235 lines
9.0 KiB
Python
235 lines
9.0 KiB
Python
import streamlit as st
|
||
import random
|
||
from pathlib import Path
|
||
|
||
# --- Import Custom Modules ---
|
||
from utils import (
|
||
load_config, save_config, load_snippets, save_snippets,
|
||
load_json, save_json, generate_templates, DEFAULTS, ALLOWED_BASE_DIR,
|
||
KEY_BATCH_DATA, KEY_PROMPT_HISTORY,
|
||
)
|
||
from tab_single import render_single_editor
|
||
from tab_batch import render_batch_processor
|
||
from tab_timeline import render_timeline_tab
|
||
from tab_timeline_wip import render_timeline_wip
|
||
from tab_comfy import render_comfy_monitor
|
||
from tab_raw import render_raw_editor
|
||
|
||
# ==========================================
|
||
# 1. PAGE CONFIGURATION
|
||
# ==========================================
|
||
st.set_page_config(layout="wide", page_title="AI Settings Manager")
|
||
|
||
# ==========================================
|
||
# 2. SESSION STATE INITIALIZATION
|
||
# ==========================================
|
||
_SESSION_DEFAULTS = {
|
||
"snippets": load_snippets,
|
||
"loaded_file": lambda: None,
|
||
"last_mtime": lambda: 0,
|
||
"edit_history_idx": lambda: None,
|
||
"single_editor_cache": lambda: DEFAULTS.copy(),
|
||
"ui_reset_token": lambda: 0,
|
||
"active_tab_name": lambda: "📝 Single Editor",
|
||
}
|
||
|
||
if 'config' not in st.session_state:
|
||
st.session_state.config = load_config()
|
||
st.session_state.current_dir = Path(st.session_state.config.get("last_dir", Path.cwd()))
|
||
|
||
for key, factory in _SESSION_DEFAULTS.items():
|
||
if key not in st.session_state:
|
||
st.session_state[key] = factory()
|
||
|
||
# ==========================================
|
||
# 3. SIDEBAR (NAVIGATOR & TOOLS)
|
||
# ==========================================
|
||
with st.sidebar:
|
||
st.header("📂 Navigator")
|
||
|
||
# --- Path Navigator ---
|
||
# Sync widget key with current_dir so the text input always reflects the actual path
|
||
if "nav_path_input" not in st.session_state:
|
||
st.session_state.nav_path_input = str(st.session_state.current_dir)
|
||
|
||
new_path = st.text_input("Current Path", key="nav_path_input")
|
||
if new_path != str(st.session_state.current_dir):
|
||
p = Path(new_path).resolve()
|
||
if p.exists() and p.is_dir():
|
||
st.session_state.current_dir = p
|
||
st.session_state.config['last_dir'] = str(p)
|
||
save_config(st.session_state.current_dir, st.session_state.config['favorites'])
|
||
st.rerun()
|
||
elif new_path.strip():
|
||
st.error(f"Path does not exist or is not a directory: {new_path}")
|
||
|
||
# --- Favorites System ---
|
||
pin_col, unpin_col = st.columns(2)
|
||
with pin_col:
|
||
if st.button("📌 Pin Folder", use_container_width=True):
|
||
if str(st.session_state.current_dir) not in st.session_state.config['favorites']:
|
||
st.session_state.config['favorites'].append(str(st.session_state.current_dir))
|
||
save_config(st.session_state.current_dir, st.session_state.config['favorites'])
|
||
st.rerun()
|
||
|
||
favorites = st.session_state.config['favorites']
|
||
if favorites:
|
||
fav_selection = st.radio(
|
||
"Jump to:",
|
||
["Select..."] + favorites,
|
||
index=0,
|
||
label_visibility="collapsed"
|
||
)
|
||
if fav_selection != "Select..." and fav_selection != str(st.session_state.current_dir):
|
||
st.session_state.current_dir = Path(fav_selection)
|
||
st.session_state.nav_path_input = fav_selection
|
||
st.rerun()
|
||
|
||
# Unpin buttons for each favorite
|
||
for fav in favorites:
|
||
fc1, fc2 = st.columns([4, 1])
|
||
fc1.caption(fav)
|
||
if fc2.button("❌", key=f"unpin_{fav}"):
|
||
st.session_state.config['favorites'].remove(fav)
|
||
save_config(st.session_state.current_dir, st.session_state.config['favorites'])
|
||
st.rerun()
|
||
|
||
st.markdown("---")
|
||
|
||
# --- Snippet Library ---
|
||
st.subheader("🧩 Snippet Library")
|
||
with st.expander("Add New Snippet"):
|
||
snip_name = st.text_input("Name", placeholder="e.g. Cinematic")
|
||
snip_content = st.text_area("Content", placeholder="4k, high quality...")
|
||
if st.button("Save Snippet"):
|
||
if snip_name and snip_content:
|
||
st.session_state.snippets[snip_name] = snip_content
|
||
save_snippets(st.session_state.snippets)
|
||
st.success(f"Saved '{snip_name}'")
|
||
st.rerun()
|
||
|
||
if st.session_state.snippets:
|
||
st.caption("Click to Append to Prompt:")
|
||
for name, content in st.session_state.snippets.items():
|
||
col_s1, col_s2 = st.columns([4, 1])
|
||
if col_s1.button(f"➕ {name}", use_container_width=True):
|
||
st.session_state.append_prompt = content
|
||
st.rerun()
|
||
if col_s2.button("🗑️", key=f"del_snip_{name}"):
|
||
del st.session_state.snippets[name]
|
||
save_snippets(st.session_state.snippets)
|
||
st.rerun()
|
||
|
||
st.markdown("---")
|
||
|
||
# --- File List & Creation ---
|
||
json_files = sorted(list(st.session_state.current_dir.glob("*.json")))
|
||
json_files = [f for f in json_files if f.name != ".editor_config.json" and f.name != ".editor_snippets.json"]
|
||
|
||
if not json_files:
|
||
if st.button("Generate Templates"):
|
||
generate_templates(st.session_state.current_dir)
|
||
st.rerun()
|
||
|
||
with st.expander("Create New JSON"):
|
||
new_filename = st.text_input("Filename", placeholder="my_prompt_vace")
|
||
is_batch = st.checkbox("Is Batch File?")
|
||
if st.button("Create"):
|
||
if not new_filename.endswith(".json"): new_filename += ".json"
|
||
path = st.session_state.current_dir / new_filename
|
||
if is_batch:
|
||
data = {KEY_BATCH_DATA: []}
|
||
else:
|
||
data = DEFAULTS.copy()
|
||
if "vace" in new_filename: data.update({"frame_to_skip": 81, "vace schedule": 1, "video file path": ""})
|
||
elif "i2v" in new_filename: data.update({"reference image path": "", "flf image path": ""})
|
||
save_json(path, data)
|
||
st.rerun()
|
||
|
||
# --- File Selector ---
|
||
if 'file_selector' not in st.session_state:
|
||
st.session_state.file_selector = json_files[0].name if json_files else None
|
||
if st.session_state.file_selector not in [f.name for f in json_files] and json_files:
|
||
st.session_state.file_selector = json_files[0].name
|
||
|
||
selected_file_name = st.radio("Select File", [f.name for f in json_files], key="file_selector")
|
||
|
||
# --- GLOBAL MONITOR TOGGLE (NEW) ---
|
||
st.markdown("---")
|
||
show_monitor = st.checkbox("Show Comfy Monitor", value=True)
|
||
|
||
# ==========================================
|
||
# 4. MAIN APP LOGIC
|
||
# ==========================================
|
||
if selected_file_name:
|
||
file_path = st.session_state.current_dir / selected_file_name
|
||
|
||
# --- FILE LOADING & AUTO-SWITCH LOGIC ---
|
||
if st.session_state.loaded_file != str(file_path):
|
||
data, mtime = load_json(file_path)
|
||
st.session_state.data_cache = data
|
||
st.session_state.last_mtime = mtime
|
||
st.session_state.loaded_file = str(file_path)
|
||
|
||
# Clear transient states
|
||
if 'append_prompt' in st.session_state: del st.session_state.append_prompt
|
||
if 'rand_seed' in st.session_state: del st.session_state.rand_seed
|
||
if 'restored_indicator' in st.session_state: del st.session_state.restored_indicator
|
||
st.session_state.edit_history_idx = None
|
||
|
||
# --- AUTO-SWITCH TAB LOGIC ---
|
||
is_batch = KEY_BATCH_DATA in data or isinstance(data, list)
|
||
if is_batch:
|
||
st.session_state.active_tab_name = "🚀 Batch Processor"
|
||
else:
|
||
st.session_state.active_tab_name = "📝 Single Editor"
|
||
|
||
else:
|
||
data = st.session_state.data_cache
|
||
|
||
st.title(f"Editing: {selected_file_name}")
|
||
|
||
# --- CONTROLLED NAVIGATION ---
|
||
# Removed "🔌 Comfy Monitor" from this list
|
||
tabs_list = [
|
||
"📝 Single Editor",
|
||
"🚀 Batch Processor",
|
||
"🕒 Timeline",
|
||
"🧪 Interactive Timeline",
|
||
"💻 Raw Editor"
|
||
]
|
||
|
||
if st.session_state.active_tab_name not in tabs_list:
|
||
st.session_state.active_tab_name = tabs_list[0]
|
||
|
||
current_tab = st.radio(
|
||
"Navigation",
|
||
tabs_list,
|
||
key="active_tab_name",
|
||
horizontal=True,
|
||
label_visibility="collapsed"
|
||
)
|
||
|
||
st.markdown("---")
|
||
|
||
# --- RENDER EDITOR TABS ---
|
||
if current_tab == "📝 Single Editor":
|
||
render_single_editor(data, file_path)
|
||
|
||
elif current_tab == "🚀 Batch Processor":
|
||
render_batch_processor(data, file_path, json_files, st.session_state.current_dir, selected_file_name)
|
||
|
||
elif current_tab == "🕒 Timeline":
|
||
render_timeline_tab(data, file_path)
|
||
|
||
elif current_tab == "🧪 Interactive Timeline":
|
||
render_timeline_wip(data, file_path)
|
||
|
||
elif current_tab == "💻 Raw Editor":
|
||
render_raw_editor(data, file_path)
|
||
|
||
# --- GLOBAL PERSISTENT MONITOR ---
|
||
if show_monitor:
|
||
st.markdown("---")
|
||
with st.expander("🔌 ComfyUI Monitor", expanded=True):
|
||
render_comfy_monitor() |