mito-ai 0.1.51__py3-none-any.whl → 0.1.52__py3-none-any.whl

mito_ai-0.1.51.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.c0b356ea4ace64bf1f03.js → mito_ai-0.1.52.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.5ec1e525d244fc8588cf.js

@@ -666,22 +666,18 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const AgentChangeControls = ({ reviewAgentChanges, app, notebookTracker, setHasCheckpoint, setDisplayedNextStepsIfAvailable, setShowRevertQuestionnaire, chatTaskpaneMessagesRef, acceptAllAICode, rejectAllAICode, getChangeCounts, getReviewProgress, setAgentReviewStatus, agentReviewStatus, }) => {
-    const [changeCounts, setChangeCounts] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(null);
+const AgentChangeControls = ({ reviewAgentChanges, app, notebookTracker, setHasCheckpoint, setDisplayedNextStepsIfAvailable, setShowRevertQuestionnaire, chatTaskpaneMessagesRef, acceptAllAICode, rejectAllAICode, getReviewProgress, setAgentReviewStatus, agentReviewStatus, }) => {
     const [reviewProgress, setReviewProgress] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(null);
     // Update counts when review starts
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
         if (agentReviewStatus === 'in-agent-code-review') {
-            const counts = getChangeCounts();
-            setChangeCounts(counts);
             const progress = getReviewProgress();
             setReviewProgress(progress);
         }
         else {
-            setChangeCounts(null);
             setReviewProgress(null);
         }
-    }, [agentReviewStatus, getChangeCounts, getReviewProgress]);
+    }, [agentReviewStatus, getReviewProgress]);
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
         // Update progress periodically during review to reflect when cells are reviewed.
         // We use polling here instead of event-driven updates because it's simpler and more robust -
@@ -702,16 +698,11 @@ const AgentChangeControls = ({ reviewAgentChanges, app, notebookTracker, setHasC
     }, [agentReviewStatus]);
     const handleReviewChanges = () => {
         reviewAgentChanges();
-        // reviewAgentChanges populates changedCellsRef synchronously, so we can get counts immediately
-        const counts = getChangeCounts();
-        // Only set status if there are actually changes to review
-        if (counts.total > 0) {
-            setAgentReviewStatus('in-agent-code-review');
-            // Scroll to show the new review controls after the UI updates
-            setTimeout(() => {
-                (0,_utils_scroll__WEBPACK_IMPORTED_MODULE_2__.scrollToDiv)(chatTaskpaneMessagesRef);
-            }, 0);
-        }
+        setAgentReviewStatus('in-agent-code-review');
+        // Scroll to show the new review controls after the UI updates
+        setTimeout(() => {
+            (0,_utils_scroll__WEBPACK_IMPORTED_MODULE_2__.scrollToDiv)(chatTaskpaneMessagesRef);
+        }, 0);
     };
     const handleAcceptAll = () => {
         acceptAllAICode();
@@ -738,22 +729,6 @@ const AgentChangeControls = ({ reviewAgentChanges, app, notebookTracker, setHasC
     if (agentReviewStatus === 'in-agent-code-review') {
         return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'message message-assistant-chat' },
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "agent-change-controls-container" },
-                changeCounts && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "agent-change-counts" },
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count" },
-                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-number agent-change-count-added" },
-                            changeCounts.added,
-                            " \u00A0"),
-                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-text" }, changeCounts.added === 1 ? 'cell added' : 'cells added')),
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count" },
-                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-number agent-change-count-modified" },
-                            changeCounts.modified,
-                            " \u00A0"),
-                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-text" }, changeCounts.modified === 1 ? 'cell modified' : 'cells modified')),
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count" },
-                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-number agent-change-count-removed" },
-                            changeCounts.removed,
-                            " \u00A0"),
-                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-text" }, changeCounts.removed === 1 ? 'cell removed' : 'cells removed')))),
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "agent-change-controls-buttons" },
                     react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "button-base button-green", title: "Accept all changes", onClick: handleAcceptAll }, "Accept all"),
                     react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "button-base button-red", onClick: handleRejectAll }, "Reject All")),
