feat: updated useSectionCollapse methods (#4807)

## 📝 Summary

<!--
Provide a concise summary of what this pull request is addressing.

If this PR fixes any issues, list them here by number (e.g., Fixes
#123).
-->
Updated useSectionCollapse to use collapseAllCells and expandAllCells.
Partially resolves #4163 by fixing useSectionCollapse bugs.

## 🔍 Description of Changes

<!--
Detail the specific changes made in this pull request. Explain the
problem addressed and how it was resolved. If applicable, provide before
and after comparisons, screenshots, or any relevant details to help
reviewers understand the changes easily.
-->
- Updates #4550 implementation by using expandAllCells and
collapseAllCells rather than running expandCell and collapseCell in a
loop
- Fixes consecutive react renders issue
- Fixes Node not in tree errors

## 📋 Checklist

- [x] I have read the [contributor
guidelines](https://github.com/marimo-team/marimo/blob/main/CONTRIBUTING.md).
- [x] For large changes, or changes that affect the public API: this
change was discussed or approved through an issue, on
[Discord](https://marimo.io/discord?ref=pr), or the community
[discussions](https://github.com/marimo-team/marimo/discussions) (Please
provide a link if applicable).
- [x] I have added tests for the changes made.
- [x] I have run the code and verified that it works as expected.

## 📜 Reviewers

<!--
Tag potential reviewers from the community or maintainers who might be
interested in reviewing this pull request.

Your PR will be reviewed more quickly if you can figure out the right
person to tag with @ -->

@mscolnick

Author: bjoaquinc

frontend/src/components/editor/actions/useNotebookActions.tsx

@@ -83,7 +83,6 @@ import { useRunAllCells } from "../cell/useRunCells";
 import { settingDialogAtom } from "@/components/app-config/state";
 import { AddDatabaseDialogContent } from "../database/add-database-form";
 import { useHideAllMarkdownCode } from "./useHideAllMarkdownCode";
-import { useSectionCollapse } from "./useSectionCollapse";
 import { Constants } from "@/core/constants";
 
 const NOOP_HANDLER = (event?: Event) => {
@@ -99,13 +98,14 @@ export function useNotebookActions() {
   const [viewState] = useAtom(viewStateAtom);
   const kioskMode = useAtomValue(kioskModeAtom);
   const hideAllMarkdownCode = useHideAllMarkdownCode();
-  const { collapseAllSections, expandAllSections } = useSectionCollapse();
 
   const {
     updateCellConfig,
     undoDeleteCell,
     clearAllCellOutputs,
     upsertSetupCell,
+    collapseAllCells,
+    expandAllCells,
   } = useCellActions();
   const restartKernel = useRestartKernel();
   const runAllCells = useRunAllCells();
@@ -381,12 +381,12 @@ export function useNotebookActions() {
     {
       icon: <ChevronRightCircleIcon size={14} strokeWidth={1.5} />,
       label: "Collapse all sections",
-      handle: collapseAllSections,
+      handle: collapseAllCells,
     },
     {
       icon: <ChevronDownCircleIcon size={14} strokeWidth={1.5} />,
       label: "Expand all sections",
-      handle: expandAllSections,
+      handle: expandAllCells,
     },
     {
       icon: <DiamondPlusIcon size={14} strokeWidth={1.5} />,