Update app.py

app.py

@@ -1,11 +1,10 @@
-import streamlit as st
-import plotly.express as px
 import pandas as pd
+import streamlit as st
+from graphviz import Digraph
 
-# Define the states and conditions of interest
-states = ["Minnesota 🌲", "Florida 🌴", "California 🌞"]
-top_n = 10
+st.set_page_config(page_title="Ideal Care Plan for Top 10 Health Conditions")
 
+# Create a pandas dataframe for the top 10 health conditions and their spending
 health_conditions = [
     {"condition": "Heart disease ❤️", "spending": 214.3},
     {"condition": "Trauma-related disorders 🤕", "spending": 198.6},
@@ -16,20 +15,40 @@ health_conditions = [
     {"condition": "Chronic obstructive pulmonary disease and asthma 🫁", "spending": 91.0},
     {"condition": "Hypertension 🩺", "spending": 83.9},
     {"condition": "Hyperlipidemia 🍔", "spending": 83.9},
-    {"condition": "Back problems 👨‍⚕️", "spending": 67.0}
+    {"condition": "Back problems 👨‍⚕️", "spending": 67.0},
 ]
 
-# Total the spending values
-total_spending = sum([hc["spending"] for hc in health_conditions])
+df = pd.DataFrame(health_conditions)
+
+# Create Graphviz graph object
+graph = Digraph(comment="Ideal Care Plan for Top 10 Health Conditions")
+graph.graph_attr["rankdir"] = "LR"  # left to right layout
 
-# Create a DataFrame from the list dictionary
-df_top_conditions = pd.DataFrame(health_conditions)
+# Add nodes to the graph for each health condition
+for i, row in df.iterrows():
+    condition = row["condition"]
+    spending = row["spending"]
 
-# Create the treemap graph using Plotly Express
-fig = px.treemap(df_top_conditions, path=["condition"], values="spending")
+    # Add node to the graph with hyperlink based on condition
+    if i < 3:
+        url = f"https://www.cdc.gov/{condition.split()[0].lower()}"
+    else:
+        url = None
+    graph.node(condition, f"{condition}\n${spending:.1f}B", href=url)
 
-# Set the title of the graph
-fig.update_layout(title=f"Top {top_n} Health Conditions in {', '.join(states)} by Spending (Total: ${total_spending}B)")
+# Add edges to the graph based on related conditions
+graph.edge("Heart disease ❤️", "Hypertension 🩺")
+graph.edge("Heart disease ❤️", "Hyperlipidemia 🍔")
+graph.edge("Trauma-related disorders 🤕", "Mental disorders 🧠")
+graph.edge("Cancer 🦀", "Heart disease ❤️")
+graph.edge("Cancer 🦀", "Trauma-related disorders 🤕")
+graph.edge("Cancer 🦀", "Osteoarthritis and joint disorders 🦴")
+graph.edge("Mental disorders 🧠", "Heart disease ❤️")
+graph.edge("Mental disorders 🧠", "Trauma-related disorders 🤕")
+graph.edge("Osteoarthritis and joint disorders 🦴", "Back problems 👨‍⚕️")
+graph.edge("Diabetes 🍬", "Heart disease ❤️")
+graph.edge("Diabetes 🍬", "Hypertension 🩺")
+graph.edge("Chronic obstructive pulmonary disease and asthma 🫁", "Heart disease ❤️")
 
-# Display the graph in Streamlit
-st.plotly_chart(fig)
+# Render the graph as an image and display in Streamlit app
+st.graphviz_chart(graph.pipe(format="svg"))