mito-ai 0.1.54__py3-none-any.whl → 0.1.55__py3-none-any.whl

mito_ai-0.1.54.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.31462f8f6a76b1cefbeb.js → mito_ai-0.1.55.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.49c79c62671528877c61.js

@@ -898,13 +898,15 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */ });
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
-/* harmony import */ var _PythonCode__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
-/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
+/* harmony import */ var _PythonCode__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
 /* harmony import */ var _style_CodeBlock_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/CodeBlock.css */ "./style/CodeBlock.css");
 /* harmony import */ var _style_AgentComponentHeader_css__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../../style/AgentComponentHeader.css */ "./style/AgentComponentHeader.css");
-/* harmony import */ var _icons_CodeIcon__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../icons/CodeIcon */ "./lib/icons/CodeIcon.js");
-/* harmony import */ var _CodeBlockToolbar__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./CodeBlockToolbar */ "./lib/Extensions/AiChat/ChatMessage/CodeBlockToolbar.js");
-/* harmony import */ var _components_AgentComponents_AgentComponentHeader__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../components/AgentComponents/AgentComponentHeader */ "./lib/components/AgentComponents/AgentComponentHeader.js");
+/* harmony import */ var _icons_CodeIcon__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../icons/CodeIcon */ "./lib/icons/CodeIcon.js");
+/* harmony import */ var _CodeBlockToolbar__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./CodeBlockToolbar */ "./lib/Extensions/AiChat/ChatMessage/CodeBlockToolbar.js");
+/* harmony import */ var _components_AgentComponents_AgentComponentHeader__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../components/AgentComponents/AgentComponentHeader */ "./lib/components/AgentComponents/AgentComponentHeader.js");
+/* harmony import */ var _icons_GoToCellIcon__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../icons/GoToCellIcon */ "./lib/icons/GoToCellIcon.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -917,21 +919,37 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const AssistantCodeBlock = ({ code, codeSummary, isCodeComplete, renderMimeRegistry, previewAICode, acceptAICode, rejectAICode, isLastAiMessage, codeReviewStatus, agentModeEnabled, isErrorFixup, }) => {
+
+
+const AssistantCodeBlock = ({ code, codeSummary, isCodeComplete, renderMimeRegistry, previewAICode, acceptAICode, rejectAICode, isLastAiMessage, codeReviewStatus, agentModeEnabled, isErrorFixup, cellId, notebookPanel, }) => {
     const [isCodeExpanded, setIsCodeExpanded] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     const shouldShowToolbar = isLastAiMessage || isCodeComplete;
+    const handleGoToCell = (e) => {
+        e.stopPropagation();
+        if (cellId && notebookPanel) {
+            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.setActiveCellByIDInNotebookPanel)(notebookPanel, cellId);
+            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.scrollToCell)(notebookPanel, cellId, undefined, 'center');
+        }
+    };
+    const renderActionButtons = () => {
+        if (!cellId || !notebookPanel) {
+            return null;
+        }
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "agent-component-header-action-button", onClick: handleGoToCell, title: "Go to cell" },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_GoToCellIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null)));
+    };
     if (agentModeEnabled) {
         // Handle regular code blocks
-        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_3__.classNames)('code-block-container', 'agent-mode', {
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)('code-block-container', 'agent-mode', {
                 'agent-mode-collapsed': !isCodeExpanded,
             }) },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_AgentComponentHeader__WEBPACK_IMPORTED_MODULE_4__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_5__["default"], null), text: codeSummary !== null && codeSummary !== void 0 ? codeSummary : 'Generated code', onClick: () => setIsCodeExpanded(!isCodeExpanded), isExpanded: isCodeExpanded, displayBorder: !isErrorFixup, className: isErrorFixup ? 'error-fixup' : undefined }),
-            isCodeExpanded && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_6__["default"], { code: code, renderMimeRegistry: renderMimeRegistry }))));
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_AgentComponentHeader__WEBPACK_IMPORTED_MODULE_6__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_7__["default"], null), text: codeSummary !== null && codeSummary !== void 0 ? codeSummary : 'Generated code', onClick: () => setIsCodeExpanded(!isCodeExpanded), isExpanded: isCodeExpanded, displayBorder: !isErrorFixup, className: isErrorFixup ? 'error-fixup' : undefined, actionButtons: renderActionButtons() }),
+            isCodeExpanded && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_8__["default"], { code: code, renderMimeRegistry: renderMimeRegistry }))));
     }
     else {
         return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "code-block-container" },
-            shouldShowToolbar && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_CodeBlockToolbar__WEBPACK_IMPORTED_MODULE_7__["default"], { code: code, isLastAiMessage: isLastAiMessage, codeReviewStatus: codeReviewStatus, onPreview: previewAICode, onAccept: acceptAICode, onReject: rejectAICode })),
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_6__["default"], { code: code, renderMimeRegistry: renderMimeRegistry })));
+            shouldShowToolbar && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_CodeBlockToolbar__WEBPACK_IMPORTED_MODULE_9__["default"], { code: code, isLastAiMessage: isLastAiMessage, codeReviewStatus: codeReviewStatus, onPreview: previewAICode, onAccept: acceptAICode, onReject: rejectAICode })),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_8__["default"], { code: code, renderMimeRegistry: renderMimeRegistry })));
     }
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AssistantCodeBlock);
@@ -952,7 +970,8 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
 /* harmony import */ var _restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../restAPI/RestAPI */ "./lib/restAPI/RestAPI.js");
-/* harmony import */ var _ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ./ChatDropdownItems */ "./lib/Extensions/AiChat/ChatMessage/ChatDropdownItems.js");
+/* harmony import */ var _ChatDropdownItems__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ./ChatDropdownItems */ "./lib/Extensions/AiChat/ChatMessage/ChatDropdownItems.js");
+/* harmony import */ var _utils_cellReferences__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../utils/cellReferences */ "./lib/utils/cellReferences.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -960,6 +979,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 const priortizeByType = (options, maxPerType) => {
     /*
     Makes sure that some of each type are displayed at the top of the dropdown so
@@ -982,7 +1002,7 @@ const priortizeByType = (options, maxPerType) => {
     // Return prioritized options first, then extras at the bottom
     return [...prioritizedOptions, ...extraOptions];
 };
-const ChatDropdown = ({ options, onSelect, filterText, isDropdownFromButton = false, onFilterChange, onClose, }) => {
+const ChatDropdown = ({ options, onSelect, filterText, isDropdownFromButton = false, onFilterChange, onClose, notebookTracker, }) => {
     const [selectedIndex, setSelectedIndex] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(0);
     const [localFilterText, setLocalFilterText] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(filterText);
     const searchInputRef = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(null);
@@ -1008,16 +1028,28 @@ const ChatDropdown = ({ options, onSelect, filterText, isDropdownFromButton = fa
     }, [isDropdownFromButton]);
     // Use local filter text when search input is shown, otherwise use prop
     const effectiveFilterText = isDropdownFromButton ? localFilterText : filterText;
+    // Get cell references if notebook tracker is available
+    const cellReferences = (notebookTracker === null || notebookTracker === void 0 ? void 0 : notebookTracker.currentWidget)
+        ? (0,_utils_cellReferences__WEBPACK_IMPORTED_MODULE_2__.getAllCellReferences)(notebookTracker.currentWidget)
+        : [];
     // Create a list of all options with the format 
     // ['type': 'variable', "expandedVariable": variable]
     // ['type': 'rule', "rule": rule]
     // ['type': 'file', "file": file]
+    // ['type': 'cell', "cellNumber": number, "cellId": string]
     const allOptions = [
         // Rules first
         ...rules.map((rule) => ({
             type: 'rule',
             rule: rule
         })),
+        // Cells second (when user types @Cell or @cell)
+        ...cellReferences.map((cell) => ({
+            type: 'cell',
+            cellNumber: cell.cellNumber,
+            cellId: cell.cellId,
+            cellType: cell.cellType
+        })),
         // Files second
         ...options
             .filter(variable => variable.file_name) // Only files
@@ -1070,6 +1102,13 @@ const ChatDropdown = ({ options, onSelect, filterText, isDropdownFromButton = fa
             return option.variable.variable_name.toLowerCase().includes(effectiveFilterText.toLowerCase()) ||
                 option.variable.value.toLowerCase().includes(effectiveFilterText.toLowerCase());
         }
+        else if (option.type === 'cell') {
+            // Match "CellN" (no space)
+            const cellText = `cell${option.cellNumber}`;
+            const numberText = String(option.cellNumber);
+            return cellText.includes(effectiveFilterText.toLowerCase()) ||
+                numberText.includes(effectiveFilterText.toLowerCase());
+        }
         else {
             return option.rule.toLowerCase().includes(effectiveFilterText.toLowerCase());
         }
@@ -1159,20 +1198,24 @@ const ChatDropdown = ({ options, onSelect, filterText, isDropdownFromButton = fa
                         const uniqueKey = option.variable.parent_df
                             ? `${option.variable.parent_df}.${option.variable.variable_name}`
                             : option.variable.variable_name;
-                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__.VariableDropdownItem, { key: uniqueKey, variable: option.variable, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_3__.VariableDropdownItem, { key: uniqueKey, variable: option.variable, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
                     }
                     case 'file': {
                         const uniqueKey = option.file.variable_name;
-                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__.FileDropdownItem, { key: uniqueKey, file: option.file, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_3__.FileDropdownItem, { key: uniqueKey, file: option.file, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
                     }
                     case 'rule': {
                         const uniqueKey = option.rule;
-                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__.RuleDropdownItem, { key: uniqueKey, rule: option.rule, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_3__.RuleDropdownItem, { key: uniqueKey, rule: option.rule, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
                     }
                     case 'db': {
                         const uniqueKey = option.variable.variable_name;
                         // You can replace VariableDropdownItem with a custom DBDropdownItem if you want
-                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__.VariableDropdownItem, { key: uniqueKey, variable: option.variable, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_3__.VariableDropdownItem, { key: uniqueKey, variable: option.variable, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                    }
+                    case 'cell': {
+                        const uniqueKey = `cell-${option.cellNumber}`;
+                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_3__.CellDropdownItem, { key: uniqueKey, cellNumber: option.cellNumber, cellId: option.cellId, cellType: option.cellType, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
                     }
                     default:
                         return null;
@@ -1192,6 +1235,7 @@ const ChatDropdown = ({ options, onSelect, filterText, isDropdownFromButton = fa
 
 __webpack_require__.r(__webpack_exports__);
 /* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   CellDropdownItem: () => (/* binding */ CellDropdownItem),
 /* harmony export */   FileDropdownItem: () => (/* binding */ FileDropdownItem),
 /* harmony export */   RuleDropdownItem: () => (/* binding */ RuleDropdownItem),
 /* harmony export */   VariableDropdownItem: () => (/* binding */ VariableDropdownItem)
@@ -1249,6 +1293,24 @@ const RuleDropdownItem = ({ rule, index, selectedIndex, onSelect }) => {
                 }
             } }, rule)));
 };
+const CellDropdownItem = ({ cellNumber, cellId, cellType, index, selectedIndex, onSelect }) => {
+    const getShortType = (type) => {
+        if (type === 'code') {
+            return 'code';
+        }
+        else if (type === 'markdown') {
+            return 'md';
+        }
+        else {
+            return 'raw';
+        }
+    };
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("li", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_1__.classNames)("chat-dropdown-item", { selected: index === selectedIndex }), onClick: () => onSelect(cellNumber, cellId), "data-testid": `chat-dropdown-item-cell-${cellNumber}` },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "chat-dropdown-item-type", title: cellType, "data-testid": `chat-dropdown-item-type-cell-${cellNumber}` }, getShortType(cellType)),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "chat-dropdown-item-name", title: `Cell ${cellNumber}`, "data-testid": `chat-dropdown-item-name-cell-${cellNumber}` },
+            "Cell",
+            cellNumber)));
+};
 
 
 /***/ }),
@@ -1265,15 +1327,16 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */ });
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
-/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _ChatDropdown__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ./ChatDropdown */ "./lib/Extensions/AiChat/ChatMessage/ChatDropdown.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
+/* harmony import */ var _ChatDropdown__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ./ChatDropdown */ "./lib/Extensions/AiChat/ChatMessage/ChatDropdown.js");
 /* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _utils_cellReferences__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/cellReferences */ "./lib/utils/cellReferences.js");
 /* harmony import */ var _style_ChatInput_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/ChatInput.css */ "./style/ChatInput.css");
 /* harmony import */ var _style_ChatDropdown_css__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../../style/ChatDropdown.css */ "./style/ChatDropdown.css");
 /* harmony import */ var _hooks_useDebouncedFunction__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../hooks/useDebouncedFunction */ "./lib/hooks/useDebouncedFunction.js");
-/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
-/* harmony import */ var _components_AttachFileButton__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../../components/AttachFileButton */ "./lib/components/AttachFileButton.js");
-/* harmony import */ var _components_DatabaseButton__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../components/DatabaseButton */ "./lib/components/DatabaseButton.js");
+/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
+/* harmony import */ var _components_AttachFileButton__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../../components/AttachFileButton */ "./lib/components/AttachFileButton.js");
+/* harmony import */ var _components_DatabaseButton__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../../components/DatabaseButton */ "./lib/components/DatabaseButton.js");
 /* harmony import */ var _utils_fileUpload__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../utils/fileUpload */ "./lib/utils/fileUpload.js");
 /*
  * Copyright (c) Saga Inc.
@@ -1290,6 +1353,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isEditing, contextManager, notebookTracker, agentModeEnabled = false, agentExecutionStatus = 'idle', operatingSystem = 'mac', displayOptimizedChatHistoryLength = 0, agentTargetNotebookPanelRef, isSignedUp = true, messageIndex, }) => {
     var _a;
     const [input, setInput] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(initialContent);
@@ -1341,20 +1405,18 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
         e.stopPropagation();
         setIsDragOver(false);
         const files = e.dataTransfer.files;
-        if (files && files.length > 0) {
-            const file = files[0];
-            if (file && !isUploading) {
-                setIsUploading(true);
-                try {
-                    // Upload file to backend using the shared utility
-                    await (0,_utils_fileUpload__WEBPACK_IMPORTED_MODULE_4__.uploadFileToBackend)(file, notebookTracker, handleFileUpload);
-                }
-                catch (error) {
-                    // Error handling is already done in the utility function
-                }
-                finally {
-                    setIsUploading(false);
-                }
+        if (files && files.length > 0 && !isUploading) {
+            setIsUploading(true);
+            try {
+                // Upload all files to backend using the shared utility
+                const uploadPromises = Array.from(files).map(file => (0,_utils_fileUpload__WEBPACK_IMPORTED_MODULE_4__.uploadFileToBackend)(file, notebookTracker, handleFileUpload));
+                // Use allSettled so we wait for all uploads to finish (success or failure)
+                // before clearing the uploading state. This avoids background uploads
+                // mutating state after isUploading becomes false.
+                await Promise.allSettled(uploadPromises);
+            }
+            finally {
+                setIsUploading(false);
             }
         }
     };
@@ -1435,6 +1497,16 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
                     }
                 ]);
             }
+            else if (option.type === 'cell') {
+                setAdditionalContext(prev => [
+                    ...prev,
+                    {
+                        type: 'cell',
+                        value: option.cellId,
+                        display: `Cell ${option.cellNumber}`
+                    }
+                ]);
+            }
             setDropdownVisible(false);
             // Use setTimeout to ensure this happens after React's state update cycle
             setTimeout(() => {
@@ -1481,16 +1553,27 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
                 { type: 'db', value: option.variable.value, display: option.variable.variable_name }
             ]);
         }
+        else if (option.type === 'cell') {
+            // For cells, add them as @CellN mentions (no space for easier filtering)
+            contextChatRepresentation = `@Cell${option.cellNumber}`;
+            // Store the stable cell ID in additionalContext, not the @CellN format
+            setAdditionalContext([
+                ...additionalContext,
+                { type: 'cell', value: option.cellId, display: `Cell ${option.cellNumber}` }
+            ]);
+        }
+        // Add a space after the selected item so user can continue typing
+        const contextChatRepresentationWithSpace = contextChatRepresentation + ' ';
         const newValue = input.slice(0, atIndex) +
-            contextChatRepresentation +
+            contextChatRepresentationWithSpace +
             textAfterCursor;
         setInput(newValue);
         setDropdownVisible(false);
-        // After updating the input value, set the cursor position after the inserted variable name
+        // After updating the input value, set the cursor position after the inserted item and space
         // We use setTimeout to ensure this happens after React's state update
         setTimeout(() => {
             if (textarea) {
-                const newCursorPosition = atIndex + contextChatRepresentation.length;
+                const newCursorPosition = atIndex + contextChatRepresentationWithSpace.length;
                 textarea.focus();
                 textarea.setSelectionRange(newCursorPosition, newCursorPosition);
             }
@@ -1507,6 +1590,10 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
             value: contextItem.value
         }));
     };
+    // Convert @Cell N references to [MITO_CELL_REF:cell_id] format before submitting
+    const processMessageForSubmission = (messageText) => {
+        return (0,_utils_cellReferences__WEBPACK_IMPORTED_MODULE_6__.convertCellReferencesToStableFormat)(messageText, notebookTracker.currentWidget);
+    };
     const getExpandedVarialbes = () => {
         const activeNotebookContext = contextManager === null || contextManager === void 0 ? void 0 : contextManager.getActiveNotebookContext();
         const expandedVariables = [
@@ -1570,6 +1657,11 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
                         display: 'Active Cell'
                     }]);
             }
+            // Remove the current notebook context item
+            const hasNotebookContext = additionalContext.some(context => context.type === 'notebook');
+            if (hasNotebookContext) {
+                setAdditionalContext(prev => prev.filter(context => context.type !== 'notebook'));
+            }
         }
         else if (agentModeEnabled) {
             // Remove active cell context when in agent mode
@@ -1577,15 +1669,24 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
             if (hasActiveCellContext) {
                 setAdditionalContext(prev => prev.filter(context => context.type !== 'active_cell'));
             }
+            const hasNotebookContext = additionalContext.some(context => context.type === 'notebook');
+            if (!hasNotebookContext) {
+                // Add a current notebook context item
+                setAdditionalContext(prev => [...prev, {
+                        type: 'notebook',
+                        value: 'Notebook',
+                        display: 'Notebook'
+                    }]);
+            }
         }
     }, [agentModeEnabled, additionalContext, activeCellCode]);
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_6__.classNames)("chat-input-container", {
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_7__.classNames)("chat-input-container", {
             "editing": isEditing,
             "drag-over": isDragOver
         }), onDragOver: handleDragOver, onDragLeave: handleDragLeave, onDrop: handleDrop },
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'context-container' },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_DatabaseButton__WEBPACK_IMPORTED_MODULE_7__["default"], { app: app }),
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AttachFileButton__WEBPACK_IMPORTED_MODULE_8__["default"], { onFileUploaded: handleFileUpload, notebookTracker: notebookTracker }),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_DatabaseButton__WEBPACK_IMPORTED_MODULE_8__["default"], { app: app }),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AttachFileButton__WEBPACK_IMPORTED_MODULE_9__["default"], { onFileUploaded: handleFileUpload, notebookTracker: notebookTracker }),
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "context-button", onClick: () => {
                     var _a;
                     setDropdownVisible(true);
@@ -1593,9 +1694,9 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
                     setIsDropdownFromButton(true);
                     (_a = textAreaRef.current) === null || _a === void 0 ? void 0 : _a.focus();
                 } }, "\uFF20 Add Context"),
-            additionalContext.map((context, index) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_9__["default"], { key: `${context.type}-${context.value}-${index}`, title: context.display || context.value, type: context.type, onRemove: () => setAdditionalContext(additionalContext.filter((_, i) => i !== index)), notebookTracker: notebookTracker, activeCellID: activeCellID })))),
+            additionalContext.map((context, index) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_10__["default"], { key: `${context.type}-${context.value}-${index}`, title: context.display || context.value, type: context.type, onRemove: () => setAdditionalContext(additionalContext.filter((_, i) => i !== index)), notebookTracker: notebookTracker, activeCellID: activeCellID, value: context.value })))),
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'chat-input-text-area-container' },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("textarea", { ref: textAreaRef, className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_6__.classNames)("message", "message-user", 'chat-input', { "agent-mode": agentModeEnabled }), placeholder: getPlaceholderText(), value: input, disabled: agentExecutionStatus === 'working' || agentExecutionStatus === 'stopping', onChange: handleInputChange, onKeyDown: (e) => {
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("textarea", { ref: textAreaRef, className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_7__.classNames)("message", "message-user", 'chat-input', { "agent-mode": agentModeEnabled }), placeholder: getPlaceholderText(), value: input, disabled: agentExecutionStatus === 'working' || agentExecutionStatus === 'stopping', onChange: handleInputChange, onKeyDown: (e) => {
                     // If dropdown is visible, only handle escape to close it
                     if (isDropdownVisible) {
                         if (e.key === 'Escape') {
@@ -1609,8 +1710,9 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
                     if (e.key === 'Enter' && !e.shiftKey) {
                         e.preventDefault();
                         adjustHeight(true);
+                        const processedMessage = processMessageForSubmission(input);
                         const additionalContextWithoutDisplayNames = getAdditionContextWithoutDisplayNames();
-                        handleSubmitUserMessage(input, messageIndex, additionalContextWithoutDisplayNames);
+                        handleSubmitUserMessage(processedMessage, messageIndex, additionalContextWithoutDisplayNames);
                         // Reset
                         setInput('');
                         setAdditionalContext([]);
@@ -1623,12 +1725,13 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
                         }
                     }
                 } }),
-            isDropdownVisible && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdown__WEBPACK_IMPORTED_MODULE_10__["default"], { options: getExpandedVarialbes(), onSelect: handleOptionSelect, filterText: dropdownFilter, isDropdownFromButton: isDropdownFromButton, onFilterChange: setDropdownFilter, onClose: handleDropdownClose }))),
+            isDropdownVisible && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdown__WEBPACK_IMPORTED_MODULE_11__["default"], { options: getExpandedVarialbes(), onSelect: handleOptionSelect, filterText: dropdownFilter, isDropdownFromButton: isDropdownFromButton, onFilterChange: setDropdownFilter, onClose: handleDropdownClose, notebookTracker: notebookTracker }))),
         isEditing &&
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "message-edit-buttons" },
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { onClick: () => {
+                        const processedMessage = processMessageForSubmission(input);
                         const additionalContextWithoutDisplayNames = getAdditionContextWithoutDisplayNames();
-                        handleSubmitUserMessage(input, messageIndex, additionalContextWithoutDisplayNames);
+                        handleSubmitUserMessage(processedMessage, messageIndex, additionalContextWithoutDisplayNames);
                     } }, "Save"),
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { onClick: onCancel }, "Cancel"))));
 };
@@ -1649,27 +1752,28 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */ });
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
-/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _UserCodeBlock__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./UserCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/UserCodeBlock.js");
-/* harmony import */ var _AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./AssistantCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js");
-/* harmony import */ var _AlertBlock__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./AlertBlock */ "./lib/Extensions/AiChat/ChatMessage/AlertBlock.js");
-/* harmony import */ var _MarkdownBlock__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ./MarkdownBlock */ "./lib/Extensions/AiChat/ChatMessage/MarkdownBlock.js");
-/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/strings */ "./lib/utils/strings.js");
-/* harmony import */ var _icons_Pencil__WEBPACK_IMPORTED_MODULE_17__ = __webpack_require__(/*! ../../../icons/Pencil */ "./lib/icons/Pencil.js");
-/* harmony import */ var _ChatInput__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ./ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
-/* harmony import */ var _components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
-/* harmony import */ var _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../../icons/PlayButtonIcon */ "./lib/icons/PlayButtonIcon.js");
-/* harmony import */ var _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_12__ = __webpack_require__(/*! ../../../icons/CopyIcon */ "./lib/icons/CopyIcon.js");
-/* harmony import */ var _utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../../../utils/copyToClipboard */ "./lib/utils/copyToClipboard.js");
-/* harmony import */ var _components_TextButton__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../../components/TextButton */ "./lib/components/TextButton.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
+/* harmony import */ var _UserCodeBlock__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./UserCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/UserCodeBlock.js");
+/* harmony import */ var _AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./AssistantCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js");
+/* harmony import */ var _AlertBlock__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./AlertBlock */ "./lib/Extensions/AiChat/ChatMessage/AlertBlock.js");
+/* harmony import */ var _MarkdownBlock__WEBPACK_IMPORTED_MODULE_16__ = __webpack_require__(/*! ./MarkdownBlock */ "./lib/Extensions/AiChat/ChatMessage/MarkdownBlock.js");
+/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../utils/strings */ "./lib/utils/strings.js");
+/* harmony import */ var _icons_Pencil__WEBPACK_IMPORTED_MODULE_18__ = __webpack_require__(/*! ../../../icons/Pencil */ "./lib/icons/Pencil.js");
+/* harmony import */ var _ChatInput__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ./ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
+/* harmony import */ var _components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
+/* harmony import */ var _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../../../icons/PlayButtonIcon */ "./lib/icons/PlayButtonIcon.js");
+/* harmony import */ var _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../../icons/CopyIcon */ "./lib/icons/CopyIcon.js");
+/* harmony import */ var _utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_12__ = __webpack_require__(/*! ../../../utils/copyToClipboard */ "./lib/utils/copyToClipboard.js");
+/* harmony import */ var _components_TextButton__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ../../../components/TextButton */ "./lib/components/TextButton.js");
 /* harmony import */ var _style_ChatMessage_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/ChatMessage.css */ "./style/ChatMessage.css");
 /* harmony import */ var _style_MarkdownMessage_css__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../../style/MarkdownMessage.css */ "./style/MarkdownMessage.css");
-/* harmony import */ var _components_AgentComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_18__ = __webpack_require__(/*! ../../../components/AgentComponents/GetCellOutputToolUI */ "./lib/components/AgentComponents/GetCellOutputToolUI.js");
-/* harmony import */ var _components_AgentComponents_AssumptionToolUI__WEBPACK_IMPORTED_MODULE_16__ = __webpack_require__(/*! ../../../components/AgentComponents/AssumptionToolUI */ "./lib/components/AgentComponents/AssumptionToolUI.js");
-/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
-/* harmony import */ var _components_AgentComponents_RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_19__ = __webpack_require__(/*! ../../../components/AgentComponents/RunAllCellsToolUI */ "./lib/components/AgentComponents/RunAllCellsToolUI.js");
-/* harmony import */ var _components_AgentComponents_CreateStreamlitAppToolUI__WEBPACK_IMPORTED_MODULE_20__ = __webpack_require__(/*! ../../../components/AgentComponents/CreateStreamlitAppToolUI */ "./lib/components/AgentComponents/CreateStreamlitAppToolUI.js");
-/* harmony import */ var _components_AgentComponents_EditStreamlitAppToolUI__WEBPACK_IMPORTED_MODULE_21__ = __webpack_require__(/*! ../../../components/AgentComponents/EditStreamlitAppToolUI */ "./lib/components/AgentComponents/EditStreamlitAppToolUI.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _components_AgentComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_19__ = __webpack_require__(/*! ../../../components/AgentComponents/GetCellOutputToolUI */ "./lib/components/AgentComponents/GetCellOutputToolUI.js");
+/* harmony import */ var _components_AgentComponents_AssumptionToolUI__WEBPACK_IMPORTED_MODULE_17__ = __webpack_require__(/*! ../../../components/AgentComponents/AssumptionToolUI */ "./lib/components/AgentComponents/AssumptionToolUI.js");
+/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
+/* harmony import */ var _components_AgentComponents_RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_20__ = __webpack_require__(/*! ../../../components/AgentComponents/RunAllCellsToolUI */ "./lib/components/AgentComponents/RunAllCellsToolUI.js");
+/* harmony import */ var _components_AgentComponents_CreateStreamlitAppToolUI__WEBPACK_IMPORTED_MODULE_21__ = __webpack_require__(/*! ../../../components/AgentComponents/CreateStreamlitAppToolUI */ "./lib/components/AgentComponents/CreateStreamlitAppToolUI.js");
+/* harmony import */ var _components_AgentComponents_EditStreamlitAppToolUI__WEBPACK_IMPORTED_MODULE_22__ = __webpack_require__(/*! ../../../components/AgentComponents/EditStreamlitAppToolUI */ "./lib/components/AgentComponents/EditStreamlitAppToolUI.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -1696,14 +1800,25 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
+const getCellIdFromCellUpdate = (cellUpdate, notebookTracker) => {
+    if (!cellUpdate) {
+        return undefined;
+    }
+    if (cellUpdate.type === 'modification') {
+        return cellUpdate.id;
+    }
+    // For 'new' type, get the cell ID by index
+    return (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.getCellIDByIndexInNotebookPanel)(notebookTracker.currentWidget, cellUpdate.index);
+};
 const ChatMessage = ({ app, message, promptType, agentResponse, messageIndex, mitoAIConnectionError, mitoAIConnectionErrorType, notebookTracker, renderMimeRegistry, isLastAiMessage, isLastMessage, operatingSystem, previewAICode, acceptAICode, rejectAICode, contextManager, codeReviewStatus, setNextSteps, agentModeEnabled, additionalContext, handleSubmitUserMessage, }) => {
     const [isEditing, setIsEditing] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     if (message.role !== 'user' && message.role !== 'assistant') {
         return null;
     }
     const editable = message.role === 'user';
-    const messageContentParts = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_3__.splitStringWithCodeBlocks)(message);
-    const messageContent = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_3__.getContentStringFromMessage)(message);
+    const messageContentParts = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_4__.splitStringWithCodeBlocks)(message);
+    const messageContent = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_4__.getContentStringFromMessage)(message);
     const handleEditClick = () => {
         setIsEditing(true);
     };
@@ -1726,11 +1841,11 @@ const ChatMessage = ({ app, message, promptType, agentResponse, messageIndex, mi
         setNextSteps(agentResponse.next_steps);
     }
     if (isEditing) {
-        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatInput__WEBPACK_IMPORTED_MODULE_4__["default"], { app: app, initialContent: message.content.replace(/```[\s\S]*?```/g, '').trim(), onCancel: handleCancel, isEditing: isEditing, contextManager: contextManager, notebookTracker: notebookTracker, agentModeEnabled: false, handleSubmitUserMessage: handleSubmitUserMessageAndCloseEditing, messageIndex: messageIndex }));
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatInput__WEBPACK_IMPORTED_MODULE_5__["default"], { app: app, initialContent: message.content.replace(/```[\s\S]*?```/g, '').trim(), onCancel: handleCancel, isEditing: isEditing, contextManager: contextManager, notebookTracker: notebookTracker, agentModeEnabled: false, handleSubmitUserMessage: handleSubmitUserMessageAndCloseEditing, messageIndex: messageIndex }));
     }
     if (mitoAIConnectionError) {
-        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)("message") },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AlertBlock__WEBPACK_IMPORTED_MODULE_6__["default"], { content: message.content, mitoAIConnectionErrorType: mitoAIConnectionErrorType })));
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_6__.classNames)("message") },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AlertBlock__WEBPACK_IMPORTED_MODULE_7__["default"], { content: message.content, mitoAIConnectionErrorType: mitoAIConnectionErrorType })));
     }
     // If the message is empty, don't render anything
     if (messageContent === undefined || messageContent === '') {
@@ -1740,28 +1855,28 @@ const ChatMessage = ({ app, message, promptType, agentResponse, messageIndex, mi
     // Users end up applying the code in the middle of streaming and it gets very confusing
     // very quickly for users. 
     let isCodeComplete = false;
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)("message", { "message-user": message.role === 'user' }, { 'message-assistant-chat': message.role === 'assistant' }) },
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_6__.classNames)("message", { "message-user": message.role === 'user' }, { 'message-assistant-chat': message.role === 'assistant' }) },
         messageContentParts.map((messagePart, index) => {
             var _a, _b;
-            if (messagePart.startsWith(_utils_strings__WEBPACK_IMPORTED_MODULE_3__.PYTHON_CODE_BLOCK_START_WITHOUT_NEW_LINE)) {
+            if (messagePart.startsWith(_utils_strings__WEBPACK_IMPORTED_MODULE_4__.PYTHON_CODE_BLOCK_START_WITHOUT_NEW_LINE)) {
                 isCodeComplete = messagePart.endsWith('```');
                 // Make sure that there is actually code in the message. 
                 // An empty code will look like this '```python  ```'
                 if (messagePart.length > 14) {
                     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null,
-                        message.role === 'user' ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_UserCodeBlock__WEBPACK_IMPORTED_MODULE_7__["default"], { code: messagePart, renderMimeRegistry: renderMimeRegistry, agentModeEnabled: agentModeEnabled })) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_8__["default"], { code: messagePart, codeSummary: (_b = (_a = agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.cell_update) === null || _a === void 0 ? void 0 : _a.code_summary) !== null && _b !== void 0 ? _b : ((agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'run_all_cells' ? 'Running all cells' : undefined), isCodeComplete: isCodeComplete, renderMimeRegistry: renderMimeRegistry, previewAICode: previewAICode, acceptAICode: acceptAICode, rejectAICode: rejectAICode, isLastAiMessage: isLastAiMessage, codeReviewStatus: codeReviewStatus, agentModeEnabled: agentModeEnabled })),
+                        message.role === 'user' ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_UserCodeBlock__WEBPACK_IMPORTED_MODULE_8__["default"], { code: messagePart, renderMimeRegistry: renderMimeRegistry, agentModeEnabled: agentModeEnabled })) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_9__["default"], { code: messagePart, codeSummary: (_b = (_a = agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.cell_update) === null || _a === void 0 ? void 0 : _a.code_summary) !== null && _b !== void 0 ? _b : ((agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'run_all_cells' ? 'Running all cells' : undefined), isCodeComplete: isCodeComplete, renderMimeRegistry: renderMimeRegistry, previewAICode: previewAICode, acceptAICode: acceptAICode, rejectAICode: rejectAICode, isLastAiMessage: isLastAiMessage, codeReviewStatus: codeReviewStatus, agentModeEnabled: agentModeEnabled, cellId: getCellIdFromCellUpdate(agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.cell_update, notebookTracker), notebookPanel: notebookTracker.currentWidget })),
                         !agentModeEnabled && isLastAiMessage && isCodeComplete && codeReviewStatus === 'chatPreview' &&
                             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'chat-message-buttons' },
