Update app.py

app.py

@@ -24,7 +24,7 @@ def get_slice_location(dicom_file):
         return 0  # Default to 0 if neither tag is available
 
 # Display the single DICOM image from the first folder
-st.header("Single DICOM Image from 00002067")
+st.header("Single DICOM Image from folder 00002067")
 
 # Check if the directory exists
 if os.path.exists(single_image_directory):
@@ -47,7 +47,7 @@ else:
     st.write("Directory 00002067 does not exist.")
 
 # Display the multiple DICOM images from the second folder using a slider
-st.header("Browse Multiple DICOM Images with slider")
+st.header("Browse Multiple DICOM Images from 00007FA6")
 
 # Check if the directory exists
 if os.path.exists(multiple_images_directory):
@@ -74,7 +74,7 @@ if os.path.exists(multiple_images_directory):
         st.pyplot(fig)
 
         # Automatic image display section
-        st.header("Automatic Image Display - view animated images (click start)")
+        st.header("Automatic Image Display (click start)")
         placeholder = st.empty()  # Placeholder for the images
 
         # Initialize session state for animation control
@@ -95,6 +95,13 @@ if os.path.exists(multiple_images_directory):
         if st.button("Resume Animation"):
             st.session_state.animation_running = True
 
+        # Display the current image based on the state
+        with placeholder:
+            fig, ax = plt.subplots()
+            ax.imshow(dicom_images[st.session_state.current_index], cmap='gray')
+            ax.axis('off')
+            st.pyplot(fig)
+
         # Run the animation loop if running
         if st.session_state.animation_running:
             while st.session_state.animation_running and st.session_state.current_index < len(dicom_images):