@@ -772,6 +747,43 @@ const AgentChangeControls = ({ reviewAgentChanges, app, notebookTracker, setHasC
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AgentChangeControls);
 
 
+/***/ }),
+
+/***/ "./lib/Extensions/AiChat/ChatMessage/AgentChangeSummary.js":
+/*!*****************************************************************!*\
+  !*** ./lib/Extensions/AiChat/ChatMessage/AgentChangeSummary.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_AgentChangeControls_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/AgentChangeControls.css */ "./style/AgentChangeControls.css");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+const AgentChangeSummary = ({ getChangeCounts, }) => {
+    const changeCounts = getChangeCounts();
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "agent-change-counts" },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count" },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-number agent-change-count-added" }, changeCounts.added),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-text" }, changeCounts.added === 1 ? 'cell added' : 'cells added')),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count" },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-number agent-change-count-modified" }, changeCounts.modified),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-text" }, changeCounts.modified === 1 ? 'cell modified' : 'cells modified')),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count" },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-number agent-change-count-removed" }, changeCounts.removed),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-change-count-text" }, changeCounts.removed === 1 ? 'cell removed' : 'cells removed'))));
+};
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AgentChangeSummary);
+
+
 /***/ }),
 
 /***/ "./lib/Extensions/AiChat/ChatMessage/AlertBlock.js":
@@ -3041,7 +3053,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                 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, displayOptimizedChatHistoryLength: displayOptimizedChatHistory.length, showRevertQuestionnaire: showRevertQuestionnaire, reviewAgentChanges: agentReview.reviewAgentChanges, acceptAllAICode: agentReview.acceptAllAICode, rejectAllAICode: agentReview.rejectAllAICode, getChangeCounts: agentReview.getChangeCounts, getReviewProgress: agentReview.getReviewProgress, 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_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 })),
         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) => {
                     if (agentModeEnabled) {
@@ -3557,8 +3569,9 @@ __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 _ChatMessage_AgentChangeControls__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../ChatMessage/AgentChangeControls */ "./lib/Extensions/AiChat/ChatMessage/AgentChangeControls.js");
-/* harmony import */ var _ChatMessage_RevertQuestionnaire__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../ChatMessage/RevertQuestionnaire */ "./lib/Extensions/AiChat/ChatMessage/RevertQuestionnaire.js");
+/* harmony import */ var _ChatMessage_AgentChangeControls__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../ChatMessage/AgentChangeControls */ "./lib/Extensions/AiChat/ChatMessage/AgentChangeControls.js");
+/* harmony import */ var _ChatMessage_AgentChangeSummary__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../ChatMessage/AgentChangeSummary */ "./lib/Extensions/AiChat/ChatMessage/AgentChangeSummary.js");
+/* harmony import */ var _ChatMessage_RevertQuestionnaire__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../ChatMessage/RevertQuestionnaire */ "./lib/Extensions/AiChat/ChatMessage/RevertQuestionnaire.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -3566,13 +3579,16 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const AgentReviewPanel = ({ agentReviewStatus, setAgentReviewStatus, hasCheckpoint, agentModeEnabled, agentExecutionStatus, displayOptimizedChatHistoryLength, showRevertQuestionnaire, reviewAgentChanges, acceptAllAICode, rejectAllAICode, getChangeCounts, getReviewProgress, setHasCheckpoint, setDisplayedNextStepsIfAvailable, setShowRevertQuestionnaire, getDuplicateChatHistoryManager, setChatHistoryManager, app, notebookTracker, chatTaskpaneMessagesRef }) => {
+
+const AgentReviewPanel = ({ agentReviewStatus, setAgentReviewStatus, hasCheckpoint, agentModeEnabled, agentExecutionStatus, showRevertQuestionnaire, reviewAgentChanges, acceptAllAICode, rejectAllAICode, getChangeCounts, getReviewProgress, hasChanges, setHasCheckpoint, setDisplayedNextStepsIfAvailable, setShowRevertQuestionnaire, getDuplicateChatHistoryManager, setChatHistoryManager, app, notebookTracker, chatTaskpaneMessagesRef }) => {
+    const agentFinished = hasCheckpoint && agentModeEnabled && agentExecutionStatus === 'idle';
+    const shouldShowAgentChangeControls = agentFinished && hasChanges();
+    const shouldShowSummary = agentFinished && (agentReviewStatus === 'pre-agent-code-review' || agentReviewStatus === 'in-agent-code-review');
     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null,
-        hasCheckpoint &&
-            agentModeEnabled &&
-            agentExecutionStatus === 'idle' &&
-            displayOptimizedChatHistoryLength > 0 && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_AgentChangeControls__WEBPACK_IMPORTED_MODULE_1__["default"], { reviewAgentChanges: reviewAgentChanges, app: app, notebookTracker: notebookTracker, setHasCheckpoint: setHasCheckpoint, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable, setShowRevertQuestionnaire: setShowRevertQuestionnaire, chatTaskpaneMessagesRef: chatTaskpaneMessagesRef, acceptAllAICode: acceptAllAICode, rejectAllAICode: rejectAllAICode, getChangeCounts: getChangeCounts, getReviewProgress: getReviewProgress, agentReviewStatus: agentReviewStatus, setAgentReviewStatus: setAgentReviewStatus })),
-        showRevertQuestionnaire && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_RevertQuestionnaire__WEBPACK_IMPORTED_MODULE_2__["default"], { onDestroy: () => setShowRevertQuestionnaire(false), getDuplicateChatHistoryManager: getDuplicateChatHistoryManager, setChatHistoryManager: setChatHistoryManager }))));
+        shouldShowSummary && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'message message-assistant-chat' },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_AgentChangeSummary__WEBPACK_IMPORTED_MODULE_1__["default"], { getChangeCounts: getChangeCounts }))),
+        shouldShowAgentChangeControls && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_AgentChangeControls__WEBPACK_IMPORTED_MODULE_2__["default"], { reviewAgentChanges: reviewAgentChanges, app: app, notebookTracker: notebookTracker, setHasCheckpoint: setHasCheckpoint, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable, setShowRevertQuestionnaire: setShowRevertQuestionnaire, chatTaskpaneMessagesRef: chatTaskpaneMessagesRef, acceptAllAICode: acceptAllAICode, rejectAllAICode: rejectAllAICode, getReviewProgress: getReviewProgress, agentReviewStatus: agentReviewStatus, setAgentReviewStatus: setAgentReviewStatus })),
+        showRevertQuestionnaire && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatMessage_RevertQuestionnaire__WEBPACK_IMPORTED_MODULE_3__["default"], { onDestroy: () => setShowRevertQuestionnaire(false), getDuplicateChatHistoryManager: getDuplicateChatHistoryManager, setChatHistoryManager: setChatHistoryManager }))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AgentReviewPanel);
 
