Update app.py

app.py

@@ -1,10 +1,8 @@
-import pandas as pd
 import streamlit as st
 from graphviz import Digraph
 
 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},
@@ -18,37 +16,36 @@ health_conditions = [
     {"condition": "Back problems 👨‍⚕️", "spending": 67.0},
 ]
 
-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
-
-# Add nodes to the graph for each health condition
-for i, row in df.iterrows():
-    condition = row["condition"]
-    spending = row["spending"]
-
-    # Add node to the graph with hyperlink based on condition
-    if i < 3:
-        url = f"https://www.cdc.gov/{condition.split()[0].lower()}"
+urls = {
+    "Heart disease ❤️": "https://www.cdc.gov/heartdisease",
+    "Trauma-related disorders 🤕": "https://www.nimh.nih.gov/health/topics/post-traumatic-stress-disorder-ptsd/index.shtml",
+    "Cancer 🦀": "https://www.cancer.gov/",
+}
+
+dot = Digraph(comment="Ideal Care Plan for Top 10 Health Conditions")
+dot.graph_attr["rankdir"] = "LR"
+for condition in health_conditions:
+    name = condition["condition"]
+    spending = condition["spending"]
+    if name in urls:
+        url = urls[name]
+        dot.node(name, f"{name}\n${spending:.1f}B", href=url)
     else:
-        url = None
-    graph.node(condition, f"{condition}\n${spending:.1f}B", href=url)
-
-# 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 ❤️")
-
-# Render the graph as an image and display in Streamlit app
-st.graphviz_chart(graph.pipe(format="svg"))
+        dot.node(name, f"{name}\n${spending:.1f}B")
+
+dot.edges([
+    ("Heart disease ❤️", "Hypertension 🩺"),
+    ("Heart disease ❤️", "Hyperlipidemia 🍔"),
+    ("Trauma-related disorders 🤕", "Mental disorders 🧠"),
+    ("Cancer 🦀", "Heart disease ❤️"),
+    ("Cancer 🦀", "Trauma-related disorders 🤕"),
+    ("Cancer 🦀", "Osteoarthritis and joint disorders 🦴"),
+    ("Mental disorders 🧠", "Heart disease ❤️"),
+    ("Mental disorders 🧠", "Trauma-related disorders 🤕"),
+    ("Osteoarthritis and joint disorders 🦴", "Back problems 👨‍⚕️"),
+    ("Diabetes 🍬", "Heart disease ❤️"),
+    ("Diabetes 🍬", "Hypertension 🩺"),
+    ("Chronic obstructive pulmonary disease and asthma 🫁", "Heart disease ❤️"),
+])
+
+st.graphviz_chart(dot.pipe(format="svg"))