import pandas as pd
import plotly.graph_objects as go

def create_sunburst_plot(df):
    fig = go.Figure(go.Sunburst(
        labels=df['labels'],
        parents=df['parents'],
        values=df['values'],
        ids=df['ids'],
        text=df['text'],
        hoverinfo="label+value",
        branchvalues="total",
    ))

    fig.update_layout(margin=dict(t=0, l=0, r=0, b=0))
    return fig

data = [
    {'ids': 'Root', 'labels': 'Root', 'parents': '', 'values': None, 'text': 'Root'},
    {'ids': 'Hip Surgery', 'labels': 'Hip Surgery', 'parents': 'Root', 'values': 30, 'text': 'Hip Surgery'},
    {'ids': 'Knee Surgery', 'labels': 'Knee Surgery', 'parents': 'Root', 'values': 40, 'text': 'Knee Surgery'},
    {'ids': 'CPT1', 'labels': 'CPT1', 'parents': 'Hip Surgery', 'values': 20, 'text': 'CPT1'},
    {'ids': 'CPT2', 'labels': 'CPT2', 'parents': 'Hip Surgery', 'values': 10, 'text': 'CPT2'},
    {'ids': 'CPT3', 'labels': 'CPT3', 'parents': 'Knee Surgery', 'values': 25, 'text': 'CPT3'},
    {'ids': 'CPT4', 'labels': 'CPT4', 'parents': 'Knee Surgery', 'values': 15, 'text': 'CPT4'},
]

df = pd.DataFrame(data)

# Filter DataFrame using a query parameter
def filter_data(df, query):
    return df.query(query)

filtered_df = filter_data(df, "parents == 'Root'")

fig = create_sunburst_plot(filtered_df)
fig.show()