Update pages/Data Collection.py

pages/Data Collection.py

@@ -462,33 +462,46 @@ elif st.session_state.current_page == "explore_images_video":
     RGB is the most common color space used in digital imaging and is ideal for screens and devices that emit light.
     """)
 
+    
+    # Subheader for the section
     st.subheader("Visualization of RGB Channels")
     
-    # Load an example image (you can use any image)
-    uploaded_file = st.file_uploader("Upload an Image", type=["jpg", "png"])
-    if uploaded_file:
+    # File uploader for the user to upload an image
+    uploaded_file = st.file_uploader("Upload an Image (JPG or PNG):", type=["jpg", "png"])
+    
+    if uploaded_file is not None:
+        # Open the uploaded image
         img = Image.open(uploaded_file)
         img_array = np.array(img)
     
-        # Split into channels
-        red_channel = img_array[:, :, 0]
-        green_channel = img_array[:, :, 1]
-        blue_channel = img_array[:, :, 2]
+        # Ensure the image has three channels (for RGB visualization)
+        if img_array.shape[-1] == 3:
+            # Split the image into its RGB channels
+            red_channel = img_array[:, :, 0]
+            green_channel = img_array[:, :, 1]
+            blue_channel = img_array[:, :, 2]
+    
+            # Plot the original image and the three channels
+            fig, ax = plt.subplots(1, 4, figsize=(15, 5))
+            ax[0].imshow(img)
+            ax[0].set_title("Original Image")
+            ax[1].imshow(red_channel, cmap="Reds")
+            ax[1].set_title("Red Channel")
+            ax[2].imshow(green_channel, cmap="Greens")
+            ax[2].set_title("Green Channel")
+            ax[3].imshow(blue_channel, cmap="Blues")
+            ax[3].set_title("Blue Channel")
+            
+            # Remove axis labels and ticks
+            for a in ax:
+                a.axis("off")
+            
+            # Render the plot in Streamlit
+            st.pyplot(fig)
+        else:
+            st.error("The uploaded image does not have 3 color channels (RGB). Please upload an RGB image.")
+    
     
-        # Plot the channels
-        fig, ax = plt.subplots(1, 4, figsize=(12, 4))
-        ax[0].imshow(img)
-        ax[0].set_title("Original Image")
-        ax[1].imshow(red_channel, cmap="Reds")
-        ax[1].set_title("Red Channel")
-        ax[2].imshow(green_channel, cmap="Greens")
-        ax[2].set_title("Green Channel")
-        ax[3].imshow(blue_channel, cmap="Blues")
-        ax[3].set_title("Blue Channel")
-        for a in ax:
-            a.axis("off")
-        st.pyplot(fig)
-
 
     
     # Go Back Button