@@ -3884,9 +3900,9 @@ __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 _AgentReviewUtils__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../AgentReviewUtils */ "./lib/Extensions/AiChat/AgentReviewUtils.js");
+/* harmony import */ var _AgentReviewUtils__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../AgentReviewUtils */ "./lib/Extensions/AiChat/AgentReviewUtils.js");
 /* harmony import */ var _utils_codeDiff__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/codeDiff */ "./lib/utils/codeDiff.js");
-/* 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_1__ = __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.
@@ -3922,47 +3938,19 @@ const useAgentReview = ({ app, agentTargetNotebookPanelRef, codeDiffStripesCompa
             total: cells.length
         };
     };
-    const acceptAICodeInAgentMode = () => {
-        var _a, _b;
-        const activeCellId = (_b = (_a = agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current) === null || _a === void 0 ? void 0 : _a.content.activeCell) === null || _b === void 0 ? void 0 : _b.model.id;
-        if (!activeCellId || !hasUnreviewedChanges(activeCellId)) {
-            return;
-        }
-        (0,_AgentReviewUtils__WEBPACK_IMPORTED_MODULE_1__.acceptSingleCellEdit)(activeCellId, agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current, notebookSnapshotAfterAgentExecutionRef.current, codeDiffStripesCompartments, changedCellsRef.current, setAgentReviewStatus);
-        updateCellToolbarButtons();
-    };
-    const rejectAICodeInAgentMode = () => {
-        var _a, _b;
-        const activeCellId = (_b = (_a = agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current) === null || _a === void 0 ? void 0 : _a.content.activeCell) === null || _b === void 0 ? void 0 : _b.model.id;
-        if (!activeCellId || !hasUnreviewedChanges(activeCellId)) {
-            return;
-        }
-        (0,_AgentReviewUtils__WEBPACK_IMPORTED_MODULE_1__.rejectSingleCellEdit)(activeCellId, agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current, codeDiffStripesCompartments, changedCellsRef.current, setAgentReviewStatus, agentTargetNotebookPanelRef || undefined);
-        updateCellToolbarButtons();
-    };
-    const acceptAllAICode = () => {
-        (0,_AgentReviewUtils__WEBPACK_IMPORTED_MODULE_1__.acceptAllCellEdits)(agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current, notebookSnapshotAfterAgentExecutionRef.current, codeDiffStripesCompartments, changedCellsRef.current);
-        updateCellToolbarButtons();
-    };
-    const rejectAllAICode = () => {
-        (0,_AgentReviewUtils__WEBPACK_IMPORTED_MODULE_1__.rejectAllCellEdits)(agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current, codeDiffStripesCompartments, changedCellsRef.current);
-        updateCellToolbarButtons();
-    };
-    const reviewAgentChanges = () => {
+    // Helper function to detect and count changes between snapshots
+    // This populates changedCellsRef and changeCountsRef without applying visual diffs
+    const detectChanges = () => {
         var _a;
         if (!(agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current)) {
-            return;
+            return [];
         }
-        // Make the notebook panel the active notebook panel
-        app.shell.activateById(agentTargetNotebookPanelRef.current.id);
-        const currentNotebookSnapshot = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_2__.getAIOptimizedCellsInNotebookPanel)(agentTargetNotebookPanelRef.current);
-        notebookSnapshotAfterAgentExecutionRef.current = currentNotebookSnapshot;
+        const currentNotebookSnapshot = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_1__.getAIOptimizedCellsInNotebookPanel)(agentTargetNotebookPanelRef.current);
         if (!notebookSnapshotPreAgentExecutionRef.current || !currentNotebookSnapshot) {
-            return;
+            return [];
         }
         // Clear and populate the changed cells array
         const changedCells = [];
-        changedCellsRef.current = changedCells;
         // Initialize counters
         let added = 0;
         let modified = 0;
@@ -4015,6 +4003,50 @@ const useAgentReview = ({ app, agentTargetNotebookPanelRef, codeDiffStripesCompa
             removed,
             total: changedCells.length
         };
+        return changedCells;
+    };
+    const hasChanges = () => {
+        detectChanges();
+        // changeCountsRef is populated by detectChanges()
+        return changeCountsRef.current.total > 0;
+    };
+    const acceptAICodeInAgentMode = () => {
+        var _a, _b;
+        const activeCellId = (_b = (_a = agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current) === null || _a === void 0 ? void 0 : _a.content.activeCell) === null || _b === void 0 ? void 0 : _b.model.id;
+        if (!activeCellId || !hasUnreviewedChanges(activeCellId)) {
+            return;
+        }
+        (0,_AgentReviewUtils__WEBPACK_IMPORTED_MODULE_2__.acceptSingleCellEdit)(activeCellId, agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current, notebookSnapshotAfterAgentExecutionRef.current, codeDiffStripesCompartments, changedCellsRef.current, setAgentReviewStatus);
+        updateCellToolbarButtons();
+    };
+    const rejectAICodeInAgentMode = () => {
+        var _a, _b;
+        const activeCellId = (_b = (_a = agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current) === null || _a === void 0 ? void 0 : _a.content.activeCell) === null || _b === void 0 ? void 0 : _b.model.id;
+        if (!activeCellId || !hasUnreviewedChanges(activeCellId)) {
+            return;
+        }
+        (0,_AgentReviewUtils__WEBPACK_IMPORTED_MODULE_2__.rejectSingleCellEdit)(activeCellId, agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current, codeDiffStripesCompartments, changedCellsRef.current, setAgentReviewStatus, agentTargetNotebookPanelRef || undefined);
+        updateCellToolbarButtons();
+    };
+    const acceptAllAICode = () => {
+        (0,_AgentReviewUtils__WEBPACK_IMPORTED_MODULE_2__.acceptAllCellEdits)(agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current, notebookSnapshotAfterAgentExecutionRef.current, codeDiffStripesCompartments, changedCellsRef.current);
+        updateCellToolbarButtons();
+    };
+    const rejectAllAICode = () => {
+        (0,_AgentReviewUtils__WEBPACK_IMPORTED_MODULE_2__.rejectAllCellEdits)(agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current, codeDiffStripesCompartments, changedCellsRef.current);
+        updateCellToolbarButtons();
+    };
+    const reviewAgentChanges = () => {
+        if (!(agentTargetNotebookPanelRef === null || agentTargetNotebookPanelRef === void 0 ? void 0 : agentTargetNotebookPanelRef.current)) {
+            return;
+        }
+        // Make the notebook panel the active notebook panel
+        app.shell.activateById(agentTargetNotebookPanelRef.current.id);
+        const currentNotebookSnapshot = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_1__.getAIOptimizedCellsInNotebookPanel)(agentTargetNotebookPanelRef.current);
+        notebookSnapshotAfterAgentExecutionRef.current = currentNotebookSnapshot;
+        // Detect changes and populate refs
+        const changedCells = detectChanges();
+        changedCellsRef.current = changedCells;
         if (changedCells.length === 0) {
             console.log('No changes detected between snapshots');
             return;
@@ -4024,16 +4056,16 @@ const useAgentReview = ({ app, agentTargetNotebookPanelRef, codeDiffStripesCompa
             // Calculate the code diffs
             const { unifiedCodeString, unifiedDiffs } = (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_3__.getCodeDiffsAndUnifiedCodeString)(change.originalCode, change.currentCode);
             // Write the unified code string to the cell
-            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_2__.writeCodeToCellByIDInNotebookPanel)(agentTargetNotebookPanelRef.current, unifiedCodeString, change.cellId);
+            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_1__.writeCodeToCellByIDInNotebookPanel)(agentTargetNotebookPanelRef.current, unifiedCodeString, change.cellId);
             // Apply diff stripes to this cell
             (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_3__.applyDiffStripesToCell)(agentTargetNotebookPanelRef.current, change.cellId, unifiedDiffs, codeDiffStripesCompartments.current);
             // Highlight the cell to draw attention
-            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_2__.highlightCodeCellInNotebookPanel)(agentTargetNotebookPanelRef.current, change.cellId);
+            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_1__.highlightCodeCellInNotebookPanel)(agentTargetNotebookPanelRef.current, change.cellId);
         });
         // Scroll to the first changed cell
         const firstChangedCell = changedCells[0];
         if (firstChangedCell && agentTargetNotebookPanelRef.current) {
-            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_2__.scrollToCell)(agentTargetNotebookPanelRef.current, firstChangedCell.cellId, undefined, 'start');
+            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_1__.scrollToCell)(agentTargetNotebookPanelRef.current, firstChangedCell.cellId, undefined, 'start');
         }
         // Update toolbar buttons to show accept/reject buttons for cells with diffs
         updateCellToolbarButtons();
@@ -4061,7 +4093,8 @@ const useAgentReview = ({ app, agentTargetNotebookPanelRef, codeDiffStripesCompa
         setNotebookSnapshotPreAgentExecution,
         hasUnreviewedChanges,
         getChangeCounts,
-        getReviewProgress
+        getReviewProgress,
+        hasChanges
     };
 };
 
@@ -5249,15 +5282,16 @@ __webpack_require__.r(__webpack_exports__);
 /* 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 _requirementsUtils__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ./requirementsUtils */ "./lib/Extensions/AppDeploy/requirementsUtils.js");
