Update app.py

app.py

@@ -1,11 +1,15 @@
 import streamlit as st
 import time
 
-# Track the current scene
+# Initialize session state variables
 if "scene" not in st.session_state:
-    st.session_state["scene"] = 1  # Start with Scene 1
+    st.session_state["scene"] = 1  # Start at Scene 1
 if "pain_increase" not in st.session_state:
-    st.session_state["pain_increase"] = False  # Track pain progression
+    st.session_state["pain_increase"] = False  # Tracks whether pain has worsened
+
+# Function to change the scene
+def change_scene(next_scene):
+    st.session_state["scene"] = next_scene
 
 # Scene 1: Patient arrives at ER
 if st.session_state["scene"] == 1:
@@ -21,21 +25,21 @@ if st.session_state["scene"] == 1:
     if choice == "Examine the leg":
         st.image("2.jpg", caption="Exam findings: Swollen leg, no significant tension", use_container_width=True)
         st.write("Findings: Swollen leg with mild tenderness. Compartment Syndrome is not confirmed.")
-        if st.button("Proceed"):
-            st.session_state["scene"] = 2
+        if st.button("Proceed to next step"):
+            change_scene(2)
 
     elif choice == "Send for X-ray":
         st.image("3.jpg", caption="X-ray: Tibial fracture observed", use_container_width=True)
         st.write("Findings: Tibial fracture confirmed. This does not rule out Compartment Syndrome.")
-        if st.button("Go Back"):
-            st.session_state["scene"] = 1
+        if st.button("Go Back to Scene 1"):
+            change_scene(1)
 
     elif choice == "Administer painkillers":
         st.image("4.jpg", caption="Painkillers administered", use_container_width=True)
         st.success("The patient reports feeling better for now.")
-        st.session_state["pain_increase"] = True  # Pain will worsen later
-        if st.button("Proceed"):
-            st.session_state["scene"] = 2
+        st.session_state["pain_increase"] = True  # Set pain to increase later
+        if st.button("Proceed to next step"):
+            change_scene(2)
 
 # Scene 2: Monitoring or Further Action
 elif st.session_state["scene"] == 2:
@@ -48,22 +52,23 @@ elif st.session_state["scene"] == 2:
 
     if choice == "Monitor and Wait":
         st.write("Monitoring the patient... Please wait.")
-        time.sleep(5)  # Simulate waiting time
+        time.sleep(5)  # Simulate waiting
         if st.session_state["pain_increase"]:
             st.warning("The patient reports severe pain again.")
-            st.session_state["scene"] = 3
+            if st.button("Proceed to next step"):
+                change_scene(3)
 
     elif choice == "Repeat Exam":
         st.image("5.jpg", caption="Repeat examination shows no major changes", use_container_width=True)
         st.write("Findings: The swelling remains mild. Compartment Pressure is not elevated yet.")
-        if st.button("Go Back"):
-            st.session_state["scene"] = 2
+        if st.button("Go Back to Scene 2"):
+            change_scene(2)
 
     elif choice == "Measure Compartment Pressure":
         st.image("6.jpg", caption="Compartment pressure measurement", use_container_width=True)
         st.write("Results: Compartment pressure is within normal limits at this time.")
-        if st.button("Proceed"):
-            st.session_state["scene"] = 2
+        if st.button("Proceed to next step"):
+            change_scene(2)
 
 # Scene 3: Progression of Symptoms
 elif st.session_state["scene"] == 3:
@@ -77,11 +82,13 @@ elif st.session_state["scene"] == 3:
     if choice == "Re-check Compartment Pressure":
         st.image("7.jpg", caption="Re-measurement of Compartment Pressure", use_container_width=True)
         st.success("Pressure >30 mmHg. Confirmed Compartment Syndrome.")
-        st.session_state["scene"] = 4
+        if st.button("Proceed to next step"):
+            change_scene(4)
 
     elif choice == "Continue Monitoring":
         st.error("The patient's condition worsens further. Necrosis has begun.")
-        st.session_state["scene"] = 4
+        if st.button("Proceed to next step"):
+            change_scene(4)
 
 # Scene 4: Treatment
 elif st.session_state["scene"] == 4: