initial commit

psychohistory.py

@@ -32,12 +32,14 @@ def generate_tree(current_x, current_y, depth, max_depth, max_nodes, x_range, G,
     return node_count_per_depth
 
 
+
 def build_graph_from_json(json_data, G):
-    """Builds a graph from JSON data."""
+    """Builds a graph from JSON data, handling subevents recursively."""
 
     def add_event(parent_id, event_data, depth):
         node_id = len(G.nodes)
         prob = event_data['probability'] / 100.0
+        # Use event_number as the z-coordinate for better visualization
         pos = (depth, prob, event_data['event_number'])
         label = event_data['name']
         G.add_node(node_id, pos=pos, label=label)
@@ -53,11 +55,10 @@ def build_graph_from_json(json_data, G):
 
     # Iterate through all top-level events
     for event_data in json_data.get('events', {}).values():
-        root_id = len(G.nodes)
-        G.add_node(root_id, pos=(0, event_data['probability'] / 100.0, event_data['event_number']), label=event_data['name'])
         add_event(None, event_data, 0)  # Add each event as a root node
 
 
+
 def find_paths(G):
     """Finds paths with highest/lowest probability and longest/shortest durations."""
     best_path, worst_path = None, None