Update pages/Data Collection.py

pages/Data Collection.py

@@ -846,6 +846,8 @@ elif st.session_state.current_page == "image_operations":
     cv2.waitKey(0)
     cv2.destroyAllWindows()""", language="python")
 
+    st.write("Once you upload an image, it will be split into its color channels (Blue, Green, and Red), with each channel displayed separately. You can then download the processed image.")
+
     # Allow user to upload an image
     uploaded_file = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
     
@@ -866,15 +868,15 @@ elif st.session_state.current_page == "image_operations":
         red_channel = cv2.merge([zeros, zeros, r])
     
         # Display the images with captions
-        st.image(blue_channel, caption="Blue Channel", channels="BGR", use_column_width=True)
-        st.image(green_channel, caption="Green Channel", channels="BGR", use_column_width=True)
-        st.image(red_channel, caption="Red Channel", channels="BGR", use_column_width=True)
+        st.image(blue_channel, caption="Blue Channel", channels="BGR", use_container_width=True)
+        st.image(green_channel, caption="Green Channel", channels="BGR", use_container_width=True)
+        st.image(red_channel, caption="Red Channel", channels="BGR", use_container_width=True)
     
         # Merge the channels back together for the original image
         original_img = cv2.merge([b, g, r])
     
         # Display the original image
-        st.image(original_img, caption="Original Image", channels="BGR", use_column_width=True)
+        st.image(original_img, caption="Original Image", channels="BGR", use_container_width=True)
     
         # Optionally, provide a download link for the processed image
         st.download_button(