-/* harmony import */ var _fileUtils__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./fileUtils */ "./lib/Extensions/AppDeploy/fileUtils.js");
+/* harmony import */ var _requirementsUtils__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./requirementsUtils */ "./lib/Extensions/AppDeploy/requirementsUtils.js");
+/* harmony import */ var _fileUtils__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./fileUtils */ "./lib/Extensions/AppDeploy/fileUtils.js");
 /* 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 _DeployAppNotification__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./DeployAppNotification */ "./lib/Extensions/AppDeploy/DeployAppNotification.js");
+/* harmony import */ var _DeployAppNotification__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./DeployAppNotification */ "./lib/Extensions/AppDeploy/DeployAppNotification.js");
 /* harmony import */ var _auth__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ./auth */ "./lib/Extensions/AppDeploy/auth.js");
 /* harmony import */ var _authPopupUtils__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ./authPopupUtils */ "./lib/Extensions/AppDeploy/authPopupUtils.js");
-/* harmony import */ var _FilesSelectorUtils__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./FilesSelectorUtils */ "./lib/Extensions/AppDeploy/FilesSelectorUtils.js");
+/* harmony import */ var _FilesSelectorUtils__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./FilesSelectorUtils */ "./lib/Extensions/AppDeploy/FilesSelectorUtils.js");
 /* harmony import */ var _utils_notebookMetadata__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../utils/notebookMetadata */ "./lib/utils/notebookMetadata.js");
