Sunburst Diagram

# === IMPORTS ===
import pandas as pd
import numpy as np
import plotly.express as px

# === USER-EDITABLE PARAMETERS ===
title = "File System Storage Usage - 500GB Drive"
figsize_width = 900
figsize_height = 700

# === EXAMPLE DATASET ===
# Hierarchical file system structure
np.random.seed(42)

data = {
    'Category': [],
    'Subcategory': [],
    'Size_GB': []
}

storage_data = {
    'Documents': [
        ('Projects', 80),
        ('Personal', 40),
    ],
    'Media': [
        ('Videos', 120),
        ('Photos', 60),
        ('Music', 20),
    ],
    'Applications': [
        ('Development Tools', 35),
        ('Office Suite', 25),
        ('Games', 20),
    ],
    'System': [
        ('Windows', 30),
        ('Drivers', 10),
        ('Cache', 10),
    ],
    'Other': [
        ('Downloads', 30),
        ('Backups', 20),
    ],
}

for category, subcats in storage_data.items():
    for subcat, size in subcats:
        data['Category'].append(category)
        data['Subcategory'].append(subcat)
        data['Size_GB'].append(size)

df = pd.DataFrame(data)

# Print summary
print("=== Storage Usage Summary ===")
print(f"\nTotal Storage Used: {df['Size_GB'].sum()}GB")
print(f"\nBy Category:")
print(df.groupby('Category')['Size_GB'].sum().sort_values(ascending=False).to_string())

# === CREATE SUNBURST ===
fig = px.sunburst(
    df,
    path=['Category', 'Subcategory'],
    values='Size_GB',
    title=title,
    color='Size_GB',
    color_continuous_scale='Reds'
)

fig.update_layout(
    width=figsize_width,
    height=figsize_height,
    margin=dict(t=50, l=10, r=10, b=10)
)

fig.update_traces(
    hovertemplate='<b>%{label}</b><br>Size: %{value}GB<br>%{percentParent:.1%} of parent<extra></extra>'
)

fig.show()
# END-OF-CODE

Total Storage: 500GB
Used Storage: 500GB
Free Storage: 0GB
Largest Category: Media (200GB, 40.0%)