Update app.py

app.py

@@ -75,83 +75,17 @@ def get_tree(imgpath):
     return get_root(node)
                  
 
-def display_tree():
-    # This function should create and return a Plotly figure of the tree
+# def display_tree():
+#     # This function should create and return a Plotly figure of the tree
 
-    # Define the nodes and edges for the graph
-    nodes = ['Node 1', 'Node 2', 'Node 3', 'Node 4']
-    edges = [(0, 1), (0, 2), (2, 3)]  # Edges are tuples of node indices
-    
-    # Define positions for the nodes (you can use a layout algorithm for more complex graphs)
-    positions = [(0, 0), (1, 2), (1, -2), (2, 0)]
-    
-    # Create traces for nodes and edges
-    edge_x = []
-    edge_y = []
-    for edge in edges:
-        x0, y0 = positions[edge[0]]
-        x1, y1 = positions[edge[1]]
-        edge_x.extend([x0, x1, None])
-        edge_y.extend([y0, y1, None])
-    
-    edge_trace = go.Scatter(
-        x=edge_x, y=edge_y,
-        line=dict(width=2, color='Black'),
-        hoverinfo='none',
-        mode='lines')
-    
-    node_x = [pos[0] for pos in positions]
-    node_y = [pos[1] for pos in positions]
-    
-    node_trace = go.Scatter(
-        x=node_x, y=node_y,
-        mode='markers+text',
-        hoverinfo='text',
-        marker=dict(showscale=False, size=10, color='Goldenrod'),
-        text=nodes,
-        textposition="top center"
-    )
-    
-    # Define the layout of the graph
-    layout = go.Layout(
-        showlegend=False,
-        hovermode='closest',
-        margin=dict(b=0, l=0, r=0, t=0),
-        xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
-        yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
-    )
+#     # Define the nodes and edges for the graph
+#     nodes = ['Node 1', 'Node 2', 'Node 3', 'Node 4']
+#     edges = [(0, 1), (0, 2), (2, 3)]  # Edges are tuples of node indices
     
-    # Create the figure
-    fig = go.Figure(data=[edge_trace, node_trace], layout=layout)
-    return fig
-
-ROOT = None
-
-# def display_tree():
-#     nodes = []
-#     edges = []
-#     positions = {}
-
-#     root = ROOT
-    
-#     def traverse(node, depth=0, index=0):
-#         # This assigns a unique index to each node, which we can use for positioning
-#         if node not in nodes:
-#             nodes.append(node)
-#         idx = nodes.index(node)
-#         positions[idx] = (depth, -len(nodes) + index * 2)  # stagger the y-positions for clarity
-
-#         # Loop through each child to connect with edges and calculate their positions
-#         for child in node.children:
-#             if child not in nodes:
-#                 nodes.append(child)
-#             child_idx = nodes.index(child)
-#             edges.append((idx, child_idx))
-#             traverse(child, depth + 1, nodes.index(child) - idx)
-
-#     # Start the traversal from the root node
-#     traverse(root)
+#     # Define positions for the nodes (you can use a layout algorithm for more complex graphs)
+#     positions = [(0, 0), (1, 2), (1, -2), (2, 0)]
     
+#     # Create traces for nodes and edges
 #     edge_x = []
 #     edge_y = []
 #     for edge in edges:
@@ -166,30 +100,93 @@ ROOT = None
 #         hoverinfo='none',
 #         mode='lines')
     
-#     node_x = [pos[0] for pos in positions.values()]
-#     node_y = [pos[1] for pos in positions.values()]
+#     node_x = [pos[0] for pos in positions]
+#     node_y = [pos[1] for pos in positions]
     
 #     node_trace = go.Scatter(
 #         x=node_x, y=node_y,
 #         mode='markers+text',
 #         hoverinfo='text',
 #         marker=dict(showscale=False, size=10, color='Goldenrod'),
-#         text=[node.name for node in nodes],
+#         text=nodes,
 #         textposition="top center"
 #     )
     
+#     # Define the layout of the graph
 #     layout = go.Layout(
-#         title="Tree Visualization",
 #         showlegend=False,
 #         hovermode='closest',
-#         margin=dict(b=0, l=0, r=0, t=40),
+#         margin=dict(b=0, l=0, r=0, t=0),
 #         xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
 #         yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
 #     )
     
+#     # Create the figure
 #     fig = go.Figure(data=[edge_trace, node_trace], layout=layout)
 #     return fig
 
+ROOT = None
+
+
+def display_tree():
+    nodes = []
+    edges = []
+    positions = {}
+
+    root = ROOT
+    
+    def traverse(node, depth=0, index=0):
+        if node not in nodes:
+            nodes.append(node)
+        idx = nodes.index(node)
+        positions[idx] = (depth * 1, index * 1 - len(nodes) / 2)  # Adjusted the multipliers for depth and index
+
+        for child in node.children:
+            if child not in nodes:
+                nodes.append(child)
+            child_idx = nodes.index(child)
+            edges.append((idx, child_idx))
+            traverse(child, depth + 1, index + len(node.children) / 2)  # Recursively traverse to set positions
+
+    traverse(root)
+    
+    edge_x = []
+    edge_y = []
+    for edge in edges:
+        x0, y0 = positions[edge[0]]
+        x1, y1 = positions[edge[1]]
+        edge_x.extend([x0, x1, None])
+        edge_y.extend([y0, y1, None])
+    
+    edge_trace = go.Scatter(
+        x=edge_x, y=edge_y,
+        line=dict(width=2, color='Black'),
+        hoverinfo='none',
+        mode='lines')
+    
+    node_x = [pos[0] for pos in positions.values()]
+    node_y = [pos[1] for pos in positions.values()]
+    
+    node_trace = go.Scatter(
+        x=node_x, y=node_y,
+        mode='markers+text',
+        hoverinfo='text',
+        marker=dict(showscale=False, size=10, color='Goldenrod'),
+        text=[node.name for node in nodes],
+        textposition="top center"
+    )
+    
+    layout = go.Layout(
+        title="Tree Visualization",
+        showlegend=False,
+        hovermode='closest',
+        margin=dict(b=0, l=0, r=0, t=40),
+        xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
+        yaxis=dict(showgrid=False, zeroline=False, showticklabels=False)
+    )
+    
+    fig = go.Figure(data=[edge_trace, node_trace], layout=layout)
+    return fig