Update app.py

app.py

@@ -31,10 +31,6 @@ with g.subgraph(name='cluster_MN') as c:
 st.graphviz_chart(g)
 
 
-
-import streamlit as st
-import graphviz as gv
-
 def render_hospital_graph(hospital):
     g = gv.Graph(format='svg')
     g.graph_attr['bgcolor'] = '#FFFFFF'
@@ -57,7 +53,7 @@ def render_hospital_graph(hospital):
                 s1.node(hospital[3])
 
     svg_output = g.pipe(format='svg')
-    st.image(svg_output, output_format='SVG')
+    return svg_output
 
 # Define hospitals data
 hospitals = [('Allina Health', 'Abbott Northwestern Hospital', 'Beds: 627', 'Specialties: Cardiovascular care, neurosciences, cancer care'),
@@ -73,9 +69,8 @@ hospitals = [('Allina Health', 'Abbott Northwestern Hospital', 'Beds: 627', 'Spe
 
 # Render a separate graph for each hospital
 for hospital in hospitals:
-    render_hospital_graph(hospital)
-
-
+    svg_output = render_hospital_graph(hospital)
+    st.write(svg_output, unsafe_allow_html=True)