Update app.py

app.py

@@ -52,15 +52,16 @@ def run_inference():
 
 # Function to read and display the DICOM image
 @st.cache_resource
-def display_dicom_image(selected_slice, dicom_files):
-    dicom_data = pydicom.dcmread(dicom_files[selected_slice])
-
-    # Display the DICOM image
+def convert_dicom_to_png(dicom_data):
+    # Convert DICOM to a PNG image
     plt.imshow(dicom_data.pixel_array, cmap=plt.cm.bone)
     plt.axis("off")
     plt.title("DICOM Image")
     plt.tight_layout()
-    return plt
+    img_buffer = BytesIO()
+    plt.savefig(img_buffer, format="png")
+    img_buffer.seek(0)
+    return img_buffer
 
 if uploaded_zip_file is not None:
     try:
@@ -83,12 +84,20 @@ if uploaded_zip_file is not None:
         if len(dicom_files) == 0:
             st.error("No DICOM files found in the ZIP archive.")
         else:
-            # Display a slider for selecting the slice
-            selected_slice = st.slider("Select a slice", 0, len(dicom_files) - 1, 0)
+            # Create a list of DICOM data
+            dicom_data_list = [pydicom.dcmread(dicom_file) for dicom_file in dicom_files]
+
+            # Convert DICOM slices to PNG frames
+            png_frames = [convert_dicom_to_png(dicom_data) for dicom_data in dicom_data_list]
+
+            # Create a list of image clips from PNG frames
+            clips = [mp.ImageClip(png_frame, duration=1) for png_frame in png_frames]
+
+            # Concatenate the image clips to create a video
+            video_clip = mp.concatenate_videoclips(clips, method="compose")
 
-            # Display the DICOM image using the cached function
-            plt = display_dicom_image(selected_slice, dicom_files)
-            st.pyplot(plt)
+            # Display the video in Streamlit
+            st.video(video_clip, format="video/mp4")
             
     except Exception as e:
         st.error(f"Error: {str(e)}")