better js

javascript/viewer.js

@@ -1,5 +1,23 @@
 window.main_viewer_height = 512;
 
+function refresh_grid() {
+    let gridContainer = document.querySelector('#final_gallery .grid-container');
+    let final_gallery = document.getElementById('final_gallery');
+
+    if (gridContainer) if (final_gallery) {
+        let rect = final_gallery.getBoundingClientRect();
+        let cols = Math.ceil((rect.width - 16.0) / rect.height);
+        if(cols < 2) cols = 2;
+        gridContainer.style.setProperty('--grid-cols', cols);
+    }
+}
+
+function refresh_grid_delayed() {
+    setTimeout(refresh_grid, 100);
+    setTimeout(refresh_grid, 500);
+    setTimeout(refresh_grid, 1000);
+}
+
 function resized() {
     let windowHeight = window.innerHeight - 260;
     let elements = document.getElementsByClassName('main_view');
@@ -11,6 +29,8 @@ function resized() {
     }
 
     window.main_viewer_height = windowHeight;
+
+    refresh_grid();
 }
 
 function viewer_to_top(delay = 100) {