Update app.py

app.py

@@ -1,125 +1,63 @@
 import streamlit as st
-import csv
-import os
-import random
 import pandas as pd
+import random
 import plotly.graph_objects as go
 import plotly.express as px
 
-def save_data(name, email, phone):
-    if not os.path.exists('community.csv'):
-        with open('community.csv', mode='w') as csv_file:
-            writer = csv.writer(csv_file)
-            writer.writerow(['Name', 'Email', 'Phone'])
-    with open('community.csv', mode='a') as csv_file:
-        fieldnames = ['Name', 'Email', 'Phone']
-        writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
-        if os.stat('community.csv').st_size == 0:
-            writer.writeheader()
-        writer.writerow({'Name': name, 'Email': email, 'Phone': phone})
-    with open(f'{name}.txt', mode='w') as file:
-        file.write(f'Name: {name}\nEmail: {email}\nPhone: {phone}\n')
-
-def reset_data():
-    os.remove('community.csv')
-    for file in os.listdir():
-        if file.endswith('.txt'):
-            os.remove(file)
-
-def show_data():
-    count = 0
-    with open('community.csv', mode='r') as csv_file:
-        csv_reader = csv.DictReader(csv_file)
-        for row in csv_reader:
-            count += 1
-            if count == 3:
-                st.write(f'{row["Name"]}: {row["Email"]} - {row["Phone"]}')
-                count = 0
-
-def app():
-    states = ["Minnesota", "Florida", "California", "Washington", "Maine", "Texas"]
+def health_game():
+    # Define the states and conditions of interest
+    states = ["Minnesota", "Florida", "California"]
     top_n = 10
 
+    # Define the list dictionary of top 10 health conditions descending by cost, with emojis, treatment recommendation and potential savings
     health_conditions = [
-        {"condition": "💔 Heart disease", "emoji": "💗", "spending": 214.3},
-        {"condition": "🤕 Trauma-related disorders", "emoji": "🚑", "spending": 198.6},
-        {"condition": "🦀 Cancer", "emoji": "🎗️", "spending": 171.0},
-        {"condition": "🧠 Mental disorders", "emoji": "🧘", "spending": 150.8},
-        {"condition": "🦴 Osteoarthritis and joint disorders", "emoji": "🏥", "spending": 142.4},
-        {"condition": "💉 Diabetes", "emoji": "🩸", "spending": 107.4},
-        {"condition": "🫁 Chronic obstructive pulmonary disease and asthma", "emoji": "🫀", "spending": 91.0},
-        {"condition": "🩺 Hypertension", "emoji": "💉", "spending": 83.9},
-        {"condition": "🔬 Hyperlipidemia", "emoji": "🔬", "spending": 83.9},
-        {"condition": "🦴 Back problems", "emoji": "🧍", "spending": 67.0}
+        {"condition": "💔 Heart disease", "emoji": "💗", "spending": 214.3, "treatment": "Regular checkups with a cardiologist", "savings": "$1000"},
+        {"condition": "🤕 Trauma-related disorders", "emoji": "🚑", "spending": 198.6, "treatment": "Counseling and physical therapy", "savings": "$500"},
+        {"condition": "🦀 Cancer", "emoji": "🎗️", "spending": 171.0, "treatment": "Early detection and treatment", "savings": "$2000"},
+        {"condition": "🧠 Mental disorders", "emoji": "🧘", "spending": 150.8, "treatment": "Therapy and medication", "savings": "$1500"},
+        {"condition": "🦴 Osteoarthritis and joint disorders", "emoji": "🏥", "spending": 142.4, "treatment": "Low-impact exercise and physical therapy", "savings": "$800"},
+        {"condition": "💉 Diabetes", "emoji": "🩸", "spending": 107.4, "treatment": "Regular checkups and medication", "savings": "$1200"},
+        {"condition": "🫁 Chronic obstructive pulmonary disease and asthma", "emoji": "🫀", "spending": 91.0, "treatment": "Inhalers and breathing exercises", "savings": "$600"},
+        {"condition": "🩺 Hypertension", "emoji": "💉", "spending": 83.9, "treatment": "Lifestyle changes and medication", "savings": "$900"},
+        {"condition": "🔬 Hyperlipidemia", "emoji": "🔬", "spending": 83.9, "treatment": "Lifestyle changes and medication", "savings": "$700"},
+        {"condition": "🦴 Back problems", "emoji": "🧍", "spending": 67.0, "treatment": "Physical therapy and exercise", "savings": "$400"}
     ]
 
-    total_spending = sum([hc["spending"] for hc in health_conditions])
-
+    # Create a DataFrame from the list dictionary
     df_top_conditions = pd.DataFrame(health_conditions)
 
