Update app.py

app.py

@@ -1,102 +1,104 @@
 import streamlit as st
 import pydicom
+import matplotlib.pyplot as plt
 import zipfile
 import os
 import subprocess
+from datetime import datetime
+import shutil
+import moviepy.video.io.ImageSequenceClip
+from io import BytesIO
+from tkinter import Tcl
+from PIL import Image
 
 subprocess_executed = False
 
-logo_image_path = '1.png'
+logo_image_path = '1.png'  # Replace with your logo image path or URL
+
+# Display the logo at the top of the page
 st.image(logo_image_path, width=150)
 st.title("Automated Abdominal Aortic Aneurysm Detection")
 
+# Upload a ZIP file containing DICOM slices
 st.write("Upload a ZIP file containing DICOM slices")
 uploaded_zip_file = st.file_uploader("Upload a .zip file", type=["zip"])
 
 @st.cache_resource
 def install_dependencies():
-    command = "chmod +x install.sh"
-    try:
-        subprocess.run(command, shell=True, check=True)
-    except subprocess.CalledProcessError as e:
-        pass
-
-    command = "./install.sh"
-    try:
-        subprocess.run(command, shell=True, check=True)
-    except subprocess.CalledProcessError as e:
-        pass
+        command = "chmod +x install.sh"
+        try:
+            subprocess.run(command, shell=True, check=True)
+            print("Script 'install.sh' has been made executable.")
+        except subprocess.CalledProcessError as e:
+            print(f"Error while making the script executable: {e}")
+
+        command = "./install.sh"
+        try:
+            subprocess.run(command, shell=True, check=True)
+            print("Script 'install.sh' has started running.")
+            install_script_executed = True
+        except subprocess.CalledProcessError as e:
+            print(f"Error while running the script: {e}")
 
 @st.cache_resource
 def run_inference():
-    command = "chmod +x inference.sh"
-    try:
-        subprocess.run(command, shell=True, check=True)
-    except subprocess.CalledProcessError as e:
-        pass
-
-    command = "./inference.sh"
-    try:
-        subprocess.run(command, shell=True, check=True)
-    except subprocess.CalledProcessError as e:
-        pass
-
-def create_zip_file():
-    outputs_directory = "/home/user/app/C2C/outputs"
-    subdirectories = [subdir for subdir in os.listdir(outputs_directory)
-                      if os.path.isdir(os.path.join(outputs_directory, subdir))]
-    first_subdirectory = subdirectories[0] if subdirectories else None
-    if first_subdirectory:
-        subdirectory_path = os.path.join(outputs_directory, first_subdirectory)
-        temp_dicom_dir_path = os.path.join(subdirectory_path, "temp_dicom_dir")
-
-        largest_slice = os.path.join(temp_dicom_dir_path, "largest_slice.png")
-        diameter_graph = os.path.join(temp_dicom_dir_path, "diameter_graph.png")
-        video_path = os.path.join(temp_dicom_dir_path, "video.mp4")
-
-        files_to_zip = [largest_slice, diameter_graph, video_path]
-
-        zip_file_path = os.path.join(outputs_directory, "outputs.zip")
-        with zipfile.ZipFile(zip_file_path, 'w') as zipf:
-            for file in files_to_zip:
-                zipf.write(file, os.path.basename(file))
-        return zip_file_path
-    else:
-        return None
+        command = "chmod +x inference.sh"
+        try:
+            subprocess.run(command, shell=True, check=True)
+            print("Script 'inference.sh' has been made executable.")
+        except subprocess.CalledProcessError as e:
+            print(f"Error while making the script executable: {e}")
+
+        command = "./inference.sh"
+        try:
+            subprocess.run(command, shell=True, check=True)
+            print("Script 'inference.sh' has started running.")
+        except subprocess.CalledProcessError as e:
+            print(f"Error while running the script: {e}")
 
 if uploaded_zip_file is not None:
-    try:
+    try:    
         install_dependencies()
-
+        
         temp_dir = "/home/user/app/C2C/temp_dicom_dir"
+        
         os.makedirs(temp_dir, exist_ok=True)
+        full_path = os.path.abspath(temp_dir)
+        st.success("Zip file uploaded successfully")
 
+        
         with zipfile.ZipFile(uploaded_zip_file, "r") as zip_ref:
             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()
+        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'
-            ds.save_as(file_path)
 
+            print("Anonymized")
+            
+            ds.save_as(file_path)
+            
     except Exception as e:
         st.error(f"Error: {str(e)}")
 
 if st.button("Analyze"):
+
     st.success("Analysis started (expected time: 5 mins)")
     run_inference()
 
+
     outputs_directory = "/home/user/app/C2C/outputs"
-    zip_file_path = create_zip_file()
-    if zip_file_path:
-        st.success("Analysis completed. Click the button below to download the outputs.")
-        st.download_button(label="Download Outputs", data=zip_file_path, file_name="outputs.zip", mime="application/zip")
+
+    
 
     subdirectories = [subdir for subdir in os.listdir(outputs_directory)
         if os.path.isdir(os.path.join(outputs_directory, subdir))]
@@ -121,10 +123,12 @@ if st.button("Analyze"):
                 st.title("Largest Slice")
                 st.image(largest_slice, use_column_width=True)
 
+            # Display the video
             if os.path.exists(video_path):
                 st.title("Video")
                 st.video(video_path, format="video/mp4")
 
+            # Display the image
             if os.path.exists(image_path):
                 st.title("Diameter Graph")
                 st.image(image_path, use_column_width=True)