Update app.py

app.py

@@ -11,13 +11,14 @@ import shutil
 subprocess_executed = False
 
 # Streamlit app title
-st.title("DICOM Image Viewer")
+st.markdown("<h1 style='text-align: center;'>A<sup>4</sup></h1>", unsafe_allow_html=True)
+st.subheader("Automated AAA Diameter Measurements")
 
 # 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])
 
@@ -28,31 +29,33 @@ def display_dicom_image(selected_slice, dicom_files):
     plt.tight_layout()
     return plt
 
+def execute_command_with_progress(command, description):
+    st.write(description)
+    process = subprocess.Popen(
+        command,
+        shell=True,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.STDOUT,
+        universal_newlines=True,
+        bufsize=1,
+    )
+
+    for line in process.stdout:
+        st.text(line)  # Display the output
+        # Assuming that the script provides a numeric progress value
+        try:
+            progress = float(line.strip())  # Extract a numeric progress value
+            st.progress(progress)
+        except ValueError:
+            pass
+
+    process.wait()
+    st.write(f"{description} has finished.")
+
 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)
@@ -75,6 +78,34 @@ if uploaded_zip_file is not None:
         st.error(f"Error: {str(e)}")
 
 if st.button("Analyze"):
+
+    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)
+
+    try:
+        execute_command_with_progress(command, "Installing required C2C packages")
+    except subprocess.CalledProcessError as e:
+        st.error(f"Error while making the script executable: {e}")
+
+
+
+
     command = "chmod +x inference.sh"
     
     try:
@@ -131,4 +162,3 @@ if st.button("Analyze"):
     # finally:
     #     shutil.rmtree(temp_dir)
 
-st.write("Upload a ZIP file containing DICOM slices to view the images.")