Remember user selection for collapsed/uncollapsed content; remove the bullet from the list

Author: svekars

_static/css/custom2.css

@@ -31,3 +31,7 @@
 .pytorch-left-menu li.toctree-l2.current a:link.reference.internal {
    color: #ee4c2c;
 }
+
+.pytorch-left-menu li.toctree-l1.current > a:before {
+   content: "";
+}

_static/js/custom.js

@@ -12,28 +12,41 @@ document.addEventListener("DOMContentLoaded", function() {
       var expandSign = document.createElement('span');
       expandSign.style.cursor = 'pointer'; // Make it look clickable
 
-      // Check if this is the "Learn the Basics" section
-      if (link.textContent.trim() === 'Learn the Basics') {
-        nestedList.style.display = 'block'; // Expand the "Learn the Basics" section
+      // Use the link text as a unique key for localStorage
+      var sectionKey = 'section_' + link.textContent.trim().replace(/\s+/g, '_');
+
+      // Retrieve the saved state from localStorage
+      var isExpanded = localStorage.getItem(sectionKey);
+
+      // If no state is saved, default to expanded for "Learn the Basics" and collapsed for others
+      if (isExpanded === null) {
+        isExpanded = (link.textContent.trim() === 'Learn the Basics') ? 'true' : 'false';
+        localStorage.setItem(sectionKey, isExpanded);
+      }
+
+      if (isExpanded === 'true') {
+        nestedList.style.display = 'block'; // Expand the section
         expandSign.textContent = '[-] '; // Show "[-]" since it's expanded
       } else {
-        nestedList.style.display = 'none'; // Collapse other sections
+        nestedList.style.display = 'none'; // Collapse the section
         expandSign.textContent = '[+] '; // Show "[+]" since it's collapsed
       }
 
-      // Insert the sign before the link
-      link.parentNode.insertBefore(expandSign, link);
-
       // Add a click event to toggle the nested list
       expandSign.addEventListener('click', function() {
         if (nestedList.style.display === 'none') {
           nestedList.style.display = 'block';
           expandSign.textContent = '[-] '; // Change to "[-]" when expanded
+          localStorage.setItem(sectionKey, 'true'); // Save state
         } else {
           nestedList.style.display = 'none';
           expandSign.textContent = '[+] '; // Change back to "[+]" when collapsed
+          localStorage.setItem(sectionKey, 'false'); // Save state
         }
       });
+
+      // Insert the sign before the link
+      link.parentNode.insertBefore(expandSign, link);
     }
   });
 });