-                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_9__["default"], { onClick: () => { previewAICode(); }, text: 'Overwrite Active Cell', icon: _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_10__["default"], title: 'Write the Ai generated code to the active cell in the jupyter notebook, replacing the current code.', variant: 'gray', width: 'fit-contents' }),
-                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_9__["default"], { onClick: () => { void (0,_utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_11__["default"])(messagePart); }, text: 'Copy', icon: _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_12__["default"], title: 'Copy the Ai generated code to your clipboard', variant: 'gray', width: 'fit-contents' })),
+                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_10__["default"], { onClick: () => { previewAICode(); }, text: 'Overwrite Active Cell', icon: _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_11__["default"], title: 'Write the Ai generated code to the active cell in the jupyter notebook, replacing the current code.', variant: 'gray', width: 'fit-contents' }),
+                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_10__["default"], { onClick: () => { void (0,_utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_12__["default"])(messagePart); }, text: 'Copy', icon: _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_13__["default"], title: 'Copy the Ai generated code to your clipboard', variant: 'gray', width: 'fit-contents' })),
                         !agentModeEnabled && isLastAiMessage && isCodeComplete && codeReviewStatus === 'codeCellPreview' &&
                             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'chat-message-buttons' },
-                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_13__["default"], { onClick: () => { acceptAICode(); }, text: `Accept code ${operatingSystem === 'mac' ? '⌘Y' : 'Ctrl+Y'}`, title: 'Accept the Ai generated code', variant: 'green', width: 'fit-contents' }),
-                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_13__["default"], { onClick: () => { rejectAICode(); }, text: `Reject code ${operatingSystem === 'mac' ? '⌘U' : 'Ctrl+U'}`, title: 'Reject the Ai generated code and revert to the previous version of the code cell', variant: 'red', width: 'fit-contents' })),
+                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_14__["default"], { onClick: () => { acceptAICode(); }, text: `Accept code ${operatingSystem === 'mac' ? '⌘Y' : 'Ctrl+Y'}`, title: 'Accept the Ai generated code', variant: 'green', width: 'fit-contents' }),
+                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_14__["default"], { onClick: () => { rejectAICode(); }, text: `Reject code ${operatingSystem === 'mac' ? '⌘U' : 'Ctrl+U'}`, title: 'Reject the Ai generated code and revert to the previous version of the code cell', variant: 'red', width: 'fit-contents' })),
                         message.role === 'user' && additionalContext && additionalContext.length > 0 &&
                             react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null, additionalContext
                                 .filter(context => context.type !== 'active_cell') // Hide active cell context in chat messages
-                                .map((context, index) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_14__["default"], { key: `${context.type}-${context.value}-${index}`, title: `${context.value}`, type: context.type, onRemove: () => { } }))))));
+                                .map((context, index) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_15__["default"], { key: `${context.type}-${context.value}-${index}`, title: `${context.value}`, type: context.type, onRemove: () => { } }))))));
                 }
                 else {
                     // Return null for empty code blocks
@@ -1769,7 +1884,7 @@ const ChatMessage = ({ app, message, promptType, agentResponse, messageIndex, mi
                 }
             }
             else {
-                return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { key: index + messagePart, className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)('markdown-message-part') },
+                return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { key: index + messagePart, className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_6__.classNames)('markdown-message-part') },
                     react__WEBPACK_IMPORTED_MODULE_0___default().createElement("p", { onDoubleClick: () => {
                             // Only allow users to edit their own messages, not the AI responses
                             if (message.role === 'user') {
@@ -1777,23 +1892,23 @@ const ChatMessage = ({ app, message, promptType, agentResponse, messageIndex, mi
                             }
                         } }, message.role === 'user' && promptType === 'smartDebug' ? (
                     /* Use a pre tag to preserve the newline and indentation of the error message */
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("pre", { className: "chat-taskpane-smart-debug-error-message" }, messagePart)) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_MarkdownBlock__WEBPACK_IMPORTED_MODULE_15__["default"], { markdown: messagePart, renderMimeRegistry: renderMimeRegistry, notebookTracker: notebookTracker }))),
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("pre", { className: "chat-taskpane-smart-debug-error-message" }, messagePart)) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_MarkdownBlock__WEBPACK_IMPORTED_MODULE_16__["default"], { markdown: messagePart, renderMimeRegistry: renderMimeRegistry, notebookTracker: notebookTracker }))),
                     (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.analysis_assumptions) &&
-                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_AssumptionToolUI__WEBPACK_IMPORTED_MODULE_16__["default"], { assumptions: agentResponse.analysis_assumptions })));
+                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_AssumptionToolUI__WEBPACK_IMPORTED_MODULE_17__["default"], { assumptions: agentResponse.analysis_assumptions })));
             }
         }),
         editable &&
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "message-action-buttons" },
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "message-start-editing-button", onClick: handleEditClick, title: "Edit message" },
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_Pencil__WEBPACK_IMPORTED_MODULE_17__["default"], null))),
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_Pencil__WEBPACK_IMPORTED_MODULE_18__["default"], null))),
         (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'get_cell_output' &&
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_18__["default"], null),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_19__["default"], null),
         (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'run_all_cells' && agentModeEnabled &&
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_19__["default"], null),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_20__["default"], null),
         (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'create_streamlit_app' && agentModeEnabled &&
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_CreateStreamlitAppToolUI__WEBPACK_IMPORTED_MODULE_20__["default"], { isRunning: isLastMessage }),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_CreateStreamlitAppToolUI__WEBPACK_IMPORTED_MODULE_21__["default"], { isRunning: isLastMessage }),
         (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'edit_streamlit_app' && agentModeEnabled &&
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_EditStreamlitAppToolUI__WEBPACK_IMPORTED_MODULE_21__["default"], { isRunning: isLastMessage })));
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_EditStreamlitAppToolUI__WEBPACK_IMPORTED_MODULE_22__["default"], { isRunning: isLastMessage })));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ChatMessage);
 
@@ -1813,7 +1928,8 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */ });
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
-/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _utils_cellReferences__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../utils/cellReferences */ "./lib/utils/cellReferences.js");
 /* harmony import */ var _style_Citation_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/Citation.css */ "./style/Citation.css");
 /*
  * Copyright (c) Saga Inc.
@@ -1822,6 +1938,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 // Helper function to get the display text for a line/range
 const getLineDisplayText = (line) => {
     if (typeof line === 'number') {
@@ -1833,15 +1950,24 @@ const getLineDisplayText = (line) => {
 };
 // Citation button component
 const Citation = ({ citationIndex, cellId, line, notebookTracker }) => {
+    // Check if the cell exists in the current notebook
+    const cellNumber = (0,_utils_cellReferences__WEBPACK_IMPORTED_MODULE_2__.getCellNumberById)(cellId, notebookTracker.currentWidget);
+    const isMissing = cellNumber === undefined;
     const handleClick = () => {
+        if (isMissing)
+            return;
         const lineStart = typeof line === 'number' ? line : line.start;
         // In order to support old citations that have just one line, we 
         // we set the end line to the start line if only a single line number is provided.
         const lineEnd = typeof line === 'number' ? line : line.end;
         // Scroll to the cell and highlight the lines
-        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_2__.scrollToAndHighlightCell)(notebookTracker.currentWidget, cellId, lineStart, lineEnd);
+        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.scrollToAndHighlightCell)(notebookTracker.currentWidget, cellId, lineStart, lineEnd);
     };
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "citation-button", onClick: handleClick, title: getLineDisplayText(line) }, citationIndex));
+    const className = isMissing ? 'citation-button citation-missing' : 'citation-button';
+    const title = isMissing
+        ? 'Cell not found (may have been deleted or is in a different notebook)'
+        : getLineDisplayText(line);
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: className, onClick: handleClick, title: title }, citationIndex));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (Citation);
 
@@ -1906,7 +2032,10 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _jupyterlab_rendermime__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_rendermime__WEBPACK_IMPORTED_MODULE_1__);
 /* harmony import */ var react_dom__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! react-dom */ "webpack/sharing/consume/default/react-dom");
 /* harmony import */ var react_dom__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(react_dom__WEBPACK_IMPORTED_MODULE_2__);
-/* harmony import */ var _Citation__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ./Citation */ "./lib/Extensions/AiChat/ChatMessage/Citation.js");
+/* harmony import */ var _Citation__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ./Citation */ "./lib/Extensions/AiChat/ChatMessage/Citation.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _hooks_useCellOrder__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../hooks/useCellOrder */ "./lib/hooks/useCellOrder.js");
+/* harmony import */ var _style_CellReference_css__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../../style/CellReference.css */ "./style/CellReference.css");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -1915,12 +2044,25 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
+
+
 const CitationPortal = ({ container, ...props }) => {
-    return (0,react_dom__WEBPACK_IMPORTED_MODULE_2__.createPortal)(react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Citation__WEBPACK_IMPORTED_MODULE_3__.Citation, { ...props }), container);
+    return (0,react_dom__WEBPACK_IMPORTED_MODULE_2__.createPortal)(react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Citation__WEBPACK_IMPORTED_MODULE_4__.Citation, { ...props }), container);
 };
 const MarkdownBlock = ({ markdown, renderMimeRegistry, notebookTracker }) => {
     const [citationPortals, setCitationPortals] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)([]);
     const containerRef = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(null);
+    // Track cell order to update cell references when cells are reordered
+    const cellOrder = (0,_hooks_useCellOrder__WEBPACK_IMPORTED_MODULE_5__.useCellOrder)(notebookTracker);
+    // Create a serialized version of cell order for dependency tracking
+    // This ensures re-renders when cells are reordered (even if count stays the same)
+    const cellOrderKey = (0,react__WEBPACK_IMPORTED_MODULE_0__.useMemo)(() => {
+        return Array.from(cellOrder.entries())
+            .sort((a, b) => a[0].localeCompare(b[0])) // Sort by cellId for stable string
+            .map(([cellId, cellNumber]) => `${cellId}:${cellNumber}`)
+            .join(',');
+    }, [cellOrder]);
     // Helper function to parse line numbers or ranges
     const parseLineNumber = (lineStr) => {
         if (lineStr.includes('-')) {
@@ -1945,22 +2087,26 @@ const MarkdownBlock = ({ markdown, renderMimeRegistry, notebookTracker }) => {
             return parseInt(lineStr, 10);
         }
     };
-    // Extract citations from the markdown, returning the markdown with the JSON citations replaced with 
-    // citation placeholders {{${id}}} and an array of citation objects.
-    const extractCitations = (0,react__WEBPACK_IMPORTED_MODULE_0__.useCallback)((text) => {
-        // Updated regex to match both single lines and line ranges: [MITO_CITATION:cell_id:line_number] or [MITO_CITATION:cell_id:start_line-end_line]
+    // Extract citations and cell references from the markdown, returning the markdown with 
+    // placeholders and arrays of citation/cell reference objects.
+    const extractCitationsAndCellRefs = (0,react__WEBPACK_IMPORTED_MODULE_0__.useCallback)((text) => {
+        // Regex for citations: [MITO_CITATION:cell_id:line_number] or [MITO_CITATION:cell_id:start_line-end_line]
         const citationRegex = /\[MITO_CITATION:([^:]+):(\d+(?:-\d+)?)\]/g;
+        // Regex for cell references: [MITO_CELL_REF:cell_id]
+        const cellRefRegex = /\[MITO_CELL_REF:([^\]]+)\]/g;
         const citations = [];
-        let counter = 0;
-        // Replace each citation with a placeholder
-        const processedMarkdown = text.replace(citationRegex, (match, cellId, lineStr) => {
+        const cellRefs = [];
+        let citationCounter = 0;
+        let cellRefCounter = 0;
+        // First, replace citations with placeholders
+        let processedMarkdown = text.replace(citationRegex, (match, cellId, lineStr) => {
             try {
-                const id = `citation-${counter++}`;
+                const id = `citation-${citationCounter++}`;
                 const line = parseLineNumber(lineStr);
                 citations.push({
                     id,
                     data: {
-                        citation_index: counter,
+                        citation_index: citationCounter,
                         cell_id: cellId,
                         line: line
                     }
@@ -1972,7 +2118,16 @@ const MarkdownBlock = ({ markdown, renderMimeRegistry, notebookTracker }) => {
                 return match;
             }
         });
-        return { processedMarkdown, citations };
+        // Then, replace cell references with placeholders
+        processedMarkdown = processedMarkdown.replace(cellRefRegex, (match, cellId) => {
+            const id = `cellref-${cellRefCounter++}`;
+            cellRefs.push({
+                id,
+                cellId: cellId.trim()
+            });
+            return `{{${id}}}`;
+        });
+        return { processedMarkdown, citations, cellRefs };
     }, []);
     // Uses the Jupyter markdowm MimeRenderer to render the markdown content as normal HTML
     const renderMarkdownContent = (0,react__WEBPACK_IMPORTED_MODULE_0__.useCallback)(async (processedMarkdown) => {
@@ -1992,13 +2147,14 @@ const MarkdownBlock = ({ markdown, renderMimeRegistry, notebookTracker }) => {
             console.error("Error rendering markdown:", error);
         }
     }, [renderMimeRegistry]);
-    // Replace the citation placeholders with Citation Components in the DOM
-    const createCitationPortals = (0,react__WEBPACK_IMPORTED_MODULE_0__.useCallback)((citations) => {
-        if (!containerRef.current || citations.length === 0)
+    // Replace the citation and cell reference placeholders with components in the DOM
+    const createPortalsFromPlaceholders = (0,react__WEBPACK_IMPORTED_MODULE_0__.useCallback)((citations, cellRefs) => {
+        if (!containerRef.current || (citations.length === 0 && cellRefs.length === 0))
             return [];
         const newPortals = [];
-        // Create a map of placeholder to citation for faster lookup
+        // Create maps for faster lookup
         const citationMap = new Map(citations.map(citation => [`{{${citation.id}}}`, citation]));
+        const cellRefMap = new Map(cellRefs.map(ref => [`{{${ref.id}}}`, ref]));
         // Find all text nodes that contain our placeholder like {{citation-id}}).
         // Since these placeholders exist within the text content (not as separate DOM elements):
         //  - Find all text nodes in the rendered markdown
@@ -2016,7 +2172,7 @@ const MarkdownBlock = ({ markdown, renderMimeRegistry, notebookTracker }) => {
                 return;
             // Check if this node contains any placeholders
             let containsPlaceholder = false;
-            for (const placeholder of citationMap.keys()) {
+            for (const placeholder of [...citationMap.keys(), ...cellRefMap.keys()]) {
                 if (node.nodeValue.includes(placeholder)) {
                     containsPlaceholder = true;
                     break;
@@ -2024,8 +2180,8 @@ const MarkdownBlock = ({ markdown, renderMimeRegistry, notebookTracker }) => {
             }
             if (!containsPlaceholder)
                 return;
-            // Create a regex to match all placeholders
-            const placeholderPattern = /\{\{citation-\d+\}\}/g;
+            // Create a regex to match all placeholders (citations and cell refs)
+            const placeholderPattern = /\{\{(citation|cellref)-\d+\}\}/g;
             const matches = [...node.nodeValue.matchAll(placeholderPattern)];
             if (matches.length === 0)
                 return;
@@ -2034,21 +2190,50 @@ const MarkdownBlock = ({ markdown, renderMimeRegistry, notebookTracker }) => {
             let lastIndex = 0;
             matches.forEach(match => {
                 const placeholder = match[0];
-                const citation = citationMap.get(placeholder);
-                if (!citation)
-                    return;
                 const startIndex = match.index;
                 // Add text before the placeholder
                 if (startIndex > lastIndex) {
                     fragment.appendChild(document.createTextNode(node.nodeValue.substring(lastIndex, startIndex)));
                 }
-                // Create span for the citation
-                const span = document.createElement('span');
-                span.classList.add('citation-container');
-                span.dataset.citationId = citation.id;
-                fragment.appendChild(span);
-                // Create React portal for this span
-                newPortals.push(react__WEBPACK_IMPORTED_MODULE_0___default().createElement(CitationPortal, { key: citation.id + '-' + matches.indexOf(match), container: span, citationIndex: citation.data.citation_index, cellId: citation.data.cell_id, line: citation.data.line, notebookTracker: notebookTracker }));
+                // Check if it's a citation or cell reference
+                const citation = citationMap.get(placeholder);
+                const cellRef = cellRefMap.get(placeholder);
+                if (citation) {
+                    // Create span for the citation
+                    const span = document.createElement('span');
+                    span.classList.add('citation-container');
+                    span.dataset.citationId = citation.id;
+                    fragment.appendChild(span);
+                    // Create React portal for this span
+                    newPortals.push(react__WEBPACK_IMPORTED_MODULE_0___default().createElement(CitationPortal, { key: citation.id + '-' + matches.indexOf(match), container: span, citationIndex: citation.data.citation_index, cellId: citation.data.cell_id, line: citation.data.line, notebookTracker: notebookTracker }));
+                }
+                else if (cellRef) {
+                    // Create clickable span for cell reference
+                    const cellNumber = cellOrder.get(cellRef.cellId);
+                    const isMissing = cellNumber === undefined;
+                    const displayText = isMissing ? 'Cell' : `Cell ${cellNumber}`;
+                    const span = document.createElement('span');
+                    span.className = isMissing ? 'cell-reference cell-reference-missing' : 'cell-reference';
+                    span.textContent = displayText;
+                    span.title = isMissing
+                        ? 'Cell not found (may have been deleted or is in a different notebook)'
+                        : `Click to navigate to ${displayText}`;
+                    // Only add click handler if cell exists
+                    if (!isMissing) {
+                        span.addEventListener('click', (e) => {
+                            e.preventDefault();
+                            e.stopPropagation();
+                            if (notebookTracker.currentWidget) {
+                                (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_6__.scrollToCell)(notebookTracker.currentWidget, cellRef.cellId, undefined, 'center');
+                                // Highlight the cell after scrolling
+                                setTimeout(() => {
+                                    (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_6__.highlightCodeCell)(notebookTracker, cellRef.cellId);
+                                }, 500);
+                            }
+                        });
+                    }
+                    fragment.appendChild(span);
+                }
                 lastIndex = startIndex + placeholder.length;
             });
             // Add any remaining text after the last placeholder
@@ -2061,20 +2246,21 @@ const MarkdownBlock = ({ markdown, renderMimeRegistry, notebookTracker }) => {
             }
         });
         return newPortals;
-    }, []);
+    }, [notebookTracker, cellOrder]);
     // Process everything in one effect, but with clear separation via helper functions
+    // cellOrderKey triggers re-render when notebook loads or cells are reordered (fixes race condition on refresh)
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
         const processMarkdown = async () => {
-            // Step 1: Extract citations and get processed markdown
-            const { processedMarkdown, citations } = extractCitations(markdown);
+            // Step 1: Extract citations and cell references, get processed markdown
+            const { processedMarkdown, citations, cellRefs } = extractCitationsAndCellRefs(markdown);
             // Step 2: Render markdown with placeholders
             await renderMarkdownContent(processedMarkdown);
-            // Step 3: Create and insert citation portals
-            const portals = createCitationPortals(citations);
+            // Step 3: Create and insert portals for citations and cell references
+            const portals = createPortalsFromPlaceholders(citations, cellRefs);
             setCitationPortals(portals);
         };
         void processMarkdown();
-    }, [markdown, extractCitations, renderMarkdownContent, createCitationPortals]);
+    }, [markdown, extractCitationsAndCellRefs, renderMarkdownContent, createPortalsFromPlaceholders, cellOrderKey]);
     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { ref: containerRef, className: "markdown-block-with-citations" }, citationPortals));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (MarkdownBlock);
@@ -2328,34 +2514,35 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! @lumino/coreutils */ "webpack/sharing/consume/default/@lumino/coreutils");
 /* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(_lumino_coreutils__WEBPACK_IMPORTED_MODULE_2__);
 /* harmony import */ var _commands__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../commands */ "./lib/commands.js");
-/* harmony import */ var _components_AgentComponents_ErrorFixupToolUI__WEBPACK_IMPORTED_MODULE_35__ = __webpack_require__(/*! ../../components/AgentComponents/ErrorFixupToolUI */ "./lib/components/AgentComponents/ErrorFixupToolUI.js");
-/* harmony import */ var _components_DropdownMenu__WEBPACK_IMPORTED_MODULE_31__ = __webpack_require__(/*! ../../components/DropdownMenu */ "./lib/components/DropdownMenu.js");
-/* harmony import */ var _components_IconButton__WEBPACK_IMPORTED_MODULE_28__ = __webpack_require__(/*! ../../components/IconButton */ "./lib/components/IconButton.js");
-/* harmony import */ var _components_LoadingCircle__WEBPACK_IMPORTED_MODULE_44__ = __webpack_require__(/*! ../../components/LoadingCircle */ "./lib/components/LoadingCircle.js");
-/* harmony import */ var _components_ModelSelector__WEBPACK_IMPORTED_MODULE_43__ = __webpack_require__(/*! ../../components/ModelSelector */ "./lib/components/ModelSelector.js");
-/* harmony import */ var _components_NextStepsPills__WEBPACK_IMPORTED_MODULE_40__ = __webpack_require__(/*! ../../components/NextStepsPills */ "./lib/components/NextStepsPills.js");
-/* harmony import */ var _components_ToggleButton__WEBPACK_IMPORTED_MODULE_42__ = __webpack_require__(/*! ../../components/ToggleButton */ "./lib/components/ToggleButton.js");
-/* harmony import */ var _icons__WEBPACK_IMPORTED_MODULE_32__ = __webpack_require__(/*! ../../icons */ "./lib/icons/index.js");
-/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_27__ = __webpack_require__(/*! ../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _utils_chatHistory__WEBPACK_IMPORTED_MODULE_26__ = __webpack_require__(/*! ../../utils/chatHistory */ "./lib/utils/chatHistory.js");
-/* harmony import */ var _utils_codeDiff__WEBPACK_IMPORTED_MODULE_25__ = __webpack_require__(/*! ../../utils/codeDiff */ "./lib/utils/codeDiff.js");
+/* harmony import */ var _components_AgentComponents_ErrorFixupToolUI__WEBPACK_IMPORTED_MODULE_36__ = __webpack_require__(/*! ../../components/AgentComponents/ErrorFixupToolUI */ "./lib/components/AgentComponents/ErrorFixupToolUI.js");
+/* harmony import */ var _components_DropdownMenu__WEBPACK_IMPORTED_MODULE_32__ = __webpack_require__(/*! ../../components/DropdownMenu */ "./lib/components/DropdownMenu.js");
+/* harmony import */ var _components_IconButton__WEBPACK_IMPORTED_MODULE_29__ = __webpack_require__(/*! ../../components/IconButton */ "./lib/components/IconButton.js");
+/* harmony import */ var _components_LoadingCircle__WEBPACK_IMPORTED_MODULE_45__ = __webpack_require__(/*! ../../components/LoadingCircle */ "./lib/components/LoadingCircle.js");
+/* harmony import */ var _components_ModelSelector__WEBPACK_IMPORTED_MODULE_44__ = __webpack_require__(/*! ../../components/ModelSelector */ "./lib/components/ModelSelector.js");
+/* harmony import */ var _components_NextStepsPills__WEBPACK_IMPORTED_MODULE_41__ = __webpack_require__(/*! ../../components/NextStepsPills */ "./lib/components/NextStepsPills.js");
+/* harmony import */ var _components_ToggleButton__WEBPACK_IMPORTED_MODULE_43__ = __webpack_require__(/*! ../../components/ToggleButton */ "./lib/components/ToggleButton.js");
+/* harmony import */ var _icons__WEBPACK_IMPORTED_MODULE_33__ = __webpack_require__(/*! ../../icons */ "./lib/icons/index.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_28__ = __webpack_require__(/*! ../../utils/classNames */ "./lib/utils/classNames.js");
+/* harmony import */ var _utils_chatHistory__WEBPACK_IMPORTED_MODULE_27__ = __webpack_require__(/*! ../../utils/chatHistory */ "./lib/utils/chatHistory.js");
+/* harmony import */ var _utils_codeDiff__WEBPACK_IMPORTED_MODULE_26__ = __webpack_require__(/*! ../../utils/codeDiff */ "./lib/utils/codeDiff.js");
 /* harmony import */ var _utils_cellOutput__WEBPACK_IMPORTED_MODULE_18__ = __webpack_require__(/*! ../../utils/cellOutput */ "./lib/utils/cellOutput.js");
-/* harmony import */ var _utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_22__ = __webpack_require__(/*! ../../utils/waitForNotebookReady */ "./lib/utils/waitForNotebookReady.js");
+/* harmony import */ var _utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_23__ = __webpack_require__(/*! ../../utils/waitForNotebookReady */ "./lib/utils/waitForNotebookReady.js");
 /* harmony import */ var _utils__WEBPACK_IMPORTED_MODULE_17__ = __webpack_require__(/*! ./utils */ "./lib/Extensions/AiChat/utils.js");
-/* harmony import */ var _restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_23__ = __webpack_require__(/*! ../../restAPI/RestAPI */ "./lib/restAPI/RestAPI.js");
-/* harmony import */ var _SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_29__ = __webpack_require__(/*! ../SettingsManager/SettingsManagerPlugin */ "./lib/Extensions/SettingsManager/SettingsManagerPlugin.js");
+/* harmony import */ var _restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_24__ = __webpack_require__(/*! ../../restAPI/RestAPI */ "./lib/restAPI/RestAPI.js");
+/* harmony import */ var _utils_sound__WEBPACK_IMPORTED_MODULE_20__ = __webpack_require__(/*! ../../utils/sound */ "./lib/utils/sound.js");
+/* harmony import */ var _SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_30__ = __webpack_require__(/*! ../SettingsManager/SettingsManagerPlugin */ "./lib/Extensions/SettingsManager/SettingsManagerPlugin.js");
 /* harmony import */ var _SettingsManager_profiler_ProfilerPage__WEBPACK_IMPORTED_MODULE_19__ = __webpack_require__(/*! ../SettingsManager/profiler/ProfilerPage */ "./lib/Extensions/SettingsManager/profiler/ProfilerPage.js");
-/* harmony import */ var _components_AgentReviewPanel__WEBPACK_IMPORTED_MODULE_38__ = __webpack_require__(/*! ./components/AgentReviewPanel */ "./lib/Extensions/AiChat/components/AgentReviewPanel.js");
-/* harmony import */ var _CTACarousel__WEBPACK_IMPORTED_MODULE_34__ = __webpack_require__(/*! ./CTACarousel */ "./lib/Extensions/AiChat/CTACarousel.js");
-/* harmony import */ var _UsageBadge__WEBPACK_IMPORTED_MODULE_30__ = __webpack_require__(/*! ./UsageBadge */ "./lib/Extensions/AiChat/UsageBadge.js");
-/* harmony import */ var _SignUpForm__WEBPACK_IMPORTED_MODULE_33__ = __webpack_require__(/*! ./SignUpForm */ "./lib/Extensions/AiChat/SignUpForm.js");
-/* harmony import */ var _FirstMessage__WEBPACK_IMPORTED_MODULE_21__ = __webpack_require__(/*! ./FirstMessage */ "./lib/Extensions/AiChat/FirstMessage.js");
-/* harmony import */ var _ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_41__ = __webpack_require__(/*! ./ChatMessage/ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
-/* harmony import */ var _ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_36__ = __webpack_require__(/*! ./ChatMessage/ChatMessage */ "./lib/Extensions/AiChat/ChatMessage/ChatMessage.js");
-/* harmony import */ var _ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_39__ = __webpack_require__(/*! ./ChatMessage/ScrollableSuggestions */ "./lib/Extensions/AiChat/ChatMessage/ScrollableSuggestions.js");
+/* harmony import */ var _components_AgentReviewPanel__WEBPACK_IMPORTED_MODULE_39__ = __webpack_require__(/*! ./components/AgentReviewPanel */ "./lib/Extensions/AiChat/components/AgentReviewPanel.js");
+/* harmony import */ var _CTACarousel__WEBPACK_IMPORTED_MODULE_35__ = __webpack_require__(/*! ./CTACarousel */ "./lib/Extensions/AiChat/CTACarousel.js");
+/* harmony import */ var _UsageBadge__WEBPACK_IMPORTED_MODULE_31__ = __webpack_require__(/*! ./UsageBadge */ "./lib/Extensions/AiChat/UsageBadge.js");
+/* harmony import */ var _SignUpForm__WEBPACK_IMPORTED_MODULE_34__ = __webpack_require__(/*! ./SignUpForm */ "./lib/Extensions/AiChat/SignUpForm.js");
+/* harmony import */ var _FirstMessage__WEBPACK_IMPORTED_MODULE_22__ = __webpack_require__(/*! ./FirstMessage */ "./lib/Extensions/AiChat/FirstMessage.js");
+/* harmony import */ var _ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_42__ = __webpack_require__(/*! ./ChatMessage/ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
+/* harmony import */ var _ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_37__ = __webpack_require__(/*! ./ChatMessage/ChatMessage */ "./lib/Extensions/AiChat/ChatMessage/ChatMessage.js");
+/* harmony import */ var _ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_40__ = __webpack_require__(/*! ./ChatMessage/ScrollableSuggestions */ "./lib/Extensions/AiChat/ChatMessage/ScrollableSuggestions.js");
 /* harmony import */ var _ChatHistoryManager__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./ChatHistoryManager */ "./lib/Extensions/AiChat/ChatHistoryManager.js");
 /* harmony import */ var _hooks_useAgentReview__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ./hooks/useAgentReview */ "./lib/Extensions/AiChat/hooks/useAgentReview.js");
-/* harmony import */ var _hooks_useAgentExecution__WEBPACK_IMPORTED_MODULE_20__ = __webpack_require__(/*! ./hooks/useAgentExecution */ "./lib/Extensions/AiChat/hooks/useAgentExecution.js");
+/* harmony import */ var _hooks_useAgentExecution__WEBPACK_IMPORTED_MODULE_21__ = __webpack_require__(/*! ./hooks/useAgentExecution */ "./lib/Extensions/AiChat/hooks/useAgentExecution.js");
 /* harmony import */ var _hooks_useUserSignup__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./hooks/useUserSignup */ "./lib/Extensions/AiChat/hooks/useUserSignup.js");
 /* harmony import */ var _hooks_useChatScroll__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./hooks/useChatScroll */ "./lib/Extensions/AiChat/hooks/useChatScroll.js");
 /* harmony import */ var _hooks_useModelConfig__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ./hooks/useModelConfig */ "./lib/Extensions/AiChat/hooks/useModelConfig.js");
@@ -2367,8 +2554,8 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _style_button_css__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../style/button.css */ "./style/button.css");
 /* harmony import */ var _style_ChatTaskpane_css__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../style/ChatTaskpane.css */ "./style/ChatTaskpane.css");
 /* harmony import */ var _style_TextButton_css__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../style/TextButton.css */ "./style/TextButton.css");
-/* harmony import */ var _components_LoadingDots__WEBPACK_IMPORTED_MODULE_37__ = __webpack_require__(/*! ../../components/LoadingDots */ "./lib/components/LoadingDots.js");
-/* harmony import */ var _utils_notebookMetadata__WEBPACK_IMPORTED_MODULE_24__ = __webpack_require__(/*! ../../utils/notebookMetadata */ "./lib/utils/notebookMetadata.js");
+/* harmony import */ var _components_LoadingDots__WEBPACK_IMPORTED_MODULE_38__ = __webpack_require__(/*! ../../components/LoadingDots */ "./lib/components/LoadingDots.js");
+/* harmony import */ var _utils_notebookMetadata__WEBPACK_IMPORTED_MODULE_25__ = __webpack_require__(/*! ../../utils/notebookMetadata */ "./lib/utils/notebookMetadata.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -2398,6 +2585,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 // Internal imports - Chat components
 
 
@@ -2433,7 +2621,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
     // User signup state
     const { isSignedUp, refreshUserSignupState } = (0,_hooks_useUserSignup__WEBPACK_IMPORTED_MODULE_7__.useUserSignup)();
     // Core chat state management
-    const { chatHistoryManager, chatHistoryManagerRef, setChatHistoryManager, loadingAIResponse, setLoadingAIResponse, codeReviewStatus, setCodeReviewStatus, agentReviewStatus, setAgentReviewStatus, nextSteps, setNextSteps, displayedNextStepsIfAvailable, setDisplayedNextStepsIfAvailable, } = (0,_hooks_useChatState__WEBPACK_IMPORTED_MODULE_8__.useChatState)(getDefaultChatHistoryManager(notebookTracker, contextManager, app));
+    const { chatHistoryManager, chatHistoryManagerRef, setChatHistoryManager, loadingStatus, setLoadingStatus, codeReviewStatus, setCodeReviewStatus, agentReviewStatus, setAgentReviewStatus, nextSteps, setNextSteps, displayedNextStepsIfAvailable, setDisplayedNextStepsIfAvailable, } = (0,_hooks_useChatState__WEBPACK_IMPORTED_MODULE_8__.useChatState)(getDefaultChatHistoryManager(notebookTracker, contextManager, app));
     // Chat scroll management
     const { chatTaskpaneMessagesRef, setAutoScrollFollowMode } = (0,_hooks_useChatScroll__WEBPACK_IMPORTED_MODULE_9__.useChatScroll)(chatHistoryManager);
     // Model configuration
@@ -2442,6 +2630,20 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
     const usageBadgeRef = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(null);
     // Streaming response management
     const { streamingContentRef, streamHandlerRef, activeRequestControllerRef } = (0,_hooks_useStreamingResponse__WEBPACK_IMPORTED_MODULE_11__.useStreamingResponse)();
+    // Audio context management
+    const audioContextRef = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(null);
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        audioContextRef.current = new (window.AudioContext || window.webkitAudioContext)();
+        return () => {
+            const audioContext = audioContextRef.current;
+            audioContextRef.current = null;
+            if (audioContext) {
+                void audioContext.close().catch(() => {
+                    // Ignore errors closing (e.g. already closed)
+                });
+            }
+        };
+    }, []);
     // Agent mode state management
     const { agentModeEnabled, agentModeEnabledRef, setAgentModeEnabled, hasCheckpoint, setHasCheckpoint, showRevertQuestionnaire, setShowRevertQuestionnaire, } = (0,_hooks_useAgentMode__WEBPACK_IMPORTED_MODULE_12__.useAgentMode)();
     // Create a shared ref for the agent target notebook panel
