Ethanfel/Comfyui-JSON-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update tab_timeline.py

Browse Source
This commit is contained in:
ethanfel
2026-01-02 13:22:03 +01:00
committed by GitHub
parent cbe2355ef6
commit ea64726c24
1 changed files with 6 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tab_timeline.py
View File

@@ -24,18 +24,15 @@ def render_timeline_tab(data, file_path):
st.markdown("---") st.markdown("---")
# 3. The "Scrubber" / Inspector # 3. The "Scrubber" / Inspector
# Since we can't click the graph directly easily, we use a selectbox as our "Time Scrubber"
col_sel, col_act = st.columns([3, 1]) col_sel, col_act = st.columns([3, 1])
all_nodes = list(htree.nodes.values()) all_nodes = list(htree.nodes.values())
all_nodes.sort(key=lambda x: x["timestamp"], reverse=True) # Newest first all_nodes.sort(key=lambda x: x["timestamp"], reverse=True) # Newest first
# Format: "Note (ID) - [Branch]"
def fmt_node(n): def fmt_node(n):
return f"{n.get('note', 'Step')} ({n['id']})" return f"{n.get('note', 'Step')} ({n['id']})"
with col_sel: with col_sel:
# Default to current HEAD
current_idx = 0 current_idx = 0
for i, n in enumerate(all_nodes): for i, n in enumerate(all_nodes):
if n["id"] == htree.head_id: if n["id"] == htree.head_id:
@@ -56,10 +53,6 @@ def render_timeline_tab(data, file_path):
st.info(f"Viewing State: **{selected_node.get('note')}** (ID: {selected_node['id']})") st.info(f"Viewing State: **{selected_node.get('note')}** (ID: {selected_node['id']})")
with st.expander("📝 View / Edit Data for this Point", expanded=True): with st.expander("📝 View / Edit Data for this Point", expanded=True):
# We show raw JSON here for total control since it's an advanced tab
# But you can edit specific keys if you want to be granular
# Editable JSON Text Area
edited_json_str = st.text_area( edited_json_str = st.text_area(
"Raw Data (JSON)", "Raw Data (JSON)",
value=json.dumps(node_data, indent=4), value=json.dumps(node_data, indent=4),
@@ -68,7 +61,7 @@ def render_timeline_tab(data, file_path):
# 5. Restore / Branch Button # 5. Restore / Branch Button
with col_act: with col_act:
st.write("") # Spacer st.write("")
st.write("") st.write("")
if st.button("⏪ Restore & Branch", type="primary", use_container_width=True): if st.button("⏪ Restore & Branch", type="primary", use_container_width=True):
try: try:
@@ -77,11 +70,7 @@ def render_timeline_tab(data, file_path):
# 1. Update the Main Data # 1. Update the Main Data
data.update(new_data_content) data.update(new_data_content)
# 2. Update the Tree (Move HEAD to this node) # 2. Move Head
# If the user EDITED the JSON, we should probably commit a NEW node automatically
# But if they just clicked restore, we move HEAD.
# Logic: Just move head to selected, then saving in future will branch.
htree.head_id = selected_node['id'] htree.head_id = selected_node['id']
# 3. Save to Disk # 3. Save to Disk
@@ -90,7 +79,9 @@ def render_timeline_tab(data, file_path):
# 4. Reset UI # 4. Reset UI
st.session_state.ui_reset_token += 1 st.session_state.ui_reset_token += 1
st.toast(f"Restored to {selected_node['id']}!", icon="↺")
# FIXED: Changed icon to standard 'Refresh' emoji
st.toast(f"Restored to {selected_node['id']}!", icon="🔄")
st.rerun() st.rerun()
except json.JSONDecodeError: except json.JSONDecodeError:
@@ -104,7 +95,7 @@ def render_timeline_tab(data, file_path):
# Fix branches pointing to this # Fix branches pointing to this
for b, tip in list(htree.branches.items()): for b, tip in list(htree.branches.items()):
if tip == selected_node['id']: if tip == selected_node['id']:
del htree.branches[b] # Prune broken branch tip del htree.branches[b]
data["history_tree"] = htree.to_dict() data["history_tree"] = htree.to_dict()
save_json(file_path, data) save_json(file_path, data)