Update app.py

app.py

@@ -18,6 +18,14 @@ st.markdown("""
 | 🍎 teaching about health | 📄 Patient info, #Education topic |
 """)
 # Create a DataFrame with CPT codes, procedures, and expected costs
+import pandas as pd
+import plotly.graph_objects as go
+import streamlit as st
+import jsonlines
+import base64
+from datetime import datetime
+
+# Create a DataFrame with Code types, values, descriptions, and expected costs
 data = {
     'Code Type': ['CPT', 'SNOMED', 'RXNORM', 'DEA', 'LOINC', 'ORI', 'ORU', 'CCD'],
     'Code Value': ['99201', 'A-12345', 'R-12345', 'D-12345', 'L-12345', 'O-12345', 'U-12345', 'C-12345'],
@@ -26,15 +34,33 @@ data = {
 }
 df = pd.DataFrame(data)
 
-# Create a heatmap with Plotly Express
-fig = px.imshow(df.corr(), color_continuous_scale='RdBu_r')
+# Create a sunburst plot with Plotly
+fig = go.Figure(go.Sunburst(
+    labels=df['Code Type'],
+    parents=['']*len(df),
+    values=df['Expected Cost'],
+    text=df['Code Description'],
+    hoverinfo="label+value+text",
+    branchvalues="total",
+))
+
+fig.update_layout(margin=dict(t=0, l=0, r=0, b=0))
 
-# Display the heatmap in Streamlit
+# Display the sunburst plot in Streamlit
 st.plotly_chart(fig)
 
 # Save DataFrame to JSONL file
-with jsonlines.open('output.jsonl', mode='w') as writer:
+timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
+filename = f"output_{timestamp}.jsonl"
+with jsonlines.open(filename, mode='w') as writer:
     writer.write(df.to_dict(orient='records'))
 
+# Function to create a download link
+def create_download_link(filename):
+    with open(filename, 'r') as file:
+        data = file.read()
+    b64 = base64.b64encode(data.encode()).decode()
+    return f'<a href="data:file/txt;base64,{b64}" download="{filename}">Download data as JSONL</a>'
+
 # Display a link to download the JSONL file
-st.markdown('[Download data as JSONL](output.jsonl)')
+st.markdown(create_download_link(filename), unsafe_allow_html=True)