Update app.py

app.py

@@ -3,13 +3,19 @@ import pydicom
 import matplotlib.pyplot as plt
 import os
 
-def visualize_dicom_sequence(file_paths):
-    for file_path in file_paths:
-        ds = pydicom.dcmread(file_path)
-        image_sequence = ds.pixel_array
-
+def visualize_dicom_sequence(file_path):
+    ds = pydicom.dcmread(file_path)
+    image_sequence = ds.pixel_array
+
+    if image_sequence.ndim == 2:
+        # Only one image in the sequence
+        fig, ax = plt.subplots()
+        ax.imshow(image_sequence, cmap=plt.cm.gray)
+        ax.axis('off')
+        st.pyplot(fig)
+    else:
+        # Multiple images in the sequence
         for i, image in enumerate(image_sequence):
-            # Plot DICOM image using Matplotlib
             fig, ax = plt.subplots()
             ax.imshow(image, cmap=plt.cm.gray)
             ax.axis('off')
@@ -18,27 +24,22 @@ def visualize_dicom_sequence(file_paths):
 def main():
     st.title("Visualizador DICOM")
 
-    # Upload DICOM files
-    uploaded_files = st.file_uploader("Selecione um ou mais arquivos DICOM", type=".dcm", multiple=True)
-
-    if uploaded_files:
-        file_paths = []
+    # Upload DICOM file
+    uploaded_file = st.file_uploader("Selecione um arquivo DICOM", type=".dcm")
 
-        # Save the uploaded files to temporary locations
-        for uploaded_file in uploaded_files:
-            file_bytes = uploaded_file.getvalue()
-            file_path = f"temp_{uploaded_file.name}"
-            file_paths.append(file_path)
+    if uploaded_file is not None:
+        # Convert uploaded file to bytes
+        file_bytes = uploaded_file.getvalue()
 
-            with open(file_path, "wb") as f:
-                f.write(file_bytes)
+        # Save the uploaded file to a temporary location
+        with open("temp.dcm", "wb") as f:
+            f.write(file_bytes)
 
-        # Visualize the DICOM image sequences
-        visualize_dicom_sequence(file_paths)
+        # Visualize the DICOM image sequence
+        visualize_dicom_sequence("temp.dcm")
 
-        # Remove the temporary files
-        for file_path in file_paths:
-            os.remove(file_path)
+        # Remove the temporary file
+        os.remove("temp.dcm")
 
 if __name__ == "__main__":
     main()