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

Update history_tree.py

Browse Source
This commit is contained in:
ethanfel
2026-01-02 13:34:50 +01:00
committed by GitHub
parent 943f964f9b
commit 8f4f049fb8
1 changed files with 34 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
history_tree.py
View File

@@ -55,47 +55,58 @@ class HistoryTree:
return {"nodes": self.nodes, "branches": self.branches, "head_id": self.head_id} return {"nodes": self.nodes, "branches": self.branches, "head_id": self.head_id}
def generate_horizontal_graph(self): def generate_horizontal_graph(self):
"""Generates a Fixed-Size, Compact Horizontal Graph.""" """Generates a Compact Graph with HTML Labels and Tooltips."""
dot = [ dot = [
'digraph History {', 'digraph History {',
' rankdir=LR;', ' rankdir=LR;',
' bgcolor="white";', # Clean background ' bgcolor="white";',
# --- COMPACT LAYOUT SETTINGS --- # TIGHT LAYOUT
' nodesep=0.15;', # Horizontal gap between nodes ' nodesep=0.2;',
' ranksep=0.25;', # Vertical gap between branches ' ranksep=0.3;',
' splines=ortho;', # Orthogonal lines (right angles) look cleaner/techier
# --- FIXED SIZE NODES (The Key Fix) --- # BASE NODE STYLE
# fixedsize=true: Ignores text length, forces the width/height below. ' node [shape=plain, fontname="Arial", fontsize=9];', # shape=plain for HTML labels
# width=1.2: Sets box width to ~100px ' edge [color="#888888", arrowsize=0.6, penwidth=1.0];'
# height=0.4: Sets box height to ~30px
' node [shape=box, style="filled,rounded", fillcolor="#f9f9f9", fontname="Arial", fontsize=9, fixedsize=true, width=1.3, height=0.45];',
' edge [color="#666666", arrowsize=0.5, penwidth=1.0];'
] ]
sorted_nodes = sorted(self.nodes.values(), key=lambda x: x["timestamp"]) sorted_nodes = sorted(self.nodes.values(), key=lambda x: x["timestamp"])
for n in sorted_nodes: for n in sorted_nodes:
nid = n["id"] nid = n["id"]
# Truncate text to fit in the 1.3 inch box
full_note = n.get('note', 'Step') full_note = n.get('note', 'Step')
short_note = (full_note[:10] + '..') if len(full_note) > 10 else full_note
label = f"{short_note}\\n{nid[:4]}" # Sanitize for HTML (replace quotes)
safe_note = full_note.replace('"', "'")
# Colors # Truncate visual label
color = "#f0f0f0" short_note = (full_note[:12] + '..') if len(full_note) > 12 else full_note
penwidth = "1"
# Determine Colors
bg_color = "#f4f4f4"
border_color = "#cccccc"
border_width = "1"
if nid == self.head_id: if nid == self.head_id:
color = "#fff6cd" # Yellow for Active bg_color = "#fff6cd" # Yellow
penwidth = "2" border_color = "#eebb00"
border_width = "2"
elif nid in self.branches.values():
bg_color = "#e6ffe6" # Green tip
border_color = "#99cc99"
if nid in self.branches.values(): # HTML Label Construction
if color == "#f0f0f0": color = "#e6ffe6" # Green for Branch Tips # This creates a tight table-like node
label = (
f'<<TABLE BORDER="{border_width}" CELLBORDER="0" CELLSPACING="0" CELLPADDING="4" BGCOLOR="{bg_color}" COLOR="{border_color}">'
f'<TR><TD><B>{short_note}</B></TD></TR>'
f'<TR><TD><FONT POINT-SIZE="8" COLOR="#666666">{nid[:4]}</FONT></TD></TR>'
f'</TABLE>>'
)
dot.append(f' "{nid}" [label="{label}", fillcolor="{color}", penwidth="{penwidth}"];') # Add node with Tooltip
dot.append(f' "{nid}" [label={label}, tooltip="{safe_note}"];')
if n["parent"] and n["parent"] in self.nodes: if n["parent"] and n["parent"] in self.nodes:
dot.append(f' "{n["parent"]}" -> "{nid}";') dot.append(f' "{n["parent"]}" -> "{nid}";')