Update app.py

app.py

@@ -69,14 +69,6 @@ def convert_dicom_to_png(dicom_data, output_dir, index):
         f.write(img_buffer.read())
     return output_path
 
-def find_dicom_files(directory):
-    dicom_files = []
-    for root, _, files in os.walk(directory):
-        for file in files:
-            if file.endswith(".dcm"):
-                dicom_files.append(os.path.join(root, file))
-    return dicom_files
-
 if uploaded_zip_file is not None:
     try:
         install_dependencies()
@@ -93,8 +85,9 @@ if uploaded_zip_file is not None:
         with zipfile.ZipFile(uploaded_zip_file, "r") as zip_ref:
             zip_ref.extractall(temp_dir)
 
-        # Find DICOM files in the directory, including subdirectories
-        dicom_files = find_dicom_files(temp_dir)
+        # 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.")