pos = nx.spring_layout(G, dim=3)  # 3D布局
edge_trace, node_trace = create_node_link_trace(G, pos)

edge_labels = nx.get_edge_attributes(G, 'relation')
edge_label_trace = create_edge_label_trace(G, pos, edge_labels)

degree_dist_fig = create_degree_distribution(G)
centrality_fig = create_centrality_plot(G)

fig = make_subplots(
    rows=2, cols=2,
    column_widths=[0.7, 0.3],
    row_heights=[0.7, 0.3],
    specs=[
        [{"type": "scene", "rowspan": 2}, {"type": "xy"}],
        [None, {"type": "xy"}]
    ],
    subplot_titles=("3D Knowledge Graph Code by AI超元域频道", "Node Degree Distribution", "Degree Centrality Distribution")
)

fig.add_trace(edge_trace, row=1, col=1)
fig.add_trace(node_trace, row=1, col=1)
fig.add_trace(edge_label_trace, row=1, col=1)

fig.add_trace(degree_dist_fig.data[0], row=1, col=2)
fig.add_trace(centrality_fig.data[0], row=2, col=2)

# 更新3D布局
fig.update_layout(
    scene=dict(
        xaxis=dict(showticklabels=False, showgrid=False, zeroline=False),
        yaxis=dict(showticklabels=False, showgrid=False, zeroline=False),
        zaxis=dict(showticklabels=False, showgrid=False, zeroline=False),
        aspectmode='cube'
    ),
    scene_camera=dict(eye=dict(x=1.5, y=1.5, z=1.5))
)

# 添加不同布局的按钮
fig.update_layout(
    updatemenus=[
        dict(
            type="buttons",
            direction="left",
            buttons=list([
                dict(args=[{"visible": [True, True, True, True, True]}], label="Show All", method="update"),
                dict(args=[{"visible": [True, True, False, True, True]}], label="Hide Edge Labels",
                     method="update"),
                dict(args=[{"visible": [False, True, False, True, True]}], label="Nodes Only", method="update")
            ]),
            pad={"r": 10, "t": 10},
            showactive=True,
            x=0.05,
            xanchor="left",
            y=1.1,
            yanchor="top"
        ),
    ]
)

# 添加节点大小的滑块
fig.update_layout(
    sliders=[dict(
        active=0,
        currentvalue={"prefix": "Node Size: "},
        pad={"t": 50},
        steps=[dict(method='update',
                    args=[{'marker.size': [i] * len(G.nodes)}],
                    label=str(i)) for i in range(5, 21, 5)]
    )]
)

# 保存图表为HTML文件
pyo.plot(fig, filename='output.html')  # 将图形保存为HTML文件