initial commit

psychohistory.py

@@ -156,19 +156,10 @@ def draw_global_tree_3d(G, filename='global_tree.png'):
     plt.savefig(filename, bbox_inches='tight')
     plt.close()
 
+
 def main(json_data):
     G = nx.DiGraph()
-    build_graph_from_json(json_data, G)
-
-    if mode == 'random':
-        generate_tree(0, 0, 0, 5, 3, 10, G)
-    elif mode == 'json' and input_file:
-        with open(input_file, 'r') as file:
-            json_data = file.read()
-        build_graph_from_json(json_data, G)
-    else:
-        print("Invalid mode or input file not provided.")
-        return
+    build_graph_from_json(json_data, G)  # Build graph from the provided JSON data
 
     draw_global_tree_3d(G, filename='global_tree.png')
 
@@ -187,8 +178,6 @@ def main(json_data):
         print(f"\nPath with the shortest duration: {' -> '.join(map(str, shortest_path))}")
         print(f"Duration: {max(G.nodes[node]['pos'][0] for node in shortest_path) - min(G.nodes[node]['pos'][0] for node in shortest_path):.2f}")
 
-    draw_global_tree_3d(G, filename='global_tree.png')
-
     if best_path:
         draw_path_3d(G, best_path, 'best_path.png', 'blue')
     if worst_path:
@@ -198,7 +187,9 @@ def main(json_data):
     if shortest_path:
         draw_path_3d(G, shortest_path, 'shortest_duration_path.png', 'purple')
 
-    return 'global_tree.png'
+    return 'global_tree.png'  # Return the filename of the global tree
+
+
 
 
 if __name__ == "__main__":