Update pages/Data Collection.py

pages/Data Collection.py

@@ -314,12 +314,11 @@ elif st.session_state.current_page == "explore_images_video":
 
 
     # Visualization for Grid-Like Structure
-    st.write("<h4 style='color: #BB3385;'>Image as a Grid Structure</h4>", unsafe_allow_html=True)
-    height, width = 6,6
+    height, width = 3,3
     grid = np.random.rand(height, width)
     
     # Adjust the figure size to make the plot very small
-    fig, ax = plt.subplots(figsize=(3, 3))  # Even smaller figure size
+    fig, ax = plt.subplots(figsize=(2,2))  # Even smaller figure size
     cax = ax.imshow(grid, cmap='viridis', interpolation='nearest')
 
     # Annotate the grid with pixel labels
@@ -328,7 +327,7 @@ elif st.session_state.current_page == "explore_images_video":
             ax.text(j, i, f'{grid[i, j]:.2f}', ha='center', va='center', color='white', fontsize=4)  # Smaller fontsize
     
     # Set axis labels and title
-    ax.set_title("Image as a Grid-Like Structure", fontsize=6)  # Smaller title size
+    ax.set_title("Image as a Grid-Like Structure", fontsize=5)  # Smaller title size
     ax.set_xlabel("Width (Pixels)", fontsize=4)  # Smaller label size
     ax.set_ylabel("Height (Pixels)", fontsize=4)  # Smaller label size
     ax.set_xticks(range(width))
@@ -338,7 +337,7 @@ elif st.session_state.current_page == "explore_images_video":
     ax.grid(color='white', linestyle='-', linewidth=0.5)  # Thinner grid lines
     
     # Color bar setup without fontsize argument
-    cbar = fig.colorbar(cax, ax=ax, orientation='vertical', label='Pixel Value')
+    cbar = fig.colorbar(cax, ax=ax, orientation='vertical')
     cbar.ax.tick_params(labelsize=4)  # Adjust the font size of the color bar
     
     # Display the plot