Update app.py

app.py

@@ -1,52 +1,103 @@
+import streamlit as st
 import plotly.graph_objects as go
+import time
+import random
 
-def create_animation(successful_treatments, failed_treatments):
-    # Create initial figure
+# Define the data
+agents = {
+    "Doctors": 5,
+    "Nurses": 10,
+    "Clinicians": 8,
+    "Patients": 200
+}
+
+patient_conditions = ["Healthy", "Mild Illness", "Chronic Illness", "Emergency"]
+doctor_emotions = ["Calm", "Stressed", "Overwhelmed"]
+nurse_emotions = ["Focused", "Fatigued", "Panicked"]
+doctor_actions = ["Prescribe Medication", "Recommend Tests", "Consult Clinician", "Schedule Surgery"]
+nurse_actions = ["Monitor Vitals", "Administer Medication", "Report to Doctor", "Assist Surgery"]
+
+# Initial values for successful and unsuccessful treatments
+doctor_successful_treatments = random.randint(0, 10)
+doctor_unsuccessful_treatments = random.randint(0, 10)
+
+# Streamlit setup
+st.title("Interactive Healthcare Animation")
+
+# Function to create interactive graph
+def create_health_graph(agents, patient_condition, doctor_emotion, nurse_emotion, doctor_action, nurse_action, success, failure):
     fig = go.Figure()
 
-    # Add bar traces
-    fig.add_trace(go.Bar(x=['Doctor', 'Nurse'],
-                         y=[successful_treatments['Doctor'], successful_treatments['Nurse']],
-                         name='Successful Treatments',
-                         marker_color='green'))
-    
-    fig.add_trace(go.Bar(x=['Doctor', 'Nurse'],
-                         y=[failed_treatments['Doctor'], failed_treatments['Nurse']],
-                         name='Failed Treatments',
-                         marker_color='red'))
-
-    # Add animation frames (if needed for more dynamic changes over time)
-    frames = [go.Frame(data=[go.Bar(x=['Doctor', 'Nurse'],
-                                    y=[s, f])],
-                       name=f'Frame {i}') 
-              for i, (s, f) in enumerate(zip(successful_treatments.values(), failed_treatments.values()))]
-
-    # Set the layout for the plot
+    # Agents count graph
+    fig.add_trace(go.Bar(
+        x=list(agents.keys()),
+        y=list(agents.values()),
+        name='Agents Count'
+    ))
+
+    # Patient condition as a pie chart
+    fig.add_trace(go.Pie(
+        labels=['Healthy', 'Mild Illness', 'Chronic Illness', 'Emergency'],
+        values=[agents['Patients'] * 0.5, agents['Patients'] * 0.2, agents['Patients'] * 0.2, agents['Patients'] * 0.1],
+        title='Patient Conditions'
+    ))
+
     fig.update_layout(
-        title="Doctor and Nurse Performance Metrics",
-        barmode='group',
-        xaxis_title="Personnel",
-        yaxis_title="Number of Treatments",
-        updatemenus=[{
-            'buttons': [
-                {'args': [None, {'frame': {'duration': 500, 'redraw': True}, 'fromcurrent': True}], 
-                 'label': 'Play',
-                 'method': 'animate'},
-                {'args': [[None], {'frame': {'duration': 0, 'redraw': True}, 'mode': 'immediate'}], 
-                 'label': 'Pause',
-                 'method': 'animate'}
-            ],
-            'type': 'buttons'
-        }]
+        title='Healthcare Agents Status',
+        template='plotly_dark',
+        barmode='group'
+    )
+    
+    # Display emotions and actions as text
+    fig.add_annotation(
+        text=f"Doctor Emotion: {doctor_emotion}\nNurse Emotion: {nurse_emotion}\nDoctor Action: {doctor_action}\nNurse Action: {nurse_action}",
+        xref='paper', yref='paper',
+        x=0.5, y=-0.2,
+        showarrow=False,
+        font=dict(size=14)
+    )
+    
+    # Display successful and unsuccessful treatments
+    fig.add_annotation(
+        text=f"Successful Treatments: {success}\nUnsuccessful Treatments: {failure}",
+        xref='paper', yref='paper',
+        x=0.5, y=-0.35,
+        showarrow=False,
+        font=dict(size=14)
     )
-
-    # Add frames to the figure for the animation
-    fig.frames = frames
 
     return fig
 
-# Example call
-successful_treatments = {'Doctor': [10, 20, 30], 'Nurse': [5, 15, 25]}
-failed_treatments = {'Doctor': [2, 5, 10], 'Nurse': [1, 3, 6]}
-fig = create_animation(successful_treatments, failed_treatments)
-fig.show()
+# Button to start the animation
+start_button = st.button("Start Animation")
+
+# Check if the start button is pressed
+if start_button:
+    # Placeholder for dynamic updates
+    placeholder = st.empty()
+
+    # Loop through each frame
+    for i in range(20):  # 20 frames for the animation
+        with placeholder.container():
+            # Randomly select conditions, emotions, and actions
+            patient_condition = random.choice(patient_conditions)
+            doctor_emotion = random.choice(doctor_emotions)
+            nurse_emotion = random.choice(nurse_emotions)
+            doctor_action = random.choice(doctor_actions)
+            nurse_action = random.choice(nurse_actions)
+
+            # Randomly adjust successful and unsuccessful treatments
+            if random.choice([True, False]):
+                doctor_successful_treatments += 1
+            else:
+                doctor_unsuccessful_treatments += 1
+
+            # Create and display the graph
+            fig = create_health_graph(agents, patient_condition, doctor_emotion, nurse_emotion, doctor_action, nurse_action, doctor_successful_treatments, doctor_unsuccessful_treatments)
+            st.plotly_chart(fig, use_container_width=True)
+
+        # Pause for 2 seconds before moving to the next frame
+        time.sleep(2)
+
+    # Display a final summary
+    st.write("**Animation Complete: Healthcare agents' statuses have been updated.**")