Update app.py

app.py

@@ -54,21 +54,6 @@ def run_inference():
         except subprocess.CalledProcessError as e:
             print(f"Error while running the script: {e}")
 
-# Function to read and 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()
-    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()
@@ -88,28 +73,6 @@ if uploaded_zip_file is not None:
         # Get a list of DICOM files in the directory
         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:
-            # 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)}")