Update tab_batch.py
This commit is contained in:
21
tab_batch.py
21
tab_batch.py
@@ -36,7 +36,6 @@ def render_batch_processor(data, file_path, json_files, current_dir, selected_fi
|
|||||||
st.button("✨ Create Batch Copy", on_click=create_batch_callback, args=(selected_file_name, data, current_dir))
|
st.button("✨ Create Batch Copy", on_click=create_batch_callback, args=(selected_file_name, data, current_dir))
|
||||||
return
|
return
|
||||||
|
|
||||||
# --- 1. RESTORED STATE INDICATOR ---
|
|
||||||
if 'restored_indicator' in st.session_state and st.session_state.restored_indicator:
|
if 'restored_indicator' in st.session_state and st.session_state.restored_indicator:
|
||||||
st.info(f"📍 Editing Restored Version: **{st.session_state.restored_indicator}**")
|
st.info(f"📍 Editing Restored Version: **{st.session_state.restored_indicator}**")
|
||||||
|
|
||||||
@@ -188,27 +187,38 @@ def render_batch_processor(data, file_path, json_files, current_dir, selected_fi
|
|||||||
seq["reference image path"] = st.text_input("Ref Img", value=seq.get("reference image path", ""), key=f"{prefix}_ri2")
|
seq["reference image path"] = st.text_input("Ref Img", value=seq.get("reference image path", ""), key=f"{prefix}_ri2")
|
||||||
seq["flf image path"] = st.text_input("FLF Img", value=seq.get("flf image path", ""), key=f"{prefix}_flfi")
|
seq["flf image path"] = st.text_input("FLF Img", value=seq.get("flf image path", ""), key=f"{prefix}_flfi")
|
||||||
|
|
||||||
# --- LoRA Settings (SMART INPUTS) ---
|
# --- LoRA Settings (SMART INPUTS + SAFE FLOAT FIX) ---
|
||||||
with st.expander("💊 LoRA Settings"):
|
with st.expander("💊 LoRA Settings"):
|
||||||
meta = st.session_state.get("comfy_meta", {})
|
meta = st.session_state.get("comfy_meta", {})
|
||||||
lora_list = meta.get("loras", [])
|
lora_list = meta.get("loras", [])
|
||||||
|
|
||||||
lc1, lc2 = st.columns(2)
|
lc1, lc2 = st.columns(2)
|
||||||
|
|
||||||
|
# HELPER: Safe conversion
|
||||||
|
def get_safe_float(val, default=1.0):
|
||||||
|
try:
|
||||||
|
return float(val)
|
||||||
|
except (ValueError, TypeError):
|
||||||
|
# If it's a string like "<lora:name>", return default 1.0
|
||||||
|
return default
|
||||||
|
|
||||||
with lc1:
|
with lc1:
|
||||||
st.caption("LoRA 1")
|
st.caption("LoRA 1")
|
||||||
seq["lora 1 high"] = render_smart_input("Model", f"{prefix}_l1h", seq.get("lora 1 high", ""), lora_list)
|
seq["lora 1 high"] = render_smart_input("Model", f"{prefix}_l1h", seq.get("lora 1 high", ""), lora_list)
|
||||||
seq["lora 1 low"] = str(st.slider("Strength", 0.0, 2.0, float(seq.get("lora 1 low", 1.0)), 0.05, key=f"{prefix}_l1l"))
|
val = get_safe_float(seq.get("lora 1 low", 1.0))
|
||||||
|
seq["lora 1 low"] = str(st.slider("Strength", 0.0, 2.0, val, 0.05, key=f"{prefix}_l1l"))
|
||||||
|
|
||||||
with lc2:
|
with lc2:
|
||||||
st.caption("LoRA 2")
|
st.caption("LoRA 2")
|
||||||
seq["lora 2 high"] = render_smart_input("Model", f"{prefix}_l2h", seq.get("lora 2 high", ""), lora_list)
|
seq["lora 2 high"] = render_smart_input("Model", f"{prefix}_l2h", seq.get("lora 2 high", ""), lora_list)
|
||||||
seq["lora 2 low"] = str(st.slider("Strength", 0.0, 2.0, float(seq.get("lora 2 low", 1.0)), 0.05, key=f"{prefix}_l2l"))
|
val = get_safe_float(seq.get("lora 2 low", 1.0))
|
||||||
|
seq["lora 2 low"] = str(st.slider("Strength", 0.0, 2.0, val, 0.05, key=f"{prefix}_l2l"))
|
||||||
|
|
||||||
with lc1:
|
with lc1:
|
||||||
st.caption("LoRA 3")
|
st.caption("LoRA 3")
|
||||||
seq["lora 3 high"] = render_smart_input("Model", f"{prefix}_l3h", seq.get("lora 3 high", ""), lora_list)
|
seq["lora 3 high"] = render_smart_input("Model", f"{prefix}_l3h", seq.get("lora 3 high", ""), lora_list)
|
||||||
seq["lora 3 low"] = str(st.slider("Strength", 0.0, 2.0, float(seq.get("lora 3 low", 1.0)), 0.05, key=f"{prefix}_l3l"))
|
val = get_safe_float(seq.get("lora 3 low", 1.0))
|
||||||
|
seq["lora 3 low"] = str(st.slider("Strength", 0.0, 2.0, val, 0.05, key=f"{prefix}_l3l"))
|
||||||
|
|
||||||
# --- CUSTOM PARAMETERS ---
|
# --- CUSTOM PARAMETERS ---
|
||||||
st.markdown("---")
|
st.markdown("---")
|
||||||
@@ -273,4 +283,3 @@ def render_batch_processor(data, file_path, json_files, current_dir, selected_fi
|
|||||||
del st.session_state.restored_indicator
|
del st.session_state.restored_indicator
|
||||||
|
|
||||||
st.toast("Batch Saved & Snapshot Created!", icon="🚀")
|
st.toast("Batch Saved & Snapshot Created!", icon="🚀")
|
||||||
st.rerun()
|
|
||||||
|
|||||||
Reference in New Issue
Block a user