Update pages/Data Collection.py

pages/Data Collection.py

@@ -314,32 +314,38 @@ elif st.session_state.current_page == "explore_images_video":
             </p>
         </div>
     """, unsafe_allow_html=True)
+    import streamlit as st
+
     # Visualization for Grid-Like Structure
     st.markdown("<h4 style='color: #BB3385;'>Image as a Grid Structure</h4>", unsafe_allow_html=True)
-    height, width = 1,1
+    height, width = 6, 6 
     grid = np.random.rand(height, width)
-    fig, ax = plt.subplots()
+    
+    # Adjust the figure size to make the plot smaller
+    fig, ax = plt.subplots(figsize=(4, 4))  # 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=3)
-
+            ax.text(j, i, f'{grid[i, j]:.2f}', ha='center', va='center', color='white', fontsize=6)  # Adjusted fontsize
+    
     # Set axis labels and title
-    ax.set_title("Image as a Grid-Like Structure", fontsize=5)
-    ax.set_xlabel("Width (Pixels)", fontsize=4)
-    ax.set_ylabel("Height (Pixels)", fontsize=4)
+    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_xticks(range(width))
     ax.set_yticks(range(height))
-    ax.set_xticklabels([f'Col {j+1}' for j in range(width)])
-    ax.set_yticklabels([f'Row {i+1}' for i in 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')
-
+    
+    # Display the plot
     st.pyplot(fig)
 
 
+
     # Go Back Button
     if st.button("Go Back"):
         navigate_to("main")