Update app.py

app.py

@@ -6,7 +6,9 @@ import os
 import subprocess
 from datetime import datetime
 import shutil
-import moviepy.editor as mp
+import moviepy.video.io.ImageSequenceClip
+from io import BytesIO
+from tkinter import Tcl
 
 
 # Flag to track if subprocess commands have been executed
@@ -53,24 +55,29 @@ def run_inference():
             print(f"Error while running the script: {e}")
 
 # 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, output_dir, index):
+    # 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)
+    output_path = os.path.join(output_dir, f"image_{index:03d}.png")
+    with open(output_path, "wb") as f:
+        f.write(img_buffer.read())
+    return output_path
 
 if uploaded_zip_file is not None:
     try:
         install_dependencies()
             
         temp_dir = "/home/user/app/C2C/temp_dicom_dir"
+        output_image_dir = "/home/user/app/C2C/images_for_video"
         
         os.makedirs(temp_dir, exist_ok=True)
+        os.makedirs(output_image_dir, exist_ok=True)
         full_path = os.path.abspath(temp_dir)
 
         st.write(f"Temporary directory path: {full_path}")
@@ -82,15 +89,27 @@ 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:
-        #     # Display a slider for selecting the slice
-        #     selected_slice = st.slider("Select a slice", 0, len(dicom_files) - 1, 0)
-
-        #     # Display the DICOM image using the cached function
-        #     plt = display_dicom_image(selected_slice, dicom_files)
-        #     st.pyplot(plt)
+        if len(dicom_files) == 0:
+            st.error("No DICOM files found in the ZIP archive.")
+        else:
+            # Convert DICOM slices to PNG frames and save them to the output folder
+            for i, dicom_file in enumerate(dicom_files):
+                dicom_data = pydicom.dcmread(dicom_file)
+                convert_dicom_to_png(dicom_data, output_image_dir, i)
+
+        image_folder = output_image_dir
+        fps = 20
+        image_files = [
+            os.path.join(image_folder, img)
+            for img in Tcl().call("lsort", "-dict", os.listdir(image_folder))
+            if img.endswith(".png")
+        ]
+        clip = moviepy.video.io.ImageSequenceClip.ImageSequenceClip(
+            image_files, fps=fps
+        )
+        clip.write_videofile("aaa.mp4")
+
+        st.video("aaa.mp4", format="video/mp4")
             
     except Exception as e:
         st.error(f"Error: {str(e)}")