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-03 01:00:35 +01:00
committed by GitHub
parent 7198ccd19b
commit ab973c6d58
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
history_tree.py
View File

@@ -54,13 +54,17 @@ class HistoryTree:
def to_dict(self): def to_dict(self):
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): # --- UPDATED GRAPH GENERATOR ---
"""Generates a Compact, Readable Horizontal Graph using HTML Labels.""" def generate_graph(self, direction="LR"):
"""
Generates Graphviz source.
direction: "LR" (Horizontal) or "TB" (Vertical)
"""
dot = [ dot = [
'digraph History {', 'digraph History {',
' rankdir=LR;', f' rankdir={direction};', # Dynamic Direction
' bgcolor="white";', ' bgcolor="white";',
' splines=ortho;', # Clean right-angle lines ' splines=ortho;',
# TIGHT SPACING # TIGHT SPACING
' nodesep=0.2;', ' nodesep=0.2;',
@@ -77,8 +81,6 @@ class HistoryTree:
nid = n["id"] nid = n["id"]
full_note = n.get('note', 'Step') full_note = n.get('note', 'Step')
# Truncate slightly for display, but keep enough to read
# HTML Labels allow distinct styling for Title vs ID
display_note = (full_note[:15] + '..') if len(full_note) > 15 else full_note display_note = (full_note[:15] + '..') if len(full_note) > 15 else full_note
# COLORS # COLORS
@@ -94,8 +96,7 @@ class HistoryTree:
bg_color = "#e6ffe6" # Green for Tips bg_color = "#e6ffe6" # Green for Tips
border_color = "#66aa66" border_color = "#66aa66"
# HTML LABEL (Table) # HTML LABEL
# This is the trick to make it compact but readable
label = ( label = (
f'<<TABLE BORDER="{border_width}" CELLBORDER="0" CELLSPACING="0" CELLPADDING="4" BGCOLOR="{bg_color}" COLOR="{border_color}">' f'<<TABLE BORDER="{border_width}" CELLBORDER="0" CELLSPACING="0" CELLPADDING="4" BGCOLOR="{bg_color}" COLOR="{border_color}">'
f'<TR><TD><B><FONT POINT-SIZE="10">{display_note}</FONT></B></TD></TR>' f'<TR><TD><B><FONT POINT-SIZE="10">{display_note}</FONT></B></TD></TR>'
@@ -103,7 +104,6 @@ class HistoryTree:
f'</TABLE>>' f'</TABLE>>'
) )
# Tooltip shows full text on hover
safe_tooltip = full_note.replace('"', "'") safe_tooltip = full_note.replace('"', "'")
dot.append(f' "{nid}" [label={label}, tooltip="{safe_tooltip}"];') dot.append(f' "{nid}" [label={label}, tooltip="{safe_tooltip}"];')