From 6b40637da31bbcc19af09334a9b99e8ce5d85dbf Mon Sep 17 00:00:00 2001
From: Nguyen Ngo <da.nguyen8744@gmail.com>
Date: Fri, 15 Aug 2025 16:17:58 -0400
Subject: [PATCH] Update export data functions

---
 static/js/export_configuration.js   | 26 ++++++++++++++++++++++++++
 templates/export_configuration.html | 12 +++++++++---
 2 files changed, 35 insertions(+), 3 deletions(-)

diff --git a/static/js/export_configuration.js b/static/js/export_configuration.js
index df38f6b..8c76040 100644
--- a/static/js/export_configuration.js
+++ b/static/js/export_configuration.js
@@ -535,6 +535,13 @@ function loadSavedPreferences() {
         const savedPrefs = localStorage.getItem('exportPreferences');
         if (!savedPrefs) {
             console.log('No saved preferences found');
+            // Check if there are already selected columns on page load and update preview
+            const alreadySelected = document.querySelectorAll('input[name="selected_columns"]:checked');
+            if (alreadySelected.length > 0) {
+                console.log('Found pre-selected columns, updating preview');
+                updateSelectedColumnsList();
+                updatePreview();
+            }
             return;
         }
         
@@ -547,6 +554,13 @@ function loadSavedPreferences() {
         if (savedDate < thirtyDaysAgo) {
             localStorage.removeItem('exportPreferences');
             console.log('Saved preferences are too old, removed');
+            // Check if there are already selected columns on page load and update preview
+            const alreadySelected = document.querySelectorAll('input[name="selected_columns"]:checked');
+            if (alreadySelected.length > 0) {
+                console.log('Found pre-selected columns, updating preview');
+                updateSelectedColumnsList();
+                updatePreview();
+            }
             return;
         }
         
@@ -585,6 +599,10 @@ function loadSavedPreferences() {
         
         console.log('Preferences loaded:', prefs);
         
+        // Update preview after loading preferences
+        updateSelectedColumnsList();
+        updatePreview();
+        
     } catch (e) {
         console.warn('Could not load saved preferences:', e);
         try {
@@ -592,6 +610,14 @@ function loadSavedPreferences() {
         } catch (removeError) {
             console.warn('Could not remove invalid preferences:', removeError);
         }
+        
+        // Check if there are already selected columns on page load and update preview
+        const alreadySelected = document.querySelectorAll('input[name="selected_columns"]:checked');
+        if (alreadySelected.length > 0) {
+            console.log('Found pre-selected columns after preference error, updating preview');
+            updateSelectedColumnsList();
+            updatePreview();
+        }
     }
 }
 
diff --git a/templates/export_configuration.html b/templates/export_configuration.html
index d3acd5b..1bb007f 100644
--- a/templates/export_configuration.html
+++ b/templates/export_configuration.html
@@ -212,11 +212,17 @@ document.addEventListener('DOMContentLoaded', function() {
     // Load saved preferences if available
     loadSavedPreferences();
     
-    // Update preview on page load
-    updatePreview();
-    
     // Initialize drag & drop
     initializeDragDrop();
+
+    // Ensure preview is updated after all initialization is complete
+    setTimeout(() => {
+        const selectedColumns = document.querySelectorAll('input[name="selected_columns"]:checked');
+        if (selectedColumns.length > 0) {
+            console.log('Final check: ensuring preview is shown for selected columns');
+            updatePreview();
+        }
+    }, 100);
 });
 
 function goBackToReport() {