Update app.py

app.py

@@ -6,6 +6,8 @@ import os
 import subprocess
 from datetime import datetime
 import shutil
+import moviepy.editor as mp
+
 
 # Flag to track if subprocess commands have been executed
 subprocess_executed = False
@@ -52,16 +54,15 @@ def run_inference():
 
 # Function to read and display the DICOM image
 @st.cache_resource
-def convert_dicom_to_png(_dicom_data):
-    # Convert DICOM to a PNG image
-    plt.imshow(_dicom_data.pixel_array, cmap=plt.cm.bone)
+def display_dicom_image(selected_slice, dicom_files):
+    dicom_data = pydicom.dcmread(dicom_files[selected_slice])
+
+    # Display the DICOM image
+    plt.imshow(dicom_data.pixel_array, cmap=plt.cm.bone)
     plt.axis("off")
     plt.title("DICOM Image")
     plt.tight_layout()
-    img_buffer = BytesIO()
-    plt.savefig(img_buffer, format="png")
-    img_buffer.seek(0)
-    return img_buffer
+    return plt
 
 if uploaded_zip_file is not None:
     try:
@@ -81,23 +82,15 @@ if uploaded_zip_file is not None:
         dicom_files = [os.path.join(temp_dir, f) for f in os.listdir(temp_dir) if f.endswith(".dcm")]
         dicom_files.sort()  # Sort the files
 
-        if len(dicom_files) == 0:
-            st.error("No DICOM files found in the ZIP archive.")
-        else:
-            # 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")
+        # 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)
 
-            # Display the video in Streamlit
-            st.video(video_clip, format="video/mp4")
+        #     # Display the DICOM image using the cached function
+        #     plt = display_dicom_image(selected_slice, dicom_files)
+        #     st.pyplot(plt)
             
     except Exception as e:
         st.error(f"Error: {str(e)}")