initial commit

psychohistory.py

@@ -34,7 +34,7 @@ def generate_tree(current_x, current_y, depth, max_depth, max_nodes, x_range, G,
 
 def build_graph_from_json(json_data, G):
     """Builds a graph from JSON data."""
-    data = json.loads(json_data)
+    # data = json.loads(json_data)  # No need to load JSON here
 
     def add_event(parent_id, event_data, depth):
         node_id = len(G.nodes)
@@ -52,7 +52,7 @@ def build_graph_from_json(json_data, G):
         for subevent in subevents:
             add_event(node_id, subevent, depth + 1)
 
-    root_event = list(data.get('events', {}).values())[0]
+    root_event = list(json_data.get('events', {}).values())[0]  # Use json_data directly
     root_id = len(G.nodes)
     G.add_node(root_id, pos=(0, root_event['probability'] / 100.0, root_event['event_number']), label=root_event['name'])
     add_event(None, root_event, 0)
@@ -188,14 +188,3 @@ def main(json_data):
         draw_path_3d(G, shortest_path, 'shortest_duration_path.png', 'purple')
 
     return 'global_tree.png'  # Return the filename of the global tree
-
-
-
-
-if __name__ == "__main__":
-    if len(sys.argv) < 2:
-        print("Usage: python script.py <mode> [input_file]")
-    else:
-        mode = sys.argv[1]
-        input_file = sys.argv[2] if len(sys.argv) > 2 else None
-        main(mode, input_file)