@@ -2522,7 +2724,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         const newChatHistoryManager = getDuplicateChatHistoryManager();
         const smartDebugMetadata = newChatHistoryManager.addSmartDebugMessage(activeThreadIdRef.current, errorMessage);
         setChatHistoryManager(newChatHistoryManager);
-        setLoadingAIResponse(true);
+        setLoadingStatus('thinking');
         // Step 2: Send the message to the AI
         const smartDebugCompletionRequest = {
             type: 'smartDebug',
@@ -2544,7 +2746,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         const newChatHistoryManager = getDuplicateChatHistoryManager();
         const agentSmartDebugMessage = newChatHistoryManager.addAgentSmartDebugMessage(activeThreadIdRef.current, errorMessage, agentTargetNotebookPanelRef.current);
         setChatHistoryManager(newChatHistoryManager);
-        setLoadingAIResponse(true);
+        setLoadingStatus('thinking');
         // Step 2: Send the message to the AI
         const smartDebugCompletionRequest = {
             type: 'agent:autoErrorFixup',
@@ -2563,7 +2765,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         const newChatHistoryManager = getDuplicateChatHistoryManager();
         const explainCodeMetadata = newChatHistoryManager.addExplainCodeMessage(activeThreadIdRef.current);
         setChatHistoryManager(newChatHistoryManager);
-        setLoadingAIResponse(true);
+        setLoadingStatus('thinking');
         // Step 2: Send the message to the AI
         const explainCompletionRequest = {
             type: 'codeExplain',
@@ -2593,7 +2795,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         }
         agentExecutionMetadata.base64EncodedActiveCellOutput = await (0,_utils__WEBPACK_IMPORTED_MODULE_17__.getBase64EncodedCellOutputInNotebook)(agentTargetNotebookPanel, sendCellIDOutput);
         setChatHistoryManager(newChatHistoryManager);
-        setLoadingAIResponse(true);
+        setLoadingStatus('thinking');
         // Step 2: Send the message to the AI
         const completionRequest = {
             type: 'agent:execution',
@@ -2619,7 +2821,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         }
         const chatMessageMetadata = await newChatHistoryManager.addChatInputMessage(input, activeThreadIdRef.current, messageIndex, additionalContext);
         setChatHistoryManager(newChatHistoryManager);
-        setLoadingAIResponse(true);
+        setLoadingStatus('thinking');
         // Yield control briefly to allow React to re-render the UI
         // A timeout of 0ms pushes the rest of the function to the next event loop cycle
         // so we don't get stuck behind the slow getActiveCellOutput function.
@@ -2651,7 +2853,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         }
     };
     const _sendMessageAndSaveResponse = async (completionRequest, newChatHistoryManager) => {
-        var _a, _b, _c, _d;
+        var _a, _b;
         // Create AbortController for this request
         const abortController = new AbortController();
         activeRequestControllerRef.current = abortController;
@@ -2681,11 +2883,13 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                         console.groupEnd();
                     }
                     addAIMessageFromResponseAndUpdateState(chunk.error.hint || chunk.error.title || "An error occurred", completionRequest.metadata.promptType, newChatHistoryManager, true, chunk.error.title);
-                    setLoadingAIResponse(false);
+                    setLoadingStatus(undefined);
                 }
                 else if (chunk.done) {
                     // Reset states to allow future messages to show the "Apply" button
                     setCodeReviewStatus('chatPreview');
+                    // Play completion sound for streaming mode
+                    (0,_utils_sound__WEBPACK_IMPORTED_MODULE_20__.playCompletionSound)(audioContextRef.current);
                 }
                 else {
                     // Use a ref to accumulate the content properly
@@ -2696,27 +2900,23 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                     setChatHistoryManager(updatedChatHistoryManager);
                     // Set loading to false after we receive the first chunk
                     if (streamingContentRef.current.length > 0) {
-                        setLoadingAIResponse(false);
+                        setLoadingStatus(undefined);
                     }
                 }
             };
-            // Store the handler for later cleanup
-            streamHandlerRef.current = streamHandler;
-            // Connect the handler
-            websocketClient.stream.connect(streamHandler, null);
             try {
-                const aiResponse = await websocketClient.sendMessage(completionRequest);
-                const content = (_b = (_a = aiResponse.items[0]) === null || _a === void 0 ? void 0 : _a.content) !== null && _b !== void 0 ? _b : '';
-                if (completionRequest.metadata.promptType === 'agent:execution' ||
-                    completionRequest.metadata.promptType === 'agent:autoErrorFixup') {
-                    // Agent:Execution prompts return a CellUpdate object that we need to parse
-                    const agentResponse = JSON.parse(content);
-                    newChatHistoryManager.addAIMessageFromAgentResponse(agentResponse);
-                }
+                // Store the handler for later cleanup
+                streamHandlerRef.current = streamHandler;
+                // Connect the handler
+                websocketClient.stream.connect(streamHandler, null);
+                // Send the message to the AI and let the stream handler handle the rest
+                await websocketClient.sendMessage(completionRequest);
             }
             catch (error) {
                 addAIMessageFromResponseAndUpdateState(error.title ? error.title : `${error}`, 'chat', newChatHistoryManager, false);
                 addAIMessageFromResponseAndUpdateState(error.hint ? error.hint : `${error}`, completionRequest.metadata.promptType, newChatHistoryManager, true);
+                // Reset loading status when an error occurs
+                setLoadingStatus(undefined);
             }
         }
         else {
@@ -2750,7 +2950,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                         : `${aiResponse.error.error_type}: ${aiResponse.error.title}`, completionRequest.metadata.promptType, newChatHistoryManager, true, aiResponse.error.title);
                 }
                 else {
-                    const content = (_d = (_c = aiResponse.items[0]) === null || _c === void 0 ? void 0 : _c.content) !== null && _d !== void 0 ? _d : '';
+                    const content = (_b = (_a = aiResponse.items[0]) === null || _a === void 0 ? void 0 : _a.content) !== null && _b !== void 0 ? _b : '';
                     if (completionRequest.metadata.promptType === 'agent:execution' || completionRequest.metadata.promptType === 'agent:autoErrorFixup') {
                         // Agent:Execution prompts return a CellUpdate object that we need to parse
                         const agentResponse = JSON.parse(content);
@@ -2777,7 +2977,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
             finally {
                 // Reset states to allow future messages to show the "Apply" button
                 setCodeReviewStatus('chatPreview');
-                setLoadingAIResponse(false);
+                setLoadingStatus(undefined);
             }
         }
         // Clean up AbortController
@@ -2799,7 +2999,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         setChatHistoryManager(chatHistoryManager);
     };
     // Initialize agent execution hook
-    const agentExecution = (0,_hooks_useAgentExecution__WEBPACK_IMPORTED_MODULE_20__.useAgentExecution)({
+    const agentExecution = (0,_hooks_useAgentExecution__WEBPACK_IMPORTED_MODULE_21__.useAgentExecution)({
         notebookTracker,
         app,
         streamlitPreviewManager,
@@ -2807,7 +3007,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         chatHistoryManagerRef,
         activeThreadIdRef,
         activeRequestControllerRef,
-        setLoadingAIResponse,
+        setLoadingStatus,
         setAutoScrollFollowMode,
         setHasCheckpoint,
         addAIMessageFromResponseAndUpdateState,
@@ -2816,7 +3016,8 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         sendAgentSmartDebugMessage,
         agentReview,
         agentTargetNotebookPanelRef,
-        setAgentReviewStatus
+        setAgentReviewStatus,
+        audioContextRef
     });
     // Main initialization effect - runs once on mount
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
@@ -2836,9 +3037,9 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                 else {
                     await startNewChat();
                 }
-                const firstMessage = (0,_FirstMessage__WEBPACK_IMPORTED_MODULE_21__.getFirstMessage)();
+                const firstMessage = (0,_FirstMessage__WEBPACK_IMPORTED_MODULE_22__.getFirstMessage)();
                 if (firstMessage) {
-                    await (0,_utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_22__.waitForNotebookReady)(notebookTracker);
+                    await (0,_utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_23__.waitForNotebookReady)(notebookTracker);
                     await startNewChat();
                     await agentExecution.startAgentExecution(firstMessage, setAgentReviewStatus);
                 }
@@ -2849,7 +3050,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                 addAIMessageFromResponseAndUpdateState(error.hint ? error.hint : `${error}`, 'chat', newChatHistoryManager, true);
             }
         };
-        void (0,_restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_23__.logEvent)('opened_ai_chat_taskpane');
+        void (0,_restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_24__.logEvent)('opened_ai_chat_taskpane');
         void initializeChatHistory();
         void refreshUserSignupState(); // Get user signup state when the component first mounts
         /****
@@ -2857,7 +3058,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
          * to specific app files
          * ****/
         const handleNotebookPanelChanged = () => {
-            (0,_utils_notebookMetadata__WEBPACK_IMPORTED_MODULE_24__.setNotebookID)(notebookTracker.currentWidget);
+            (0,_utils_notebookMetadata__WEBPACK_IMPORTED_MODULE_25__.setNotebookID)(notebookTracker.currentWidget);
         };
         // Event fires every time the active notebook panel changes
         notebookTracker.currentChanged.connect(handleNotebookPanelChanged);
@@ -2936,14 +3137,14 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
             className: 'text-button-mito-ai button-base button-green',
             caption: 'Accept Code',
             execute: acceptAICode,
-            isVisible: () => (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_25__.shouldShowDiffToolbarButtons)(notebookTracker, cellStateBeforeDiff.current, agentReview.changedCellsRef.current)
+            isVisible: () => (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_26__.shouldShowDiffToolbarButtons)(notebookTracker, cellStateBeforeDiff.current, agentReview.changedCellsRef.current)
         });
         app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_13__.COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE, {
             label: `Reject ${operatingSystem === 'mac' ? '⌘U' : 'Ctrl+U'}`,
             className: 'text-button-mito-ai button-base button-red',
             caption: 'Reject Code',
             execute: rejectAICode,
-            isVisible: () => (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_25__.shouldShowDiffToolbarButtons)(notebookTracker, cellStateBeforeDiff.current, agentReview.changedCellsRef.current)
+            isVisible: () => (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_26__.shouldShowDiffToolbarButtons)(notebookTracker, cellStateBeforeDiff.current, agentReview.changedCellsRef.current)
         });
     }, []);
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
@@ -2997,7 +3198,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
     // In agent mode, we group consecutive error messages together. 
     // In chat mode, we display messages individually as they were sent
     if (agentModeEnabled) {
-        processedDisplayOptimizedChatHistory = (0,_utils_chatHistory__WEBPACK_IMPORTED_MODULE_26__.processChatHistoryForErrorGrouping)(chatHistoryManager.getDisplayOptimizedHistory());
+        processedDisplayOptimizedChatHistory = (0,_utils_chatHistory__WEBPACK_IMPORTED_MODULE_27__.processChatHistoryForErrorGrouping)(chatHistoryManager.getDisplayOptimizedHistory());
     }
     else {
         processedDisplayOptimizedChatHistory = chatHistoryManager.getDisplayOptimizedHistory();
@@ -3008,20 +3209,20 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
     };
     return (
     // We disable the chat taskpane if the user is not signed up AND there are no chat history items
-    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_27__.classNames)('chat-taskpane', { 'disabled': !(isSignedUp || displayOptimizedChatHistory.length > 0) }) },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_28__.classNames)('chat-taskpane', { 'disabled': !(isSignedUp || displayOptimizedChatHistory.length > 0) }) },
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-taskpane-header" },
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-taskpane-header-left" },
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_28__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_1__.settingsIcon.react, null), title: "Mito AI Settings", onClick: () => {
-                        void app.commands.execute(_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_29__.COMMAND_MITO_AI_SETTINGS);
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_29__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_1__.settingsIcon.react, null), title: "Mito AI Settings", onClick: () => {
+                        void app.commands.execute(_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_30__.COMMAND_MITO_AI_SETTINGS);
                     } }),
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_UsageBadge__WEBPACK_IMPORTED_MODULE_30__["default"], { app: app, ref: usageBadgeRef })),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_UsageBadge__WEBPACK_IMPORTED_MODULE_31__["default"], { app: app, ref: usageBadgeRef })),
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-taskpane-header-right" },
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_28__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_1__.addIcon.react, null), title: "Start New Chat", onClick: async () => { await startNewChat(); } }),
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_DropdownMenu__WEBPACK_IMPORTED_MODULE_31__["default"], { trigger: react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "icon-button", title: "Chat Threads", onClick: fetchChatThreads },
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_29__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_1__.addIcon.react, null), title: "Start New Chat", onClick: async () => { await startNewChat(); } }),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_DropdownMenu__WEBPACK_IMPORTED_MODULE_32__["default"], { trigger: react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "icon-button", title: "Chat Threads", onClick: fetchChatThreads },
                         react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_1__.historyIcon.react, null)), items: chatThreads.length > 0
                         ? chatThreads.map(thread => ({
                             label: thread.name,
-                            primaryIcon: activeThreadIdRef.current === thread.thread_id ? _icons__WEBPACK_IMPORTED_MODULE_32__.OpenIndicatorLabIcon.react : undefined,
+                            primaryIcon: activeThreadIdRef.current === thread.thread_id ? _icons__WEBPACK_IMPORTED_MODULE_33__.OpenIndicatorLabIcon.react : undefined,
                             onClick: () => fetchChatHistoryAndSetActiveThread(thread.thread_id),
                             secondaryActions: [
                                 {
@@ -3039,23 +3240,27 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-messages", ref: chatTaskpaneMessagesRef },
             displayOptimizedChatHistory.length === 0 &&
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-empty-message" }, isSignedUp === false
-                    ? react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_SignUpForm__WEBPACK_IMPORTED_MODULE_33__["default"], { onSignUpSuccess: refreshUserSignupState })
-                    : react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_CTACarousel__WEBPACK_IMPORTED_MODULE_34__["default"], { app: app })),
+                    ? react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_SignUpForm__WEBPACK_IMPORTED_MODULE_34__["default"], { onSignUpSuccess: refreshUserSignupState })
+                    : react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_CTACarousel__WEBPACK_IMPORTED_MODULE_35__["default"], { app: app })),
             processedDisplayOptimizedChatHistory.map((displayOptimizedChat, index) => {
                 if (isGroupedErrorMessages(displayOptimizedChat)) {
-                    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_ErrorFixupToolUI__WEBPACK_IMPORTED_MODULE_35__["default"], { key: index, messages: displayOptimizedChat, renderMimeRegistry: renderMimeRegistry }));
+                    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_ErrorFixupToolUI__WEBPACK_IMPORTED_MODULE_36__["default"], { key: index, messages: displayOptimizedChat, renderMimeRegistry: renderMimeRegistry, notebookTracker: notebookTracker }));
                 }
                 else {
-                    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_36__["default"], { key: index, message: displayOptimizedChat.message, promptType: displayOptimizedChat.promptType, agentResponse: displayOptimizedChat.agentResponse, codeCellID: displayOptimizedChat.codeCellID, mitoAIConnectionError: displayOptimizedChat.type === 'connection error', mitoAIConnectionErrorType: displayOptimizedChat.mitoAIConnectionErrorType || null, messageIndex: index, notebookTracker: notebookTracker, renderMimeRegistry: renderMimeRegistry, app: app, isLastAiMessage: index === lastAIMessagesIndex, isLastMessage: index === displayOptimizedChatHistory.length - 1, operatingSystem: operatingSystem, previewAICode: previewAICodeToActiveCell, acceptAICode: acceptAICode, rejectAICode: rejectAICode, handleSubmitUserMessage: handleSubmitUserMessage, contextManager: contextManager, codeReviewStatus: codeReviewStatus, setNextSteps: setNextSteps, agentModeEnabled: agentModeEnabled, additionalContext: displayOptimizedChat.additionalContext }));
+                    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_37__["default"], { key: index, message: displayOptimizedChat.message, promptType: displayOptimizedChat.promptType, agentResponse: displayOptimizedChat.agentResponse, codeCellID: displayOptimizedChat.codeCellID, mitoAIConnectionError: displayOptimizedChat.type === 'connection error', mitoAIConnectionErrorType: displayOptimizedChat.mitoAIConnectionErrorType || null, messageIndex: index, notebookTracker: notebookTracker, renderMimeRegistry: renderMimeRegistry, app: app, isLastAiMessage: index === lastAIMessagesIndex, isLastMessage: index === displayOptimizedChatHistory.length - 1, operatingSystem: operatingSystem, previewAICode: previewAICodeToActiveCell, acceptAICode: acceptAICode, rejectAICode: rejectAICode, handleSubmitUserMessage: handleSubmitUserMessage, contextManager: contextManager, codeReviewStatus: codeReviewStatus, setNextSteps: setNextSteps, agentModeEnabled: agentModeEnabled, additionalContext: displayOptimizedChat.additionalContext }));
                 }
             }).filter(message => message !== null),
-            loadingAIResponse &&
+            loadingStatus === 'thinking' &&
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-loading-message" },
                     "Thinking ",
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_LoadingDots__WEBPACK_IMPORTED_MODULE_37__["default"], null)),
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentReviewPanel__WEBPACK_IMPORTED_MODULE_38__["default"], { hasCheckpoint: hasCheckpoint, agentModeEnabled: agentModeEnabled, agentExecutionStatus: agentExecution.agentExecutionStatus, showRevertQuestionnaire: showRevertQuestionnaire, reviewAgentChanges: agentReview.reviewAgentChanges, acceptAllAICode: agentReview.acceptAllAICode, rejectAllAICode: agentReview.rejectAllAICode, getChangeCounts: agentReview.getChangeCounts, getReviewProgress: agentReview.getReviewProgress, hasChanges: agentReview.hasChanges, setHasCheckpoint: setHasCheckpoint, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable, setShowRevertQuestionnaire: setShowRevertQuestionnaire, getDuplicateChatHistoryManager: getDuplicateChatHistoryManager, setChatHistoryManager: setChatHistoryManager, app: app, notebookTracker: notebookTracker, chatTaskpaneMessagesRef: chatTaskpaneMessagesRef, agentReviewStatus: agentReviewStatus, setAgentReviewStatus: setAgentReviewStatus })),
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_LoadingDots__WEBPACK_IMPORTED_MODULE_38__["default"], null)),
+            loadingStatus === 'running-code' &&
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-loading-message" },
+                    "Running code ",
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_LoadingDots__WEBPACK_IMPORTED_MODULE_38__["default"], null)),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentReviewPanel__WEBPACK_IMPORTED_MODULE_39__["default"], { hasCheckpoint: hasCheckpoint, agentModeEnabled: agentModeEnabled, agentExecutionStatus: agentExecution.agentExecutionStatus, showRevertQuestionnaire: showRevertQuestionnaire, reviewAgentChanges: agentReview.reviewAgentChanges, acceptAllAICode: agentReview.acceptAllAICode, rejectAllAICode: agentReview.rejectAllAICode, getChangeCounts: agentReview.getChangeCounts, getReviewProgress: agentReview.getReviewProgress, hasChanges: agentReview.hasChanges, setHasCheckpoint: setHasCheckpoint, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable, setShowRevertQuestionnaire: setShowRevertQuestionnaire, getDuplicateChatHistoryManager: getDuplicateChatHistoryManager, setChatHistoryManager: setChatHistoryManager, app: app, notebookTracker: notebookTracker, chatTaskpaneMessagesRef: chatTaskpaneMessagesRef, agentReviewStatus: agentReviewStatus, setAgentReviewStatus: setAgentReviewStatus })),
         displayOptimizedChatHistory.length === 0 && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "suggestions-container" },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_39__["default"], { onSelectSuggestion: (prompt) => {
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_40__["default"], { onSelectSuggestion: (prompt) => {
                     if (agentModeEnabled) {
                         void agentExecution.startAgentExecution(prompt, setAgentReviewStatus);
                     }
@@ -3064,11 +3269,11 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                     }
                 } }))),
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: `connected-input-container ${nextSteps.length > 0 ? 'has-next-steps' : ''}` },
-            nextSteps.length > 0 && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_NextStepsPills__WEBPACK_IMPORTED_MODULE_40__["default"], { nextSteps: nextSteps, onSelectNextStep: agentExecution.startAgentExecution, displayedNextStepsIfAvailable: displayedNextStepsIfAvailable, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable, setAgentReviewStatus: setAgentReviewStatus })),
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_41__["default"], { app: app, initialContent: '', handleSubmitUserMessage: handleSubmitUserMessage, onCancel: undefined, isEditing: false, contextManager: contextManager, notebookTracker: notebookTracker, agentModeEnabled: agentModeEnabled, agentExecutionStatus: agentExecution.agentExecutionStatus, operatingSystem: operatingSystem, displayOptimizedChatHistoryLength: displayOptimizedChatHistory.length, agentTargetNotebookPanelRef: agentTargetNotebookPanelRef, isSignedUp: isSignedUp })),
+            nextSteps.length > 0 && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_NextStepsPills__WEBPACK_IMPORTED_MODULE_41__["default"], { nextSteps: nextSteps, onSelectNextStep: agentExecution.startAgentExecution, displayedNextStepsIfAvailable: displayedNextStepsIfAvailable, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable, setAgentReviewStatus: setAgentReviewStatus })),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_42__["default"], { app: app, initialContent: '', handleSubmitUserMessage: handleSubmitUserMessage, onCancel: undefined, isEditing: false, contextManager: contextManager, notebookTracker: notebookTracker, agentModeEnabled: agentModeEnabled, agentExecutionStatus: agentExecution.agentExecutionStatus, operatingSystem: operatingSystem, displayOptimizedChatHistoryLength: displayOptimizedChatHistory.length, agentTargetNotebookPanelRef: agentTargetNotebookPanelRef, isSignedUp: isSignedUp })),
         agentExecution.agentExecutionStatus !== 'working' && agentExecution.agentExecutionStatus !== 'stopping' && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-controls" },
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-controls-left" },
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_ToggleButton__WEBPACK_IMPORTED_MODULE_42__["default"], { leftText: "Chat", leftTooltip: "Chat mode suggests an edit to the active cell and let's you decide to accept or reject it.", rightText: "Agent", rightTooltip: "Agent mode writes and executes code until it's finished your request.", isLeftSelected: !agentModeEnabled, onChange: async (isLeftSelected) => {
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_ToggleButton__WEBPACK_IMPORTED_MODULE_43__["default"], { leftText: "Chat", leftTooltip: "Chat mode suggests an edit to the active cell and let's you decide to accept or reject it.", rightText: "Agent", rightTooltip: "Agent mode writes and executes code until it's finished your request.", isLeftSelected: !agentModeEnabled, onChange: async (isLeftSelected) => {
                         await startNewChat(); // TODO: delete thread instead of starting new chat
                         setAgentModeEnabled(!isLeftSelected);
                         // Clear agent checkpoint when switching modes
@@ -3080,7 +3285,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                             chatInput.focus();
                         }
                     } }),
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_ModelSelector__WEBPACK_IMPORTED_MODULE_43__["default"], { onConfigChange: (config) => {
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_ModelSelector__WEBPACK_IMPORTED_MODULE_44__["default"], { onConfigChange: (config) => {
                         // Just update the backend
                         void updateModelOnBackend(config.model);
                     } })),
@@ -3104,7 +3309,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                 " \u23CE"))),
         (agentExecution.agentExecutionStatus === 'working' || agentExecution.agentExecutionStatus === 'stopping') && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "button-base button-red stop-agent-button", onClick: () => void agentExecution.markAgentForStopping('userStop'), disabled: agentExecution.agentExecutionStatus === 'stopping', "data-testid": "stop-agent-button" }, agentExecution.agentExecutionStatus === 'stopping' ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "stop-agent-button-content" },
             "Stopping",
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_LoadingCircle__WEBPACK_IMPORTED_MODULE_44__["default"], null),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_LoadingCircle__WEBPACK_IMPORTED_MODULE_45__["default"], null),
             " ")) : ('Stop Agent')))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ChatTaskpane);
@@ -3609,11 +3814,12 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
 /* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! @lumino/coreutils */ "webpack/sharing/consume/default/@lumino/coreutils");
 /* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_lumino_coreutils__WEBPACK_IMPORTED_MODULE_1__);
-/* harmony import */ var _utils_checkpoint__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/checkpoint */ "./lib/utils/checkpoint.js");
-/* harmony import */ var _utils_agentActions__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/agentActions */ "./lib/utils/agentActions.js");
-/* harmony import */ var _utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../utils/blacklistedWords */ "./lib/utils/blacklistedWords.js");
-/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../utils/strings */ "./lib/utils/strings.js");
-/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _utils_checkpoint__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../utils/checkpoint */ "./lib/utils/checkpoint.js");
+/* harmony import */ var _utils_agentActions__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../utils/agentActions */ "./lib/utils/agentActions.js");
+/* harmony import */ var _utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/blacklistedWords */ "./lib/utils/blacklistedWords.js");
+/* harmony import */ var _utils_sound__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../utils/sound */ "./lib/utils/sound.js");
+/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../utils/strings */ "./lib/utils/strings.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -3625,8 +3831,9 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 const AGENT_EXECUTION_DEPTH_LIMIT = 20;
-const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, websocketClient, chatHistoryManagerRef, activeThreadIdRef, activeRequestControllerRef, setLoadingAIResponse, setAutoScrollFollowMode, setHasCheckpoint, addAIMessageFromResponseAndUpdateState, getDuplicateChatHistoryManager, sendAgentExecutionMessage, sendAgentSmartDebugMessage, agentReview, agentTargetNotebookPanelRef }) => {
+const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, websocketClient, chatHistoryManagerRef, activeThreadIdRef, activeRequestControllerRef, setLoadingStatus, setAutoScrollFollowMode, setHasCheckpoint, addAIMessageFromResponseAndUpdateState, getDuplicateChatHistoryManager, sendAgentExecutionMessage, sendAgentSmartDebugMessage, agentReview, agentTargetNotebookPanelRef, audioContextRef, }) => {
     // Agent execution state
     const [agentExecutionStatus, setAgentExecutionStatus] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)('idle');
     const shouldContinueAgentExecution = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(true);
@@ -3635,7 +3842,7 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
         shouldContinueAgentExecution.current = false;
         // Update state/UI
         setAgentExecutionStatus('idle');
-        setLoadingAIResponse(false);
+        setLoadingStatus(undefined);
         if (reason === 'userStop') {
             // Immediately abort any ongoing requests
             if (activeRequestControllerRef.current) {
@@ -3656,13 +3863,13 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
                 stream: false
             });
         }
-        return;
+        (0,_utils_sound__WEBPACK_IMPORTED_MODULE_2__.playCompletionSound)(audioContextRef.current);
     };
     const startAgentExecution = async (input, setAgentReviewStatus, messageIndex, additionalContext) => {
         agentTargetNotebookPanelRef.current = notebookTracker.currentWidget;
         agentReview.acceptAllAICode();
-        agentReview.setNotebookSnapshotPreAgentExecution((0,_utils_notebook__WEBPACK_IMPORTED_MODULE_2__.getAIOptimizedCellsInNotebookPanel)(agentTargetNotebookPanelRef.current));
-        await (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_3__.createCheckpoint)(app, setHasCheckpoint);
+        agentReview.setNotebookSnapshotPreAgentExecution((0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.getAIOptimizedCellsInNotebookPanel)(agentTargetNotebookPanelRef.current));
+        await (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_4__.createCheckpoint)(app, setHasCheckpoint);
         setAgentExecutionStatus('working');
         setAgentReviewStatus('pre-agent-code-review');
         // Enable follow mode when user starts agent execution
@@ -3703,9 +3910,9 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
             const aiDisplayOptimizedChatItem = chatHistoryManagerRef.current.getLastAIDisplayOptimizedChatItem();
             // # TODO: Make this is a helper function so we can also use it in the auto error fixup! 
             if (aiDisplayOptimizedChatItem) {
-                const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_4__.getCodeBlockFromMessage)(aiDisplayOptimizedChatItem.message);
+                const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_5__.getCodeBlockFromMessage)(aiDisplayOptimizedChatItem.message);
                 if (aiGeneratedCode) {
-                    const securityCheck = (0,_utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_5__.checkForBlacklistedWords)(aiGeneratedCode);
+                    const securityCheck = (0,_utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_6__.checkForBlacklistedWords)(aiGeneratedCode);
                     if (!securityCheck.safe) {
                         console.error('Security Warning:', securityCheck.reason);
                         addAIMessageFromResponseAndUpdateState(`I cannot execute this code without your approval because this code did not pass my security checks. ${securityCheck.reason}. For your safety, I am stopping execution of this plan.`, 'agent:execution', chatHistoryManagerRef.current);
@@ -3747,15 +3954,21 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
             }
             if (agentResponse.type === 'cell_update' && agentResponse.cell_update) {
                 // Run the code and handle any errors
-                await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_6__.acceptAndRunCellUpdate)(agentResponse.cell_update, agentTargetNotebookPanelRef.current);
-                const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_6__.retryIfExecutionError)(agentTargetNotebookPanelRef.current, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef);
+                setLoadingStatus('running-code');
+                try {
+                    await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_7__.acceptAndRunCellUpdate)(agentResponse.cell_update, agentTargetNotebookPanelRef.current);
+                }
+                finally {
+                    setLoadingStatus(undefined);
+                }
+                const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_7__.retryIfExecutionError)(agentTargetNotebookPanelRef.current, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef, setLoadingStatus);
                 if (status === 'interupted') {
                     break;
                 }
-                // If we were not able to run the code, break out of the loop 
+                // If we were not able to run the code, break out of the loop
                 // so we don't continue to execute the plan. Instead, we encourage
-                // the user to update the plan and try again. 
-                // TODO: Save this message in backend also even if there is not another message sent. 
+                // the user to update the plan and try again.
+                // TODO: Save this message in backend also even if there is not another message sent.
                 // TODO: Move this into the retryIfExecutionError function?
                 if (status === 'failure') {
                     addAIMessageFromResponseAndUpdateState("I apologize, but I was unable to fix the error after 3 attempts. You may want to try rephrasing your request or providing more context.", 'agent:execution', chatHistoryManagerRef.current);
@@ -3763,17 +3976,24 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
                 }
             }
             if (agentResponse.type === 'get_cell_output' && agentResponse.get_cell_output_cell_id !== null && agentResponse.get_cell_output_cell_id !== undefined) {
-                // Mark that we should send the cell output to the agent 
+                // Mark that we should send the cell output to the agent
                 // in the next loop iteration
                 sendCellIDOutput = agentResponse.get_cell_output_cell_id;
             }
             if (agentResponse.type === 'run_all_cells') {
-                const result = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_6__.runAllCells)(app, agentTargetNotebookPanelRef.current);
+                setLoadingStatus('running-code');
+                let result;
+                try {
+                    result = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_7__.runAllCells)(app, agentTargetNotebookPanelRef.current);
+                }
+                finally {
+                    setLoadingStatus(undefined);
+                }
                 // If run_all_cells resulted in an error, handle it through the error fixup process
                 if (!result.success && result.errorMessage && result.errorCellId) {
                     // Set the error cell as active so the error retry logic can work with it
-                    (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_2__.setActiveCellByIDInNotebookPanel)(agentTargetNotebookPanelRef.current, result.errorCellId);
-                    const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_6__.retryIfExecutionError)(agentTargetNotebookPanelRef.current, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef);
+                    (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.setActiveCellByIDInNotebookPanel)(agentTargetNotebookPanelRef.current, result.errorCellId);
+                    const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_7__.retryIfExecutionError)(agentTargetNotebookPanelRef.current, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef, setLoadingStatus);
                     if (status === 'interupted') {
                         break;
                     }
@@ -4209,7 +4429,7 @@ __webpack_require__.r(__webpack_exports__);
  * Manages:
  * - chatHistoryManager: The main chat history manager instance
  * - chatHistoryManagerRef: Ref to access the latest chatHistoryManager value (for use in callbacks)
- * - loadingAIResponse: Whether an AI response is currently being loaded
+ * - loadingStatus: Current loading status (undefined, 'thinking', or 'running-code')
  * - codeReviewStatus: Current status of code review (preview, applied, etc.)
  * - agentReviewStatus: Current status of agent review
  * - nextSteps: Array of suggested next steps from the AI
@@ -4222,7 +4442,7 @@ __webpack_require__.r(__webpack_exports__);
 const useChatState = (initialChatHistoryManager) => {
     const [chatHistoryManager, setChatHistoryManager] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(initialChatHistoryManager);
     const chatHistoryManagerRef = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(chatHistoryManager);
-    const [loadingAIResponse, setLoadingAIResponse] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    const [loadingStatus, setLoadingStatus] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(undefined);
     const [codeReviewStatus, setCodeReviewStatus] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)('chatPreview');
     const [agentReviewStatus, setAgentReviewStatus] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)('pre-agent-code-review');
     const [nextSteps, setNextSteps] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)([]);