+/* harmony import */ var _AppPreview_utils__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../AppPreview/utils */ "./lib/Extensions/AppPreview/utils.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -5271,6 +5305,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 /*
 This function generates the requirements.txt file needed to host the streamlit app,
 and deploys it!
@@ -5305,16 +5340,17 @@ const deployStreamlitApp = async (notebookPanel, appDeployService, appManagerSer
     }
     const notebookPath = notebookPanel.context.path;
     const notebookID = (0,_utils_notebookMetadata__WEBPACK_IMPORTED_MODULE_4__.getNotebookIDAndSetIfNonexistant)(notebookPanel);
+    const appFilename = notebookID ? (0,_AppPreview_utils__WEBPACK_IMPORTED_MODULE_5__.getAppNameFromNotebookID)(notebookID) : 'app.py';
     const notificationId = _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__.Notification.emit('Step 1/7: Gathering requirements...', 'in-progress', {
         autoClose: false
     });
     // Build the requirements.txt file
-    const requirementsContent = await (0,_requirementsUtils__WEBPACK_IMPORTED_MODULE_5__.generateRequirementsTxt)(notebookPanel);
+    const requirementsContent = await (0,_requirementsUtils__WEBPACK_IMPORTED_MODULE_6__.generateRequirementsTxt)(notebookPanel, appFilename);
     // Save the files to the current directory
