Update app.py

app.py

@@ -36,7 +36,6 @@ def run_inference():
     command = "./inference.sh"
     subprocess.run(command, shell=True, check=True)
 
-# Function to zip and download files
 def zip_and_download(directory, zip_filename):
     zipf = zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED)
     for root, _, files in os.walk(directory):
@@ -48,7 +47,6 @@ def zip_and_download(directory, zip_filename):
         bytes_data = f.read()
     st.download_button(label="Download Outputs", data=bytes_data, file_name=zip_filename, mime="application/zip")
 
-# Function to search for files in all subdirectories
 def search_file(start_path, target_file):
     for root, _, files in os.walk(start_path):
         if target_file in files:
@@ -92,10 +90,6 @@ def main():
                         image_path = search_file(temp_dicom_dir_path, "diameter_graph.png")
                         largest_slice = search_file(temp_dicom_dir_path, "out.png")
 
-                        st.write(f"Video Path: {video_path}")
-                        st.write(f"Image Path: {image_path}")
-                        st.write(f"Largest Slice Path: {largest_slice}")
-
                         if video_path and image_path and largest_slice:
                             zip_filename = os.path.join(temp_dir, "outputs.zip")
                             zip_and_download(temp_dicom_dir_path, zip_filename)
@@ -109,11 +103,11 @@ def main():
                             st.title("Diameter Graph")
                             st.image(image_path, use_column_width=True)
                         else:
-                            st.error("Output files not found.")
+                            st.error("Output files not found")
                     else:
-                        st.warning("No subdirectories found inside 'outputs'.")
+                        st.warning("Ouput files not found")
                 else:
-                    st.error("The 'outputs' directory does not exist.")
+                    st.error("Output files not found")
         except Exception as e:
             st.error(f"Error: {str(e)}")