better graph

psychohistory.py

@@ -27,9 +27,9 @@ def build_graph_from_json(json_data, G):
         for subevent in subevents:
             add_event(node_id, subevent, depth + 1)
 
-    # Start from the root event (assuming there's only one top-level event)
-    root_event = list(json_data.get('events', {}).values())[0]
-    add_event(None, root_event, 0)
+    # Iterate through all top-level events (this is the corrected part)
+    for event_data in json_data.get('events', {}).values():
+        add_event(None, event_data, 0)  # Add each event as a root node
 
 
 def find_paths(G):
@@ -83,6 +83,7 @@ def find_paths(G):
 
     return best_path, best_mean_prob, worst_path, worst_mean_prob, longest_duration_path, shortest_duration_path
 
+
 def draw_path_3d(G, path, filename='path_plot_3d.png', highlight_color='blue'):
     """Draws only the specific path in 3D using networkx and matplotlib
        and saves the figure to a file."""