@@ -4254,8 +4474,8 @@ const useChatState = (initialChatHistoryManager) => {
         chatHistoryManager,
         chatHistoryManagerRef,
         setChatHistoryManager,
-        loadingAIResponse,
-        setLoadingAIResponse,
+        loadingStatus,
+        setLoadingStatus,
         codeReviewStatus,
         setCodeReviewStatus,
         agentReviewStatus,
@@ -7282,7 +7502,7 @@ const CollapsibleWarningBlock = ({ message }) => {
     const [isExpanded, setIsExpanded] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     const lines = message.split('\n');
     const warningCount = lines.filter(line => line.includes("Warning: ")).length;
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "output-block" },
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "output-block mito-warning-block" },
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { style: { display: 'flex', alignItems: 'center', gap: '8px', cursor: 'pointer' }, onClick: () => setIsExpanded(!isExpanded) },
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "collapse-button" }, isExpanded ? '▼' : '▶'),
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("pre", { style: { margin: 0, display: 'inline' } },
@@ -7540,6 +7760,248 @@ const getFullErrorMessageFromTraceback = (tracebackArray) => {
 };
 
 
+/***/ }),
+
+/***/ "./lib/Extensions/MitoThemeLight/index.js":
+/*!************************************************!*\
+  !*** ./lib/Extensions/MitoThemeLight/index.js ***!
+  \************************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! @jupyterlab/apputils */ "webpack/sharing/consume/default/@jupyterlab/apputils");
+/* harmony import */ var _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _jupyterlab_translation__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! @jupyterlab/translation */ "webpack/sharing/consume/default/@jupyterlab/translation");
+/* harmony import */ var _jupyterlab_translation__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_translation__WEBPACK_IMPORTED_MODULE_1__);
+/* harmony import */ var _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! @jupyterlab/notebook */ "webpack/sharing/consume/default/@jupyterlab/notebook");
+/* harmony import */ var _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_2__);
+/* harmony import */ var _jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! @jupyterlab/ui-components */ "webpack/sharing/consume/default/@jupyterlab/ui-components");
+/* harmony import */ var _jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_3___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_3__);
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_4___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_4__);
+/* harmony import */ var _components_RunCellButton__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../components/RunCellButton */ "./lib/components/RunCellButton.js");
+/* harmony import */ var _utils_lineNumbers__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../utils/lineNumbers */ "./lib/utils/lineNumbers.js");
+/* harmony import */ var _style_RunCellButton_css__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../style/RunCellButton.css */ "./style/RunCellButton.css");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+
+
+
+
+
+
+/**
+ * Updates cell numbers for all cells in a notebook.
+ * Uses notebook.widgets which is always in the correct order.
+ */
+function updateAllCellNumbers(notebookPanel) {
+    const notebook = notebookPanel.content;
+    notebook.widgets.forEach((cell, index) => {
+        const header = cell.node.querySelector('.jp-Cell-header');
+        if (header) {
+            // 1-indexed for display (Cell 1, Cell 2, etc.)
+            header.setAttribute('data-cell-number', String(index + 1));
+        }
+    });
+}
+/**
+ * Sets up cell numbering for a notebook panel.
+ * Returns cleanup function to disconnect handlers.
+ */
+function setupCellNumbering(notebookPanel) {
+    var _a;
+    const notebook = notebookPanel.content;
+    // Store handler references so they can be disconnected on cleanup
+    const handleCellsChanged = () => {
+        updateAllCellNumbers(notebookPanel);
+    };
+    const handleActiveCellChanged = () => {
+        updateAllCellNumbers(notebookPanel);
+    };
+    // Update when cells are added, removed, or moved
+    (_a = notebook.model) === null || _a === void 0 ? void 0 : _a.cells.changed.connect(handleCellsChanged);
+    // Update when active cell changes (often happens when scrolling)
+    notebook.activeCellChanged.connect(handleActiveCellChanged);
+    // Use MutationObserver to handle virtualization (cells being attached/detached)
+    const observer = new MutationObserver(() => {
+        updateAllCellNumbers(notebookPanel);
+    });
+    observer.observe(notebook.node, { childList: true, subtree: true });
+    // Return cleanup function
+    return () => {
+        var _a;
+        (_a = notebook.model) === null || _a === void 0 ? void 0 : _a.cells.changed.disconnect(handleCellsChanged);
+        notebook.activeCellChanged.disconnect(handleActiveCellChanged);
+        observer.disconnect();
+        // Remove cell numbers from all cells
+        notebook.widgets.forEach((cell) => {
+            const header = cell.node.querySelector('.jp-Cell-header');
+            if (header) {
+                header.removeAttribute('data-cell-number');
+            }
+        });
+    };
+}
+/**
+ * A plugin for the Mito Light Theme.
+ *
+ * The Run Cell Button, cell numbering, and hidden default toolbar buttons only apply
+ * when the Mito Light theme is active.
+ */
+const plugin = {
+    id: 'mito_ai:theme',
+    description: 'Adds the Mito Light theme.',
+    requires: [_jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__.IThemeManager, _jupyterlab_translation__WEBPACK_IMPORTED_MODULE_1__.ITranslator, _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_2__.INotebookTracker],
+    activate: (app, manager, translator, notebookTracker) => {
+        const trans = translator.load('jupyterlab');
+        const style = 'mito_ai/index.css';
+        // Store connection for cleanup
+        let widgetAddedConnection = null;
+        // Store cell numbering cleanup functions for each notebook
+        const cellNumberingCleanups = new Map();
+        // Add Run Cell button to notebook toolbar
+        const addRunCellButton = (notebookPanel) => {
+            const toolbar = notebookPanel.toolbar;
+            if (!toolbar) {
+                return;
+            }
+            // Check if button already exists
+            if (toolbar.node.querySelector('.mito-run-cell-button-widget')) {
+                return;
+            }
+            // Create React widget with the specific notebook panel
+            class RunCellButtonWidget extends _jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_3__.ReactWidget {
+                constructor(panel) {
+                    super();
+                    this.panel = panel;
+                    this.addClass('mito-run-cell-button-widget');
+                }
+                render() {
+                    return react__WEBPACK_IMPORTED_MODULE_4___default().createElement(_components_RunCellButton__WEBPACK_IMPORTED_MODULE_6__["default"], { notebookPanel: this.panel });
+                }
+            }
+            const runCellWidget = new RunCellButtonWidget(notebookPanel);
+            // Add to the right side of the toolbar by inserting after spacer or at the end
+            try {
+                toolbar.insertAfter('spacer', 'mito-run-cell-button', runCellWidget);
+            }
+            catch (_a) {
+                // If spacer doesn't exist, add at the end
+                toolbar.addItem('mito-run-cell-button', runCellWidget);
+            }
+        };
+        // Remove Run Cell button from notebook toolbar
+        const removeRunCellButton = (notebookPanel) => {
+            const toolbar = notebookPanel.toolbar;
+            if (!toolbar) {
+                return;
+            }
+            // Find and remove the button widget by iterating toolbar items
+            for (const name of toolbar.names()) {
+                if (name === 'mito-run-cell-button') {
+                    // Hide the widget (disposal happens automatically when panel is disposed)
+                    const widget = Array.from(toolbar.children()).find(w => w.hasClass('mito-run-cell-button-widget'));
+                    if (widget) {
+                        widget.dispose();
+                    }
+                    break;
+                }
+            }
+        };
+        // Remove cell numbering from a notebook panel
+        const removeCellNumbering = (notebookPanel) => {
+            const cleanup = cellNumberingCleanups.get(notebookPanel);
+            if (cleanup) {
+                cleanup();
+                cellNumberingCleanups.delete(notebookPanel);
+            }
+        };
+        // Add buttons and cell numbering to all notebooks
+        const addButtonsToAllNotebooks = () => {
+            notebookTracker.forEach(widget => {
+                addRunCellButton(widget);
+                // Enable line numbers if needed
+                void (0,_utils_lineNumbers__WEBPACK_IMPORTED_MODULE_7__.enableLineNumbersIfNeeded)(app, widget);
+                // Setup cell numbering
+                const cleanup = setupCellNumbering(widget);
+                if (cleanup) {
+                    cellNumberingCleanups.set(widget, cleanup);
+                    // Also cleanup when notebook is disposed
+                    widget.disposed.connect(() => {
+                        cellNumberingCleanups.delete(widget);
+                    });
+                }
+            });
+            // Connect to new notebooks
+            widgetAddedConnection = (sender, widget) => {
+                setTimeout(() => {
+                    // Check if widget is still valid before proceeding
+                    if (widget.isDisposed) {
+                        return;
+                    }
+                    // Only add if Mito Light theme is still active
+                    if (manager.theme === 'Mito Light') {
+                        addRunCellButton(widget);
+                        // Enable line numbers if needed
+                        void (0,_utils_lineNumbers__WEBPACK_IMPORTED_MODULE_7__.enableLineNumbersIfNeeded)(app, widget);
+                        // Setup cell numbering
+                        const cleanup = setupCellNumbering(widget);
+                        if (cleanup) {
+                            cellNumberingCleanups.set(widget, cleanup);
+                            // Also cleanup when notebook is disposed
+                            widget.disposed.connect(() => {
+                                cellNumberingCleanups.delete(widget);
+                            });
+                        }
+                    }
+                }, 100);
+            };
+            notebookTracker.widgetAdded.connect(widgetAddedConnection);
+        };
+        // Remove buttons and cell numbering from all notebooks
+        const removeButtonsFromAllNotebooks = () => {
+            // Disconnect from new notebooks
+            if (widgetAddedConnection) {
+                notebookTracker.widgetAdded.disconnect(widgetAddedConnection);
+                widgetAddedConnection = null;
+            }
+            // Remove from all existing notebooks
+            notebookTracker.forEach(widget => {
+                removeRunCellButton(widget);
+                removeCellNumbering(widget);
+            });
+            // Clear all cleanup functions
+            cellNumberingCleanups.clear();
+        };
+        manager.register({
+            name: 'Mito Light',
+            displayName: trans.__('Mito Light'),
+            isLight: true,
+            themeScrollbars: false,
+            load: async () => {
+                // Load theme CSS (hides default buttons)
+                await manager.loadCSS(style);
+                // Add Run Cell buttons to all notebooks and enable line numbers
+                addButtonsToAllNotebooks();
+            },
+            unload: async () => {
+                // Remove Run Cell buttons from all notebooks
+                removeButtonsFromAllNotebooks();
+            }
+        });
+    },
+    autoStart: true
+};
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (plugin);
+
+
 /***/ }),
 
 /***/ "./lib/Extensions/NotebookFooter/NotebookFooter.js":
@@ -7561,6 +8023,7 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _components_LoadingCircle__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../components/LoadingCircle */ "./lib/components/LoadingCircle.js");
 /* harmony import */ var _icons_NotebookFooter_CodeIcon__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../icons/NotebookFooter/CodeIcon */ "./lib/icons/NotebookFooter/CodeIcon.js");
 /* harmony import */ var _icons_NotebookFooter_TextIcon__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../icons/NotebookFooter/TextIcon */ "./lib/icons/NotebookFooter/TextIcon.js");
+/* harmony import */ var _icons_NotebookFooter_SpreadsheetIcon__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../icons/NotebookFooter/SpreadsheetIcon */ "./lib/icons/NotebookFooter/SpreadsheetIcon.js");
 /* harmony import */ var _utils_userSignupEvents__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../utils/userSignupEvents */ "./lib/utils/userSignupEvents.js");
 /* harmony import */ var _AiChat_hooks_useUserSignup__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../AiChat/hooks/useUserSignup */ "./lib/Extensions/AiChat/hooks/useUserSignup.js");
 /*
@@ -7577,15 +8040,21 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 const NotebookFooter = ({ notebookTracker, app }) => {
     var _a;
     const notebook = (_a = notebookTracker.currentWidget) === null || _a === void 0 ? void 0 : _a.content;
     const [inputValue, setInputValue] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)('');
     const [isGenerating, setIsGenerating] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    const [hasMitosheet, setHasMitosheet] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     const { isSignedUp, refreshUserSignupState } = (0,_AiChat_hooks_useUserSignup__WEBPACK_IMPORTED_MODULE_3__.useUserSignup)();
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
         void refreshUserSignupState();
     }, []);
+    // Check if mitosheet command exists
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        setHasMitosheet(app.commands.hasCommand('mitosheet:create-empty-mitosheet'));
+    }, [app]);
     // Listen for signup success events from other components
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
         const handleSignupSuccess = () => {
@@ -7619,6 +8088,9 @@ const NotebookFooter = ({ notebookTracker, app }) => {
         if (cellType === 'code') {
             _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_1__.NotebookActions.insertBelow(notebook);
         }
+        else if (cellType === 'spreadsheet') {
+            void app.commands.execute('mitosheet:create-empty-mitosheet');
+        }
         else {
             _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_1__.NotebookActions.insertBelow(notebook);
             // Change the cell type after insertion
@@ -7684,7 +8156,12 @@ const NotebookFooter = ({ notebookTracker, app }) => {
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "button-content" },
                     react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "button-icon" },
                         react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_NotebookFooter_TextIcon__WEBPACK_IMPORTED_MODULE_8__["default"], null)),
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "button-label" }, "Text"))))));
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "button-label" }, "Text"))),
+            hasMitosheet && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { onClick: () => addCell('spreadsheet'), className: "footer-button", onMouseDown: (e) => e.stopPropagation() },
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "button-content" },
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "button-icon" },
+                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_NotebookFooter_SpreadsheetIcon__WEBPACK_IMPORTED_MODULE_9__["default"], null)),
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "button-label" }, "Spreadsheet")))))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (NotebookFooter);
 
@@ -9240,7 +9717,7 @@ const ToolbarButtonsPlugin = {
         commands.addCommand('toolbar-button:preview-as-streamlit', {
             label: 'App Mode',
             caption: 'Preview notebook as app and turn on App Mode',
-            className: 'text-button-mito-ai button-base button-purple button-small',
+            className: 'text-button-mito-ai button-base button-blue button-small',
             execute: async () => {
                 void app.commands.execute(_commands__WEBPACK_IMPORTED_MODULE_2__.COMMAND_MITO_AI_PREVIEW_AS_STREAMLIT);
             },
@@ -9275,14 +9752,12 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */ __webpack_require__.d(__webpack_exports__, {
 /* harmony export */   versionCheckPlugin: () => (/* binding */ versionCheckPlugin)
 /* harmony export */ });
-/* harmony import */ var _utils_version_check__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../utils/version_check */ "./lib/utils/version_check.js");
-/* harmony import */ var _utils_user__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../../utils/user */ "./lib/utils/user.js");
+/* harmony import */ var _utils_version_check__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../../utils/version_check */ "./lib/utils/version_check.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
-
 // Check interval - once per day in milliseconds
 const CHECK_INTERVAL = 24 * 60 * 60 * 1000;
 // Cache key for storing the last check time
@@ -9355,13 +9830,7 @@ function updateLastCheckTime() {
  * Performs the version check
  */
 async function performVersionCheck(app) {
-    // Skip version check when running in Electron (desktop app).
-    // Desktop users only need a single notification - when a new desktop app version is available.
-    // That new version of the desktop app will handle updating mito-ai automatically.
-    if ((0,_utils_user__WEBPACK_IMPORTED_MODULE_0__.isElectronBasedFrontend)()) {
-        return;
-    }
-    return (0,_utils_version_check__WEBPACK_IMPORTED_MODULE_1__.checkForUpdates)(app.serviceManager.serverSettings);
+    return (0,_utils_version_check__WEBPACK_IMPORTED_MODULE_0__.checkForUpdates)(app.serviceManager.serverSettings);
 }
 /**
  * Sets up periodic checks for updates
@@ -9766,6 +10235,8 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 const MITO_AI = 'mito_ai';
+// Note that this command is also used by the mitosheet package to check 
+// if the Mito AI extension is enabled. Make sure to update it there if you change it here.
 const COMMAND_MITO_AI_OPEN_CHAT = `${MITO_AI}:open-chat`;
 const COMMAND_MITO_AI_PREVIEW_LATEST_CODE = `${MITO_AI}:preview-latest-code`;
 const COMMAND_MITO_AI_APPLY_LATEST_CODE = `${MITO_AI}:apply-latest-code`;
@@ -9803,7 +10274,7 @@ __webpack_require__.r(__webpack_exports__);
  */
 
 
-const AgentComponentHeader = ({ icon, text, onClick, isExpanded, displayBorder = true, className }) => {
+const AgentComponentHeader = ({ icon, text, onClick, isExpanded, displayBorder = true, className, actionButtons }) => {
     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { onClick: onClick, className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_1__.classNames)('agent-component-header', {
             expanded: isExpanded,
             'no-border': !displayBorder
@@ -9811,10 +10282,12 @@ const AgentComponentHeader = ({ icon, text, onClick, isExpanded, displayBorder =
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-component-header-content" },
             icon,
             text),
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_1__.classNames)('agent-component-header-expand-icon', {
-                expanded: isExpanded
-            }), width: "16", height: "16", viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M4 6L8 10L12 6", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round", transform: isExpanded ? 'rotate(90 8 8)' : 'rotate(0 8 8)' }))));
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "agent-component-header-actions" },
+            actionButtons,
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_1__.classNames)('agent-component-header-expand-icon', {
+                    expanded: isExpanded
+                }), width: "16", height: "16", viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M4 6L8 10L12 6", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round", transform: isExpanded ? 'rotate(90 8 8)' : 'rotate(0 8 8)' })))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AgentComponentHeader);
 
@@ -9936,15 +10409,16 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */ });
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
-/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../utils/strings */ "./lib/utils/strings.js");
-/* harmony import */ var _Extensions_AiChat_ChatMessage_PythonCode__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../Extensions/AiChat/ChatMessage/PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
-/* harmony import */ var _Extensions_AiChat_ChatMessage_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../Extensions/AiChat/ChatMessage/AssistantCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js");
-/* harmony import */ var _icons_AlertIcon__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../icons/AlertIcon */ "./lib/icons/AlertIcon.js");
-/* harmony import */ var _AgentComponentHeader__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ./AgentComponentHeader */ "./lib/components/AgentComponents/AgentComponentHeader.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../utils/classNames */ "./lib/utils/classNames.js");
+/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../utils/strings */ "./lib/utils/strings.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _Extensions_AiChat_ChatMessage_PythonCode__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../Extensions/AiChat/ChatMessage/PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
+/* harmony import */ var _Extensions_AiChat_ChatMessage_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../Extensions/AiChat/ChatMessage/AssistantCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js");
+/* harmony import */ var _icons_AlertIcon__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../icons/AlertIcon */ "./lib/icons/AlertIcon.js");
+/* harmony import */ var _AgentComponentHeader__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./AgentComponentHeader */ "./lib/components/AgentComponents/AgentComponentHeader.js");
 /* harmony import */ var _style_ErrorFixupToolUI_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../style/ErrorFixupToolUI.css */ "./style/ErrorFixupToolUI.css");
 /* harmony import */ var _style_AgentComponentHeader_css__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../style/AgentComponentHeader.css */ "./style/AgentComponentHeader.css");
-/* harmony import */ var _RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./RunAllCellsToolUI */ "./lib/components/AgentComponents/RunAllCellsToolUI.js");
+/* harmony import */ var _RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./RunAllCellsToolUI */ "./lib/components/AgentComponents/RunAllCellsToolUI.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -9959,6 +10433,17 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
+const getCellIdFromCellUpdate = (cellUpdate, notebookTracker) => {
+    if (!cellUpdate) {
+        return undefined;
+    }
+    if (cellUpdate.type === 'modification') {
+        return cellUpdate.id;
+    }
+    // For 'new' type, get the cell ID by index
+    return (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.getCellIDByIndexInNotebookPanel)(notebookTracker.currentWidget, cellUpdate.index);
+};
 const parsePythonErrorType = (content) => {
     if (!content)
         return 'Error';
@@ -9975,31 +10460,31 @@ const processErrorContent = (content) => {
 };
 const ErrorDetectedBlock = ({ errorMessage, renderMimeRegistry, }) => {
     const [expandedError, setExpandedError] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
-    const rawErrorContent = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_3__.getContentStringFromMessage)(errorMessage);
+    const rawErrorContent = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_4__.getContentStringFromMessage)(errorMessage);
     const errorContent = processErrorContent(rawErrorContent);
     const errorType = parsePythonErrorType(errorContent);
     const toggleError = () => {
         setExpandedError(!expandedError);
     };
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_4__.classNames)('error-fixup-container', { expanded: expandedError }) },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AgentComponentHeader__WEBPACK_IMPORTED_MODULE_5__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_AlertIcon__WEBPACK_IMPORTED_MODULE_6__["default"], null), text: `${errorType} Detected`, onClick: toggleError, isExpanded: expandedError, displayBorder: false }),
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)('error-fixup-container', { expanded: expandedError }) },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AgentComponentHeader__WEBPACK_IMPORTED_MODULE_6__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_AlertIcon__WEBPACK_IMPORTED_MODULE_7__["default"], null), text: `${errorType} Detected`, onClick: toggleError, isExpanded: expandedError, displayBorder: false }),
         expandedError && errorContent && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "error-fixup-expanded" },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Extensions_AiChat_ChatMessage_PythonCode__WEBPACK_IMPORTED_MODULE_7__["default"], { code: errorContent, renderMimeRegistry: renderMimeRegistry })))));
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Extensions_AiChat_ChatMessage_PythonCode__WEBPACK_IMPORTED_MODULE_8__["default"], { code: errorContent, renderMimeRegistry: renderMimeRegistry })))));
 };
-const GroupedErrorsAndFixes = ({ messages, renderMimeRegistry, }) => {
+const GroupedErrorsAndFixes = ({ messages, renderMimeRegistry, notebookTracker, }) => {
     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "error-fixup-root" },
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "error-fixup-header" },
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("p", null, "Fixing an error")),
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "error-fixup-messages-container" }, messages.map((messageItem, index) => {
-            var _a, _b, _c, _d;
+            var _a, _b, _c, _d, _e;
             const isUserMessage = messageItem.message.role === 'user';
             if (isUserMessage) {
                 return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(ErrorDetectedBlock, { key: `error-${index}`, errorMessage: messageItem.message, renderMimeRegistry: renderMimeRegistry }));
             }
             if (((_a = messageItem.agentResponse) === null || _a === void 0 ? void 0 : _a.type) === 'run_all_cells') {
-                return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_8__["default"], { key: `run-all-cells-${index}`, inErrorFixup: true }));
+                return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_9__["default"], { key: `run-all-cells-${index}`, inErrorFixup: true }));
             }
-            return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Extensions_AiChat_ChatMessage_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_9__["default"], { key: `assistant-${index}`, code: messageItem.message.content, codeSummary: (_d = (_c = (_b = messageItem.agentResponse) === null || _b === void 0 ? void 0 : _b.cell_update) === null || _c === void 0 ? void 0 : _c.code_summary) !== null && _d !== void 0 ? _d : 'Fixing error', isCodeComplete: true, renderMimeRegistry: renderMimeRegistry, previewAICode: () => { }, acceptAICode: () => { }, rejectAICode: () => { }, isLastAiMessage: false, codeReviewStatus: "chatPreview", agentModeEnabled: true, isErrorFixup: true }));
+            return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Extensions_AiChat_ChatMessage_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_10__["default"], { key: `assistant-${index}`, code: messageItem.message.content, codeSummary: (_d = (_c = (_b = messageItem.agentResponse) === null || _b === void 0 ? void 0 : _b.cell_update) === null || _c === void 0 ? void 0 : _c.code_summary) !== null && _d !== void 0 ? _d : 'Fixing error', isCodeComplete: true, renderMimeRegistry: renderMimeRegistry, previewAICode: () => { }, acceptAICode: () => { }, rejectAICode: () => { }, isLastAiMessage: false, codeReviewStatus: "chatPreview", agentModeEnabled: true, isErrorFixup: true, cellId: getCellIdFromCellUpdate((_e = messageItem.agentResponse) === null || _e === void 0 ? void 0 : _e.cell_update, notebookTracker), notebookPanel: notebookTracker.currentWidget }));
         }))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (GroupedErrorsAndFixes);
@@ -10435,22 +10920,39 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+const GOOD_FOR_FAST = [
+    'Quick data exploration',
+    'Pandas operations',
+    'Basic data cleaning',
+    'Fast code iterations'
+];
+const GOOD_FOR_SMART = [
+    'Complex data analysis',
+    'Advanced debugging',
+    'Statistical analysis and modeling',
+    'Multi-step data workflows'
+];
 const MODEL_MAPPINGS = [
     {
-        displayName: 'GPT 4.1',
-        fullName: 'gpt-4.1',
+        displayName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.GPT_4_1_DISPLAY_NAME,
+        fullName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.GPT_4_1_MODEL_NAME,
         type: 'smart',
-        goodFor: [
-            'Complex data analysis',
-            'Advanced debugging',
-            'Statistical analysis and modeling',
-            'Multi-step data workflows'
-        ],
+        goodFor: [...GOOD_FOR_SMART],
         provider: 'OpenAI',
         tokenLimit: '1M',
         speed: 'Medium',
         complexityHandling: 'High'
     },
+    {
+        displayName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.GPT_5_2_DISPLAY_NAME,
+        fullName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.GPT_5_2_MODEL_NAME,
+        type: 'fast',
+        goodFor: [...GOOD_FOR_FAST],
+        provider: 'OpenAI',
+        tokenLimit: '400K',
+        speed: 'Fast',
+        complexityHandling: 'Medium'
+    },
     {
         displayName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.CLAUDE_HAIKU_DISPLAY_NAME,
         fullName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.CLAUDE_HAIKU_MODEL_NAME,
@@ -10470,42 +10972,27 @@ const MODEL_MAPPINGS = [
         displayName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.CLAUDE_SONNET_DISPLAY_NAME,
         fullName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.CLAUDE_SONNET_MODEL_NAME,
         type: 'smart',
-        goodFor: [
-            'Complex data analysis',
-            'Advanced debugging',
-            'Statistical analysis and modeling',
-            'Multi-step data workflows'
-        ],
+        goodFor: [...GOOD_FOR_SMART],
         provider: 'Anthropic',
         tokenLimit: '1M',
         speed: 'Medium',
         complexityHandling: 'High'
     },
     {
-        displayName: 'Gemini 2.5 Pro',
-        fullName: 'gemini-2.5-pro',
-        type: 'smart',
-        goodFor: [
-            'Complex data analysis',
-            'Advanced debugging',
-            'Statistical analysis and modeling',
-            'Multi-step data workflows'
-        ],
+        displayName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.GEMINI_3_FLASH_DISPLAY_NAME,
+        fullName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.GEMINI_3_FLASH_MODEL_NAME,
+        type: 'fast',
+        goodFor: [...GOOD_FOR_FAST],
         provider: 'Google',
         tokenLimit: '1M',
-        speed: 'Medium',
-        complexityHandling: 'High'
+        speed: 'Fast',
+        complexityHandling: 'Medium'
     },
     {
-        displayName: 'Gemini 3 Pro',
-        fullName: 'gemini-3-pro-preview',
+        displayName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.GEMINI_3_PRO_DISPLAY_NAME,
+        fullName: _utils_models__WEBPACK_IMPORTED_MODULE_3__.GEMINI_3_PRO_MODEL_NAME,
         type: 'smart',
-        goodFor: [
-            'Most complex data analysis',
-            'Advanced debugging',
-            'Statistical analysis and modeling',
-            'Multi-step data workflows'
-        ],
+        goodFor: [...GOOD_FOR_SMART],
         provider: 'Google',
         tokenLimit: '1M',
         speed: 'Slow',
@@ -10697,10 +11184,10 @@ const NextStepsPills = ({ nextSteps, onSelectNextStep, displayedNextStepsIfAvail
 
 /***/ }),
 
-/***/ "./lib/components/SelectedContextContainer.js":
-/*!****************************************************!*\
-  !*** ./lib/components/SelectedContextContainer.js ***!
-  \****************************************************/
+/***/ "./lib/components/RunCellButton.js":
+/*!*****************************************!*\
+  !*** ./lib/components/RunCellButton.js ***!
+  \*****************************************/
 /***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
 
 __webpack_require__.r(__webpack_exports__);
@@ -10709,12 +11196,19 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */ });
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
-/* harmony import */ var _style_SelectedContextContainer_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../style/SelectedContextContainer.css */ "./style/SelectedContextContainer.css");
-/* harmony import */ var _icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../icons/RuleIcon */ "./lib/icons/RuleIcon.js");
-/* harmony import */ var _icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../icons/CodeIcon */ "./lib/icons/CodeIcon.js");
-/* harmony import */ var _icons_DatabaseIcon__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../icons/DatabaseIcon */ "./lib/icons/DatabaseIcon.js");
-/* harmony import */ var _icons_PhotoIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../icons/PhotoIcon */ "./lib/icons/PhotoIcon.js");
-/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! @jupyterlab/notebook */ "webpack/sharing/consume/default/@jupyterlab/notebook");
+/* harmony import */ var _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_1__);
+/* harmony import */ var _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! @jupyterlab/apputils */ "webpack/sharing/consume/default/@jupyterlab/apputils");
+/* harmony import */ var _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_2__);
+/* harmony import */ var _icons_ChevronIcon__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../icons/ChevronIcon */ "./lib/icons/ChevronIcon.js");
+/* harmony import */ var _icons_RunAllIcon__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../icons/RunAllIcon */ "./lib/icons/RunAllIcon.js");
+/* harmony import */ var _icons_RestartAndRunIcon__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../icons/RestartAndRunIcon */ "./lib/icons/RestartAndRunIcon.js");
+/* harmony import */ var _icons_SimplePlayIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../icons/SimplePlayIcon */ "./lib/icons/SimplePlayIcon.js");
+/* harmony import */ var _icons_RestartIcon__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../icons/RestartIcon */ "./lib/icons/RestartIcon.js");
+/* harmony import */ var _icons_StopIcon__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../icons/StopIcon */ "./lib/icons/StopIcon.js");
+/* harmony import */ var _icons_ClearIcon__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../icons/ClearIcon */ "./lib/icons/ClearIcon.js");
+/* harmony import */ var _LoadingCircle__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ./LoadingCircle */ "./lib/components/LoadingCircle.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../utils/classNames */ "./lib/utils/classNames.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -10726,23 +11220,298 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const SelectedContextContainer = ({ title, type, onRemove, onClick, notebookTracker, activeCellID }) => {
-    const [isHovered, setIsHovered] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
-    let icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__["default"], null);
-    if (type.startsWith('image/')) {
-        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_PhotoIcon__WEBPACK_IMPORTED_MODULE_3__["default"], null);
-    }
-    else if (type === 'rule') {
-        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__["default"], null);
-    }
-    else if (type === 'variable') {
-        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null);
-    }
-    else if (type === 'db') {
-        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_DatabaseIcon__WEBPACK_IMPORTED_MODULE_5__["default"], null);
-    }
-    else if (type === 'active_cell') {
-        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null);
+
+
+
+
+
+const RunCellButton = ({ notebookPanel }) => {
+    const handleRunCurrentCell = () => {
+        var _a;
+        const notebook = notebookPanel.content;
+        const sessionContext = (_a = notebookPanel.context) === null || _a === void 0 ? void 0 : _a.sessionContext;
+        void _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_1__.NotebookActions.run(notebook, sessionContext);
+    };
+    const handleRunAllCells = () => {
+        var _a;
+        const notebook = notebookPanel.content;
+        const sessionContext = (_a = notebookPanel.context) === null || _a === void 0 ? void 0 : _a.sessionContext;
+        void _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_1__.NotebookActions.runAll(notebook, sessionContext);
+    };
+    const handleRestart = async () => {
+        var _a;
+        const sessionContext = (_a = notebookPanel.context) === null || _a === void 0 ? void 0 : _a.sessionContext;
+        if (!sessionContext) {
+            return;
+        }
+        // Use SessionContextDialogs.restart() which handles the restart dialog
+        // and waits for the kernel to be ready, matching Jupyter Lab core behavior
+        const dialogs = new _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_2__.SessionContextDialogs();
+        await dialogs.restart(sessionContext);
+    };
+    const handleRestartAndRunAll = async () => {
+        var _a;
+        const sessionContext = (_a = notebookPanel.context) === null || _a === void 0 ? void 0 : _a.sessionContext;
+        if (!sessionContext) {
+            return;
+        }
+        // Use SessionContextDialogs.restart() which handles the restart dialog
+        // and waits for the kernel to be ready, matching Jupyter Lab core behavior
+        const dialogs = new _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_2__.SessionContextDialogs();
+        const restarted = await dialogs.restart(sessionContext);
+        if (restarted) {
+            const notebook = notebookPanel.content;
+            void _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_1__.NotebookActions.runAll(notebook, sessionContext);
+        }
+    };
+    const handleStop = () => {
+        var _a, _b;
+        const sessionContext = (_a = notebookPanel.context) === null || _a === void 0 ? void 0 : _a.sessionContext;
+        const kernel = (_b = sessionContext === null || sessionContext === void 0 ? void 0 : sessionContext.session) === null || _b === void 0 ? void 0 : _b.kernel;
+        if (kernel) {
+            void kernel.interrupt();
+        }
+    };
+    const handleClearAllOutputs = () => {
+        const notebook = notebookPanel.content;
+        _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_1__.NotebookActions.clearAllOutputs(notebook);
+    };
+    const [isDropdownOpen, setIsDropdownOpen] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    const [isRunning, setIsRunning] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    const dropdownRef = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(null);
+    const executionCountRef = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(0);
+    // Track execution state for this specific notebook panel
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        var _a;
+        const sessionContext = (_a = notebookPanel.context) === null || _a === void 0 ? void 0 : _a.sessionContext;
+        if (!sessionContext) {
+            setIsRunning(false);
+            return;
+        }
+        // Listen to iopub messages to track execution
+        const handleIOPubMessage = (sender, msg) => {
+            const msgType = msg.header.msg_type;
+            // When a cell starts executing
+            if (msgType === 'execute_input') {
+                executionCountRef.current += 1;
+                setIsRunning(true);
+                setIsDropdownOpen(false); // Close dropdown when execution starts
+            }
+            // When a cell finishes executing
+            if (msgType === 'execute_reply') {
+                executionCountRef.current = Math.max(0, executionCountRef.current - 1);
+                // Only set to not running if no cells are executing
+                if (executionCountRef.current === 0) {
+                    setIsRunning(false);
+                }
+            }
+        };
+        // Listen to kernel status changes - this is the primary way to detect completion
+        const handleStatusChange = (sender, status) => {
+            // When kernel becomes idle, execution has finished - reset to "Run all" state
+            if (status === 'idle') {
+                executionCountRef.current = 0;
+                setIsRunning(false);
+            }
+        };
+        // Handle kernel disconnection
+        const handleKernelChange = () => {
+            var _a;
+            const kernel = (_a = sessionContext.session) === null || _a === void 0 ? void 0 : _a.kernel;
+            if (!kernel) {
+                executionCountRef.current = 0;
+                setIsRunning(false);
+            }
+        };
+        sessionContext.iopubMessage.connect(handleIOPubMessage);
+        sessionContext.statusChanged.connect(handleStatusChange);
+        sessionContext.kernelChanged.connect(handleKernelChange);
+        return () => {
+            sessionContext.iopubMessage.disconnect(handleIOPubMessage);
+            sessionContext.statusChanged.disconnect(handleStatusChange);
+            sessionContext.kernelChanged.disconnect(handleKernelChange);
+        };
+    }, [notebookPanel]);
+    // Close dropdown when clicking outside
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        const handleClickOutside = (event) => {
+            if (dropdownRef.current &&
+                !dropdownRef.current.contains(event.target)) {
+                setIsDropdownOpen(false);
+            }
+        };
+        if (isDropdownOpen) {
+            document.addEventListener('mousedown', handleClickOutside);
+            return () => document.removeEventListener('mousedown', handleClickOutside);
+        }
+        return undefined;
+    }, [isDropdownOpen]);
+    const handleMainButtonClick = () => {
+        // When running, clicking opens the dropdown (so user can Stop)
+        if (isRunning) {
+            setIsDropdownOpen(!isDropdownOpen);
+            return;
+        }
+        // Otherwise, run the current cell
+        handleRunCurrentCell();
+    };
+    const handleDropdownButtonClick = () => {
+        setIsDropdownOpen(!isDropdownOpen);
+    };
+    const menuSections = [
+        {
+            title: 'Run Code',
+            items: [
+                {
+                    label: 'Run Current Cell',
+                    icon: _icons_SimplePlayIcon__WEBPACK_IMPORTED_MODULE_3__["default"],
+                    shortcut: 'Shift+Enter',
+                    tooltip: 'Run the currently selected cell',
+                    onClick: () => {
+                        handleRunCurrentCell();
+                        setIsDropdownOpen(false);
+                    }
+                },
+                {
+                    label: 'Run All Cells',
+                    icon: _icons_RunAllIcon__WEBPACK_IMPORTED_MODULE_4__["default"],
+                    tooltip: 'Run all cells in the notebook from top to bottom',
+                    onClick: () => {
+                        handleRunAllCells();
+                        setIsDropdownOpen(false);
+                    }
+                },
+                {
+                    label: 'Restart and Run All',
+                    icon: _icons_RestartAndRunIcon__WEBPACK_IMPORTED_MODULE_5__["default"],
+                    tooltip: 'Restart the kernel to clear all variables and state, and then run all cells in the notebook',
+                    onClick: () => {
+                        void handleRestartAndRunAll();
+                        setIsDropdownOpen(false);
+                    }
+                }
+            ]
+        },
+        {
+            title: 'Kernel',
+            items: [
+                {
+                    label: 'Restart',
+                    icon: _icons_RestartIcon__WEBPACK_IMPORTED_MODULE_6__["default"],
+                    shortcut: '0, 0',
+                    tooltip: 'Restart the kernel, clearing all variables and state',
+                    onClick: () => {
+                        void handleRestart();
+                        setIsDropdownOpen(false);
+                    }
+                },
+                {
+                    label: 'Stop',
+                    icon: _icons_StopIcon__WEBPACK_IMPORTED_MODULE_7__["default"],
+                    shortcut: 'I, I',
+                    tooltip: 'Interrupt the kernel to stop the currently running cell',
+                    onClick: () => {
+                        handleStop();
+                        setIsDropdownOpen(false);
+                    }
+                }
+            ]
+        },
+        {
+            title: 'Notebook',
+            items: [
+                {
+                    label: 'Clear All Outputs',
+                    icon: _icons_ClearIcon__WEBPACK_IMPORTED_MODULE_8__["default"],
+                    tooltip: 'Clear all cell outputs in the notebook',
+                    onClick: () => {
+                        handleClearAllOutputs();
+                        setIsDropdownOpen(false);
+                    }
+                }
+            ]
+        }
+    ];
+    const trigger = (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "mito-run-cell-button-container", ref: dropdownRef },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_9__.classNames)('mito-run-cell-button-group', { 'mito-run-cell-button-running': isRunning }) },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "mito-run-cell-button mito-run-cell-button-main", onClick: handleMainButtonClick, title: isRunning ? "Running Cells" : "Run Active Cell" }, isRunning ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null,
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_LoadingCircle__WEBPACK_IMPORTED_MODULE_10__["default"], null),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "mito-run-cell-button-text" }, "Running Cells"))) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null,
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_SimplePlayIcon__WEBPACK_IMPORTED_MODULE_3__["default"], null),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "mito-run-cell-button-text" }, "Run Active Cell")))),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "mito-run-cell-button-divider" }),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "mito-run-cell-button mito-run-cell-button-dropdown", onClick: handleDropdownButtonClick, title: "More actions" },
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_ChevronIcon__WEBPACK_IMPORTED_MODULE_11__["default"], { direction: "down" }))),
+        isDropdownOpen && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "mito-run-cell-dropdown-menu" }, menuSections.map((section, sectionIndex) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { key: sectionIndex, className: "mito-run-cell-dropdown-section" },
+            sectionIndex > 0 && react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "mito-run-cell-dropdown-separator" }),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "mito-run-cell-dropdown-section-header" }, section.title),
+            section.items.map((item, itemIndex) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { key: itemIndex, className: "mito-run-cell-dropdown-item", title: item.tooltip, onClick: (e) => {
+                    e.preventDefault();
+                    e.stopPropagation();
+                    item.onClick();
+                } },
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "mito-run-cell-dropdown-item-icon" }, item.icon && react__WEBPACK_IMPORTED_MODULE_0___default().createElement(item.icon)),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "mito-run-cell-dropdown-item-label" }, item.label),
+                item.shortcut && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "mito-run-cell-dropdown-item-shortcut" }, item.shortcut))))))))))));
+    return trigger;
+};
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (RunCellButton);
+
+
+/***/ }),
+
+/***/ "./lib/components/SelectedContextContainer.js":
+/*!****************************************************!*\
+  !*** ./lib/components/SelectedContextContainer.js ***!
+  \****************************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _style_SelectedContextContainer_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../style/SelectedContextContainer.css */ "./style/SelectedContextContainer.css");
+/* harmony import */ var _icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../icons/RuleIcon */ "./lib/icons/RuleIcon.js");
+/* harmony import */ var _icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../icons/CodeIcon */ "./lib/icons/CodeIcon.js");
+/* harmony import */ var _icons_DatabaseIcon__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../icons/DatabaseIcon */ "./lib/icons/DatabaseIcon.js");
+/* harmony import */ var _icons_PhotoIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../icons/PhotoIcon */ "./lib/icons/PhotoIcon.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../utils/notebook */ "./lib/utils/notebook.js");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+
+
+
+
+
+const SelectedContextContainer = ({ title, type, onRemove, onClick, notebookTracker, activeCellID, value }) => {
+    const [isHovered, setIsHovered] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    let icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__["default"], null);
+    if (type.startsWith('image/')) {
+        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_PhotoIcon__WEBPACK_IMPORTED_MODULE_3__["default"], null);
+    }
+    else if (type === 'rule') {
+        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__["default"], null);
+    }
+    else if (type === 'variable') {
+        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null);
+    }
+    else if (type === 'db') {
+        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_DatabaseIcon__WEBPACK_IMPORTED_MODULE_5__["default"], null);
+    }
+    else if (type === 'active_cell') {
+        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null);
+    }
+    else if (type === 'notebook') {
+        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null);
+    }
+    else if (type === 'cell') {
+        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null);
     }
     const handleClick = () => {
         var _a;
@@ -10762,6 +11531,16 @@ const SelectedContextContainer = ({ title, type, onRemove, onClick, notebookTrac
             }
             // If notebookTracker or activeCellID are not available, do nothing
         }
+        else if (type === 'cell' && notebookTracker && value) {
+            // Handle cell context click - scroll to the cell
+            if (notebookTracker.currentWidget) {
+                (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_6__.scrollToCell)(notebookTracker.currentWidget, value, undefined, 'center');
+            }
+            // Highlight the cell
+            setTimeout(() => {
+                (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_6__.highlightCodeCell)(notebookTracker, value);
+            }, 500);
+        }
         else if (onClick) {
             // Call the custom onClick handler for other context types
             onClick();
@@ -10771,7 +11550,7 @@ const SelectedContextContainer = ({ title, type, onRemove, onClick, notebookTrac
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: `icon`, onClick: (e) => {
                 e.stopPropagation(); // Prevent triggering the button's onClick
                 onRemove();
-            }, title: isHovered ? "Remove rule" : "Selected rule" }, isHovered && type !== 'active_cell' ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "remove-icon" }, "X")) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "icon" }, icon))),
+            }, title: isHovered ? "Remove rule" : "Selected rule" }, isHovered && type !== 'active_cell' && type !== 'notebook' ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "remove-icon" }, "X")) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "icon" }, icon))),
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "rule-name" }, title)));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (SelectedContextContainer);
@@ -10888,6 +11667,86 @@ const ToggleButton = ({ leftText, rightText, leftTooltip, rightTooltip, leftIcon
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ToggleButton);
 
 
+/***/ }),
+
+/***/ "./lib/hooks/useCellOrder.js":
+/*!***********************************!*\
+  !*** ./lib/hooks/useCellOrder.js ***!
+  \***********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   useCellOrder: () => (/* binding */ useCellOrder)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+/**
+ * Hook that tracks the current cell order in the notebook.
+ * Returns a map of cellId → cellNumber (1-indexed).
+ * Updates automatically when cells are added, removed, or reordered.
+ *
+ * @param notebookTracker - The notebook tracker to monitor
+ * @returns A map of cellId to cellNumber (1-indexed)
+ */
+const useCellOrder = (notebookTracker) => {
+    var _a, _b;
+    const [cellOrderKey, setCellOrderKey] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(0);
+    // Track current widget ID to detect notebook switches
+    const currentWidgetId = (_b = (_a = notebookTracker.currentWidget) === null || _a === void 0 ? void 0 : _a.id) !== null && _b !== void 0 ? _b : null;
+    // Compute the cell order mapping
+    const cellOrder = (0,react__WEBPACK_IMPORTED_MODULE_0__.useMemo)(() => {
+        const orderMap = new Map();
+        const notebookPanel = notebookTracker.currentWidget;
+        if (!notebookPanel) {
+            return orderMap;
+        }
+        const notebook = notebookPanel.content;
+        notebook.widgets.forEach((cell, index) => {
+            // 1-indexed cell numbers for display
+            orderMap.set(cell.model.id, index + 1);
+        });
+        return orderMap;
+    }, [notebookTracker, cellOrderKey, currentWidgetId]);
+    // Listen to cell changes to trigger re-computation
+    // Include currentWidgetId in dependencies so listener re-attaches when switching notebooks
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        var _a;
+        const notebookPanel = notebookTracker.currentWidget;
+        if (!notebookPanel) {
+            return;
+        }
+        const notebook = notebookPanel.content;
+        // Update when cells are added, removed, or reordered
+        const handleCellChange = () => {
+            setCellOrderKey(prev => prev + 1);
+        };
+        // Listen to cell model changes (fires when cells are added, removed, or reordered)
+        (_a = notebook.model) === null || _a === void 0 ? void 0 : _a.cells.changed.connect(handleCellChange);
+        return () => {
+            var _a;
+            (_a = notebook.model) === null || _a === void 0 ? void 0 : _a.cells.changed.disconnect(handleCellChange);
+        };
+    }, [notebookTracker, currentWidgetId]);
+    // Also update when the current notebook changes
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        const handleNotebookChange = () => {
+            setCellOrderKey(prev => prev + 1);
+        };
+        notebookTracker.currentChanged.connect(handleNotebookChange);
+        return () => {
+            notebookTracker.currentChanged.disconnect(handleNotebookChange);
+        };
+    }, [notebookTracker]);
+    return cellOrder;
+};
+
+
 /***/ }),
 
 /***/ "./lib/hooks/useDebouncedFunction.js":
