Update pages/Data Collection.py

pages/Data Collection.py

@@ -321,25 +321,25 @@ elif st.session_state.current_page == "explore_images_video":
     height, width = 6, 6 
     grid = np.random.rand(height, width)
     
-    # Adjust the figure size to make the plot smaller
-    fig, ax = plt.subplots(figsize=(4, 4))  # Smaller figure size
+    # Adjust the figure size to make the plot very small
+    fig, ax = plt.subplots(figsize=(3, 3))  # Even smaller figure size
     cax = ax.imshow(grid, cmap='viridis')
     
     # Annotate the grid with pixel labels
     for i in range(height):
         for j in range(width):
-            ax.text(j, i, f'{grid[i, j]:.2f}', ha='center', va='center', color='white', fontsize=6)  # Adjusted fontsize
+            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)  # Adjusted title size
-    ax.set_xlabel("Width (Pixels)", fontsize=6)  # Adjusted label size
-    ax.set_ylabel("Height (Pixels)", fontsize=6)  # Adjusted label size
+    ax.set_title("Image as a Grid-Like Structure", fontsize=6)  # 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))
     ax.set_yticks(range(height))
-    ax.set_xticklabels([f'Col {j+1}' for j in range(width)], fontsize=5)  # Adjusted label size
-    ax.set_yticklabels([f'Row {i+1}' for i in range(height)], fontsize=5)  # Adjusted label size
-    ax.grid(color='white', linestyle='-', linewidth=1)
-    fig.colorbar(cax, ax=ax, orientation='vertical', label='Pixel Value')
+    ax.set_xticklabels([f'Col {j+1}' for j in range(width)], fontsize=4)  # Smaller label size
+    ax.set_yticklabels([f'Row {i+1}' for i in range(height)], fontsize=4)  # Smaller label size
+    ax.grid(color='white', linestyle='-', linewidth=0.5)  # Thinner grid lines
+    fig.colorbar(cax, ax=ax, orientation='vertical', label='Pixel Value', fontsize=4)
     
     # Display the plot
     st.pyplot(fig)