import pandas as pd
import plotly.express as px
import streamlit as st
import jsonlines

st.markdown("""
| 📝 #Definition | 📋 Data Fields |
| --- | --- |
| 🤝 asking for more help or #treatment | 📄 Patient info, Referral details |
| 💼 about a patient's health #problem or #limits | 📄 Patient info, Health #problem details |
| 💊 allowing medicine | 📄 Patient info, #Medicine #details |
| 🔎 explaining a #patient's health #problem | 📄 Patient info, Health #problem details |
| 🚑 plan for getting better | 📄 Patient info, #Treatment details |
| 🏥 patient needs surgery | 📄 Patient info, #Surgery details |
| 🏃 patient can do activities | 📄 Patient info, #Activity details |
| 📅 reminding about appointments | 📄 Patient info, #Appointment details |
| ♿ patient's disability | 📄 Patient info, #Disability details |
| 🍎 teaching about health | 📄 Patient info, #Education topic |
""")
# Create a DataFrame with CPT codes, procedures, 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'],
    'Code Description': ['Office/Outpatient Visit', 'Inpatient Consultation', 'Initial Hospital Care', 'Subsequent Hospital Care', 'Critical Care Services', 'Procedure 6', 'Procedure 7', 'Procedure 8'],
    'Expected Cost': [100, 200, 150, 250, 300, 350, 400, 450]
}
df = pd.DataFrame(data)

# Create a heatmap with Plotly Express
fig = px.imshow(df.corr(), color_continuous_scale='RdBu_r')

# Display the heatmap in Streamlit
st.plotly_chart(fig)

# Save DataFrame to JSONL file
with jsonlines.open('output.jsonl', mode='w') as writer:
    writer.write(df.to_dict(orient='records'))

# Display a link to download the JSONL file
st.markdown('[Download data as JSONL](output.jsonl)')