Update app.py

app.py

@@ -69,6 +69,14 @@ 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()
@@ -85,9 +93,8 @@ if uploaded_zip_file is not None:
         with zipfile.ZipFile(uploaded_zip_file, "r") as zip_ref:
             zip_ref.extractall(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
+        # Find DICOM files in the directory, including subdirectories
+        dicom_files = find_dicom_files(temp_dir)
 
         if len(dicom_files) == 0:
             st.error("No DICOM files found in the ZIP archive.")