Update app.py

app.py

@@ -70,60 +70,62 @@ def create_gauge(title, value, max_value, domain, gauge_color):
     fig.update_layout(template='plotly_dark')
     return fig
 
-# Prepare data
-distribute_patients()
-total_treatments, successful_treatments, unsuccessful_treatments, patients_waiting = generate_treatment_metrics()
-doctor_actions = generate_doctor_actions(total_treatments)
-penalties = generate_penalties(total_treatments)
-feedback = generate_feedback(total_treatments)
-
 # Streamlit setup
 st.title("Interactive Healthcare Metrics")
 
-# Slider to choose the animation
-animation_index = st.slider("Select Animation", min_value=1, max_value=5, value=1)
-
-# Animation 1: Patient Conditions
-if animation_index == 1:
-    st.subheader("Patients by Condition")
-    for condition, count in condition_counts.items():
-        fig = create_gauge(f"Patients with {condition}", count, agents["Patients"], {'x': [0, 1], 'y': [0, 1]}, 'blue')
-        st.plotly_chart(fig, use_container_width=True)
-        st.markdown("---")
-
-# Animation 2: Treatment Metrics
-elif animation_index == 2:
-    st.subheader("Treatment Metrics")
-    fig1 = create_gauge("Total Patients", agents["Patients"], agents["Patients"], {'x': [0, 0.5], 'y': [0, 1]}, 'green')
-    fig2 = create_gauge("Successful Treatments", successful_treatments, total_treatments, {'x': [0.5, 1], 'y': [0, 0.5]}, 'orange')
-    fig3 = create_gauge("Unsuccessful Treatments", unsuccessful_treatments, total_treatments, {'x': [0.5, 1], 'y': [0.5, 1]}, 'red')
-    st.plotly_chart(fig1, use_container_width=True)
-    st.plotly_chart(fig2, use_container_width=True)
-    st.plotly_chart(fig3, use_container_width=True)
-    st.markdown("---")
-
-# Animation 3: Doctor Actions
-elif animation_index == 3:
-    st.subheader("Doctor Actions")
-    for action, count in doctor_actions.items():
-        fig = create_gauge(f"Doctor Action: {action}", count, total_treatments, {'x': [0, 1], 'y': [0, 1]}, 'purple')
-        st.plotly_chart(fig, use_container_width=True)
-        st.markdown("---")
-
-# Animation 4: Penalties
-elif animation_index == 4:
-    st.subheader("Penalties")
-    for penalty, count in penalties.items():
-        fig = create_gauge(f"Penalty: {penalty}", count, total_treatments // 6, {'x': [0, 1], 'y': [0, 1]}, 'red')
-        st.plotly_chart(fig, use_container_width=True)
-        st.markdown("---")
-
-# Animation 5: Patient Feedback
-elif animation_index == 5:
-    st.subheader("Patient Feedback")
-    for fb, count in feedback.items():
-        fig = create_gauge(f"Feedback: {fb}", count, total_treatments, {'x': [0, 1], 'y': [0, 1]}, 'blue')
-        st.plotly_chart(fig, use_container_width=True)
+# Button to run the animations
+if st.button("Run Animations"):
+    # Prepare data
+    distribute_patients()
+    total_treatments, successful_treatments, unsuccessful_treatments, patients_waiting = generate_treatment_metrics()
+    doctor_actions = generate_doctor_actions(total_treatments)
+    penalties = generate_penalties(total_treatments)
+    feedback = generate_feedback(total_treatments)
+    
+    # Slider to choose the animation
+    animation_index = st.slider("Select Animation", min_value=1, max_value=5, value=1)
+
+    # Animation 1: Patient Conditions
+    if animation_index == 1:
+        st.subheader("Patients by Condition")
+        for condition, count in condition_counts.items():
+            fig = create_gauge(f"Patients with {condition}", count, agents["Patients"], {'x': [0, 1], 'y': [0, 1]}, 'blue')
+            st.plotly_chart(fig, use_container_width=True)
+            st.markdown("---")
+
+    # Animation 2: Treatment Metrics
+    elif animation_index == 2:
+        st.subheader("Treatment Metrics")
+        fig1 = create_gauge("Total Patients", agents["Patients"], agents["Patients"], {'x': [0, 0.5], 'y': [0, 1]}, 'green')
+        fig2 = create_gauge("Successful Treatments", successful_treatments, total_treatments, {'x': [0.5, 1], 'y': [0, 0.5]}, 'orange')
+        fig3 = create_gauge("Unsuccessful Treatments", unsuccessful_treatments, total_treatments, {'x': [0.5, 1], 'y': [0.5, 1]}, 'red')
+        st.plotly_chart(fig1, use_container_width=True)
+        st.plotly_chart(fig2, use_container_width=True)
+        st.plotly_chart(fig3, use_container_width=True)
         st.markdown("---")
 
-st.write("**End of Animations.**")
+    # Animation 3: Doctor Actions
+    elif animation_index == 3:
+        st.subheader("Doctor Actions")
+        for action, count in doctor_actions.items():
+            fig = create_gauge(f"Doctor Action: {action}", count, total_treatments, {'x': [0, 1], 'y': [0, 1]}, 'purple')
+            st.plotly_chart(fig, use_container_width=True)
+            st.markdown("---")
+
+    # Animation 4: Penalties
+    elif animation_index == 4:
+        st.subheader("Penalties")
+        for penalty, count in penalties.items():
+            fig = create_gauge(f"Penalty: {penalty}", count, total_treatments // 6, {'x': [0, 1], 'y': [0, 1]}, 'red')
+            st.plotly_chart(fig, use_container_width=True)
+            st.markdown("---")
+
+    # Animation 5: Patient Feedback
+    elif animation_index == 5:
+        st.subheader("Patient Feedback")
+        for fb, count in feedback.items():
+            fig = create_gauge(f"Feedback: {fb}", count, total_treatments, {'x': [0, 1], 'y': [0, 1]}, 'blue')
+            st.plotly_chart(fig, use_container_width=True)
+            st.markdown("---")
+
+    st.write("**End of Animations.**")