Update pages/Data Collection.py

pages/Data Collection.py

@@ -747,7 +747,7 @@ elif st.session_state.current_page == "image_operations":
     
     # Heading for cv2.merge() function
     st.markdown("""
-        <h3 style="color: #9400d3;">Merging Color Channels</h3>
+        <h3 style="color: #e25822;">Merging Color Channels</h3>
     """, unsafe_allow_html=True)
     
     # About cv2.merge() function
@@ -802,7 +802,7 @@ elif st.session_state.current_page == "image_operations":
 
     # Heading for Splitting Channels
     st.markdown("""
-        <h3 style="color: #9400d3;">Splitting Channels</h3>
+        <h3 style="color: #e25822;">Splitting Channels</h3>
     """, unsafe_allow_html=True)
     
     # About cv2.split() function
@@ -845,6 +845,47 @@ elif st.session_state.current_page == "image_operations":
     cv2.waitKey(0)
     cv2.destroyAllWindows()""", language="python")
 
+    # Allow user to upload an image
+    uploaded_file = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
+    
+    if uploaded_file is not None:
+        # Convert the uploaded image to an OpenCV-compatible format
+        image = np.array(bytearray(uploaded_file.read()), dtype=np.uint8)
+        img = cv2.imdecode(image, 1)  # Decode into an image
+    
+        # Split the image into Blue, Green, and Red channels
+        b, g, r = cv2.split(img)
+    
+        # Create a zeros array to hold the empty channels
+        zeros = np.zeros(img.shape[:-1], dtype=np.uint8)
+    
+        # Merge the Blue channel with zeros for Green and Red
+        blue_channel = cv2.merge([b, zeros, zeros])
+        green_channel = cv2.merge([zeros, g, zeros])
+        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)
+    
+        # 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)
+    
+        # Optionally, provide a download link for the processed image
+        st.download_button(
+            label="Download Merged Image",
+            data=cv2.imencode('.jpg', original_img)[1].tobytes(),
+            file_name="merged_image.jpg",
+            mime="image/jpeg"
+        )
+    else:
+        st.write("Please upload an image to proceed.")
+        
+
 elif st.session_state.current_page == "explore_audio":
     st.markdown("""
         <h3 style="color: #e25822;">Exploring Audio</h3>