Update app.py

app.py

@@ -1,6 +1,7 @@
 import pandas as pd
 import streamlit as st
 import plotly.graph_objects as go
+import time
 
 def create_sunburst_plot(df):
     fig = go.Figure(go.Sunburst(
@@ -16,23 +17,30 @@ def create_sunburst_plot(df):
     fig.update_layout(margin=dict(t=0, l=0, r=0, b=0))
     return fig
 
-# New data reflecting the CPT codes and expected costs
 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': '99213', 'labels': 'CPT 99213', 'parents': 'Hip Surgery', 'values': 300, 'text': 'Office Visit'},
-    {'ids': '99214', 'labels': 'CPT 99214', 'parents': 'Hip Surgery', 'values': 400, 'text': 'Detailed Exam'},
-    {'ids': '99284', 'labels': 'CPT 99284', 'parents': 'Knee Surgery', 'values': 250, 'text': 'Emergency Department Visit'},
-    {'ids': '99285', 'labels': 'CPT 99285', 'parents': 'Knee Surgery', 'values': 450, 'text': 'Emergency Department Visit, High Complexity'},
+    {'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)
+# Function to update the data
+def update_data(df):
+    # Here you can add your logic to vary the cost data
+    df['values'] = df['values'] + 5
+    return df
 
-filtered_df = filter_data(df, "parents == 'Root'")
+# Create a placeholder for the plot
+plot_placeholder = st.empty()
 
-st.plotly_chart(create_sunburst_plot(filtered_df))
+# Loop to animate
+for i in range(10):  # Loop for 10 iterations
+    df = update_data(df)
+    fig = create_sunburst_plot(df)
+    plot_placeholder.plotly_chart(fig)
+    time.sleep(1)  # Sleep for 1 second