Update app.py

app.py

@@ -32,44 +32,43 @@ st.graphviz_chart(g)
 
 
 
+def render_hospital_graph(hospital):
+    g = gv.Graph(format='svg')
+    g.graph_attr['bgcolor'] = '#FFFFFF'
+    g.graph_attr['outputorder'] = 'edgesfirst'
+    g.graph_attr['size'] = '10,10'
+    g.node_attr['style'] = 'filled'
+    g.node_attr['shape'] = 'box'
+    g.node_attr['fillcolor'] = '#FFDAB9'
 
-
-# Define the cluster relations graph using gvmap
-g = gv.Graph(format='svg')
-g.graph_attr['bgcolor'] = '#FFFFFF'
-g.graph_attr['outputorder'] = 'edgesfirst'
-g.graph_attr['size'] = '10,10'
-g.node_attr['style'] = 'filled'
-g.node_attr['shape'] = 'box'
-g.node_attr['fillcolor'] = '#FFDAB9'
-
-with g.subgraph(name='cluster_MN') as c:
-    c.graph_attr['bgcolor'] = '#ADD8E6'
-    c.node_attr['color'] = '#000000'
-    c.node_attr['fontcolor'] = '#000000'
-    c.attr(label='Minnesota', fontsize='24')
-    for hospital in [('Allina Health', 'Abbott Northwestern Hospital', 'Beds: 627', 'Specialties: Cardiovascular care, neurosciences, cancer care'),
-                     ('Fairview Health Services', 'University of Minnesota Medical Center', 'Beds: 914', 'Specialties: Cancer care, transplant services, orthopedics'),
-                     ('HealthPartners', 'Regions Hospital', 'Beds: 454', 'Specialties: Level I Trauma Center, heart care, cancer care'),
-                     ('Mayo Clinic', 'Saint Marys Hospital', 'Beds: 1,265', 'Specialties: Cardiology, neurology, gastroenterology'),
-                     ('CentraCare', 'St. Cloud Hospital', 'Beds: 489', 'Specialties: Cancer care, heart care, neuroscience'),
-                     ('Essentia Health', 'St. Mary’s Medical Center', 'Beds: 330', 'Specialties: Cancer care, heart care, orthopedics'),
-                     ('Hennepin Healthcare', 'HCMC', 'Beds: 497', 'Specialties: Level I Trauma Center, burn center, cancer care'),
-                     ("Children's Minnesota", "Children's Hospitals and Clinics of Minnesota", 'Beds: 381', 'Specialties: Pediatrics, heart care, neonatology'),
-                     ('Sanford Health', 'Sanford Bemidji Medical Center', 'Beds: 161', 'Specialties: Cancer care, heart care, orthopedics'),
-                     ("St. Luke's Hospital", "St. Luke's Hospital", 'Beds: 267', 'Specialties: Cancer care, heart care, orthopedics')]:
-        hospital_name, hospital_desc, bed_count, specialties = hospital
-        with c.subgraph(name=f'cluster_{hospital_name}') as s:
-            s.attr(label=hospital_name, fontsize='16')
-            s.node(hospital_desc, fontsize='12')
-            with s.subgraph(name=f'cluster_{hospital_desc}') as s1:
+    with g.subgraph(name=f'cluster_{hospital[0]}') as c:
+        c.graph_attr['bgcolor'] = '#ADD8E6'
+        c.node_attr['color'] = '#000000'
+        c.node_attr['fontcolor'] = '#000000'
+        c.attr(label=hospital[0], fontsize='24')
+        with c.subgraph(name=f'cluster_{hospital[1]}') as s:
+            s.attr(label=hospital[1], fontsize='16')
+            s.node(hospital[2], fontsize='12')
+            with s.subgraph(name=f'cluster_{hospital[2]}') as s1:
                 s1.attr(fontsize='10')
-                s1.node(bed_count)
-                s1.node(specialties)
+                s1.node(hospital[3])
 
-# Use neato to rearrange the clusters
-g.layout(prog='neato')
+    g.layout(prog='neato')
+    st.graphviz_chart(g)
 
-# Render the graph using streamlit
-st.graphviz_chart(g)
+# Define hospitals data
+hospitals = [('Allina Health', 'Abbott Northwestern Hospital', 'Beds: 627', 'Specialties: Cardiovascular care, neurosciences, cancer care'),
+             ('Fairview Health Services', 'University of Minnesota Medical Center', 'Beds: 914', 'Specialties: Cancer care, transplant services, orthopedics'),
+             ('HealthPartners', 'Regions Hospital', 'Beds: 454', 'Specialties: Level I Trauma Center, heart care, cancer care'),
+             ('Mayo Clinic', 'Saint Marys Hospital', 'Beds: 1,265', 'Specialties: Cardiology, neurology, gastroenterology'),
+             ('CentraCare', 'St. Cloud Hospital', 'Beds: 489', 'Specialties: Cancer care, heart care, neuroscience'),
+             ('Essentia Health', 'St. Mary’s Medical Center', 'Beds: 330', 'Specialties: Cancer care, heart care, orthopedics'),
+             ('Hennepin Healthcare', 'HCMC', 'Beds: 497', 'Specialties: Level I Trauma Center, burn center, cancer care'),
+             ("Children's Minnesota", "Children's Hospitals and Clinics of Minnesota", 'Beds: 381', 'Specialties: Pediatrics, heart care, neonatology'),
+             ('Sanford Health', 'Sanford Bemidji Medical Center', 'Beds: 161', 'Specialties: Cancer care, heart care, orthopedics'),
+             ("St. Luke's Hospital", "St. Luke's Hospital", 'Beds: 267', 'Specialties: Cancer care, heart care, orthopedics')]
+
+# Render a separate graph for each hospital
+for hospital in hospitals:
+    render_hospital_graph(hospital)