@@ -11036,6 +11895,71 @@ const BrainIcon = ({ height = 14, width = 14, fill = 'currentColor' }) => (react
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (BrainIcon);
 
 
+/***/ }),
+
+/***/ "./lib/icons/ChevronIcon.js":
+/*!**********************************!*\
+  !*** ./lib/icons/ChevronIcon.js ***!
+  \**********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const ChevronIcon = ({ direction = 'down', className = '', width = '16', height = '16' }) => {
+    // Calculate rotation based on direction
+    const getRotation = () => {
+        switch (direction) {
+            case 'up':
+                return 'rotate(180 8 8)';
+            case 'right':
+                return 'rotate(-90 8 8)';
+            case 'left':
+                return 'rotate(90 8 8)';
+            case 'down':
+            default:
+                return 'rotate(0 8 8)';
+        }
+    };
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { className: className, width: width, height: height, viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M4 6L8 10L12 6", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round", transform: getRotation() })));
+};
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ChevronIcon);
+
+
+/***/ }),
+
+/***/ "./lib/icons/ClearIcon.js":
+/*!********************************!*\
+  !*** ./lib/icons/ClearIcon.js ***!
+  \********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const ClearIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "16", height: "16", viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("circle", { cx: "8", cy: "8", r: "6", stroke: "currentColor", strokeWidth: "1.5", fill: "none" }),
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M5 5L11 11M11 5L5 11", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round" })));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ClearIcon);
+
+
 /***/ }),
 
 /***/ "./lib/icons/CodeIcon.js":
@@ -11163,6 +12087,30 @@ const ExpandIcon = ({ isExpanded }) => (react__WEBPACK_IMPORTED_MODULE_0___defau
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ExpandIcon);
 
 
+/***/ }),
+
+/***/ "./lib/icons/GoToCellIcon.js":
+/*!***********************************!*\
+  !*** ./lib/icons/GoToCellIcon.js ***!
+  \***********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const GoToCellIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "16", height: "16", viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M3 8H11M11 8L8 5M11 8L8 11M13 3V13", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round" })));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (GoToCellIcon);
+
+
 /***/ }),
 
 /***/ "./lib/icons/LightningIcon.js":
@@ -11302,6 +12250,35 @@ const CodeIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createEleme
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (CodeIcon);
 
 
+/***/ }),
+
+/***/ "./lib/icons/NotebookFooter/SpreadsheetIcon.js":
+/*!*****************************************************!*\
+  !*** ./lib/icons/NotebookFooter/SpreadsheetIcon.js ***!
+  \*****************************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const SpreadsheetIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "16", height: "12", viewBox: "0 0 16 12", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: 'currentColor' },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { fillRule: "evenodd", clipRule: "evenodd", d: "M15.5 9.75V1.75C15.5 0.784 14.717 0 13.75 0H1.75C0.783 0 0 0.784 0 1.75V9.75C0 10.716 0.783 11.5 1.75 11.5H13.75C14.717 11.5 15.5 10.716 15.5 9.75ZM14 1.75V9.75C14 9.888 13.888 10 13.75 10H1.75C1.612 10 1.5 9.888 1.5 9.75V1.75C1.5 1.612 1.612 1.5 1.75 1.5H13.75C13.888 1.5 14 1.612 14 1.75Z" }),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { fillRule: "evenodd", clipRule: "evenodd", d: "M0.75 8.25H14.75C15.164 8.25 15.5 7.914 15.5 7.5C15.5 7.086 15.164 6.75 14.75 6.75H0.75C0.336 6.75 0 7.086 0 7.5C0 7.914 0.336 8.25 0.75 8.25Z" }),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { fillRule: "evenodd", clipRule: "evenodd", d: "M9.5 0.75V10.75C9.5 11.164 9.836 11.5 10.25 11.5C10.664 11.5 11 11.164 11 10.75V0.75C11 0.336 10.664 0 10.25 0C9.836 0 9.5 0.336 9.5 0.75Z" }),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { fillRule: "evenodd", clipRule: "evenodd", d: "M0.75 4.75H14.75C15.164 4.75 15.5 4.414 15.5 4C15.5 3.586 15.164 3.25 14.75 3.25H0.75C0.336 3.25 0 3.586 0 4C0 4.414 0.336 4.75 0.75 4.75Z" }),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { fillRule: "evenodd", clipRule: "evenodd", d: "M4.5 0.75V10.75C4.5 11.164 4.836 11.5 5.25 11.5C5.664 11.5 6 11.164 6 10.75V0.75C6 0.336 5.664 0 5.25 0C4.836 0 4.5 0.336 4.5 0.75Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (SpreadsheetIcon);
+
+
 /***/ }),
 
 /***/ "./lib/icons/NotebookFooter/TextIcon.js":
@@ -11397,18 +12374,145 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
-const PencilIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "14", height: "14", viewBox: "0 0 10 10", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
-    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: 'currentColor' },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M9.88308 1.82456C9.68326 1.49798 9.43085 1.18192 9.1259 0.876387C8.8209 0.570854 8.5054 0.317998 8.18991 0.117862C7.88492 -0.0717913 7.48531 -0.0296413 7.22238 0.223215L1.15425 6.29155C1.12269 6.32317 1.10167 6.35476 1.08065 6.39691L0.0289843 9.38893C-0.0341088 9.55749 0.00796491 9.73662 0.134151 9.86302C0.218298 9.94732 0.333977 10 0.449651 10C0.502235 10 0.554818 9.98947 0.596894 9.97894L3.58363 8.92541C3.6257 8.91488 3.65723 8.88326 3.6888 8.85168L9.76744 2.78325C10.0304 2.51987 10.0724 2.13008 9.88311 1.82454L9.88308 1.82456ZM1.56439 6.80777C1.86938 6.98685 2.18488 7.2397 2.47934 7.53471C2.77382 7.82972 3.01568 8.14578 3.19449 8.45128L1.46976 9.06235C1.41717 8.957 1.34357 8.85164 1.25942 8.76734C1.16476 8.67252 1.0701 8.60931 0.964937 8.55664L1.56439 6.80777Z" }))));
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (PencilIcon);
+const PencilIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "14", height: "14", viewBox: "0 0 10 10", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: 'currentColor' },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M9.88308 1.82456C9.68326 1.49798 9.43085 1.18192 9.1259 0.876387C8.8209 0.570854 8.5054 0.317998 8.18991 0.117862C7.88492 -0.0717913 7.48531 -0.0296413 7.22238 0.223215L1.15425 6.29155C1.12269 6.32317 1.10167 6.35476 1.08065 6.39691L0.0289843 9.38893C-0.0341088 9.55749 0.00796491 9.73662 0.134151 9.86302C0.218298 9.94732 0.333977 10 0.449651 10C0.502235 10 0.554818 9.98947 0.596894 9.97894L3.58363 8.92541C3.6257 8.91488 3.65723 8.88326 3.6888 8.85168L9.76744 2.78325C10.0304 2.51987 10.0724 2.13008 9.88311 1.82454L9.88308 1.82456ZM1.56439 6.80777C1.86938 6.98685 2.18488 7.2397 2.47934 7.53471C2.77382 7.82972 3.01568 8.14578 3.19449 8.45128L1.46976 9.06235C1.41717 8.957 1.34357 8.85164 1.25942 8.76734C1.16476 8.67252 1.0701 8.60931 0.964937 8.55664L1.56439 6.80777Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (PencilIcon);
+
+
+/***/ }),
+
+/***/ "./lib/icons/PhotoIcon.js":
+/*!********************************!*\
+  !*** ./lib/icons/PhotoIcon.js ***!
+  \********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const PhotoIcon = () => {
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "14", height: "14", viewBox: "0 0 98 76", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { fillRule: "evenodd", clipRule: "evenodd", d: "M14.85 0.129997H83.15C91.32 0.129997 98 6.82 98 14.98V61.02C98 69.18 91.32 75.87 83.15 75.87H14.85C6.68 75.87 0 69.18 0 61.02V14.98C0 6.82 6.68 0.129997 14.85 0.129997ZM70.01 11.03C74.21 11.03 77.61 14.43 77.61 18.63C77.61 22.83 74.21 26.24 70.01 26.24C65.81 26.24 62.4 22.83 62.4 18.63C62.4 14.43 65.81 11.03 70.01 11.03ZM21.03 52.61L38.48 24.28C39.94 22.6 43.25 22.6 44.71 24.28L56.04 42.66L60.53 35.37C61.57 34.17 63.92 34.17 64.96 35.37L77.35 55.49C79.69 59.29 77.22 63.95 73.08 63.95C57.73 63.95 42.38 63.95 27.03 63.95C21.14 63.95 17.73 57.96 21.03 52.61Z" }))));
+};
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (PhotoIcon);
+
+
+/***/ }),
+
+/***/ "./lib/icons/PlayButtonIcon.js":
+/*!*************************************!*\
+  !*** ./lib/icons/PlayButtonIcon.js ***!
+  \*************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const PlayButtonIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "1em", height: "1em", viewBox: "0 0 10 11", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { fillRule: "evenodd", clipRule: "evenodd", d: "M5 0.5C6.38068 0.5 7.63072 1.05968 8.53558 1.9646C9.44034 2.86935 10 4.11938 10 5.50005C10 6.88071 9.44031 8.13075 8.53558 9.03559C7.63071 9.94034 6.38058 10.5 5 10.5C3.61942 10.5 2.36928 9.94032 1.46442 9.03559C0.559662 8.13073 0 6.88062 0 5.50005C0 4.11948 0.55969 2.86935 1.46461 1.9645C2.36937 1.05975 3.61942 0.500096 5.0001 0.500096L5 0.5ZM7.31017 5.70697C7.34646 5.68679 7.37783 5.65673 7.40006 5.61816C7.46534 5.50465 7.42621 5.35975 7.31271 5.29447L3.89591 3.32169C3.85701 3.29401 3.80938 3.27775 3.75805 3.27775C3.62653 3.27775 3.51992 3.38435 3.51992 3.51588L3.51988 5.50003V7.48417H3.52085C3.52085 7.52427 3.53099 7.56497 3.55237 7.60234C3.61764 7.71585 3.76256 7.75501 3.87606 7.6897L5.59438 6.69758L7.31017 5.70697Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (PlayButtonIcon);
+
+
+/***/ }),
+
+/***/ "./lib/icons/RejectIcon.js":
+/*!*********************************!*\
+  !*** ./lib/icons/RejectIcon.js ***!
+  \*********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const RejectIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "1em", height: "1em", viewBox: "0 0 21 20", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M10.2245 0C7.57229 0 5.02869 1.0536 3.15349 2.929C1.27817 4.8044 0.224487 7.348 0.224487 10C0.224487 12.652 1.27809 15.1958 3.15349 17.071C5.02889 18.9463 7.57249 20 10.2245 20C12.8765 20 15.4203 18.9464 17.2955 17.071C19.1708 15.1956 20.2245 12.652 20.2245 10C20.2245 8.2446 19.7624 6.5202 18.8847 5C18.0071 3.47984 16.7447 2.2174 15.2245 1.3398C13.7044 0.46206 11.9799 3.9978e-05 10.2245 3.9978e-05L10.2245 0ZM13.9315 12.293C14.1244 12.4794 14.2344 12.7354 14.2367 13.0036C14.2391 13.2718 14.1335 13.5298 13.9438 13.7194C13.7542 13.9091 13.4962 14.0146 13.228 14.0123C12.9598 14.01 12.7038 13.9 12.5174 13.7071L10.2244 11.4141L7.93143 13.7071C7.67737 13.9524 7.31283 14.0455 6.97229 13.9521C6.63175 13.8588 6.36565 13.5927 6.27229 13.2521C6.17893 12.9116 6.27205 12.5471 6.51737 12.293L8.81037 10L6.51737 7.707C6.27205 7.45294 6.17893 7.0884 6.27229 6.74786C6.36565 6.40732 6.63175 6.14122 6.97229 6.04786C7.31283 5.9545 7.67737 6.04763 7.93143 6.29294L10.2244 8.58594L12.5174 6.29294V6.29286C12.7715 6.04754 13.136 5.95442 13.4766 6.04778C13.8171 6.14114 14.0832 6.40724 14.1766 6.74778C14.2699 7.08832 14.1768 7.45286 13.9315 7.70692L11.6385 9.99992L13.9315 12.293Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (RejectIcon);
+
+
+/***/ }),
+
+/***/ "./lib/icons/RestartAndRunIcon.js":
+/*!****************************************!*\
+  !*** ./lib/icons/RestartAndRunIcon.js ***!
+  \****************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const RestartAndRunIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "16", height: "16", viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("rect", { x: "3", y: "3", width: "10", height: "10", stroke: "currentColor", strokeWidth: "1.5", fill: "none", rx: "1" }),
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M6 6L8 8L6 10M9 6L11 8L9 10", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round", fill: "none" })));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (RestartAndRunIcon);
+
+
+/***/ }),
+
+/***/ "./lib/icons/RestartIcon.js":
+/*!**********************************!*\
+  !*** ./lib/icons/RestartIcon.js ***!
+  \**********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const RestartIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "16", height: "16", viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M8 3C10.7614 3 13 5.23858 13 8C13 10.7614 10.7614 13 8 13", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", fill: "none" }),
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M6 2L4 4L6 6", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round", fill: "none" })));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (RestartIcon);
 
 
 /***/ }),
 
-/***/ "./lib/icons/PhotoIcon.js":
-/*!********************************!*\
-  !*** ./lib/icons/PhotoIcon.js ***!
-  \********************************/
+/***/ "./lib/icons/RuleIcon.js":
+/*!*******************************!*\
+  !*** ./lib/icons/RuleIcon.js ***!
+  \*******************************/
 /***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
 
 __webpack_require__.r(__webpack_exports__);
@@ -11422,20 +12526,20 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
-const PhotoIcon = () => {
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "14", height: "14", viewBox: "0 0 98 76", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+const RuleIcon = () => {
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "9", height: "12", viewBox: "0 0 9 12", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { fillRule: "evenodd", clipRule: "evenodd", d: "M14.85 0.129997H83.15C91.32 0.129997 98 6.82 98 14.98V61.02C98 69.18 91.32 75.87 83.15 75.87H14.85C6.68 75.87 0 69.18 0 61.02V14.98C0 6.82 6.68 0.129997 14.85 0.129997ZM70.01 11.03C74.21 11.03 77.61 14.43 77.61 18.63C77.61 22.83 74.21 26.24 70.01 26.24C65.81 26.24 62.4 22.83 62.4 18.63C62.4 14.43 65.81 11.03 70.01 11.03ZM21.03 52.61L38.48 24.28C39.94 22.6 43.25 22.6 44.71 24.28L56.04 42.66L60.53 35.37C61.57 34.17 63.92 34.17 64.96 35.37L77.35 55.49C79.69 59.29 77.22 63.95 73.08 63.95C57.73 63.95 42.38 63.95 27.03 63.95C21.14 63.95 17.73 57.96 21.03 52.61Z" }))));
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M1.22727 0C0.554444 0 0 0.580846 0 1.28571V10.7143C0 11.4192 0.554444 12 1.22727 12H7.77273C8.44556 12 9 11.4192 9 10.7143V3.42857H6.95455C6.28172 3.42857 5.72727 2.84773 5.72727 2.14286V0H1.22727ZM6.54545 0V2.14286C6.54545 2.38768 6.72085 2.57143 6.95455 2.57143H9C9 2.45777 8.95688 2.3488 8.88013 2.26843L6.83468 0.125577C6.75797 0.0451748 6.65396 0 6.54545 0ZM2.45455 4.71429H6.54545C6.77139 4.71429 6.95455 4.90616 6.95455 5.14286C6.95455 5.37955 6.77139 5.57143 6.54545 5.57143H2.45455C2.22861 5.57143 2.04545 5.37955 2.04545 5.14286C2.04545 4.90616 2.22861 4.71429 2.45455 4.71429ZM2.45455 6.85714H6.54545C6.77139 6.85714 6.95455 7.04902 6.95455 7.28571C6.95455 7.52241 6.77139 7.71429 6.54545 7.71429H2.45455C2.22861 7.71429 2.04545 7.52241 2.04545 7.28571C2.04545 7.04902 2.22861 6.85714 2.45455 6.85714ZM2.45455 9H6.54545C6.77139 9 6.95455 9.19187 6.95455 9.42857C6.95455 9.66527 6.77139 9.85714 6.54545 9.85714H2.45455C2.22861 9.85714 2.04545 9.66527 2.04545 9.42857C2.04545 9.19187 2.22861 9 2.45455 9Z" }))));
 };
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (PhotoIcon);
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (RuleIcon);
 
 
 /***/ }),
 
-/***/ "./lib/icons/PlayButtonIcon.js":
-/*!*************************************!*\
-  !*** ./lib/icons/PlayButtonIcon.js ***!
-  \*************************************/
+/***/ "./lib/icons/RunAllIcon.js":
+/*!*********************************!*\
+  !*** ./lib/icons/RunAllIcon.js ***!
+  \*********************************/
 /***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
 
 __webpack_require__.r(__webpack_exports__);
@@ -11449,18 +12553,18 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
-const PlayButtonIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "1em", height: "1em", viewBox: "0 0 10 11", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
-    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { fillRule: "evenodd", clipRule: "evenodd", d: "M5 0.5C6.38068 0.5 7.63072 1.05968 8.53558 1.9646C9.44034 2.86935 10 4.11938 10 5.50005C10 6.88071 9.44031 8.13075 8.53558 9.03559C7.63071 9.94034 6.38058 10.5 5 10.5C3.61942 10.5 2.36928 9.94032 1.46442 9.03559C0.559662 8.13073 0 6.88062 0 5.50005C0 4.11948 0.55969 2.86935 1.46461 1.9645C2.36937 1.05975 3.61942 0.500096 5.0001 0.500096L5 0.5ZM7.31017 5.70697C7.34646 5.68679 7.37783 5.65673 7.40006 5.61816C7.46534 5.50465 7.42621 5.35975 7.31271 5.29447L3.89591 3.32169C3.85701 3.29401 3.80938 3.27775 3.75805 3.27775C3.62653 3.27775 3.51992 3.38435 3.51992 3.51588L3.51988 5.50003V7.48417H3.52085C3.52085 7.52427 3.53099 7.56497 3.55237 7.60234C3.61764 7.71585 3.76256 7.75501 3.87606 7.6897L5.59438 6.69758L7.31017 5.70697Z" }))));
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (PlayButtonIcon);
+const RunAllIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "16", height: "16", viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M3 4L7 8L3 12", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round" }),
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M8 4L12 8L8 12", stroke: "currentColor", strokeWidth: "1.5", strokeLinecap: "round", strokeLinejoin: "round" })));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (RunAllIcon);
 
 
 /***/ }),
 
-/***/ "./lib/icons/RejectIcon.js":
-/*!*********************************!*\
-  !*** ./lib/icons/RejectIcon.js ***!
-  \*********************************/
+/***/ "./lib/icons/SimplePlayIcon.js":
+/*!*************************************!*\
+  !*** ./lib/icons/SimplePlayIcon.js ***!
+  \*************************************/
 /***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
 
 __webpack_require__.r(__webpack_exports__);
@@ -11474,17 +12578,16 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
-const RejectIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "1em", height: "1em", viewBox: "0 0 21 20", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
-    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M10.2245 0C7.57229 0 5.02869 1.0536 3.15349 2.929C1.27817 4.8044 0.224487 7.348 0.224487 10C0.224487 12.652 1.27809 15.1958 3.15349 17.071C5.02889 18.9463 7.57249 20 10.2245 20C12.8765 20 15.4203 18.9464 17.2955 17.071C19.1708 15.1956 20.2245 12.652 20.2245 10C20.2245 8.2446 19.7624 6.5202 18.8847 5C18.0071 3.47984 16.7447 2.2174 15.2245 1.3398C13.7044 0.46206 11.9799 3.9978e-05 10.2245 3.9978e-05L10.2245 0ZM13.9315 12.293C14.1244 12.4794 14.2344 12.7354 14.2367 13.0036C14.2391 13.2718 14.1335 13.5298 13.9438 13.7194C13.7542 13.9091 13.4962 14.0146 13.228 14.0123C12.9598 14.01 12.7038 13.9 12.5174 13.7071L10.2244 11.4141L7.93143 13.7071C7.67737 13.9524 7.31283 14.0455 6.97229 13.9521C6.63175 13.8588 6.36565 13.5927 6.27229 13.2521C6.17893 12.9116 6.27205 12.5471 6.51737 12.293L8.81037 10L6.51737 7.707C6.27205 7.45294 6.17893 7.0884 6.27229 6.74786C6.36565 6.40732 6.63175 6.14122 6.97229 6.04786C7.31283 5.9545 7.67737 6.04763 7.93143 6.29294L10.2244 8.58594L12.5174 6.29294V6.29286C12.7715 6.04754 13.136 5.95442 13.4766 6.04778C13.8171 6.14114 14.0832 6.40724 14.1766 6.74778C14.2699 7.08832 14.1768 7.45286 13.9315 7.70692L11.6385 9.99992L13.9315 12.293Z" }))));
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (RejectIcon);
+const SimplePlayIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "16", height: "16", viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M6 4L12 8L6 12V4Z", fill: "currentColor" })));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (SimplePlayIcon);
 
 
 /***/ }),
 
-/***/ "./lib/icons/RuleIcon.js":
+/***/ "./lib/icons/StopIcon.js":
 /*!*******************************!*\
-  !*** ./lib/icons/RuleIcon.js ***!
+  !*** ./lib/icons/StopIcon.js ***!
   \*******************************/
 /***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
 
@@ -11499,12 +12602,9 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
-const RuleIcon = () => {
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "9", height: "12", viewBox: "0 0 9 12", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M1.22727 0C0.554444 0 0 0.580846 0 1.28571V10.7143C0 11.4192 0.554444 12 1.22727 12H7.77273C8.44556 12 9 11.4192 9 10.7143V3.42857H6.95455C6.28172 3.42857 5.72727 2.84773 5.72727 2.14286V0H1.22727ZM6.54545 0V2.14286C6.54545 2.38768 6.72085 2.57143 6.95455 2.57143H9C9 2.45777 8.95688 2.3488 8.88013 2.26843L6.83468 0.125577C6.75797 0.0451748 6.65396 0 6.54545 0ZM2.45455 4.71429H6.54545C6.77139 4.71429 6.95455 4.90616 6.95455 5.14286C6.95455 5.37955 6.77139 5.57143 6.54545 5.57143H2.45455C2.22861 5.57143 2.04545 5.37955 2.04545 5.14286C2.04545 4.90616 2.22861 4.71429 2.45455 4.71429ZM2.45455 6.85714H6.54545C6.77139 6.85714 6.95455 7.04902 6.95455 7.28571C6.95455 7.52241 6.77139 7.71429 6.54545 7.71429H2.45455C2.22861 7.71429 2.04545 7.52241 2.04545 7.28571C2.04545 7.04902 2.22861 6.85714 2.45455 6.85714ZM2.45455 9H6.54545C6.77139 9 6.95455 9.19187 6.95455 9.42857C6.95455 9.66527 6.77139 9.85714 6.54545 9.85714H2.45455C2.22861 9.85714 2.04545 9.66527 2.04545 9.42857C2.04545 9.19187 2.22861 9 2.45455 9Z" }))));
-};
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (RuleIcon);
+const StopIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "16", height: "16", viewBox: "0 0 16 16", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("rect", { x: "4", y: "4", width: "8", height: "8", fill: "currentColor", rx: "1" })));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (StopIcon);
 
 
 /***/ }),