-    await (0,_fileUtils__WEBPACK_IMPORTED_MODULE_6__.saveFileWithKernel)(notebookPanel, './requirements.txt', requirementsContent);
+    await (0,_fileUtils__WEBPACK_IMPORTED_MODULE_7__.saveFileWithKernel)(notebookPanel, './requirements.txt', requirementsContent);
     try {
         _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__.Notification.dismiss(notificationId);
-        selectedFiles = await (0,_FilesSelectorUtils__WEBPACK_IMPORTED_MODULE_7__.fileSelectorPopup)(notebookPanel);
+        selectedFiles = await (0,_FilesSelectorUtils__WEBPACK_IMPORTED_MODULE_8__.fileSelectorPopup)(notebookPanel);
     }
     catch (error) {
         const errorMsg = 'Files selection failed: ' + error;
@@ -5359,7 +5395,7 @@ const deployStreamlitApp = async (notebookPanel, appDeployService, appManagerSer
         else {
             console.log("App deployment response:", response);
             const url = response.url;
-            (0,_DeployAppNotification__WEBPACK_IMPORTED_MODULE_8__.deployAppNotification)(url, appManagerService, newNotificationId);
+            (0,_DeployAppNotification__WEBPACK_IMPORTED_MODULE_9__.deployAppNotification)(url, appManagerService, newNotificationId);
         }
     }
     catch (error) {
@@ -5704,9 +5740,9 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
-const userPoolIdDev = 'us-east-1_Kk0f9mOfx';
-const userPoolClientIdDev = '6ara3u3l8sss738hrhbq1qtiqf';
-// TODO: modify to prod user ID and client ID on after creating teh cognito resources in prod
+const userPoolIdDev = 'us-east-1_s2V9RiyOa';
+const userPoolClientIdDev = '2sunerv2m6gp1qk3hib4t8oblh';
+// TODO: modify to prod user ID and client ID on after creating the cognito resources in prod
 const userPoolId = userPoolIdDev;
 const userPoolClientId = userPoolClientIdDev;
 const configureAmplify = () => {
@@ -5808,37 +5844,47 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 // Function to generate requirements.txt content using the kernel with pipreqs
-const generateRequirementsTxt = async (notebookPanel) => {
+const generateRequirementsTxt = async (notebookPanel, appFilename) => {
     // Initialize with fallback requirements in case kernel execution fails
     let requirementsContent = 'streamlit>=1.28.0';
     try {
         // Use the kernel to execute Python code using pipreqs
         const session = notebookPanel.sessionContext.session;
         if (session) {
-            const appPyPath = `app.py`;
-            // Create Python code to run pipreqs on the app.py file
+            const sanitizedAppFilename = JSON.stringify(appFilename);
+            // Create Python code to run pipreqs on the app file
             const pythonCode = `
 import subprocess
 import os
+import tempfile
+import shutil
 
-# Check if app.py exists in the notebook directory
-app_py_path = os.path.join(os.getcwd(), "${appPyPath}")
+# Check if the Streamlit app file exists in the notebook directory
+app_py_filename = ${sanitizedAppFilename}
+app_py_path = os.path.join(os.getcwd(), app_py_filename)
 if not os.path.exists(app_py_path):
-    print(f"Error: app.py not found at {app_py_path}")
+    print(f"Error: {app_py_filename} not found at {app_py_path}")
     exit(1)
+notebook_dir = os.path.dirname(app_py_path)
 
-# Make sure pipreqs is installed. Then
-# 1. Create a requirements.in file
-# 2. From the requirements.in file, generate the requirements.txt file with the canonical PyPI name of the packages
-# and the versions as they exist on the user's terminal
 try:
-    # Run pipreqs on the directory containing app.py
-    notebook_dir = os.path.dirname(app_py_path)
-    generate_req_in_file = subprocess.run(
-        ['pipreqs', '--encoding=utf-8', '--savepath', 'requirements.in', '--force', notebook_dir],
-        capture_output=True, 
-        text=True
-    )
+    # Create a temporary directory and copy the app file into it
+    with tempfile.TemporaryDirectory() as tmpdir:
+        tmp_app_path = os.path.join(tmpdir, app_py_filename)
+        shutil.copy(app_py_path, tmp_app_path)
+
+        # Run pipreqs on the temporary directory
+        generate_req_in_file = subprocess.run(
+            [
+                'pipreqs',
+                '--encoding=utf-8',
+                '--savepath', os.path.join(notebook_dir, 'requirements.in'),
+                '--force',
+                tmpdir
+            ],
+            capture_output=True,
+            text=True
+        )
 
     print("Log: ", generate_req_in_file.stderr)
 
@@ -10247,7 +10293,8 @@ __webpack_require__.r(__webpack_exports__);
 const MODEL_MAPPINGS = [
     { displayName: 'GPT 4.1', fullName: 'gpt-4.1' },
     { displayName: _utils_models__WEBPACK_IMPORTED_MODULE_2__.CLAUDE_SONNET_DISPLAY_NAME, fullName: _utils_models__WEBPACK_IMPORTED_MODULE_2__.CLAUDE_SONNET_MODEL_NAME },
-    { displayName: 'Gemini 2.5 Pro', fullName: 'gemini-2.5-pro-preview-03-25' }
+    { displayName: 'Gemini 2.5 Pro', fullName: 'gemini-2.5-pro-preview-03-25' },
+    { displayName: 'Gemini 3 Pro', fullName: 'gemini-3-pro-preview' }
 ];
 const ALL_MODEL_DISPLAY_NAMES = MODEL_MAPPINGS.map(mapping => mapping.displayName);
 // Maximum length for displayed model name before truncating
@@ -14442,25 +14489,27 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
     font-size: 12px;
     margin-bottom: 4px;
     display: flex;
-    flex-direction: column;
-    gap: 8px;
-    align-items: start;
+    flex-direction: row;
+    gap: 16px;
+    align-items: center;
 }
 
 .agent-change-count {
     display: inline-flex;
-    align-items: center;
+    align-items: baseline;
+    gap: 4px;
 }
 
 .agent-change-count-number {
-    font-weight: 500;
+    font-weight: bold;
+    font-size: 14px;
     display: inline-block;
-    min-width: 24px;
     text-align: left;
 }
 
 .agent-change-count-text {
     color: var(--muted-text-color);
+    font-size: 11px;
 }
 
 .agent-change-count-added {
@@ -14497,7 +14546,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
     align-items: center;
 }
 
-`, "",{"version":3,"sources":["webpack://./style/AgentChangeControls.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,eAAe;IACf,kBAAkB;IAClB,aAAa;IACb,sBAAsB;IACtB,QAAQ;IACR,kBAAkB;AACtB;;AAEA;IACI,oBAAoB;IACpB,mBAAmB;AACvB;;AAEA;IACI,gBAAgB;IAChB,qBAAqB;IACrB,eAAe;IACf,gBAAgB;AACpB;;AAEA;IACI,8BAA8B;AAClC;;AAEA;IACI,uBAAuB;AAC3B;;AAEA;IACI,wBAAwB;AAC5B;;AAEA;IACI,qBAAqB;AACzB;;AAEA;IACI,8BAA8B;IAC9B,aAAa;AACjB;;AAEA;IACI,eAAe;IACf,8BAA8B;AAClC;;AAEA;IACI,aAAa;IACb,sBAAsB;IACtB,QAAQ;AACZ;;AAEA;IACI,aAAa;IACb,QAAQ;IACR,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.agent-change-counts {\n    font-size: 12px;\n    margin-bottom: 4px;\n    display: flex;\n    flex-direction: column;\n    gap: 8px;\n    align-items: start;\n}\n\n.agent-change-count {\n    display: inline-flex;\n    align-items: center;\n}\n\n.agent-change-count-number {\n    font-weight: 500;\n    display: inline-block;\n    min-width: 24px;\n    text-align: left;\n}\n\n.agent-change-count-text {\n    color: var(--muted-text-color);\n}\n\n.agent-change-count-added {\n    color: var(--green-700);\n}\n\n.agent-change-count-modified {\n    color: var(--yellow-600);\n}\n\n.agent-change-count-removed {\n    color: var(--red-700);\n}\n\n.agent-change-count-separator {\n    color: var(--muted-text-color);\n    margin: 0 4px;\n}\n\n.agent-review-progress {\n    font-size: 12px;\n    color: var(--muted-text-color);\n}\n\n.agent-change-controls-container {\n    display: flex;\n    flex-direction: column;\n    gap: 8px;\n}\n\n.agent-change-controls-buttons {\n    display: flex;\n    gap: 8px;\n    align-items: center;\n}\n\n"],"sourceRoot":""}]);
+`, "",{"version":3,"sources":["webpack://./style/AgentChangeControls.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,eAAe;IACf,kBAAkB;IAClB,aAAa;IACb,mBAAmB;IACnB,SAAS;IACT,mBAAmB;AACvB;;AAEA;IACI,oBAAoB;IACpB,qBAAqB;IACrB,QAAQ;AACZ;;AAEA;IACI,iBAAiB;IACjB,eAAe;IACf,qBAAqB;IACrB,gBAAgB;AACpB;;AAEA;IACI,8BAA8B;IAC9B,eAAe;AACnB;;AAEA;IACI,uBAAuB;AAC3B;;AAEA;IACI,wBAAwB;AAC5B;;AAEA;IACI,qBAAqB;AACzB;;AAEA;IACI,8BAA8B;IAC9B,aAAa;AACjB;;AAEA;IACI,eAAe;IACf,8BAA8B;AAClC;;AAEA;IACI,aAAa;IACb,sBAAsB;IACtB,QAAQ;AACZ;;AAEA;IACI,aAAa;IACb,QAAQ;IACR,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.agent-change-counts {\n    font-size: 12px;\n    margin-bottom: 4px;\n    display: flex;\n    flex-direction: row;\n    gap: 16px;\n    align-items: center;\n}\n\n.agent-change-count {\n    display: inline-flex;\n    align-items: baseline;\n    gap: 4px;\n}\n\n.agent-change-count-number {\n    font-weight: bold;\n    font-size: 14px;\n    display: inline-block;\n    text-align: left;\n}\n\n.agent-change-count-text {\n    color: var(--muted-text-color);\n    font-size: 11px;\n}\n\n.agent-change-count-added {\n    color: var(--green-700);\n}\n\n.agent-change-count-modified {\n    color: var(--yellow-600);\n}\n\n.agent-change-count-removed {\n    color: var(--red-700);\n}\n\n.agent-change-count-separator {\n    color: var(--muted-text-color);\n    margin: 0 4px;\n}\n\n.agent-review-progress {\n    font-size: 12px;\n    color: var(--muted-text-color);\n}\n\n.agent-change-controls-container {\n    display: flex;\n    flex-direction: column;\n    gap: 8px;\n}\n\n.agent-change-controls-buttons {\n    display: flex;\n    gap: 8px;\n    align-items: center;\n}\n\n"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -21843,4 +21892,4 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
 /***/ })
 
 }]);
-//# sourceMappingURL=lib_index_js.c0b356ea4ace64bf1f03.js.map
+//# sourceMappingURL=lib_index_js.5ec1e525d244fc8588cf.js.map