Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import plotly.graph_objects as go
 import random
-import time
 
 # Initialize session state
 if 'initialized' not in st.session_state:
@@ -88,60 +87,50 @@ if st.button("Run Animations"):
     st.session_state.initialized = True
 
 if st.session_state.initialized:
+    # Slider to choose the animation
+    animation_index = st.slider("Select Animation", min_value=1, max_value=5, value=1)
+
     # Animation 1: Patient Conditions
-    placeholder1 = st.empty()
-    st.subheader("Patients by Condition")
-    for condition, count in st.session_state.condition_counts.items():
-        fig = create_gauge(f"Patients with {condition}", count, 200, {'x': [0, 1], 'y': [0, 1]}, 'blue')
-        placeholder1.plotly_chart(fig, use_container_width=True)
-        time.sleep(3)  # Wait for 3 seconds before showing the next frame
-        placeholder1.empty()  # Clear the previous frame
-    time.sleep(3)  # Additional pause before moving to the next animation
+    if animation_index == 1:
+        st.subheader("Patients by Condition")
+        for condition, count in st.session_state.condition_counts.items():
+            fig = create_gauge(f"Patients with {condition}", count, 200, {'x': [0, 1], 'y': [0, 1]}, 'blue')
+            st.plotly_chart(fig, use_container_width=True)
+            st.markdown("---")
 
     # Animation 2: Treatment Metrics
-    placeholder2 = st.empty()
-    st.subheader("Treatment Metrics")
-    fig1 = create_gauge("Total Patients", 200, 200, {'x': [0, 0.5], 'y': [0, 1]}, 'green')
-    fig2 = create_gauge("Successful Treatments", st.session_state.successful_treatments, st.session_state.total_treatments, {'x': [0.5, 1], 'y': [0, 0.5]}, 'orange')
-    fig3 = create_gauge("Unsuccessful Treatments", st.session_state.unsuccessful_treatments, st.session_state.total_treatments, {'x': [0.5, 1], 'y': [0.5, 1]}, 'red')
-    placeholder2.plotly_chart(fig1, use_container_width=True)
-    time.sleep(3)  # Wait for 3 seconds before showing the next frame
-    placeholder2.empty()  # Clear the previous frame
-    placeholder2.plotly_chart(fig2, use_container_width=True)
-    time.sleep(3)  # Wait for 3 seconds before showing the next frame
-    placeholder2.empty()  # Clear the previous frame
-    placeholder2.plotly_chart(fig3, use_container_width=True)
-    time.sleep(3)  # Additional pause before moving to the next animation
+    elif animation_index == 2:
+        st.subheader("Treatment Metrics")
+        fig1 = create_gauge("Total Patients", 200, 200, {'x': [0, 0.5], 'y': [0, 1]}, 'green')
+        fig2 = create_gauge("Successful Treatments", st.session_state.successful_treatments, st.session_state.total_treatments, {'x': [0.5, 1], 'y': [0, 0.5]}, 'orange')
+        fig3 = create_gauge("Unsuccessful Treatments", st.session_state.unsuccessful_treatments, st.session_state.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
-    placeholder3 = st.empty()
-    st.subheader("Doctor Actions")
-    for action, count in st.session_state.doctor_actions.items():
-        fig = create_gauge(f"Doctor Action: {action}", count, st.session_state.total_treatments, {'x': [0, 1], 'y': [0, 1]}, 'purple')
-        placeholder3.plotly_chart(fig, use_container_width=True)
-        time.sleep(3)  # Wait for 3 seconds before showing the next frame
-        placeholder3.empty()  # Clear the previous frame
-    time.sleep(3)  # Additional pause before moving to the next animation
+    elif animation_index == 3:
+        st.subheader("Doctor Actions")
+        for action, count in st.session_state.doctor_actions.items():
+            fig = create_gauge(f"Doctor Action: {action}", count, st.session_state.total_treatments, {'x': [0, 1], 'y': [0, 1]}, 'purple')
+            st.plotly_chart(fig, use_container_width=True)
+            st.markdown("---")
 
     # Animation 4: Penalties
-    placeholder4 = st.empty()
-    st.subheader("Penalties")
-    for penalty, count in st.session_state.penalties.items():
-        fig = create_gauge(f"Penalty: {penalty}", count, st.session_state.total_treatments // 6, {'x': [0, 1], 'y': [0, 1]}, 'red')
-        placeholder4.plotly_chart(fig, use_container_width=True)
-        time.sleep(3)  # Wait for 3 seconds before showing the next frame
-        placeholder4.empty()  # Clear the previous frame
-    time.sleep(3)  # Additional pause before moving to the next animation
+    elif animation_index == 4:
+        st.subheader("Penalties")
+        for penalty, count in st.session_state.penalties.items():
+            fig = create_gauge(f"Penalty: {penalty}", count, st.session_state.total_treatments // 6, {'x': [0, 1], 'y': [0, 1]}, 'red')
+            st.plotly_chart(fig, use_container_width=True)
+            st.markdown("---")
 
     # Animation 5: Patient Feedback
-    placeholder5 = st.empty()
-    st.subheader("Patient Feedback")
-    for fb, count in st.session_state.feedback.items():
-        fig = create_gauge(f"Feedback: {fb}", count, st.session_state.total_treatments, {'x': [0, 1], 'y': [0, 1]}, 'blue')
-        placeholder5.plotly_chart(fig, use_container_width=True)
-        time.sleep(3)  # Wait for 3 seconds before showing the next frame
-        placeholder5.empty()  # Clear the previous frame
-    st.write("**End of Animations.**")
+    elif animation_index == 5:
+        st.subheader("Patient Feedback")
+        for fb, count in st.session_state.feedback.items():
+            fig = create_gauge(f"Feedback: {fb}", count, st.session_state.total_treatments, {'x': [0, 1], 'y': [0, 1]}, 'blue')
+            st.plotly_chart(fig, use_container_width=True)
+            st.markdown("---")
 
-    # Reset session state to allow re-running the animations if needed
-    st.session_state.initialized = False
+    st.write("**End of Animations.**")