Update app.py

app.py

@@ -11,14 +11,13 @@ import shutil
 subprocess_executed = False
 
 # Streamlit app title
-st.markdown("<h1 style='text-align: center;'>A<sup>4</sup></h1>", unsafe_allow_html=True)
-st.subheader("Automated AAA Diameter Measurements")
+st.title("DICOM Image Viewer")
 
 # Upload a ZIP file containing DICOM slices
 uploaded_zip_file = st.file_uploader("Upload a ZIP file containing DICOM slices", type=["zip"])
-st.write("Upload a ZIP file containing DICOM slices to view the images.")
 
 # Function to read and display the DICOM image
+@st.cache(suppress_st_warning=True, show_spinner=False)
 def display_dicom_image(selected_slice, dicom_files):
     dicom_data = pydicom.dcmread(dicom_files[selected_slice])
 
@@ -32,10 +31,28 @@ def display_dicom_image(selected_slice, dicom_files):
 if uploaded_zip_file is not None:
     try:
         # Create a temporary directory to unzip the files
+
+
+        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.")
+        except subprocess.CalledProcessError as e:
+            print(f"Error while running the script: {e}")
+            
         temp_dir = "/home/user/app/C2C/temp_dicom_dir"
         
         os.makedirs(temp_dir, exist_ok=True)
         full_path = os.path.abspath(temp_dir)
+
+        st.write(f"Temporary directory path: {full_path}")
         
         with zipfile.ZipFile(uploaded_zip_file, "r") as zip_ref:
             zip_ref.extractall(temp_dir)
@@ -54,68 +71,65 @@ if uploaded_zip_file is not None:
             plt = display_dicom_image(selected_slice, dicom_files)
             st.pyplot(plt)
 
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
-
-if st.button("Analyze"):
+        if not subprocess_executed:
 
-    subprocess_executed = True
-
-    if subprocess_executed:
-
-        command = "chmod +x install.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 making the script executable: {e}")
+            command = "chmod +x inference.sh"
     
-        command = "./install.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 making the script executable: {e}")
     
-        try:
-            subprocess.run(command, shell=True, check=True)
-            print("Script 'inference.sh' has started running.")
-        except subprocess.CalledProcessError as e:
-            print(f"Error while making the script executable: {e}")
-    
-        # Set the flag to indicate that subprocess commands have been executed
-        subprocess_executed = True
-    
-        # 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))]
+            command = "./inference.sh"
     
-        # 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
-            video_path = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/aaa.mp4")
-            image_path = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/diameter_graph.png")
-            largest_slice = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/out.png")
-        
-            if os.path.exists(largest_slice):
-                st.title("Largest Slice")
-                st.image(largest_slice, use_column_width=True)
+            try:
+                subprocess.run(command, shell=True, check=True)
+                print("Script 'inference.sh' has started running.")
+            except subprocess.CalledProcessError as e:
+                print(f"Error while making the script executable: {e}")
+
+            # Set the flag to indicate that subprocess commands have been executed
+            subprocess_executed = True
+
+            # 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
+                video_path = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/aaa.mp4")
+                image_path = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/diameter_graph.png")
+                largest_slice = os.path.join(outputs_directory, first_subdirectory, "temp_dicom_dir/images/summary/out.png")
             
-            # Display the video generated by the inference script
-            if os.path.exists(video_path):
-                st.title("Video")
-                st.video(video_path, format="video/mp4")
+                if os.path.exists(largest_slice):
+                    st.title("Largest Slice")
+                    st.image(largest_slice, use_column_width=True)
+                
+                # Display the video generated by the inference script
+                if os.path.exists(video_path):
+                    st.title("Video")
+                    st.video(video_path, format="video/mp4")
             
-            # Display the image generated by the inference script
-            if os.path.exists(image_path):
-                st.title("Diameter Graph")
-                st.image(image_path, use_column_width=True)
+                # Display the image generated by the inference script
+                if os.path.exists(image_path):
+                    st.title("Diameter Graph")
+                    st.image(image_path, use_column_width=True)
+            else:
+                st.warning("No subdirectories or folders found inside the 'outputs' directory.")
 
 
 
 
-    
+    except Exception as e:
+        st.error(f"Error: {str(e)}")
     # finally:
     #     shutil.rmtree(temp_dir)
 
+st.write("Upload a ZIP file containing DICOM slices to view the images.")