@@ -11606,6 +12706,7 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _Extensions_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./Extensions/SettingsManager/SettingsManagerPlugin */ "./lib/Extensions/SettingsManager/SettingsManagerPlugin.js");
 /* harmony import */ var _Extensions_VersionCheck__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./Extensions/VersionCheck */ "./lib/Extensions/VersionCheck/index.js");
 /* harmony import */ var _Extensions_NotebookFooter__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ./Extensions/NotebookFooter */ "./lib/Extensions/NotebookFooter/index.js");
+/* harmony import */ var _Extensions_MitoThemeLight__WEBPACK_IMPORTED_MODULE_12__ = __webpack_require__(/*! ./Extensions/MitoThemeLight */ "./lib/Extensions/MitoThemeLight/index.js");
 /* harmony import */ var _Extensions_AppManager_ManageAppsPlugin__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ./Extensions/AppManager/ManageAppsPlugin */ "./lib/Extensions/AppManager/ManageAppsPlugin.js");
 /*
  * Copyright (c) Saga Inc.
@@ -11623,6 +12724,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 // This is the main entry point to the mito-ai extension. It must export all of the top level
 // extensions that we want to load.
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = ([
@@ -11637,7 +12739,8 @@ __webpack_require__.r(__webpack_exports__);
     _Extensions_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_8__["default"],
     _Extensions_VersionCheck__WEBPACK_IMPORTED_MODULE_9__.versionCheckPlugin,
     _Extensions_NotebookFooter__WEBPACK_IMPORTED_MODULE_10__["default"],
-    _Extensions_AppManager_ManageAppsPlugin__WEBPACK_IMPORTED_MODULE_11__["default"]
+    _Extensions_AppManager_ManageAppsPlugin__WEBPACK_IMPORTED_MODULE_11__["default"],
+    _Extensions_MitoThemeLight__WEBPACK_IMPORTED_MODULE_12__["default"]
 ]);
 
 
@@ -11990,7 +13093,7 @@ const acceptAndRunCellUpdate = async (cellUpdate, notebookPanel) => {
     // gets the most up to date data.
     await (0,_sleep__WEBPACK_IMPORTED_MODULE_2__.sleep)(1000);
 };
-const retryIfExecutionError = async (notebookPanel, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef) => {
+const retryIfExecutionError = async (notebookPanel, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef, setLoadingStatus) => {
     var _a;
     const cell = notebookPanel.content.activeCell;
     // Note: If you update the max retries, update the message we display on each failure
@@ -12021,7 +13124,13 @@ const retryIfExecutionError = async (notebookPanel, app, sendAgentSmartDebugMess
         if (agentResponse.type === 'cell_update') {
             const cellUpdate = agentResponse.cell_update;
             if (cellUpdate !== undefined && cellUpdate !== null) {
-                await acceptAndRunCellUpdate(cellUpdate, notebookPanel);
+                setLoadingStatus('running-code');
+                try {
+                    await acceptAndRunCellUpdate(cellUpdate, notebookPanel);
+                }
+                finally {
+                    setLoadingStatus(undefined);
+                }
             }
         }
         else if (agentResponse.type === 'run_all_cells') {
@@ -12032,7 +13141,14 @@ const retryIfExecutionError = async (notebookPanel, app, sendAgentSmartDebugMess
             }
             runAllCellsAttempts++;
             // Execute runAllCells to fix NameError issues
-            const result = await runAllCells(app, notebookPanel);
+            setLoadingStatus('running-code');
+            let result;
+            try {
+                result = await runAllCells(app, notebookPanel);
+            }
+            finally {
+                setLoadingStatus(undefined);
+            }
             if (!result.success) {
                 // If run_all_cells resulted in an error, we should continue with error handling
                 // The error will be caught in the main loop
@@ -12181,10 +13297,6 @@ const checkForBlacklistedWords = (code) => {
             pattern: /\bdrop\s+database\b/i,
             message: "This code contains an SQL DROP DATABASE command that could delete your entire database"
         },
-        {
-            pattern: /\bformat\s*\(/,
-            message: "This code contains a format command that could potentially format your drives or storage devices"
-        },
         {
             pattern: /\bsystem\s*\(/,
             message: "This code contains a system() call that could execute arbitrary system commands, which is a security risk"
@@ -12350,6 +13462,136 @@ const getCellOutputByIDInNotebook = async (notebookPanel, codeCellID) => {
 };
 
 
+/***/ }),
+
+/***/ "./lib/utils/cellReferences.js":
+/*!*************************************!*\
+  !*** ./lib/utils/cellReferences.js ***!
+  \*************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   convertCellReferencesToStableFormat: () => (/* binding */ convertCellReferencesToStableFormat),
+/* harmony export */   getAllCellReferences: () => (/* binding */ getAllCellReferences),
+/* harmony export */   getCellNumberById: () => (/* binding */ getCellNumberById),
+/* harmony export */   parseCellReferences: () => (/* binding */ parseCellReferences)
+/* harmony export */ });
+/* harmony import */ var _notebook__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ./notebook */ "./lib/utils/notebook.js");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+/**
+ * Parses cell references from a message text.
+ * Matches patterns like "@Cell 1", "@Cell 2", "@cell 5", etc.
+ *
+ * @param messageText - The message text to parse
+ * @param notebookPanel - The notebook panel to resolve cell IDs
+ * @returns Array of cell references found in the message
+ */
+function parseCellReferences(messageText, notebookPanel) {
+    const references = [];
+    if (!notebookPanel) {
+        return references;
+    }
+    // Match @CellN or @cellN (case insensitive, no space)
+    const cellReferenceRegex = /@[Cc]ell(\d+)/g;
+    let match;
+    while ((match = cellReferenceRegex.exec(messageText)) !== null) {
+        if (!match[1]) {
+            continue;
+        }
+        const cellNumber = parseInt(match[1], 10);
+        const startIndex = match.index;
+        const endIndex = startIndex + match[0].length;
+        // Convert 1-indexed cell number to 0-indexed for lookup
+        const cellIndex = cellNumber - 1;
+        const cellId = (0,_notebook__WEBPACK_IMPORTED_MODULE_0__.getCellIDByIndexInNotebookPanel)(notebookPanel, cellIndex);
+        if (cellId) {
+            references.push({
+                cellNumber,
+                cellId,
+                startIndex,
+                endIndex
+            });
+        }
+    }
+    return references;
+}
+/**
+ * Converts @Cell N references in a message to [MITO_CELL_REF:cell_id] format.
+ * This ensures cell references are stable even if cells are reordered.
+ *
+ * @param messageText - The original message text containing @Cell N references
+ * @param notebookPanel - The notebook panel to resolve cell IDs
+ * @returns The message with @Cell N replaced by [MITO_CELL_REF:cell_id]
+ */
+function convertCellReferencesToStableFormat(messageText, notebookPanel) {
+    if (!notebookPanel) {
+        return messageText;
+    }
+    // Find all @Cell N references
+    const references = parseCellReferences(messageText, notebookPanel);
+    if (references.length === 0) {
+        return messageText;
+    }
+    // Replace from end to start to preserve indices
+    let processedMessage = messageText;
+    for (let i = references.length - 1; i >= 0; i--) {
+        const ref = references[i];
+        if (!ref)
+            continue;
+        const before = processedMessage.substring(0, ref.startIndex);
+        const after = processedMessage.substring(ref.endIndex);
+        processedMessage = before + `[MITO_CELL_REF:${ref.cellId}]` + after;
+    }
+    return processedMessage;
+}
+/**
+ * Gets the current cell number for a given cell ID.
+ *
+ * @param cellId - The cell ID to look up
+ * @param notebookPanel - The notebook panel
+ * @returns The 1-indexed cell number, or undefined if not found
+ */
+function getCellNumberById(cellId, notebookPanel) {
+    if (!notebookPanel) {
+        return undefined;
+    }
+    const notebook = notebookPanel.content;
+    const cellIndex = notebook.widgets.findIndex(cell => cell.model.id === cellId);
+    if (cellIndex === -1) {
+        return undefined;
+    }
+    // Return 1-indexed cell number
+    return cellIndex + 1;
+}
+/**
+ * Gets all available cells with their numbers and IDs.
+ * Useful for populating dropdowns or autocomplete.
+ *
+ * @param notebookPanel - The notebook panel
+ * @returns Array of { cellNumber, cellId, cellType } objects
+ */
+function getAllCellReferences(notebookPanel) {
+    if (!notebookPanel) {
+        return [];
+    }
+    const notebook = notebookPanel.content;
+    const cells = [];
+    notebook.widgets.forEach((cell, index) => {
+        cells.push({
+            cellNumber: index + 1,
+            cellId: cell.model.id,
+            cellType: cell.model.type
+        });
+    });
+    return cells;
+}
+
+
 /***/ }),
 
 /***/ "./lib/utils/chatHistory.js":
@@ -13026,6 +14268,63 @@ const uploadFileToBackend = async (file, notebookTracker, onFileUploaded) => {
 };
 
 
+/***/ }),
+
+/***/ "./lib/utils/lineNumbers.js":
+/*!**********************************!*\
+  !*** ./lib/utils/lineNumbers.js ***!
+  \**********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   areLineNumbersEnabled: () => (/* binding */ areLineNumbersEnabled),
+/* harmony export */   enableLineNumbersIfNeeded: () => (/* binding */ enableLineNumbersIfNeeded)
+/* harmony export */ });
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+/**
+ * Check if line numbers are currently enabled in a notebook panel.
+ * Returns true if at least one code cell has line numbers visible.
+ */
+const areLineNumbersEnabled = (notebookPanel) => {
+    const notebook = notebookPanel.content;
+    // Check if any code cell has line numbers visible
+    for (const cell of notebook.widgets) {
+        if (cell.model.type === 'code' && cell.editor) {
+            // Check if line numbers gutter exists in the DOM
+            const editorNode = cell.node.querySelector('.jp-Editor');
+            if (editorNode) {
+                // In CodeMirror 6, line numbers are in a gutter with class 'cm-lineNumbers'
+                const lineNumbersGutter = editorNode.querySelector('.cm-lineNumbers');
+                if (lineNumbersGutter) {
+                    return true;
+                }
+            }
+        }
+    }
+    return false;
+};
+/**
+ * Enable line numbers for a notebook panel if they're not already enabled.
+ */
+const enableLineNumbersIfNeeded = async (app, notebookPanel) => {
+    // Check if line numbers are already enabled
+    if (areLineNumbersEnabled(notebookPanel)) {
+        return;
+    }
+    // If not enabled, toggle them on
+    try {
+        await app.commands.execute('notebook:toggle-all-cell-line-numbers');
+    }
+    catch (error) {
+        console.warn('Failed to enable line numbers:', error);
+    }
+};
+
+
 /***/ }),
 
 /***/ "./lib/utils/models.js":
@@ -13040,8 +14339,14 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */   CLAUDE_HAIKU_MODEL_NAME: () => (/* binding */ CLAUDE_HAIKU_MODEL_NAME),
 /* harmony export */   CLAUDE_SONNET_DISPLAY_NAME: () => (/* binding */ CLAUDE_SONNET_DISPLAY_NAME),
 /* harmony export */   CLAUDE_SONNET_MODEL_NAME: () => (/* binding */ CLAUDE_SONNET_MODEL_NAME),
+/* harmony export */   GEMINI_3_FLASH_DISPLAY_NAME: () => (/* binding */ GEMINI_3_FLASH_DISPLAY_NAME),
+/* harmony export */   GEMINI_3_FLASH_MODEL_NAME: () => (/* binding */ GEMINI_3_FLASH_MODEL_NAME),
+/* harmony export */   GEMINI_3_PRO_DISPLAY_NAME: () => (/* binding */ GEMINI_3_PRO_DISPLAY_NAME),
+/* harmony export */   GEMINI_3_PRO_MODEL_NAME: () => (/* binding */ GEMINI_3_PRO_MODEL_NAME),
 /* harmony export */   GPT_4_1_DISPLAY_NAME: () => (/* binding */ GPT_4_1_DISPLAY_NAME),
-/* harmony export */   GPT_4_1_MODEL_NAME: () => (/* binding */ GPT_4_1_MODEL_NAME)
+/* harmony export */   GPT_4_1_MODEL_NAME: () => (/* binding */ GPT_4_1_MODEL_NAME),
+/* harmony export */   GPT_5_2_DISPLAY_NAME: () => (/* binding */ GPT_5_2_DISPLAY_NAME),
+/* harmony export */   GPT_5_2_MODEL_NAME: () => (/* binding */ GPT_5_2_MODEL_NAME)
 /* harmony export */ });
 /*
  * Copyright (c) Saga Inc.
@@ -13053,6 +14358,12 @@ const CLAUDE_HAIKU_DISPLAY_NAME = 'Claude Haiku 4.5';
 const CLAUDE_HAIKU_MODEL_NAME = 'claude-haiku-4-5-20251001';
 const GPT_4_1_DISPLAY_NAME = 'GPT 4.1';
 const GPT_4_1_MODEL_NAME = 'gpt-4.1';
+const GPT_5_2_DISPLAY_NAME = 'GPT 5.2';
+const GPT_5_2_MODEL_NAME = 'gpt-5.2';
+const GEMINI_3_FLASH_DISPLAY_NAME = 'Gemini 3 Flash';
+const GEMINI_3_FLASH_MODEL_NAME = 'gemini-3-flash-preview';
+const GEMINI_3_PRO_DISPLAY_NAME = 'Gemini 3 Pro';
+const GEMINI_3_PRO_MODEL_NAME = 'gemini-3-pro-preview';
 
 
 /***/ }),
@@ -13079,6 +14390,8 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */   getCellByIDInNotebookPanel: () => (/* binding */ getCellByIDInNotebookPanel),
 /* harmony export */   getCellCodeByID: () => (/* binding */ getCellCodeByID),
 /* harmony export */   getCellCodeByIDInNotebookPanel: () => (/* binding */ getCellCodeByIDInNotebookPanel),
+/* harmony export */   getCellIDByIndex: () => (/* binding */ getCellIDByIndex),
+/* harmony export */   getCellIDByIndexInNotebookPanel: () => (/* binding */ getCellIDByIndexInNotebookPanel),
 /* harmony export */   getCellIndexByID: () => (/* binding */ getCellIndexByID),
 /* harmony export */   getCellIndexByIDInNotebookPanel: () => (/* binding */ getCellIndexByIDInNotebookPanel),
 /* harmony export */   getNotebookName: () => (/* binding */ getNotebookName),
@@ -13164,6 +14477,18 @@ const getCellIndexByIDInNotebookPanel = (notebookPanel, cellID) => {
     }
     return undefined;
 };
+const getCellIDByIndex = (notebookTracker, index) => {
+    const notebookPanel = notebookTracker.currentWidget;
+    return getCellIDByIndexInNotebookPanel(notebookPanel, index);
+};
+const getCellIDByIndexInNotebookPanel = (notebookPanel, index) => {
+    var _a, _b;
+    const cellList = (_a = notebookPanel === null || notebookPanel === void 0 ? void 0 : notebookPanel.model) === null || _a === void 0 ? void 0 : _a.cells;
+    if (cellList === undefined || index < 0 || index >= cellList.length) {
+        return undefined;
+    }
+    return (_b = cellList.get(index)) === null || _b === void 0 ? void 0 : _b.id;
+};
 const setActiveCellByID = (notebookTracker, cellID) => {
     const notebookPanel = notebookTracker.currentWidget;
     setActiveCellByIDInNotebookPanel(notebookPanel, cellID);
@@ -13655,6 +14980,52 @@ function sleep(ms) {
 }
 
 
+/***/ }),
+
+/***/ "./lib/utils/sound.js":
+/*!****************************!*\
+  !*** ./lib/utils/sound.js ***!
+  \****************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   playCompletionSound: () => (/* binding */ playCompletionSound)
+/* harmony export */ });
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+/**
+ * Plays a completion sound to notify the user that the agent has finished.
+ * Uses the Web Audio API to generate a pleasant two-tone notification sound.
+ */
+const playCompletionSound = (audioContext) => {
+    if (!audioContext) {
+        return;
+    }
+    // Create a pleasant two-note completion sound
+    const playTone = (frequency, startTime, duration) => {
+        const oscillator = audioContext.createOscillator();
+        const gainNode = audioContext.createGain();
+        oscillator.connect(gainNode);
+        gainNode.connect(audioContext.destination);
+        oscillator.frequency.value = frequency;
+        oscillator.type = 'sine';
+        // Fade in and out for a smoother sound
+        gainNode.gain.setValueAtTime(0, startTime);
+        gainNode.gain.linearRampToValueAtTime(0.3, startTime + 0.05);
+        gainNode.gain.linearRampToValueAtTime(0, startTime + duration);
+        oscillator.start(startTime);
+        oscillator.stop(startTime + duration);
+    };
+    const now = audioContext.currentTime;
+    // Play two ascending tones (C5 and E5) for a pleasant "complete" sound
+    playTone(523.25, now, 0.15); // C5
+    playTone(659.25, now + 0.12, 0.2); // E5
+};
+
+
 /***/ }),
 
 /***/ "./lib/utils/strings.js":
@@ -13919,6 +15290,7 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__);
 /* harmony import */ var semver__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! semver */ "webpack/sharing/consume/default/semver/semver");
 /* harmony import */ var semver__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(semver__WEBPACK_IMPORTED_MODULE_2__);
+/* harmony import */ var _user__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ./user */ "./lib/utils/user.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -13926,9 +15298,11 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 const VERSION_CHECK_TIMEOUT = 5000;
 const VERSION_CHECK_ENDPOINT = 'mito-ai/version-check';