-    locations = ["CA", "FL", "MN", "WA", "ME", "TX"]
-    fig_map = px.choropleth(locations=locations, locationmode="USA-states", color=[1, 2, 3, 4, 5, 6], scope="usa")
-    #fig_map.update_layout(geo=dict(bgcolor="rgba(0,0,0,0)", lakecolor="rgb(255, 255, 255)")
-    
-    fig_map.update_layout(geo=dict(bgcolor= "rgba(0, 0, 0, 0)", lakecolor="rgb(255, 255, 255)"))
-
-    fig_bar = go.Figure(go.Bar(x=df_top_conditions["emoji"] + " " + df_top_conditions["condition"], y=df_top_conditions["spending"], marker_color="#1f77b4"))
-    fig_bar.update_layout(title=f"Top {top_n} Health Conditions in {', '.join(states)} by Spending (Total: ${total_spending}B)",
-                      yaxis_title="Spending ($B)", plot_bgcolor='rgba(0,0,0,0)', xaxis=dict(showgrid=False), yaxis=dict(showgrid=False))
-
-
-   # fig_bar.update_layout(title=f"Top {top_n} Health Conditions in {', '.join(states)} by Spending (Total: ${total_spending}B)",
-    #                      yaxis_title="Spending ($B)", plot_bgcolor='rgba(0,0,0,0)', xaxis=dict(showgrid=False), yaxis=dict(showgrid=False))
-    
-    col1, col2 = st.columns([2, 1])
-    with col1:
-        st.plotly_chart(fig_map, use_container_width=True)
-    with col2:
-        st.plotly_chart(fig_bar, use_container_width=True)
-
-st.title('Community Hub Form')
+    # Calculate the total spending
+    total_spending = round(df_top_conditions["spending"].sum(), 1)
 
-name = st.text_input('Name', key='name_input')
-email = st.text_input('Email', key='email_input')
-phone = st.text_input('Phone', key='phone_input')
+    # Define the roll function
+    def roll():
+        rolls = [random.randint(1, 10) for _ in range(1000)]
+        frequencies = [rolls.count(i) for i in range(1, 11)]
+        return frequencies
 
-if st.button('Submit', key='submit_button'):
-    save_data(name, email, phone)
-    st.success('Form submitted!')
+    # Define the sunburst chart
+    fig_sunburst = go.Figure(go.Sunburst(
+        labels=df_top_conditions["emoji"] + " " + df_top_conditions["condition"],
+        parents=[""] * top_n,
+        values=df_top_conditions["spending"],
+        maxdepth=2
+    ))
 
-if st.button('Reset', key='reset_button'):
-    reset_data()
-    st.success('Data reset!')
+    # Customize the layout of the sunburst chart
+   
+    fig_sunburst.update_layout(title=f"Top {top_n} Health Conditions in {', '.join(states)} by Spending (Total: ${total_spending}B)")
 
-if st.button('Show data', key='show_data_button'):
-    show_data()
+    # Display the sunburst chart and variants per condition in the Streamlit app
+    st.plotly_chart(fig_sunburst)
+    condition_idx = st.selectbox("Select your current health condition", df_top_conditions.index)
+    row = df_top_conditions.loc[condition_idx]
 
-st.write('Emails/SMS')
-with open('community.csv', mode='r') as csv_file:
-    csv_reader = csv.DictReader(csv_file)
-    for row in csv_reader:
-        name = row['Name']
-        email = row['Email']
-        phone = row['Phone']
-        st.write(f'{name}: {email} - {phone}')
-        if st.button('Reply', key=f'reply_{name}'):
-            st.text_input(f'Reply to {name}', key=f'reply_input_{name}')
-            if st.button(f'Vote up {name}', key=f'vote_up_{name}'):
-                st.success(f'{name} has been voted up!')
-            if st.button(f'No thanks to {name}', key=f'vote_down_{name}'):
-                st.warning(f'{name} has been voted down!')
-                with open(f'{name}.txt', mode='w') as file:
-                    file.write(f'Life points: 0\n')
-        st.write('')
+    st.write(f"Based on the severity of your {row['condition']}, we recommend {row['treatment']} for early treatment. This could save you up to {row['savings']} in healthcare costs.")
 
-st.write('Add tip')
-tip = st.text_input('Tip', key='tip_input')
-if st.button('Submit tip', key='submit_tip_button'):
-    emoji = random.choice(['👍', '👌', '👏', '💡'])
-    with open(f'{name}.txt', mode='a') as file:
-        file.write(f'Tip' + ': {tip} {emoji}\nLife points: 10\n')
-    st.success('Tip submitted!')
+    frequencies = roll()
+    fig_bar = px.bar(x=[f"Variant {i}" for i in range(1, 11)], y=frequencies[:10], labels={'x': 'Variant', 'y': 'Frequency'})
+    fig_bar.update_layout(title=f"Variants of {row['condition']} ({row['emoji']})")
+    st.plotly_chart(fig_bar)
 
-if name == 'main':
-    app()
 
+health_game()