Update app.py

app.py

@@ -71,7 +71,21 @@ if uploaded_zip_file is not None:
             zip_ref.extractall(temp_dir)
 
         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
+        dicom_files.sort()  
+
+        # Anonymize DICOM files
+        for file_path in dicom_files:
+            ds = pydicom.dcmread(file_path)
+
+            # Anonymize personal information
+            if 'PatientName' in ds:
+                ds.PatientName = 'Anonymized'
+            if 'PatientID' in ds:
+                ds.PatientID = '00000000'
+
+            print("Anonymized")
+            
+            ds.save_as(file_path)
             
     except Exception as e:
         st.error(f"Error: {str(e)}")
@@ -81,26 +95,20 @@ if st.button("Analyze"):
     st.success("Analysis started (expected time: 5 mins)")
     run_inference()
 
-    # Assuming the 'outputs' directory is located at "/home/user/app/C2C/comp2comp/outputs"
     outputs_directory = "/home/user/app/C2C/outputs"
 
-    # Get a list of subdirectories inside the 'outputs' directory
     subdirectories = [subdir for subdir in os.listdir(outputs_directory)
         if os.path.isdir(os.path.join(outputs_directory, subdir))]
 
-    # Use the first (or only) subdirectory, or None if there are no subdirectories
     first_subdirectory = subdirectories[0] if subdirectories else None
 
     if first_subdirectory:
-        # Construct the paths to the video and image files inside the first subdirectory
         subdirectory_path = os.path.join(outputs_directory, first_subdirectory)
         temp_dicom_dir_path = os.path.join(subdirectory_path, "temp_dicom_dir")
 
-        # Get a list of subdirectories inside the 'temp_dicom_dir' directory
         dicom_subdirectories = [subdir for subdir in os.listdir(temp_dicom_dir_path)
             if os.path.isdir(os.path.join(temp_dicom_dir_path, subdir))]
 
-        # Use the first (or only) subdirectory inside 'temp_dicom_dir', or None if there are none
         first_dicom_subdirectory = dicom_subdirectories[0] if dicom_subdirectories else None
 
         if first_dicom_subdirectory:
@@ -112,12 +120,12 @@ if st.button("Analyze"):
                 st.title("Largest Slice")
                 st.image(largest_slice, use_column_width=True)
 
-            # Display the video generated by the inference script
+            # Display the video
             if os.path.exists(video_path):
                 st.title("Video")
                 st.video(video_path, format="video/mp4")
 
-            # Display the image generated by the inference script
+            # Display the image
             if os.path.exists(image_path):
                 st.title("Diameter Graph")
                 st.image(image_path, use_column_width=True)