-const DOCS_UPDATE_URL = 'https://docs.trymito.io/getting-started/installing-mito';
+const DESKTOP_DOCS_UPDATE_URL = 'https://docs.trymito.io/getting-started/installing-mito/upgrading-mito-desktop#how-can-i-update-mito-ai';
+const PIP_DOCS_UPDATE_URL = 'https://docs.trymito.io/how-to/upgrading-mito';
 const RELEASE_NOTES_URL = 'https://docs.trymito.io/misc/release-notes';
 /**
  * Display a notification when Mito AI is outdated
@@ -13940,7 +15314,7 @@ function showVersionOutdatedNotification(currentVersion, latestVersion) {
             {
                 label: 'Learn how to update',
                 callback: () => {
-                    window.open(DOCS_UPDATE_URL, '_blank');
+                    window.open((0,_user__WEBPACK_IMPORTED_MODULE_3__.isElectronBasedFrontend)() ? DESKTOP_DOCS_UPDATE_URL : PIP_DOCS_UPDATE_URL, '_blank');
                 }
             },
             {
@@ -14937,6 +16311,33 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
     flex-shrink: 0;
 }
 
+.agent-component-header-actions {
+    display: flex;
+    align-items: center;
+    gap: 4px;
+    flex-shrink: 0;
+}
+
+.agent-component-header-action-button {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    width: 24px;
+    height: 24px;
+    border: none;
+    background: transparent;
+    cursor: pointer;
+    border-radius: 4px;
+    color: var(--jp-ui-font-color2);
+    transition: background-color 0.15s ease, color 0.15s ease;
+    padding: 0;
+}
+
+.agent-component-header-action-button:hover {
+    background-color: var(--jp-layout-color3);
+    color: var(--jp-ui-font-color1);
+}
+
 .agent-component-header-expand-icon {
     flex-shrink: 0;
     transition: transform 0.2s ease;
@@ -14944,7 +16345,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 
 .agent-component-header-expand-icon.expanded {
     transform: rotate(90deg);
-} `, "",{"version":3,"sources":["webpack://./style/AgentComponentHeader.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,eAAe;IACf,+DAA+D;IAC/D,+BAA+B;IAC/B,aAAa;IACb,8BAA8B;IAC9B,mBAAmB;IACnB,oDAAoD;IACpD,sDAAsD;IACtD,8DAA8D;AAClE;;AAEA;IACI,yCAAyC;AAC7C;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,YAAY;IACZ,0BAA0B;IAC1B,yCAAyC;AAC7C;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,yCAAyC;AAC7C;;AAEA;IACI,aAAa;IACb,mBAAmB;IACnB,SAAS;IACT,OAAO;IACP,YAAY;AAChB;;AAEA;IACI,cAAc;AAClB;;AAEA;IACI,cAAc;IACd,+BAA+B;AACnC;;AAEA;IACI,wBAAwB;AAC5B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.agent-component-header {\n    cursor: pointer;\n    padding: 10px var(--chat-taskpane-tool-call-horizontal-padding);\n    color: var(--jp-ui-font-color2);\n    display: flex;\n    justify-content: space-between;\n    align-items: center;\n    border: 1px solid var(--jp-cell-editor-border-color);\n    border-radius: var(--chat-taskpane-item-border-radius);\n    transition: background-color 0.2s ease, border-color 0.2s ease;\n}\n\n.agent-component-header:hover {\n    background-color: var(--jp-layout-color3);\n}\n\n.agent-component-header.no-border {\n    border: none;\n}\n\n.agent-component-header.expanded {\n    border: none;\n    border-radius: 5px 5px 0 0;\n    background-color: var(--jp-layout-color2);\n}\n\n.agent-component-header.error-fixup {\n    border: none;\n}\n\n.agent-component-header.error-fixup.expanded {\n    background-color: var(--jp-layout-color2);\n}\n\n.agent-component-header-content {\n    display: flex;\n    align-items: center;\n    gap: 10px;\n    flex: 1;\n    min-width: 0;\n}\n\n.agent-component-header-content svg {\n    flex-shrink: 0;\n}\n\n.agent-component-header-expand-icon {\n    flex-shrink: 0;\n    transition: transform 0.2s ease;\n}\n\n.agent-component-header-expand-icon.expanded {\n    transform: rotate(90deg);\n} "],"sourceRoot":""}]);
+} `, "",{"version":3,"sources":["webpack://./style/AgentComponentHeader.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,eAAe;IACf,+DAA+D;IAC/D,+BAA+B;IAC/B,aAAa;IACb,8BAA8B;IAC9B,mBAAmB;IACnB,oDAAoD;IACpD,sDAAsD;IACtD,8DAA8D;AAClE;;AAEA;IACI,yCAAyC;AAC7C;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,YAAY;IACZ,0BAA0B;IAC1B,yCAAyC;AAC7C;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,yCAAyC;AAC7C;;AAEA;IACI,aAAa;IACb,mBAAmB;IACnB,SAAS;IACT,OAAO;IACP,YAAY;AAChB;;AAEA;IACI,cAAc;AAClB;;AAEA;IACI,aAAa;IACb,mBAAmB;IACnB,QAAQ;IACR,cAAc;AAClB;;AAEA;IACI,aAAa;IACb,mBAAmB;IACnB,uBAAuB;IACvB,WAAW;IACX,YAAY;IACZ,YAAY;IACZ,uBAAuB;IACvB,eAAe;IACf,kBAAkB;IAClB,+BAA+B;IAC/B,yDAAyD;IACzD,UAAU;AACd;;AAEA;IACI,yCAAyC;IACzC,+BAA+B;AACnC;;AAEA;IACI,cAAc;IACd,+BAA+B;AACnC;;AAEA;IACI,wBAAwB;AAC5B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.agent-component-header {\n    cursor: pointer;\n    padding: 10px var(--chat-taskpane-tool-call-horizontal-padding);\n    color: var(--jp-ui-font-color2);\n    display: flex;\n    justify-content: space-between;\n    align-items: center;\n    border: 1px solid var(--jp-cell-editor-border-color);\n    border-radius: var(--chat-taskpane-item-border-radius);\n    transition: background-color 0.2s ease, border-color 0.2s ease;\n}\n\n.agent-component-header:hover {\n    background-color: var(--jp-layout-color3);\n}\n\n.agent-component-header.no-border {\n    border: none;\n}\n\n.agent-component-header.expanded {\n    border: none;\n    border-radius: 5px 5px 0 0;\n    background-color: var(--jp-layout-color2);\n}\n\n.agent-component-header.error-fixup {\n    border: none;\n}\n\n.agent-component-header.error-fixup.expanded {\n    background-color: var(--jp-layout-color2);\n}\n\n.agent-component-header-content {\n    display: flex;\n    align-items: center;\n    gap: 10px;\n    flex: 1;\n    min-width: 0;\n}\n\n.agent-component-header-content svg {\n    flex-shrink: 0;\n}\n\n.agent-component-header-actions {\n    display: flex;\n    align-items: center;\n    gap: 4px;\n    flex-shrink: 0;\n}\n\n.agent-component-header-action-button {\n    display: flex;\n    align-items: center;\n    justify-content: center;\n    width: 24px;\n    height: 24px;\n    border: none;\n    background: transparent;\n    cursor: pointer;\n    border-radius: 4px;\n    color: var(--jp-ui-font-color2);\n    transition: background-color 0.15s ease, color 0.15s ease;\n    padding: 0;\n}\n\n.agent-component-header-action-button:hover {\n    background-color: var(--jp-layout-color3);\n    color: var(--jp-ui-font-color1);\n}\n\n.agent-component-header-expand-icon {\n    flex-shrink: 0;\n    transition: transform 0.2s ease;\n}\n\n.agent-component-header-expand-icon.expanded {\n    transform: rotate(90deg);\n} "],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -16037,27 +17438,87 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
     box-sizing: border-box;
 }
 
-.cta-carousel-dot.active {
-    background-color: var(--purple-500);
+.cta-carousel-dot.active {
+    background-color: var(--purple-500);
+}
+
+.cta-carousel-button {
+    margin-top: 8px;
+}
+
+.cta-logo-container {
+    margin: 0 auto 8px;
+    display: block;
+    text-align: center;
+}
+
+.cta-title {
+    display: block;
+    text-align: center;
+    font-weight: bold;
+    font-size: 20px;
+    margin-bottom: 15px;
+}`, "",{"version":3,"sources":["webpack://./style/CTACarousel.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,WAAW;AACf;;AAEA;IACI,aAAa;IACb,mBAAmB;IACnB,uBAAuB;IACvB,iBAAiB;AACrB;;AAEA;IACI,cAAc;IACd,kBAAkB;IAClB,eAAe;IACf,+BAA+B;IAC/B,mBAAmB;IACnB,OAAO;IACP,gBAAgB;IAChB,iBAAiB;IACjB,cAAc;IACd,UAAU;IACV,oCAAoC;AACxC;;AAEA;IACI,aAAa;IACb,uBAAuB;IACvB,QAAQ;IACR,eAAe;AACnB;;AAEA;IACI,WAAW;IACX,YAAY;IACZ,kBAAkB;IAClB,yCAAyC;IACzC,eAAe;IACf,sCAAsC;IACtC,sBAAsB;AAC1B;;AAEA;IACI,mCAAmC;AACvC;;AAEA;IACI,eAAe;AACnB;;AAEA;IACI,kBAAkB;IAClB,cAAc;IACd,kBAAkB;AACtB;;AAEA;IACI,cAAc;IACd,kBAAkB;IAClB,iBAAiB;IACjB,eAAe;IACf,mBAAmB;AACvB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.cta-carousel {\n    width: 100%;\n}\n\n.cta-carousel-container {\n    display: flex;\n    align-items: center;\n    justify-content: center;\n    min-height: 100px;\n}\n\n.cta-message {\n    display: block;\n    text-align: center;\n    font-size: 14px;\n    color: var(--jp-ui-font-color3);\n    margin-bottom: 15px;\n    flex: 1;\n    line-height: 1.3;\n    min-height: 2.5em;\n    padding: 0 8px;\n    opacity: 1;\n    transition: opacity 0.5s ease-in-out;\n}\n\n.cta-carousel-dots {\n    display: flex;\n    justify-content: center;\n    gap: 8px;\n    margin-top: 4px;\n}\n\n.cta-carousel-dot {\n    width: 12px;\n    height: 12px;\n    border-radius: 50%;\n    background-color: var(--jp-layout-color3);\n    cursor: pointer;\n    transition: background-color 0.3s ease;\n    box-sizing: border-box;\n}\n\n.cta-carousel-dot.active {\n    background-color: var(--purple-500);\n}\n\n.cta-carousel-button {\n    margin-top: 8px;\n}\n\n.cta-logo-container {\n    margin: 0 auto 8px;\n    display: block;\n    text-align: center;\n}\n\n.cta-title {\n    display: block;\n    text-align: center;\n    font-weight: bold;\n    font-size: 20px;\n    margin-bottom: 15px;\n}"],"sourceRoot":""}]);
+// Exports
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
+
+
+/***/ }),
+
+/***/ "./node_modules/css-loader/dist/cjs.js!./style/CellReference.css":
+/*!***********************************************************************!*\
+  !*** ./node_modules/css-loader/dist/cjs.js!./style/CellReference.css ***!
+  \***********************************************************************/
+/***/ ((module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/sourceMaps.js */ "./node_modules/css-loader/dist/runtime/sourceMaps.js");
+/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/api.js */ "./node_modules/css-loader/dist/runtime/api.js");
+/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__);
+// Imports
+
+
+var ___CSS_LOADER_EXPORT___ = _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default()((_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default()));
+// Module
+___CSS_LOADER_EXPORT___.push([module.id, `/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+/* Cell reference links in chat messages - styled as standard links */
+.cell-reference {
+    color: var(--purple-600);
+    text-decoration: underline;
+    cursor: pointer;
+    font-weight: 500;
+    padding: 1px 5px;
+    border-radius: 3px;
+    position: relative;
+    vertical-align: baseline;
+}
+
+.cell-reference:hover {
+    background-color: var(--purple-300);
+    color: var(--purple-700);
 }
 
-.cta-carousel-button {
-    margin-top: 8px;
+/* Greyed out style for missing/unresolved cell references (deleted or in different notebook) */
+.cell-reference.cell-reference-missing {
+    color: var(--jp-ui-font-color2);
+    text-decoration: none;
+    cursor: default;
+    opacity: 0.6;
 }
 
-.cta-logo-container {
-    margin: 0 auto 8px;
-    display: block;
-    text-align: center;
+.cell-reference.cell-reference-missing:hover {
+    background-color: var(--jp-layout-color3);
+    color: var(--jp-ui-font-color2);
 }
-
-.cta-title {
-    display: block;
-    text-align: center;
-    font-weight: bold;
-    font-size: 20px;
-    margin-bottom: 15px;
-}`, "",{"version":3,"sources":["webpack://./style/CTACarousel.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,WAAW;AACf;;AAEA;IACI,aAAa;IACb,mBAAmB;IACnB,uBAAuB;IACvB,iBAAiB;AACrB;;AAEA;IACI,cAAc;IACd,kBAAkB;IAClB,eAAe;IACf,+BAA+B;IAC/B,mBAAmB;IACnB,OAAO;IACP,gBAAgB;IAChB,iBAAiB;IACjB,cAAc;IACd,UAAU;IACV,oCAAoC;AACxC;;AAEA;IACI,aAAa;IACb,uBAAuB;IACvB,QAAQ;IACR,eAAe;AACnB;;AAEA;IACI,WAAW;IACX,YAAY;IACZ,kBAAkB;IAClB,yCAAyC;IACzC,eAAe;IACf,sCAAsC;IACtC,sBAAsB;AAC1B;;AAEA;IACI,mCAAmC;AACvC;;AAEA;IACI,eAAe;AACnB;;AAEA;IACI,kBAAkB;IAClB,cAAc;IACd,kBAAkB;AACtB;;AAEA;IACI,cAAc;IACd,kBAAkB;IAClB,iBAAiB;IACjB,eAAe;IACf,mBAAmB;AACvB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.cta-carousel {\n    width: 100%;\n}\n\n.cta-carousel-container {\n    display: flex;\n    align-items: center;\n    justify-content: center;\n    min-height: 100px;\n}\n\n.cta-message {\n    display: block;\n    text-align: center;\n    font-size: 14px;\n    color: var(--jp-ui-font-color3);\n    margin-bottom: 15px;\n    flex: 1;\n    line-height: 1.3;\n    min-height: 2.5em;\n    padding: 0 8px;\n    opacity: 1;\n    transition: opacity 0.5s ease-in-out;\n}\n\n.cta-carousel-dots {\n    display: flex;\n    justify-content: center;\n    gap: 8px;\n    margin-top: 4px;\n}\n\n.cta-carousel-dot {\n    width: 12px;\n    height: 12px;\n    border-radius: 50%;\n    background-color: var(--jp-layout-color3);\n    cursor: pointer;\n    transition: background-color 0.3s ease;\n    box-sizing: border-box;\n}\n\n.cta-carousel-dot.active {\n    background-color: var(--purple-500);\n}\n\n.cta-carousel-button {\n    margin-top: 8px;\n}\n\n.cta-logo-container {\n    margin: 0 auto 8px;\n    display: block;\n    text-align: center;\n}\n\n.cta-title {\n    display: block;\n    text-align: center;\n    font-weight: bold;\n    font-size: 20px;\n    margin-bottom: 15px;\n}"],"sourceRoot":""}]);
+`, "",{"version":3,"sources":["webpack://./style/CellReference.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF,qEAAqE;AACrE;IACI,wBAAwB;IACxB,0BAA0B;IAC1B,eAAe;IACf,gBAAgB;IAChB,gBAAgB;IAChB,kBAAkB;IAClB,kBAAkB;IAClB,wBAAwB;AAC5B;;AAEA;IACI,mCAAmC;IACnC,wBAAwB;AAC5B;;AAEA,+FAA+F;AAC/F;IACI,+BAA+B;IAC/B,qBAAqB;IACrB,eAAe;IACf,YAAY;AAChB;;AAEA;IACI,yCAAyC;IACzC,+BAA+B;AACnC","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n/* Cell reference links in chat messages - styled as standard links */\n.cell-reference {\n    color: var(--purple-600);\n    text-decoration: underline;\n    cursor: pointer;\n    font-weight: 500;\n    padding: 1px 5px;\n    border-radius: 3px;\n    position: relative;\n    vertical-align: baseline;\n}\n\n.cell-reference:hover {\n    background-color: var(--purple-300);\n    color: var(--purple-700);\n}\n\n/* Greyed out style for missing/unresolved cell references (deleted or in different notebook) */\n.cell-reference.cell-reference-missing {\n    color: var(--jp-ui-font-color2);\n    text-decoration: none;\n    cursor: default;\n    opacity: 0.6;\n}\n\n.cell-reference.cell-reference-missing:hover {\n    background-color: var(--jp-layout-color3);\n    color: var(--jp-ui-font-color2);\n}\n"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -16375,10 +17836,6 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   padding: 0 var(--chat-taskpane-item-indent);
 }
 
-.message:not(.chat-input) {
-  margin-bottom: 20px;
-}
-
 .message:hover .message-start-editing-button {
   opacity: 1 !important;
 }
@@ -16387,6 +17844,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   background-color: var(--chat-user-message-background-color);
   color: var(--chat-user-message-font-color);
   padding: 10px;
+  margin-bottom: 5px;
 }
 
 .message-assistant-chat {
@@ -16499,7 +17957,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 
 .message-edit-buttons button:hover {
   background-color: var(--jp-layout-color2);
-}`, "",{"version":3,"sources":["webpack://./style/ChatMessage.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,kBAAkB;EAClB,mBAAmB;EACnB,sBAAsB;EACtB,WAAW;EACX,eAAe;EACf,sDAAsD;EACtD,2CAA2C;AAC7C;;AAEA;EACE,mBAAmB;AACrB;;AAEA;EACE,qBAAqB;AACvB;;AAEA;EACE,2DAA2D;EAC3D,0CAA0C;EAC1C,aAAa;AACf;;AAEA;EACE,+CAA+C;AACjD;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,QAAQ;EACR,gBAAgB;EAChB,kBAAkB;EAClB,eAAe;AACjB;;AAEA;EACE,gBAAgB;EAChB,sBAAsB;EACtB,SAAS;EACT,gBAAgB;AAClB;;AAEA;EACE,mBAAmB;AACrB;;AAEA,qCAAqC;AACrC;EACE,kBAAkB;EAClB,WAAW;EACX,UAAU;EACV,aAAa;EACb,kBAAkB;EAClB,gBAAgB,EAAE,0DAA0D;AAC9E;;AAEA,qCAAqC;AACrC;;EAEE,eAAe;EACf,UAAU;EACV;;qCAEmC;EACnC,yCAAyC;EACzC,YAAY;EACZ,aAAa;EACb,mBAAmB;EACnB,uBAAuB;EACvB,oCAAoC;AACtC;;AAEA,gCAAgC;AAChC;EACE,WAAW;EACX,YAAY;EACZ,UAAU;EACV,gBAAgB,EAAE,oCAAoC;AACxD;;AAEA,yCAAyC;AACzC;EACE,+CAA+C;AACjD;;AAEA,8FAA8F;AAC9F;;EAEE,gBAAgB;AAClB;;AAEA,2BAA2B;AAC3B;EACE,kBAAkB;EAClB,YAAY;EACZ,kBAAkB;AACpB;;AAEA,mBAAmB;AACnB;EACE,oCAAoC;AACtC;;AAEA,iBAAiB;AACjB;;EAEE,yCAAyC;AAC3C;;AAEA,kCAAkC;AAClC;;EAEE,UAAU;AACZ;;AAEA;EACE,aAAa;EACb,QAAQ;EACR,YAAY;AACd;;AAEA;EACE,iBAAiB;EACjB,kBAAkB;EAClB,yCAAyC;EACzC,oCAAoC;EACpC,yCAAyC;EACzC,eAAe;AACjB;;AAEA;EACE,yCAAyC;AAC3C","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.message {\n  position: relative;\n  height: min-content;\n  box-sizing: border-box;\n  width: 100%;\n  font-size: 14px;\n  border-radius: var(--chat-taskpane-item-border-radius);\n  padding: 0 var(--chat-taskpane-item-indent);\n}\n\n.message:not(.chat-input) {\n  margin-bottom: 20px;\n}\n\n.message:hover .message-start-editing-button {\n  opacity: 1 !important;\n}\n\n.message-user {\n  background-color: var(--chat-user-message-background-color);\n  color: var(--chat-user-message-font-color);\n  padding: 10px;\n}\n\n.message-assistant-chat {\n  color: var(--chat-assistant-message-font-color);\n}\n\n.chat-message-buttons {\n  display: flex;\n  flex-direction: row;\n  gap: 8px;\n  margin-top: 10px;\n  margin-bottom: 4px;\n  flex-wrap: wrap;\n}\n\n.chat-taskpane-smart-debug-error-message {\n  white-space: pre;\n  font-family: monospace;\n  margin: 0;\n  overflow-x: auto;\n}\n\n.message-text {\n  align-items: center;\n}\n\n/* Message Action Buttons Container */\n.message-action-buttons {\n  position: absolute;\n  bottom: 8px;\n  right: 8px;\n  display: flex;\n  border-radius: 4px;\n  overflow: hidden; /* Ensures inner buttons don't break the rounded corners */\n}\n\n/* Common styles for action buttons */\n.message-action-buttons button,\n.message-start-editing-button {\n  cursor: pointer;\n  opacity: 0;\n  transition:\n    opacity 0.2s ease-in-out,\n    background-color 0.2s ease-in-out;\n  background-color: var(--jp-layout-color3);\n  border: none;\n  display: flex;\n  align-items: center;\n  justify-content: center;\n  color: var(--jp-content-font-color1);\n}\n\n/* Action buttons in the group */\n.message-action-buttons button {\n  width: 28px;\n  height: 28px;\n  padding: 0;\n  border-radius: 0; /* Remove individual button radius */\n}\n\n/* Add subtle separator between buttons */\n.message-action-buttons button:not(:last-child) {\n  border-right: 1px solid var(--jp-border-color1);\n}\n\n/* Remove the individual button radius styles since we're handling it at the container level */\n.message-action-buttons button:first-child,\n.message-action-buttons button:last-child {\n  border-radius: 0;\n}\n\n/* Standalone edit button */\n.message-start-editing-button {\n  position: relative;\n  padding: 4px;\n  border-radius: 4px;\n}\n\n/* SVG icon color */\n.message-action-buttons button svg {\n  color: var(--jp-content-font-color1);\n}\n\n/* Hover states */\n.message-action-buttons button:hover,\n.message-start-editing-button:hover {\n  background-color: var(--jp-layout-color4);\n}\n\n/* Show buttons on message hover */\n.message:hover .message-action-buttons button,\n.message:hover .message-start-editing-button {\n  opacity: 1;\n}\n\n.message-edit-buttons {\n  display: flex;\n  gap: 8px;\n  padding: 5px;\n}\n\n.message-edit-buttons button {\n  padding: 4px 12px;\n  border-radius: 4px;\n  background-color: var(--jp-layout-color1);\n  color: var(--jp-content-font-color1);\n  border: 1px solid var(--jp-border-color1);\n  cursor: pointer;\n}\n\n.message-edit-buttons button:hover {\n  background-color: var(--jp-layout-color2);\n}"],"sourceRoot":""}]);
+}`, "",{"version":3,"sources":["webpack://./style/ChatMessage.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,kBAAkB;EAClB,mBAAmB;EACnB,sBAAsB;EACtB,WAAW;EACX,eAAe;EACf,sDAAsD;EACtD,2CAA2C;AAC7C;;AAEA;EACE,qBAAqB;AACvB;;AAEA;EACE,2DAA2D;EAC3D,0CAA0C;EAC1C,aAAa;EACb,kBAAkB;AACpB;;AAEA;EACE,+CAA+C;AACjD;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,QAAQ;EACR,gBAAgB;EAChB,kBAAkB;EAClB,eAAe;AACjB;;AAEA;EACE,gBAAgB;EAChB,sBAAsB;EACtB,SAAS;EACT,gBAAgB;AAClB;;AAEA;EACE,mBAAmB;AACrB;;AAEA,qCAAqC;AACrC;EACE,kBAAkB;EAClB,WAAW;EACX,UAAU;EACV,aAAa;EACb,kBAAkB;EAClB,gBAAgB,EAAE,0DAA0D;AAC9E;;AAEA,qCAAqC;AACrC;;EAEE,eAAe;EACf,UAAU;EACV;;qCAEmC;EACnC,yCAAyC;EACzC,YAAY;EACZ,aAAa;EACb,mBAAmB;EACnB,uBAAuB;EACvB,oCAAoC;AACtC;;AAEA,gCAAgC;AAChC;EACE,WAAW;EACX,YAAY;EACZ,UAAU;EACV,gBAAgB,EAAE,oCAAoC;AACxD;;AAEA,yCAAyC;AACzC;EACE,+CAA+C;AACjD;;AAEA,8FAA8F;AAC9F;;EAEE,gBAAgB;AAClB;;AAEA,2BAA2B;AAC3B;EACE,kBAAkB;EAClB,YAAY;EACZ,kBAAkB;AACpB;;AAEA,mBAAmB;AACnB;EACE,oCAAoC;AACtC;;AAEA,iBAAiB;AACjB;;EAEE,yCAAyC;AAC3C;;AAEA,kCAAkC;AAClC;;EAEE,UAAU;AACZ;;AAEA;EACE,aAAa;EACb,QAAQ;EACR,YAAY;AACd;;AAEA;EACE,iBAAiB;EACjB,kBAAkB;EAClB,yCAAyC;EACzC,oCAAoC;EACpC,yCAAyC;EACzC,eAAe;AACjB;;AAEA;EACE,yCAAyC;AAC3C","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.message {\n  position: relative;\n  height: min-content;\n  box-sizing: border-box;\n  width: 100%;\n  font-size: 14px;\n  border-radius: var(--chat-taskpane-item-border-radius);\n  padding: 0 var(--chat-taskpane-item-indent);\n}\n\n.message:hover .message-start-editing-button {\n  opacity: 1 !important;\n}\n\n.message-user {\n  background-color: var(--chat-user-message-background-color);\n  color: var(--chat-user-message-font-color);\n  padding: 10px;\n  margin-bottom: 5px;\n}\n\n.message-assistant-chat {\n  color: var(--chat-assistant-message-font-color);\n}\n\n.chat-message-buttons {\n  display: flex;\n  flex-direction: row;\n  gap: 8px;\n  margin-top: 10px;\n  margin-bottom: 4px;\n  flex-wrap: wrap;\n}\n\n.chat-taskpane-smart-debug-error-message {\n  white-space: pre;\n  font-family: monospace;\n  margin: 0;\n  overflow-x: auto;\n}\n\n.message-text {\n  align-items: center;\n}\n\n/* Message Action Buttons Container */\n.message-action-buttons {\n  position: absolute;\n  bottom: 8px;\n  right: 8px;\n  display: flex;\n  border-radius: 4px;\n  overflow: hidden; /* Ensures inner buttons don't break the rounded corners */\n}\n\n/* Common styles for action buttons */\n.message-action-buttons button,\n.message-start-editing-button {\n  cursor: pointer;\n  opacity: 0;\n  transition:\n    opacity 0.2s ease-in-out,\n    background-color 0.2s ease-in-out;\n  background-color: var(--jp-layout-color3);\n  border: none;\n  display: flex;\n  align-items: center;\n  justify-content: center;\n  color: var(--jp-content-font-color1);\n}\n\n/* Action buttons in the group */\n.message-action-buttons button {\n  width: 28px;\n  height: 28px;\n  padding: 0;\n  border-radius: 0; /* Remove individual button radius */\n}\n\n/* Add subtle separator between buttons */\n.message-action-buttons button:not(:last-child) {\n  border-right: 1px solid var(--jp-border-color1);\n}\n\n/* Remove the individual button radius styles since we're handling it at the container level */\n.message-action-buttons button:first-child,\n.message-action-buttons button:last-child {\n  border-radius: 0;\n}\n\n/* Standalone edit button */\n.message-start-editing-button {\n  position: relative;\n  padding: 4px;\n  border-radius: 4px;\n}\n\n/* SVG icon color */\n.message-action-buttons button svg {\n  color: var(--jp-content-font-color1);\n}\n\n/* Hover states */\n.message-action-buttons button:hover,\n.message-start-editing-button:hover {\n  background-color: var(--jp-layout-color4);\n}\n\n/* Show buttons on message hover */\n.message:hover .message-action-buttons button,\n.message:hover .message-start-editing-button {\n  opacity: 1;\n}\n\n.message-edit-buttons {\n  display: flex;\n  gap: 8px;\n  padding: 5px;\n}\n\n.message-edit-buttons button {\n  padding: 4px 12px;\n  border-radius: 4px;\n  background-color: var(--jp-layout-color1);\n  color: var(--jp-content-font-color1);\n  border: 1px solid var(--jp-border-color1);\n  cursor: pointer;\n}\n\n.message-edit-buttons button:hover {\n  background-color: var(--jp-layout-color2);\n}"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -16790,26 +18248,38 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
+/* Citation links - styled as superscript with background for visibility */
 .citation-button {
-    background-color: var(--purple-400);
     color: var(--purple-700);
-    height: 14px;
-    width: 14px;
-    border-radius: 2px;
-    font-size: 12px;
+    background-color: var(--purple-200);
+    text-decoration: underline;
     cursor: pointer;
-    display: inline-flex;
-    align-items: center;
-    justify-content: center;
+    font-size: 11px;
+    font-weight: 600;
+    padding: 1px 5px;
+    border-radius: 3px;
     position: relative;
-    top: -0.5em; /* Makes it appear as superscript */
+    top: -0.4em;
     vertical-align: baseline;
-    margin-right: 2px;
+    margin: 0 2px;
 }
 
 .citation-button:hover {
-    background-color: var(--purple-500);
-}`, "",{"version":3,"sources":["webpack://./style/Citation.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,mCAAmC;IACnC,wBAAwB;IACxB,YAAY;IACZ,WAAW;IACX,kBAAkB;IAClB,eAAe;IACf,eAAe;IACf,oBAAoB;IACpB,mBAAmB;IACnB,uBAAuB;IACvB,kBAAkB;IAClB,WAAW,EAAE,mCAAmC;IAChD,wBAAwB;IACxB,iBAAiB;AACrB;;AAEA;IACI,mCAAmC;AACvC","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.citation-button {\n    background-color: var(--purple-400);\n    color: var(--purple-700);\n    height: 14px;\n    width: 14px;\n    border-radius: 2px;\n    font-size: 12px;\n    cursor: pointer;\n    display: inline-flex;\n    align-items: center;\n    justify-content: center;\n    position: relative;\n    top: -0.5em; /* Makes it appear as superscript */\n    vertical-align: baseline;\n    margin-right: 2px;\n}\n\n.citation-button:hover {\n    background-color: var(--purple-500);\n}"],"sourceRoot":""}]);
+    background-color: var(--purple-300);
+}
+
+/* Greyed out style for missing/unresolved citations (deleted or in different notebook) */
+.citation-button.citation-missing {
+    color: var(--jp-ui-font-color2);
+    text-decoration: none;
+    cursor: default;
+    opacity: 0.6;
+}
+
+.citation-button.citation-missing:hover {
+    background-color: var(--jp-layout-color3);
+}
+`, "",{"version":3,"sources":["webpack://./style/Citation.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF,0EAA0E;AAC1E;IACI,wBAAwB;IACxB,mCAAmC;IACnC,0BAA0B;IAC1B,eAAe;IACf,eAAe;IACf,gBAAgB;IAChB,gBAAgB;IAChB,kBAAkB;IAClB,kBAAkB;IAClB,WAAW;IACX,wBAAwB;IACxB,aAAa;AACjB;;AAEA;IACI,mCAAmC;AACvC;;AAEA,yFAAyF;AACzF;IACI,+BAA+B;IAC/B,qBAAqB;IACrB,eAAe;IACf,YAAY;AAChB;;AAEA;IACI,yCAAyC;AAC7C","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n/* Citation links - styled as superscript with background for visibility */\n.citation-button {\n    color: var(--purple-700);\n    background-color: var(--purple-200);\n    text-decoration: underline;\n    cursor: pointer;\n    font-size: 11px;\n    font-weight: 600;\n    padding: 1px 5px;\n    border-radius: 3px;\n    position: relative;\n    top: -0.4em;\n    vertical-align: baseline;\n    margin: 0 2px;\n}\n\n.citation-button:hover {\n    background-color: var(--purple-300);\n}\n\n/* Greyed out style for missing/unresolved citations (deleted or in different notebook) */\n.citation-button.citation-missing {\n    color: var(--jp-ui-font-color2);\n    text-decoration: none;\n    cursor: default;\n    opacity: 0.6;\n}\n\n.citation-button.citation-missing:hover {\n    background-color: var(--jp-layout-color3);\n}\n"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -19039,130 +20509,391 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
     margin-left: auto;
 }
 
-.revert-questionnaire-button:hover::after {
-    opacity: 1;
-    transform: translateX(0);
-}`, "",{"version":3,"sources":["webpack://./style/RevertQuestionnaire.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,aAAa;IACb,sBAAsB;IACtB,QAAQ;IACR,kBAAkB;IAClB,eAAe;IACf,mBAAmB;IACnB,2DAA2D;IAC3D,yCAAyC;IACzC,sBAAsB;IACtB,wBAAwB;IACxB,6BAA6B;AACjC;;AAEA;IACI,eAAe;IACf,+BAA+B;IAC/B,mBAAmB;IACnB,qBAAqB;IACrB,yBAAyB;IACzB,eAAe;AACnB;;AAEA;IACI,oBAAoB;IACpB,mBAAmB;IACnB,8BAA8B;IAC9B,QAAQ;IACR,iBAAiB;IACjB,mCAAmC;IACnC,yCAAyC;IACzC,mBAAmB;IACnB,+BAA+B;IAC/B,eAAe;IACf,eAAe;IACf,iDAAiD;IACjD,gBAAgB;IAChB,WAAW;IACX,gBAAgB;IAChB,kBAAkB;IAClB,kBAAkB;IAClB,sBAAsB;IACtB,qBAAqB;IACrB,yBAAyB;IACzB,mBAAmB;IACnB,gBAAgB;IAChB,eAAe;AACnB;;AAEA;IACI,WAAW;IACX,kBAAkB;IAClB,MAAM;IACN,OAAO;IACP,QAAQ;IACR,SAAS;IACT,sGAAsG;IACtG,mBAAmB;IACnB,UAAU;IACV,6BAA6B;AACjC;;AAEA;IACI,mCAAmC;IACnC,uCAAuC;IACvC,yDAAyD;IACzD,oCAAoC;AACxC;;AAEA;IACI,UAAU;AACd;;AAEA;IACI,oCAAoC;IACpC,yBAAyB;AAC7B;;AAEA;IACI,YAAY;IACZ,6BAA6B;IAC7B,iDAAiD;IACjD,UAAU;IACV,2BAA2B;IAC3B,cAAc;IACd,iBAAiB;AACrB;;AAEA;IACI,UAAU;IACV,wBAAwB;AAC5B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.revert-questionnaire-message {\n    display: flex;\n    flex-direction: column;\n    gap: 2px;\n    padding: 12px 16px;\n    margin: 8px 8px;\n    border-radius: 12px;\n    background-color: var(--chat-user-message-background-color);\n    border: 1px solid var(--jp-border-color2);\n    box-sizing: border-box;\n    width: calc(100% - 16px);\n    max-width: calc(100vw - 32px);\n}\n\n.revert-questionnaire-message-text {\n    font-size: 12px;\n    color: var(--jp-ui-font-color1);\n    margin-bottom: 20px;\n    word-wrap: break-word;\n    overflow-wrap: break-word;\n    max-width: 100%;\n}\n\n.revert-questionnaire-button {\n    display: inline-flex;\n    align-items: center;\n    justify-content: space-between;\n    gap: 8px;\n    padding: 8px 12px;\n    background: var(--jp-layout-color1);\n    border: 1px solid var(--jp-border-color2);\n    border-radius: 20px;\n    color: var(--jp-ui-font-color2);\n    font-size: 12px;\n    cursor: pointer;\n    transition: all 0.2s cubic-bezier(0.4, 0, 0.2, 1);\n    text-align: left;\n    width: 100%;\n    overflow: hidden;\n    position: relative;\n    margin-bottom: 6px;\n    box-sizing: border-box;\n    word-wrap: break-word;\n    overflow-wrap: break-word;\n    white-space: normal;\n    min-height: 36px;\n    max-width: 100%;\n}\n\n.revert-questionnaire-button::before {\n    content: '';\n    position: absolute;\n    top: 0;\n    left: 0;\n    right: 0;\n    bottom: 0;\n    background: linear-gradient(135deg, var(--jp-brand-color3) 0%, rgba(var(--jp-brand-color1), 0.1) 100%);\n    border-radius: 20px;\n    opacity: 0;\n    transition: opacity 0.2s ease;\n}\n\n.revert-questionnaire-button:hover {\n    background: var(--jp-layout-color2);\n    transform: translateY(-1px) scale(1.02);\n    box-shadow: 0 4px 12px rgba(var(--jp-brand-color1), 0.15);\n    border-color: var(--jp-brand-color1);\n}\n\n.revert-questionnaire-button:hover::before {\n    opacity: 1;\n}\n\n.revert-questionnaire-button:active {\n    transform: translateY(0) scale(0.98);\n    transition: all 0.1s ease;\n}\n\n.revert-questionnaire-button::after {\n    content: '→';\n    color: var(--jp-brand-color1);\n    transition: all 0.2s cubic-bezier(0.4, 0, 0.2, 1);\n    opacity: 0;\n    transform: translateX(-4px);\n    flex-shrink: 0;\n    margin-left: auto;\n}\n\n.revert-questionnaire-button:hover::after {\n    opacity: 1;\n    transform: translateX(0);\n}"],"sourceRoot":""}]);
-// Exports
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
-
+.revert-questionnaire-button:hover::after {
+    opacity: 1;
+    transform: translateX(0);
+}`, "",{"version":3,"sources":["webpack://./style/RevertQuestionnaire.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,aAAa;IACb,sBAAsB;IACtB,QAAQ;IACR,kBAAkB;IAClB,eAAe;IACf,mBAAmB;IACnB,2DAA2D;IAC3D,yCAAyC;IACzC,sBAAsB;IACtB,wBAAwB;IACxB,6BAA6B;AACjC;;AAEA;IACI,eAAe;IACf,+BAA+B;IAC/B,mBAAmB;IACnB,qBAAqB;IACrB,yBAAyB;IACzB,eAAe;AACnB;;AAEA;IACI,oBAAoB;IACpB,mBAAmB;IACnB,8BAA8B;IAC9B,QAAQ;IACR,iBAAiB;IACjB,mCAAmC;IACnC,yCAAyC;IACzC,mBAAmB;IACnB,+BAA+B;IAC/B,eAAe;IACf,eAAe;IACf,iDAAiD;IACjD,gBAAgB;IAChB,WAAW;IACX,gBAAgB;IAChB,kBAAkB;IAClB,kBAAkB;IAClB,sBAAsB;IACtB,qBAAqB;IACrB,yBAAyB;IACzB,mBAAmB;IACnB,gBAAgB;IAChB,eAAe;AACnB;;AAEA;IACI,WAAW;IACX,kBAAkB;IAClB,MAAM;IACN,OAAO;IACP,QAAQ;IACR,SAAS;IACT,sGAAsG;IACtG,mBAAmB;IACnB,UAAU;IACV,6BAA6B;AACjC;;AAEA;IACI,mCAAmC;IACnC,uCAAuC;IACvC,yDAAyD;IACzD,oCAAoC;AACxC;;AAEA;IACI,UAAU;AACd;;AAEA;IACI,oCAAoC;IACpC,yBAAyB;AAC7B;;AAEA;IACI,YAAY;IACZ,6BAA6B;IAC7B,iDAAiD;IACjD,UAAU;IACV,2BAA2B;IAC3B,cAAc;IACd,iBAAiB;AACrB;;AAEA;IACI,UAAU;IACV,wBAAwB;AAC5B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.revert-questionnaire-message {\n    display: flex;\n    flex-direction: column;\n    gap: 2px;\n    padding: 12px 16px;\n    margin: 8px 8px;\n    border-radius: 12px;\n    background-color: var(--chat-user-message-background-color);\n    border: 1px solid var(--jp-border-color2);\n    box-sizing: border-box;\n    width: calc(100% - 16px);\n    max-width: calc(100vw - 32px);\n}\n\n.revert-questionnaire-message-text {\n    font-size: 12px;\n    color: var(--jp-ui-font-color1);\n    margin-bottom: 20px;\n    word-wrap: break-word;\n    overflow-wrap: break-word;\n    max-width: 100%;\n}\n\n.revert-questionnaire-button {\n    display: inline-flex;\n    align-items: center;\n    justify-content: space-between;\n    gap: 8px;\n    padding: 8px 12px;\n    background: var(--jp-layout-color1);\n    border: 1px solid var(--jp-border-color2);\n    border-radius: 20px;\n    color: var(--jp-ui-font-color2);\n    font-size: 12px;\n    cursor: pointer;\n    transition: all 0.2s cubic-bezier(0.4, 0, 0.2, 1);\n    text-align: left;\n    width: 100%;\n    overflow: hidden;\n    position: relative;\n    margin-bottom: 6px;\n    box-sizing: border-box;\n    word-wrap: break-word;\n    overflow-wrap: break-word;\n    white-space: normal;\n    min-height: 36px;\n    max-width: 100%;\n}\n\n.revert-questionnaire-button::before {\n    content: '';\n    position: absolute;\n    top: 0;\n    left: 0;\n    right: 0;\n    bottom: 0;\n    background: linear-gradient(135deg, var(--jp-brand-color3) 0%, rgba(var(--jp-brand-color1), 0.1) 100%);\n    border-radius: 20px;\n    opacity: 0;\n    transition: opacity 0.2s ease;\n}\n\n.revert-questionnaire-button:hover {\n    background: var(--jp-layout-color2);\n    transform: translateY(-1px) scale(1.02);\n    box-shadow: 0 4px 12px rgba(var(--jp-brand-color1), 0.15);\n    border-color: var(--jp-brand-color1);\n}\n\n.revert-questionnaire-button:hover::before {\n    opacity: 1;\n}\n\n.revert-questionnaire-button:active {\n    transform: translateY(0) scale(0.98);\n    transition: all 0.1s ease;\n}\n\n.revert-questionnaire-button::after {\n    content: '→';\n    color: var(--jp-brand-color1);\n    transition: all 0.2s cubic-bezier(0.4, 0, 0.2, 1);\n    opacity: 0;\n    transform: translateX(-4px);\n    flex-shrink: 0;\n    margin-left: auto;\n}\n\n.revert-questionnaire-button:hover::after {\n    opacity: 1;\n    transform: translateX(0);\n}"],"sourceRoot":""}]);
+// Exports
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
+
+
+/***/ }),
+
+/***/ "./node_modules/css-loader/dist/cjs.js!./style/RulesForm.css":
+/*!*******************************************************************!*\
+  !*** ./node_modules/css-loader/dist/cjs.js!./style/RulesForm.css ***!
+  \*******************************************************************/
+/***/ ((module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/sourceMaps.js */ "./node_modules/css-loader/dist/runtime/sourceMaps.js");
+/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/api.js */ "./node_modules/css-loader/dist/runtime/api.js");
+/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__);
+// Imports
+
+
+var ___CSS_LOADER_EXPORT___ = _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default()((_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default()));
+// Module
+___CSS_LOADER_EXPORT___.push([module.id, `/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+.rules-form textarea {
+    width: 100%;
+    box-sizing: border-box;
+    padding: 8px 12px;
+    border: 1px solid var(--jp-border-color1);
+    border-radius: 4px;
+    background: var(--jp-layout-color0);
+    color: var(--jp-ui-font-color1);
+    font-size: 14px;
+    transition: border-color 0.2s;
+    font-family: var(--jp-ui-font-family);
+    resize: vertical;
+    min-height: 120px;
+}
+
+.rules-form textarea:focus {
+    outline: none;
+    border-color: var(--purple-500);
+    box-shadow: 0 0 0 2px rgba(124, 58, 237, 0.1);
+}
+
+.rules-form .form-group sub {
+    color: var(--jp-ui-font-color2);
+    margin-top: 4px;
+    margin-bottom: 4px;
+    display: block;
+}
+
+/* Rules List Styling */
+.rules-list {
+    margin-top: 24px;
+    display: flex;
+    flex-direction: column;
+    gap: 16px;
+}
+
+.rules-list .rule-item {
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    padding: 20px;
+    border: 1px solid var(--jp-border-color1);
+    border-radius: 8px;
+    background: var(--jp-layout-color0);
+    transition: border-color 0.2s, box-shadow 0.2s;
+    cursor: pointer;
+}
+
+.rules-list .rule-item:hover {
+    border-color: var(--jp-border-color2);
+    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
+}
+
+.rules-list .rule-content {
+    flex: 1;
+    margin-right: 16px;
+}
+
+.rules-list .rule-name {
+    margin: 0 0 8px 0;
+    color: var(--jp-ui-font-color1);
+    font-size: 14px;
+    font-weight: 500;
+    line-height: 1;
+}
+
+.rules-list .rule-description {
+    margin: 0;
+    color: var(--jp-ui-font-color2);
+    font-size: 14px;
+    line-height: 1.5;
+}
+
+.rules-list .rule-actions {
+    display: flex;
+    gap: 8px;
+}
+
+.rules-list .empty-state {
+    text-align: center;
+    padding: 40px 20px;
+    color: var(--jp-ui-font-color2);
+    font-style: italic;
+}
+
+.rules-list .empty-state p {
+    margin: 0;
+    font-size: 16px;
+}
+`, "",{"version":3,"sources":["webpack://./style/RulesForm.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,WAAW;IACX,sBAAsB;IACtB,iBAAiB;IACjB,yCAAyC;IACzC,kBAAkB;IAClB,mCAAmC;IACnC,+BAA+B;IAC/B,eAAe;IACf,6BAA6B;IAC7B,qCAAqC;IACrC,gBAAgB;IAChB,iBAAiB;AACrB;;AAEA;IACI,aAAa;IACb,+BAA+B;IAC/B,6CAA6C;AACjD;;AAEA;IACI,+BAA+B;IAC/B,eAAe;IACf,kBAAkB;IAClB,cAAc;AAClB;;AAEA,uBAAuB;AACvB;IACI,gBAAgB;IAChB,aAAa;IACb,sBAAsB;IACtB,SAAS;AACb;;AAEA;IACI,aAAa;IACb,8BAA8B;IAC9B,mBAAmB;IACnB,aAAa;IACb,yCAAyC;IACzC,kBAAkB;IAClB,mCAAmC;IACnC,8CAA8C;IAC9C,eAAe;AACnB;;AAEA;IACI,qCAAqC;IACrC,wCAAwC;AAC5C;;AAEA;IACI,OAAO;IACP,kBAAkB;AACtB;;AAEA;IACI,iBAAiB;IACjB,+BAA+B;IAC/B,eAAe;IACf,gBAAgB;IAChB,cAAc;AAClB;;AAEA;IACI,SAAS;IACT,+BAA+B;IAC/B,eAAe;IACf,gBAAgB;AACpB;;AAEA;IACI,aAAa;IACb,QAAQ;AACZ;;AAEA;IACI,kBAAkB;IAClB,kBAAkB;IAClB,+BAA+B;IAC/B,kBAAkB;AACtB;;AAEA;IACI,SAAS;IACT,eAAe;AACnB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.rules-form textarea {\n    width: 100%;\n    box-sizing: border-box;\n    padding: 8px 12px;\n    border: 1px solid var(--jp-border-color1);\n    border-radius: 4px;\n    background: var(--jp-layout-color0);\n    color: var(--jp-ui-font-color1);\n    font-size: 14px;\n    transition: border-color 0.2s;\n    font-family: var(--jp-ui-font-family);\n    resize: vertical;\n    min-height: 120px;\n}\n\n.rules-form textarea:focus {\n    outline: none;\n    border-color: var(--purple-500);\n    box-shadow: 0 0 0 2px rgba(124, 58, 237, 0.1);\n}\n\n.rules-form .form-group sub {\n    color: var(--jp-ui-font-color2);\n    margin-top: 4px;\n    margin-bottom: 4px;\n    display: block;\n}\n\n/* Rules List Styling */\n.rules-list {\n    margin-top: 24px;\n    display: flex;\n    flex-direction: column;\n    gap: 16px;\n}\n\n.rules-list .rule-item {\n    display: flex;\n    justify-content: space-between;\n    align-items: center;\n    padding: 20px;\n    border: 1px solid var(--jp-border-color1);\n    border-radius: 8px;\n    background: var(--jp-layout-color0);\n    transition: border-color 0.2s, box-shadow 0.2s;\n    cursor: pointer;\n}\n\n.rules-list .rule-item:hover {\n    border-color: var(--jp-border-color2);\n    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);\n}\n\n.rules-list .rule-content {\n    flex: 1;\n    margin-right: 16px;\n}\n\n.rules-list .rule-name {\n    margin: 0 0 8px 0;\n    color: var(--jp-ui-font-color1);\n    font-size: 14px;\n    font-weight: 500;\n    line-height: 1;\n}\n\n.rules-list .rule-description {\n    margin: 0;\n    color: var(--jp-ui-font-color2);\n    font-size: 14px;\n    line-height: 1.5;\n}\n\n.rules-list .rule-actions {\n    display: flex;\n    gap: 8px;\n}\n\n.rules-list .empty-state {\n    text-align: center;\n    padding: 40px 20px;\n    color: var(--jp-ui-font-color2);\n    font-style: italic;\n}\n\n.rules-list .empty-state p {\n    margin: 0;\n    font-size: 16px;\n}\n"],"sourceRoot":""}]);
+// Exports
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
+
+
+/***/ }),
+
+/***/ "./node_modules/css-loader/dist/cjs.js!./style/RunCellButton.css":
+/*!***********************************************************************!*\
+  !*** ./node_modules/css-loader/dist/cjs.js!./style/RunCellButton.css ***!
+  \***********************************************************************/
+/***/ ((module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/sourceMaps.js */ "./node_modules/css-loader/dist/runtime/sourceMaps.js");
+/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/api.js */ "./node_modules/css-loader/dist/runtime/api.js");
+/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__);
+// Imports
+
+
+var ___CSS_LOADER_EXPORT___ = _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default()((_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default()));
+// Module
+___CSS_LOADER_EXPORT___.push([module.id, `/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+.mito-run-cell-button-container {
+  position: relative;
+  display: block;
+  width: 160px;
+}
+
+/* Button group - contains both buttons */
+.mito-run-cell-button-group {
+  display: flex;
+  align-items: center;
+  height: var(--toolbar-button-height);
+  width: 100%;
+  border-radius: 4px;
+  overflow: hidden;
+  background-color: var(--blue-300);
+  border: 0.5px solid var(--blue-900);
+}
+
+/* Shared button styles */
+.mito-run-cell-button {
+  display: flex;
+  align-items: center;
+  border: none;
+  padding: 0;
+  border-radius: 0;
+  cursor: pointer;
+  color: var(--blue-900);
+  font-size: 13px;
+  font-weight: 400;
+  transition: background-color 0.15s ease;
+  height: 100%;
+  background-color: transparent;
+}
+
+.mito-run-cell-button:hover {
+  background-color: var(--blue-400);
+}
+
+/* Main button (left side - play icon + text) */
+.mito-run-cell-button-main {
+  flex: 1;
+  gap: 6px;
+  padding: 0 10px;
+}
+
+/* Divider between buttons */
+.mito-run-cell-button-divider {
+  width: 1px;
+  height: 14px;
+  background: linear-gradient(
+    to bottom,
+    transparent,
+    var(--blue-900) 20%,
+    var(--blue-900) 80%,
+    transparent
+  );
+  opacity: 0.4;
+  flex-shrink: 0;
+}
+
+/* Dropdown button (right side - chevron) */
+.mito-run-cell-button-dropdown {
+  padding: 0 6px;
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  border-radius: 0;
+}
+
+/* Play icon styling */
+.mito-run-cell-button-main > svg {
+  color: var(--blue-900);
+  fill: var(--blue-900);
+  width: 12px;
+  height: 12px;
+  flex-shrink: 0;
+}
+
+/* Chevron styling */
+.mito-run-cell-button-dropdown svg {
+  width: 14px;
+  height: 14px;
+  color: var(--blue-900);
+}
+
+/* Ensure chevron path only uses stroke, not fill */
+.mito-run-cell-button-dropdown svg path {
+  fill: none;
+  stroke: currentColor;
+}
+
+.mito-run-cell-button-text {
+  white-space: nowrap;
+}
+
+/* Running state - applied to the group */
+.mito-run-cell-button-running {
+  border-color: var(--green-900);
+}
 
