set categorical variable selection in world_map.html based on stored …

…value

seshat/apps/core/templates/core/world_map.html

@@ -199,13 +199,17 @@ <h3>Base Map</h3>
     <script src="{% static 'core/js/map_functions.js' %}"></script>
     <script>
 
+        var categorical_variables = {{ categorical_variables | safe }};
+        updateCategoricalVariableSelection('language');  // Set the default categorical variable selection
+
         // If there's a variable stored in the local storage, set the variable selection to that
         if (localStorage.getItem('variable')) {
             document.getElementById('chooseVariable').value = localStorage.getItem('variable');
+            // If it's a categorical variable, set the categorical variable selection to that
+            if (localStorage.getItem('variable') in categorical_variables) {
+                updateCategoricalVariableSelection(localStorage.getItem('variable')); 
+            };
         }
-
-        var categorical_variables = {{ categorical_variables | safe }};
-        updateCategoricalVariableSelection('language');  // Set the default categorical variable selection
 
         var displayedShapes = [];
         var polityBorderWeight = 0;