diff --git a/tab_timeline_wip.py b/tab_timeline_wip.py index d496dc0..3a2a2c9 100644 --- a/tab_timeline_wip.py +++ b/tab_timeline_wip.py @@ -12,7 +12,7 @@ def render_timeline_wip(data, file_path): htree = HistoryTree(tree_data) - # --- 1. BUILD GRAPH DATA --- + # --- 1. BUILD GRAPH --- nodes = [] edges = [] @@ -21,18 +21,15 @@ def render_timeline_wip(data, file_path): for n in sorted_nodes: nid = n["id"] note = n.get('note', 'Step') - # Visual styling short_note = (note[:15] + '..') if len(note) > 15 else note color = "#ffffff" border = "#666666" - # Highlight Current Head if nid == htree.head_id: color = "#fff6cd" border = "#eebb00" - # Highlight Branch Tips if nid in htree.branches.values(): if color == "#ffffff": color = "#e6ffe6" @@ -57,12 +54,11 @@ def render_timeline_wip(data, file_path): type="STRAIGHT" )) - # --- 2. CONFIGURATION --- config = Config( width="100%", height="400px", directed=True, - physics=False, # Keep false for stability + physics=False, hierarchical=True, layout={ "hierarchical": { @@ -79,14 +75,13 @@ def render_timeline_wip(data, file_path): st.subheader("✨ Interactive Timeline") st.caption("Click a node to view its settings below.") - # --- 3. RENDER GRAPH --- - # KEY FIX: Adding a unique key ensures Streamlit tracks clicks correctly + # Render Graph selected_id = agraph(nodes=nodes, edges=edges, config=config) st.markdown("---") - # --- 4. DETERMINE TARGET NODE --- - # If nothing is clicked, default to the current HEAD + # --- 2. DETERMINE TARGET --- + # Default to HEAD if nothing clicked target_node_id = selected_id if selected_id else htree.head_id if target_node_id and target_node_id in htree.nodes: @@ -100,8 +95,8 @@ def render_timeline_wip(data, file_path): # Restore Button with c_h2: - st.write("") - if st.button("⏪ Restore This Version", type="primary", use_container_width=True): + st.write(""); st.write("") + if st.button("⏪ Restore This Version", type="primary", use_container_width=True, key=f"rst_{target_node_id}"): data.update(node_data) htree.head_id = target_node_id @@ -115,52 +110,52 @@ def render_timeline_wip(data, file_path): st.toast(f"Restored {target_node_id}!", icon="🔄") st.rerun() - # --- 5. PREVIEW PANELS (READ ONLY) --- + # --- 3. PREVIEW PANELS (DYNAMIC KEYS FIX) --- + # We append target_node_id to every key to force a hard refresh # A. Prompts p_col1, p_col2 = st.columns(2) with p_col1: val_gp = node_data.get("general_prompt", "") - st.text_area("General Positive", value=val_gp, height=80, disabled=True, key="p_gp") + st.text_area("General Positive", value=val_gp, height=80, disabled=True, key=f"p_gp_{target_node_id}") val_sp = node_data.get("current_prompt", "") or node_data.get("prompt", "") - st.text_area("Specific Positive", value=val_sp, height=80, disabled=True, key="p_sp") + st.text_area("Specific Positive", value=val_sp, height=80, disabled=True, key=f"p_sp_{target_node_id}") with p_col2: val_gn = node_data.get("general_negative", "") - st.text_area("General Negative", value=val_gn, height=80, disabled=True, key="p_gn") + st.text_area("General Negative", value=val_gn, height=80, disabled=True, key=f"p_gn_{target_node_id}") val_sn = node_data.get("negative", "") - st.text_area("Specific Negative", value=val_sn, height=80, disabled=True, key="p_sn") + st.text_area("Specific Negative", value=val_sn, height=80, disabled=True, key=f"p_sn_{target_node_id}") # B. Key Settings st.caption("⚙️ Core Settings") s_col1, s_col2, s_col3 = st.columns(3) - s_col1.text_input("Camera", value=str(node_data.get("camera", "static")), disabled=True, key="p_cam") - s_col2.text_input("FLF", value=str(node_data.get("flf", "0.0")), disabled=True, key="p_flf") - s_col3.text_input("Seed", value=str(node_data.get("seed", "-1")), disabled=True, key="p_seed") + s_col1.text_input("Camera", value=str(node_data.get("camera", "static")), disabled=True, key=f"p_cam_{target_node_id}") + s_col2.text_input("FLF", value=str(node_data.get("flf", "0.0")), disabled=True, key=f"p_flf_{target_node_id}") + s_col3.text_input("Seed", value=str(node_data.get("seed", "-1")), disabled=True, key=f"p_seed_{target_node_id}") # C. LoRAs with st.expander("💊 LoRA Configuration", expanded=False): l1, l2, l3 = st.columns(3) with l1: - st.text_input("LoRA 1 Name", value=node_data.get("lora 1 high", ""), disabled=True, key="p_l1h") - st.text_input("LoRA 1 Str", value=str(node_data.get("lora 1 low", "")), disabled=True, key="p_l1l") + st.text_input("LoRA 1 Name", value=node_data.get("lora 1 high", ""), disabled=True, key=f"p_l1h_{target_node_id}") + st.text_input("LoRA 1 Str", value=str(node_data.get("lora 1 low", "")), disabled=True, key=f"p_l1l_{target_node_id}") with l2: - st.text_input("LoRA 2 Name", value=node_data.get("lora 2 high", ""), disabled=True, key="p_l2h") - st.text_input("LoRA 2 Str", value=str(node_data.get("lora 2 low", "")), disabled=True, key="p_l2l") + st.text_input("LoRA 2 Name", value=node_data.get("lora 2 high", ""), disabled=True, key=f"p_l2h_{target_node_id}") + st.text_input("LoRA 2 Str", value=str(node_data.get("lora 2 low", "")), disabled=True, key=f"p_l2l_{target_node_id}") with l3: - st.text_input("LoRA 3 Name", value=node_data.get("lora 3 high", ""), disabled=True, key="p_l3h") - st.text_input("LoRA 3 Str", value=str(node_data.get("lora 3 low", "")), disabled=True, key="p_l3l") + st.text_input("LoRA 3 Name", value=node_data.get("lora 3 high", ""), disabled=True, key=f"p_l3h_{target_node_id}") + st.text_input("LoRA 3 Str", value=str(node_data.get("lora 3 low", "")), disabled=True, key=f"p_l3l_{target_node_id}") # D. VACE / I2V Specifics - # Check if any VACE keys exist in this history node vace_keys = ["frame_to_skip", "vace schedule", "video file path"] has_vace = any(k in node_data for k in vace_keys) if has_vace: with st.expander("🎞️ VACE / I2V Settings", expanded=True): v1, v2, v3 = st.columns(3) - v1.text_input("Skip Frames", value=str(node_data.get("frame_to_skip", "")), disabled=True, key="p_fts") - v2.text_input("Schedule", value=str(node_data.get("vace schedule", "")), disabled=True, key="p_vsc") - v3.text_input("Video Path", value=str(node_data.get("video file path", "")), disabled=True, key="p_vid") + v1.text_input("Skip Frames", value=str(node_data.get("frame_to_skip", "")), disabled=True, key=f"p_fts_{target_node_id}") + v2.text_input("Schedule", value=str(node_data.get("vace schedule", "")), disabled=True, key=f"p_vsc_{target_node_id}") + v3.text_input("Video Path", value=str(node_data.get("video file path", "")), disabled=True, key=f"p_vid_{target_node_id}")