-/***/ }),
+.mito-run-cell-button-running .mito-run-cell-button,
+.mito-run-cell-button-running .mito-run-cell-button-dropdown {
+  background-color: var(--green-300);
+  color: var(--green-900);
+}
 
-/***/ "./node_modules/css-loader/dist/cjs.js!./style/RulesForm.css":
-/*!*******************************************************************!*\
-  !*** ./node_modules/css-loader/dist/cjs.js!./style/RulesForm.css ***!
-  \*******************************************************************/
-/***/ ((module, __webpack_exports__, __webpack_require__) => {
+.mito-run-cell-button-running .mito-run-cell-button-dropdown:hover {
+  background-color: var(--green-500);
+}
 
-__webpack_require__.r(__webpack_exports__);
-/* harmony export */ __webpack_require__.d(__webpack_exports__, {
-/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
-/* harmony export */ });
-/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/sourceMaps.js */ "./node_modules/css-loader/dist/runtime/sourceMaps.js");
-/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__);
-/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/api.js */ "./node_modules/css-loader/dist/runtime/api.js");
-/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__);
-// Imports
+.mito-run-cell-button-running .mito-run-cell-button-divider {
+  background: linear-gradient(
+    to bottom,
+    transparent,
+    var(--green-900) 20%,
+    var(--green-900) 80%,
+    transparent
+  );
+  opacity: 0.5;
+}
 
+/* LoadingCircle styling when running */
+.mito-run-cell-button-running .mito-run-cell-button-main > svg {
+  color: var(--green-900);
+  fill: var(--green-900);
+  width: 16px;
+  height: 16px;
+}
 
-var ___CSS_LOADER_EXPORT___ = _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default()((_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default()));
-// Module
-___CSS_LOADER_EXPORT___.push([module.id, `/*
- * Copyright (c) Saga Inc.
- * Distributed under the terms of the GNU Affero General Public License v3.0 License.
- */
+.mito-run-cell-button-running .mito-run-cell-button-text {
+  color: var(--green-900);
+}
 
-.rules-form textarea {
-    width: 100%;
-    box-sizing: border-box;
-    padding: 8px 12px;
-    border: 1px solid var(--jp-border-color1);
-    border-radius: 4px;
-    background: var(--jp-layout-color0);
-    color: var(--jp-ui-font-color1);
-    font-size: 14px;
-    transition: border-color 0.2s;
-    font-family: var(--jp-ui-font-family);
-    resize: vertical;
-    min-height: 120px;
+.mito-run-cell-button-running .mito-run-cell-button-dropdown svg {
+  color: var(--green-900);
 }
 
-.rules-form textarea:focus {
-    outline: none;
-    border-color: var(--purple-500);
-    box-shadow: 0 0 0 2px rgba(124, 58, 237, 0.1);
+.mito-run-cell-dropdown-menu {
+  position: absolute;
+  top: calc(100% + 4px);
+  right: 0;
+  z-index: 1000;
+  background-color: var(--jp-layout-color1);
+  border: 1px solid var(--jp-layout-color2);
+  border-radius: 3px;
+  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);
+  min-width: 200px;
+  white-space: nowrap;
+  padding: 4px 0;
 }
 
-.rules-form .form-group sub {
-    color: var(--jp-ui-font-color2);
-    margin-top: 4px;
-    margin-bottom: 4px;
-    display: block;
+.mito-run-cell-dropdown-section {
+  display: flex;
+  flex-direction: column;
 }
 
-/* Rules List Styling */
-.rules-list {
-    margin-top: 24px;
-    display: flex;
-    flex-direction: column;
-    gap: 16px;
+.mito-run-cell-dropdown-section-header {
+  padding: 6px 12px 4px 12px;
+  font-size: 11px;
+  font-weight: 600;
+  color: var(--jp-ui-font-color2);
+  text-transform: uppercase;
+  letter-spacing: 0.5px;
 }
 
-.rules-list .rule-item {
-    display: flex;
-    justify-content: space-between;
-    align-items: center;
-    padding: 20px;
-    border: 1px solid var(--jp-border-color1);
-    border-radius: 8px;
-    background: var(--jp-layout-color0);
-    transition: border-color 0.2s, box-shadow 0.2s;
-    cursor: pointer;
+.mito-run-cell-dropdown-separator {
+  height: 1px;
+  background-color: var(--jp-layout-color2);
+  margin: 4px 0;
 }
 
-.rules-list .rule-item:hover {
-    border-color: var(--jp-border-color2);
-    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
+.mito-run-cell-dropdown-item {
+  display: flex;
+  align-items: center;
+  width: 100%;
+  padding: 6px 12px;
+  text-align: left;
+  border: none;
+  background: none;
+  color: var(--jp-ui-font-color1);
+  cursor: pointer;
+  font-size: 13px;
+  font-weight: 400;
+  gap: 8px;
 }
 
-.rules-list .rule-content {
-    flex: 1;
-    margin-right: 16px;
+.mito-run-cell-dropdown-item:hover {
+  background-color: var(--jp-layout-color2);
 }
 
-.rules-list .rule-name {
-    margin: 0 0 8px 0;
-    color: var(--jp-ui-font-color1);
-    font-size: 14px;
-    font-weight: 500;
-    line-height: 1;
+.mito-run-cell-dropdown-item-icon {
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  width: 20px; /* Fixed width to align all icons */
+  height: 16px;
+  flex-shrink: 0;
+  color: var(--jp-ui-font-color1);
+  line-height: 0; /* Remove extra line height for better alignment */
 }
 
-.rules-list .rule-description {
-    margin: 0;
-    color: var(--jp-ui-font-color2);
-    font-size: 14px;
-    line-height: 1.5;
+.mito-run-cell-dropdown-item-icon svg {
+  width: 16px;
+  height: 16px;
+  display: block; /* Remove inline spacing */
+  margin: 0 auto; /* Center the icon within the fixed width container */
+  color: var(--jp-ui-font-color1); 
 }
 
-.rules-list .rule-actions {
-    display: flex;
-    gap: 8px;
+.mito-run-cell-dropdown-item-label {
+  flex: 1;
 }
 
-.rules-list .empty-state {
-    text-align: center;
-    padding: 40px 20px;
-    color: var(--jp-ui-font-color2);
-    font-style: italic;
+.mito-run-cell-dropdown-item-shortcut {
+  color: var(--jp-ui-font-color2);
+  font-size: 11px;
+  margin-left: 10px;
+  opacity: 0.7;
 }
 
-.rules-list .empty-state p {
-    margin: 0;
-    font-size: 16px;
+/* Position the button widget on the right side of the toolbar */
+.jp-Notebook-toolbar .mito-run-cell-button-widget {
+  margin-left: auto;
+  order: 9999; /* Ensure it appears last */
 }
-`, "",{"version":3,"sources":["webpack://./style/RulesForm.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,WAAW;IACX,sBAAsB;IACtB,iBAAiB;IACjB,yCAAyC;IACzC,kBAAkB;IAClB,mCAAmC;IACnC,+BAA+B;IAC/B,eAAe;IACf,6BAA6B;IAC7B,qCAAqC;IACrC,gBAAgB;IAChB,iBAAiB;AACrB;;AAEA;IACI,aAAa;IACb,+BAA+B;IAC/B,6CAA6C;AACjD;;AAEA;IACI,+BAA+B;IAC/B,eAAe;IACf,kBAAkB;IAClB,cAAc;AAClB;;AAEA,uBAAuB;AACvB;IACI,gBAAgB;IAChB,aAAa;IACb,sBAAsB;IACtB,SAAS;AACb;;AAEA;IACI,aAAa;IACb,8BAA8B;IAC9B,mBAAmB;IACnB,aAAa;IACb,yCAAyC;IACzC,kBAAkB;IAClB,mCAAmC;IACnC,8CAA8C;IAC9C,eAAe;AACnB;;AAEA;IACI,qCAAqC;IACrC,wCAAwC;AAC5C;;AAEA;IACI,OAAO;IACP,kBAAkB;AACtB;;AAEA;IACI,iBAAiB;IACjB,+BAA+B;IAC/B,eAAe;IACf,gBAAgB;IAChB,cAAc;AAClB;;AAEA;IACI,SAAS;IACT,+BAA+B;IAC/B,eAAe;IACf,gBAAgB;AACpB;;AAEA;IACI,aAAa;IACb,QAAQ;AACZ;;AAEA;IACI,kBAAkB;IAClB,kBAAkB;IAClB,+BAA+B;IAC/B,kBAAkB;AACtB;;AAEA;IACI,SAAS;IACT,eAAe;AACnB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.rules-form textarea {\n    width: 100%;\n    box-sizing: border-box;\n    padding: 8px 12px;\n    border: 1px solid var(--jp-border-color1);\n    border-radius: 4px;\n    background: var(--jp-layout-color0);\n    color: var(--jp-ui-font-color1);\n    font-size: 14px;\n    transition: border-color 0.2s;\n    font-family: var(--jp-ui-font-family);\n    resize: vertical;\n    min-height: 120px;\n}\n\n.rules-form textarea:focus {\n    outline: none;\n    border-color: var(--purple-500);\n    box-shadow: 0 0 0 2px rgba(124, 58, 237, 0.1);\n}\n\n.rules-form .form-group sub {\n    color: var(--jp-ui-font-color2);\n    margin-top: 4px;\n    margin-bottom: 4px;\n    display: block;\n}\n\n/* Rules List Styling */\n.rules-list {\n    margin-top: 24px;\n    display: flex;\n    flex-direction: column;\n    gap: 16px;\n}\n\n.rules-list .rule-item {\n    display: flex;\n    justify-content: space-between;\n    align-items: center;\n    padding: 20px;\n    border: 1px solid var(--jp-border-color1);\n    border-radius: 8px;\n    background: var(--jp-layout-color0);\n    transition: border-color 0.2s, box-shadow 0.2s;\n    cursor: pointer;\n}\n\n.rules-list .rule-item:hover {\n    border-color: var(--jp-border-color2);\n    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);\n}\n\n.rules-list .rule-content {\n    flex: 1;\n    margin-right: 16px;\n}\n\n.rules-list .rule-name {\n    margin: 0 0 8px 0;\n    color: var(--jp-ui-font-color1);\n    font-size: 14px;\n    font-weight: 500;\n    line-height: 1;\n}\n\n.rules-list .rule-description {\n    margin: 0;\n    color: var(--jp-ui-font-color2);\n    font-size: 14px;\n    line-height: 1.5;\n}\n\n.rules-list .rule-actions {\n    display: flex;\n    gap: 8px;\n}\n\n.rules-list .empty-state {\n    text-align: center;\n    padding: 40px 20px;\n    color: var(--jp-ui-font-color2);\n    font-style: italic;\n}\n\n.rules-list .empty-state p {\n    margin: 0;\n    font-size: 16px;\n}\n"],"sourceRoot":""}]);
+
+/* Ensure toolbar uses flexbox for proper alignment */
+.jp-Notebook-toolbar {
+  display: flex;
+  align-items: center;
+}`, "",{"version":3,"sources":["webpack://./style/RunCellButton.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,kBAAkB;EAClB,cAAc;EACd,YAAY;AACd;;AAEA,yCAAyC;AACzC;EACE,aAAa;EACb,mBAAmB;EACnB,oCAAoC;EACpC,WAAW;EACX,kBAAkB;EAClB,gBAAgB;EAChB,iCAAiC;EACjC,mCAAmC;AACrC;;AAEA,yBAAyB;AACzB;EACE,aAAa;EACb,mBAAmB;EACnB,YAAY;EACZ,UAAU;EACV,gBAAgB;EAChB,eAAe;EACf,sBAAsB;EACtB,eAAe;EACf,gBAAgB;EAChB,uCAAuC;EACvC,YAAY;EACZ,6BAA6B;AAC/B;;AAEA;EACE,iCAAiC;AACnC;;AAEA,+CAA+C;AAC/C;EACE,OAAO;EACP,QAAQ;EACR,eAAe;AACjB;;AAEA,4BAA4B;AAC5B;EACE,UAAU;EACV,YAAY;EACZ;;;;;;GAMC;EACD,YAAY;EACZ,cAAc;AAChB;;AAEA,2CAA2C;AAC3C;EACE,cAAc;EACd,aAAa;EACb,mBAAmB;EACnB,uBAAuB;EACvB,gBAAgB;AAClB;;AAEA,sBAAsB;AACtB;EACE,sBAAsB;EACtB,qBAAqB;EACrB,WAAW;EACX,YAAY;EACZ,cAAc;AAChB;;AAEA,oBAAoB;AACpB;EACE,WAAW;EACX,YAAY;EACZ,sBAAsB;AACxB;;AAEA,mDAAmD;AACnD;EACE,UAAU;EACV,oBAAoB;AACtB;;AAEA;EACE,mBAAmB;AACrB;;AAEA,yCAAyC;AACzC;EACE,8BAA8B;AAChC;;AAEA;;EAEE,kCAAkC;EAClC,uBAAuB;AACzB;;AAEA;EACE,kCAAkC;AACpC;;AAEA;EACE;;;;;;GAMC;EACD,YAAY;AACd;;AAEA,uCAAuC;AACvC;EACE,uBAAuB;EACvB,sBAAsB;EACtB,WAAW;EACX,YAAY;AACd;;AAEA;EACE,uBAAuB;AACzB;;AAEA;EACE,uBAAuB;AACzB;;AAEA;EACE,kBAAkB;EAClB,qBAAqB;EACrB,QAAQ;EACR,aAAa;EACb,yCAAyC;EACzC,yCAAyC;EACzC,kBAAkB;EAClB,wCAAwC;EACxC,gBAAgB;EAChB,mBAAmB;EACnB,cAAc;AAChB;;AAEA;EACE,aAAa;EACb,sBAAsB;AACxB;;AAEA;EACE,0BAA0B;EAC1B,eAAe;EACf,gBAAgB;EAChB,+BAA+B;EAC/B,yBAAyB;EACzB,qBAAqB;AACvB;;AAEA;EACE,WAAW;EACX,yCAAyC;EACzC,aAAa;AACf;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,WAAW;EACX,iBAAiB;EACjB,gBAAgB;EAChB,YAAY;EACZ,gBAAgB;EAChB,+BAA+B;EAC/B,eAAe;EACf,eAAe;EACf,gBAAgB;EAChB,QAAQ;AACV;;AAEA;EACE,yCAAyC;AAC3C;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,uBAAuB;EACvB,WAAW,EAAE,mCAAmC;EAChD,YAAY;EACZ,cAAc;EACd,+BAA+B;EAC/B,cAAc,EAAE,kDAAkD;AACpE;;AAEA;EACE,WAAW;EACX,YAAY;EACZ,cAAc,EAAE,0BAA0B;EAC1C,cAAc,EAAE,qDAAqD;EACrE,+BAA+B;AACjC;;AAEA;EACE,OAAO;AACT;;AAEA;EACE,+BAA+B;EAC/B,eAAe;EACf,iBAAiB;EACjB,YAAY;AACd;;AAEA,gEAAgE;AAChE;EACE,iBAAiB;EACjB,WAAW,EAAE,2BAA2B;AAC1C;;AAEA,qDAAqD;AACrD;EACE,aAAa;EACb,mBAAmB;AACrB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.mito-run-cell-button-container {\n  position: relative;\n  display: block;\n  width: 160px;\n}\n\n/* Button group - contains both buttons */\n.mito-run-cell-button-group {\n  display: flex;\n  align-items: center;\n  height: var(--toolbar-button-height);\n  width: 100%;\n  border-radius: 4px;\n  overflow: hidden;\n  background-color: var(--blue-300);\n  border: 0.5px solid var(--blue-900);\n}\n\n/* Shared button styles */\n.mito-run-cell-button {\n  display: flex;\n  align-items: center;\n  border: none;\n  padding: 0;\n  border-radius: 0;\n  cursor: pointer;\n  color: var(--blue-900);\n  font-size: 13px;\n  font-weight: 400;\n  transition: background-color 0.15s ease;\n  height: 100%;\n  background-color: transparent;\n}\n\n.mito-run-cell-button:hover {\n  background-color: var(--blue-400);\n}\n\n/* Main button (left side - play icon + text) */\n.mito-run-cell-button-main {\n  flex: 1;\n  gap: 6px;\n  padding: 0 10px;\n}\n\n/* Divider between buttons */\n.mito-run-cell-button-divider {\n  width: 1px;\n  height: 14px;\n  background: linear-gradient(\n    to bottom,\n    transparent,\n    var(--blue-900) 20%,\n    var(--blue-900) 80%,\n    transparent\n  );\n  opacity: 0.4;\n  flex-shrink: 0;\n}\n\n/* Dropdown button (right side - chevron) */\n.mito-run-cell-button-dropdown {\n  padding: 0 6px;\n  display: flex;\n  align-items: center;\n  justify-content: center;\n  border-radius: 0;\n}\n\n/* Play icon styling */\n.mito-run-cell-button-main > svg {\n  color: var(--blue-900);\n  fill: var(--blue-900);\n  width: 12px;\n  height: 12px;\n  flex-shrink: 0;\n}\n\n/* Chevron styling */\n.mito-run-cell-button-dropdown svg {\n  width: 14px;\n  height: 14px;\n  color: var(--blue-900);\n}\n\n/* Ensure chevron path only uses stroke, not fill */\n.mito-run-cell-button-dropdown svg path {\n  fill: none;\n  stroke: currentColor;\n}\n\n.mito-run-cell-button-text {\n  white-space: nowrap;\n}\n\n/* Running state - applied to the group */\n.mito-run-cell-button-running {\n  border-color: var(--green-900);\n}\n\n.mito-run-cell-button-running .mito-run-cell-button,\n.mito-run-cell-button-running .mito-run-cell-button-dropdown {\n  background-color: var(--green-300);\n  color: var(--green-900);\n}\n\n.mito-run-cell-button-running .mito-run-cell-button-dropdown:hover {\n  background-color: var(--green-500);\n}\n\n.mito-run-cell-button-running .mito-run-cell-button-divider {\n  background: linear-gradient(\n    to bottom,\n    transparent,\n    var(--green-900) 20%,\n    var(--green-900) 80%,\n    transparent\n  );\n  opacity: 0.5;\n}\n\n/* LoadingCircle styling when running */\n.mito-run-cell-button-running .mito-run-cell-button-main > svg {\n  color: var(--green-900);\n  fill: var(--green-900);\n  width: 16px;\n  height: 16px;\n}\n\n.mito-run-cell-button-running .mito-run-cell-button-text {\n  color: var(--green-900);\n}\n\n.mito-run-cell-button-running .mito-run-cell-button-dropdown svg {\n  color: var(--green-900);\n}\n\n.mito-run-cell-dropdown-menu {\n  position: absolute;\n  top: calc(100% + 4px);\n  right: 0;\n  z-index: 1000;\n  background-color: var(--jp-layout-color1);\n  border: 1px solid var(--jp-layout-color2);\n  border-radius: 3px;\n  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);\n  min-width: 200px;\n  white-space: nowrap;\n  padding: 4px 0;\n}\n\n.mito-run-cell-dropdown-section {\n  display: flex;\n  flex-direction: column;\n}\n\n.mito-run-cell-dropdown-section-header {\n  padding: 6px 12px 4px 12px;\n  font-size: 11px;\n  font-weight: 600;\n  color: var(--jp-ui-font-color2);\n  text-transform: uppercase;\n  letter-spacing: 0.5px;\n}\n\n.mito-run-cell-dropdown-separator {\n  height: 1px;\n  background-color: var(--jp-layout-color2);\n  margin: 4px 0;\n}\n\n.mito-run-cell-dropdown-item {\n  display: flex;\n  align-items: center;\n  width: 100%;\n  padding: 6px 12px;\n  text-align: left;\n  border: none;\n  background: none;\n  color: var(--jp-ui-font-color1);\n  cursor: pointer;\n  font-size: 13px;\n  font-weight: 400;\n  gap: 8px;\n}\n\n.mito-run-cell-dropdown-item:hover {\n  background-color: var(--jp-layout-color2);\n}\n\n.mito-run-cell-dropdown-item-icon {\n  display: flex;\n  align-items: center;\n  justify-content: center;\n  width: 20px; /* Fixed width to align all icons */\n  height: 16px;\n  flex-shrink: 0;\n  color: var(--jp-ui-font-color1);\n  line-height: 0; /* Remove extra line height for better alignment */\n}\n\n.mito-run-cell-dropdown-item-icon svg {\n  width: 16px;\n  height: 16px;\n  display: block; /* Remove inline spacing */\n  margin: 0 auto; /* Center the icon within the fixed width container */\n  color: var(--jp-ui-font-color1); \n}\n\n.mito-run-cell-dropdown-item-label {\n  flex: 1;\n}\n\n.mito-run-cell-dropdown-item-shortcut {\n  color: var(--jp-ui-font-color2);\n  font-size: 11px;\n  margin-left: 10px;\n  opacity: 0.7;\n}\n\n/* Position the button widget on the right side of the toolbar */\n.jp-Notebook-toolbar .mito-run-cell-button-widget {\n  margin-left: auto;\n  order: 9999; /* Ensure it appears last */\n}\n\n/* Ensure toolbar uses flexbox for proper alignment */\n.jp-Notebook-toolbar {\n  display: flex;\n  align-items: center;\n}"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -20029,6 +21760,17 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   background-color: var(--purple-400);
 }
 
+.button-blue {
+  background-color: var(--blue-300);
+  color: var(--blue-900) !important;
+  border: 0.5px solid var(--blue-900);
+  border-radius: 4px;
+}
+
+.button-blue:hover {
+  background-color: var(--blue-400);
+}
+
 .button-dark-purple {
   background-color: var(--purple-500);
   color: var(--white) !important;
@@ -20047,11 +21789,11 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 }
 
 .button-small {
-  height: 20px;
+  height: var(--toolbar-button-height);
   font-size: 12px;
   padding: 2px 5px;
 }
-`, "",{"version":3,"sources":["webpack://./style/button.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;;;;CAIC;;AAED;EACE,YAAY;EACZ,kBAAkB;EAClB,eAAe;EACf,eAAe;EACf,2EAA2E;EAC3E,mBAAmB;EACnB,iBAAiB;AACnB;;AAEA;EACE,kCAAkC;EAClC,kCAAkC;AACpC;;AAEA;EACE,kCAAkC;AACpC;;AAEA;EACE,gCAAgC;EAChC,gCAAgC;AAClC;;AAEA;EACE,gCAAgC;AAClC;;AAEA;EACE,yCAAyC;EACzC,+CAA+C;AACjD;;AAEA;EACE,yCAAyC;AAC3C;;AAEA;EACE,mCAAmC;EACnC,mCAAmC;AACrC;;AAEA;EACE,mCAAmC;AACrC;;AAEA;EACE,mCAAmC;EACnC,8BAA8B;AAChC;;AAEA;EACE,mCAAmC;AACrC;;AAEA;EACE,WAAW;AACb;;AAEA;EACE,kBAAkB;AACpB;;AAEA;EACE,YAAY;EACZ,eAAe;EACf,gBAAgB;AAClB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n/* \n    Classes that can be used for any button, making it easier to keep \n    the theme consistent. For example, use them to make a textButton or\n    textAndIconButton green.\n*/\n\n.button-base {\n  border: none;\n  border-radius: 3px;\n  cursor: pointer;\n  font-size: 14px;\n  /* Make sure the button grows large enough so that the text does not wrap */\n  white-space: nowrap;\n  padding: 5px 10px;\n}\n\n.button-green {\n  background-color: var(--green-400);\n  color: var(--green-900) !important;\n}\n\n.button-green:hover {\n  background-color: var(--green-500);\n}\n\n.button-red {\n  background-color: var(--red-400);\n  color: var(--red-900) !important;\n}\n\n.button-red:hover {\n  background-color: var(--red-500);\n}\n\n.button-gray {\n  background-color: var(--jp-layout-color2);\n  color: var(--jp-content-font-color1) !important;\n}\n\n.button-gray:hover {\n  background-color: var(--jp-layout-color3);\n}\n\n.button-purple {\n  background-color: var(--purple-300);\n  color: var(--purple-700) !important;\n}\n\n.button-purple:hover {\n  background-color: var(--purple-400);\n}\n\n.button-dark-purple {\n  background-color: var(--purple-500);\n  color: var(--white) !important;\n}\n\n.button-dark-purple:hover {\n  background-color: var(--purple-600);\n}\n\n.button-width-block {\n  width: 100%;\n}\n\n.button-width-fit-contents {\n  width: fit-content;\n}\n\n.button-small {\n  height: 20px;\n  font-size: 12px;\n  padding: 2px 5px;\n}\n"],"sourceRoot":""}]);
+`, "",{"version":3,"sources":["webpack://./style/button.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;;;;CAIC;;AAED;EACE,YAAY;EACZ,kBAAkB;EAClB,eAAe;EACf,eAAe;EACf,2EAA2E;EAC3E,mBAAmB;EACnB,iBAAiB;AACnB;;AAEA;EACE,kCAAkC;EAClC,kCAAkC;AACpC;;AAEA;EACE,kCAAkC;AACpC;;AAEA;EACE,gCAAgC;EAChC,gCAAgC;AAClC;;AAEA;EACE,gCAAgC;AAClC;;AAEA;EACE,yCAAyC;EACzC,+CAA+C;AACjD;;AAEA;EACE,yCAAyC;AAC3C;;AAEA;EACE,mCAAmC;EACnC,mCAAmC;AACrC;;AAEA;EACE,mCAAmC;AACrC;;AAEA;EACE,iCAAiC;EACjC,iCAAiC;EACjC,mCAAmC;EACnC,kBAAkB;AACpB;;AAEA;EACE,iCAAiC;AACnC;;AAEA;EACE,mCAAmC;EACnC,8BAA8B;AAChC;;AAEA;EACE,mCAAmC;AACrC;;AAEA;EACE,WAAW;AACb;;AAEA;EACE,kBAAkB;AACpB;;AAEA;EACE,oCAAoC;EACpC,eAAe;EACf,gBAAgB;AAClB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n/* \n    Classes that can be used for any button, making it easier to keep \n    the theme consistent. For example, use them to make a textButton or\n    textAndIconButton green.\n*/\n\n.button-base {\n  border: none;\n  border-radius: 3px;\n  cursor: pointer;\n  font-size: 14px;\n  /* Make sure the button grows large enough so that the text does not wrap */\n  white-space: nowrap;\n  padding: 5px 10px;\n}\n\n.button-green {\n  background-color: var(--green-400);\n  color: var(--green-900) !important;\n}\n\n.button-green:hover {\n  background-color: var(--green-500);\n}\n\n.button-red {\n  background-color: var(--red-400);\n  color: var(--red-900) !important;\n}\n\n.button-red:hover {\n  background-color: var(--red-500);\n}\n\n.button-gray {\n  background-color: var(--jp-layout-color2);\n  color: var(--jp-content-font-color1) !important;\n}\n\n.button-gray:hover {\n  background-color: var(--jp-layout-color3);\n}\n\n.button-purple {\n  background-color: var(--purple-300);\n  color: var(--purple-700) !important;\n}\n\n.button-purple:hover {\n  background-color: var(--purple-400);\n}\n\n.button-blue {\n  background-color: var(--blue-300);\n  color: var(--blue-900) !important;\n  border: 0.5px solid var(--blue-900);\n  border-radius: 4px;\n}\n\n.button-blue:hover {\n  background-color: var(--blue-400);\n}\n\n.button-dark-purple {\n  background-color: var(--purple-500);\n  color: var(--white) !important;\n}\n\n.button-dark-purple:hover {\n  background-color: var(--purple-600);\n}\n\n.button-width-block {\n  width: 100%;\n}\n\n.button-width-fit-contents {\n  width: fit-content;\n}\n\n.button-small {\n  height: var(--toolbar-button-height);\n  font-size: 12px;\n  padding: 2px 5px;\n}\n"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -20578,6 +22320,60 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
        /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (_node_modules_css_loader_dist_cjs_js_CTACarousel_css__WEBPACK_IMPORTED_MODULE_6__["default"] && _node_modules_css_loader_dist_cjs_js_CTACarousel_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals ? _node_modules_css_loader_dist_cjs_js_CTACarousel_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals : undefined);
 
 
+/***/ }),
+
+/***/ "./style/CellReference.css":
+/*!*********************************!*\
+  !*** ./style/CellReference.css ***!
+  \*********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/injectStylesIntoStyleTag.js */ "./node_modules/style-loader/dist/runtime/injectStylesIntoStyleTag.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/styleDomAPI.js */ "./node_modules/style-loader/dist/runtime/styleDomAPI.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/insertBySelector.js */ "./node_modules/style-loader/dist/runtime/insertBySelector.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/setAttributesWithoutAttributes.js */ "./node_modules/style-loader/dist/runtime/setAttributesWithoutAttributes.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/insertStyleElement.js */ "./node_modules/style-loader/dist/runtime/insertStyleElement.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/styleTagTransform.js */ "./node_modules/style-loader/dist/runtime/styleTagTransform.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5__);
+/* harmony import */ var _node_modules_css_loader_dist_cjs_js_CellReference_css__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! !!../node_modules/css-loader/dist/cjs.js!./CellReference.css */ "./node_modules/css-loader/dist/cjs.js!./style/CellReference.css");
+
+      
+      
+      
+      
+      
+      
+      
+      
+      
+
+var options = {};
+
+options.styleTagTransform = (_node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5___default());
+options.setAttributes = (_node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3___default());
+
+      options.insert = _node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2___default().bind(null, "head");
+    
+options.domAPI = (_node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1___default());
+options.insertStyleElement = (_node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4___default());
+
+var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0___default()(_node_modules_css_loader_dist_cjs_js_CellReference_css__WEBPACK_IMPORTED_MODULE_6__["default"], options);
+
+
+
+
+       /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (_node_modules_css_loader_dist_cjs_js_CellReference_css__WEBPACK_IMPORTED_MODULE_6__["default"] && _node_modules_css_loader_dist_cjs_js_CellReference_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals ? _node_modules_css_loader_dist_cjs_js_CellReference_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals : undefined);
+
+
 /***/ }),
 
 /***/ "./style/ChatDropdown.css":
@@ -21766,6 +23562,60 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
        /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (_node_modules_css_loader_dist_cjs_js_RulesForm_css__WEBPACK_IMPORTED_MODULE_6__["default"] && _node_modules_css_loader_dist_cjs_js_RulesForm_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals ? _node_modules_css_loader_dist_cjs_js_RulesForm_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals : undefined);
 
 
+/***/ }),
+
+/***/ "./style/RunCellButton.css":
+/*!*********************************!*\
+  !*** ./style/RunCellButton.css ***!
+  \*********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/injectStylesIntoStyleTag.js */ "./node_modules/style-loader/dist/runtime/injectStylesIntoStyleTag.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/styleDomAPI.js */ "./node_modules/style-loader/dist/runtime/styleDomAPI.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/insertBySelector.js */ "./node_modules/style-loader/dist/runtime/insertBySelector.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/setAttributesWithoutAttributes.js */ "./node_modules/style-loader/dist/runtime/setAttributesWithoutAttributes.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/insertStyleElement.js */ "./node_modules/style-loader/dist/runtime/insertStyleElement.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/styleTagTransform.js */ "./node_modules/style-loader/dist/runtime/styleTagTransform.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5__);
+/* harmony import */ var _node_modules_css_loader_dist_cjs_js_RunCellButton_css__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! !!../node_modules/css-loader/dist/cjs.js!./RunCellButton.css */ "./node_modules/css-loader/dist/cjs.js!./style/RunCellButton.css");
+
+      
+      
+      
+      
+      
+      
+      
+      
+      
+
+var options = {};
+
+options.styleTagTransform = (_node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5___default());
+options.setAttributes = (_node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3___default());
+
+      options.insert = _node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2___default().bind(null, "head");
+    
+options.domAPI = (_node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1___default());
+options.insertStyleElement = (_node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4___default());
+
+var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0___default()(_node_modules_css_loader_dist_cjs_js_RunCellButton_css__WEBPACK_IMPORTED_MODULE_6__["default"], options);
+
+
+
+
+       /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (_node_modules_css_loader_dist_cjs_js_RunCellButton_css__WEBPACK_IMPORTED_MODULE_6__["default"] && _node_modules_css_loader_dist_cjs_js_RunCellButton_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals ? _node_modules_css_loader_dist_cjs_js_RunCellButton_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals : undefined);
+
+
 /***/ }),
 
 /***/ "./style/SelectedContextContainer.css":
@@ -22363,4 +24213,4 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
 /***/ })
 
 }]);
-//# sourceMappingURL=lib_index_js.31462f8f6a76b1cefbeb.js.map
+//# sourceMappingURL=lib_index_js.49c79c62671528877c61.js.map