Update pages/Data Collection.py

pages/Data Collection.py

@@ -415,21 +415,21 @@ elif st.session_state.current_page == "explore_images_video":
     Grayscale eliminates color information, focusing entirely on the intensity of light in an image. Each pixel has only one intensity value, making it a simpler and more compact representation compared to color images.
     """)
 
-    # Section: Grayscale Representation
-    st.subheader("Grayscale Representation")
-    
     # Create grayscale gradient with labeled intensity values
     gradient = np.linspace(0, 255, 256)  # Generate gradient values
     gradient = np.tile(gradient, (10, 1))  # Repeat the gradient to make it visually clear
     
     # Plot the gradient
-    fig, ax = plt.subplots(figsize=(8, 2))
+    fig, ax = plt.subplots(figsize=(8, 2), facecolor='none')  # Set transparent facecolor
     ax.imshow(gradient, cmap='gray', aspect='auto')
     ax.set_xticks(np.linspace(0, 255, 11))  # Set ticks for every 25.5 (0, 25, ..., 255)
     ax.set_xticklabels([str(int(x)) for x in np.linspace(0, 255, 11)], fontsize=10, color='red')  # Add labels in red
     ax.set_yticks([])  # Remove y-axis ticks
     ax.set_title("Grayscale Representation", fontsize=12)
     
+    # Save the figure with a transparent background
+    plt.savefig('grayscale_representation.png', transparent=True)
+    
     # Render the plot in Streamlit
     st.pyplot(fig)