pip install --upgrade pip
pip install --upgrade pyvis
pip install --upgrade networkx

import os
import json

# List of folders to be excluded from the folder structure
excluded_folders = [".git", "node_modules", ".bin"]

def create_folder_structure_json(dir_path):
    folder_structure = {}
    for root, dirs, files in os.walk(dir_path):
        # Check if any excluded folders are in the current directory
        dirs[:] = [d for d in dirs if d not in excluded_folders]

        current_folder = folder_structure
        folders = os.path.relpath(root, dir_path).split(os.path.sep)
        for folder in folders:
            if folder not in excluded_folders:
                current_folder = current_folder.setdefault(folder, {})
    return folder_structure

def save_to_json(data, output_file):
    with open(output_file, 'w') as json_file:
        json.dump(data, json_file, indent=4)

if __name__ == "__main__":
    directory_path = input("Enter the directory path: ")
    if os.path.exists(directory_path) and os.path.isdir(directory_path):
        folder_structure = create_folder_structure_json(directory_path)
        output_file = "folder_structure.json"
        save_to_json(folder_structure, output_file)
        print(f"Folder structure saved to {output_file}")
    else:
        print("Invalid directory path or directory does not exist.")

import json
import os
import random
from pyvis.network import Network
import networkx as nx

def generate_random_color():
    """Generate a random hex color."""
    return "#{:06x}".format(random.randint(0, 0xFFFFFF))

# Path to the JSON file
file_path = os.path.join(os.path.dirname(__file__), 'folder_structure.json')

# Load data from JSON file
with open(file_path, 'r') as file:
    data = json.load(file)

# Create a NetworkX graph
G = nx.Graph()

# Function to recursively add nodes and edges to the graph
def add_nodes_and_edges(parent, node_dict):
    for node, sub_dict in node_dict.items():
        full_node = os.path.join(parent, node) if parent != '.' else node
        G.add_node(full_node)
        if sub_dict:
            for sub_node in sub_dict:
                G.add_edge(full_node, os.path.join(full_node, sub_node))
            add_nodes_and_edges(full_node, sub_dict)

# Start by adding the root node
G.add_node('.')
add_nodes_and_edges('.', data)

# Create a PyVis network from the NetworkX graph
# https://pyvis.readthedocs.io/en/latest/_modules/pyvis/network.html
net = Network(
    notebook=False,
    bgcolor="#1a1a1a",
    #cdn_resources="remote",
    height="93vh",
    width="100%",
    select_menu=True,
    font_color="#cccccc",
    # filter_menu=True,
    directed=False,
    layout="hierarchical"
)

# Assign different random colors to different nodes
for node in G.nodes():
    color = generate_random_color()
    # Get the folder name (last part of the path)
    folder_name = os.path.basename(node)
    net.add_node(node, folder_name, color=color)

# Add edges
for edge in G.edges():
    net.add_edge(edge[0], edge[1])

net.repulsion(node_distance=256, spring_length=512)
net.show_buttons()

# Save the network
net.write_html("folder_structure_network.html", local=True, notebook=False, open_browser=True)