Update app.py

app.py

@@ -105,25 +105,28 @@ if uploaded_zip_file is not None:
                 # 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)
+                
                 # 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.image(image_path, caption="Diameter Graph", use_column_width=True)
+                    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:
-        # Clean up by removing the temporary directory
-        for file in dicom_files:
-            os.remove(file)
-        os.rmdir(temp_dir)
 
 st.write("Upload a ZIP file containing DICOM slices to view the images.")