Update app.py

app.py

@@ -9,8 +9,11 @@ def visualize_image(file_path):
     # Get the number of bands in the image
     num_bands = dataset.count
 
+    # Calculate the number of rows required for subplots
+    num_rows = (num_bands + 1) // 2
+
     # Create subplots to visualize each band
-    fig, axes = plt.subplots(2, 5, figsize=(30, 30))
+    fig, axes = plt.subplots(num_rows, 2, figsize=(10, 6 * num_rows))
 
     # Loop through each band and plot it
     for band_index in range(1, num_bands + 1):
@@ -18,13 +21,17 @@ def visualize_image(file_path):
         band = dataset.read(band_index)
 
         # Determine the subplot position
-        row_index = (band_index - 1) // 5
-        col_index = (band_index - 1) % 5
+        row_index = (band_index - 1) // 2
+        col_index = (band_index - 1) % 2
 
         # Plot the band on the corresponding subplot
-        axes[row_index, col_index].imshow(band, cmap='jet')
+        axes[row_index, col_index].imshow(band, cmap='gray')
         axes[row_index, col_index].set_title(f"Band {band_index}")
 
+    # Remove any empty subplots if there are odd number of bands
+    if num_bands % 2 != 0:
+        fig.delaxes(axes[num_rows - 1, 1])
+
     # Adjust the layout and display the visualization
     plt.tight_layout()