mito-ai 0.1.47__py3-none-any.whl → 0.1.48__py3-none-any.whl

mito_ai-0.1.47.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.2db61d2b629817845901.js → mito_ai-0.1.48.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.5c7d84a45ddeb5704b61.js

@@ -360,6 +360,7 @@ class ChatHistoryManager {
         const streamlitAppIsOpen = this.checkIfStreamlitAppIsOpen();
         const agentExecutionMetadata = {
             promptType: 'agent:execution',
+            activeCellId: (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_1__.getActiveCellIDInNotebookPanel)(notebookPanel) || '',
             variables: (notebookContext === null || notebookContext === void 0 ? void 0 : notebookContext.variables) || [],
             files: (notebookContext === null || notebookContext === void 0 ? void 0 : notebookContext.files) || [],
             aiOptimizedCells: aiOptimizedCells,
@@ -1022,15 +1023,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_5__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _ChatDropdown__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./ChatDropdown */ "./lib/Extensions/AiChat/ChatMessage/ChatDropdown.js");
+/* 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_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.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_4__ = __webpack_require__(/*! ../../../hooks/useDebouncedFunction */ "./lib/hooks/useDebouncedFunction.js");
-/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
-/* harmony import */ var _components_AttachFileButton__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../components/AttachFileButton */ "./lib/components/AttachFileButton.js");
-/* harmony import */ var _components_DatabaseButton__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../components/DatabaseButton */ "./lib/components/DatabaseButton.js");
+/* 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 _utils_fileUpload__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../utils/fileUpload */ "./lib/utils/fileUpload.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -1045,6 +1047,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 const ChatInput = ({ app, initialContent, onSave, onCancel, isEditing, contextManager, notebookTracker, agentModeEnabled = false, agentExecutionStatus = 'idle', operatingSystem = 'mac', displayOptimizedChatHistoryLength = 0, agentTargetNotebookPanelRef, isSignedUp = true, }) => {
     var _a;
     const [input, setInput] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(initialContent);
@@ -1055,6 +1058,8 @@ const ChatInput = ({ app, initialContent, onSave, onCancel, isEditing, contextMa
     const [dropdownFilter, setDropdownFilter] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)('');
     const [additionalContext, setAdditionalContext] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)([]);
     const [isDropdownFromButton, setIsDropdownFromButton] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    const [isDragOver, setIsDragOver] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    const [isUploading, setIsUploading] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     const handleFileUpload = (file) => {
         let uploadType;
         if (file.type.startsWith('image/')) {
@@ -1075,9 +1080,45 @@ const ChatInput = ({ app, initialContent, onSave, onCancel, isEditing, contextMa
             }
         ]);
     };
+    // Drag and drop handlers
+    const handleDragOver = (e) => {
+        e.preventDefault();
+        e.stopPropagation();
+        // Only show drag over state if not currently uploading
+        if (!isUploading) {
+            setIsDragOver(true);
+        }
+    };
+    const handleDragLeave = (e) => {
+        e.preventDefault();
+        e.stopPropagation();
+        setIsDragOver(false);
+    };
+    const handleDrop = async (e) => {
+        e.preventDefault();
+        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);
+                }
+            }
+        }
+    };
     // Debounce the active cell ID change to avoid multiple rerenders. 
     // We use this to avoid a flickering screen when the active cell changes. 
-    const debouncedSetActiveCellID = (0,_hooks_useDebouncedFunction__WEBPACK_IMPORTED_MODULE_4__.useDebouncedFunction)((newID) => {
+    const debouncedSetActiveCellID = (0,_hooks_useDebouncedFunction__WEBPACK_IMPORTED_MODULE_5__.useDebouncedFunction)((newID) => {
         setActiveCellID(newID);
     }, 100);
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
@@ -1305,10 +1346,13 @@ const ChatInput = ({ app, initialContent, onSave, onCancel, isEditing, contextMa
             }
         }
     }, [agentModeEnabled, additionalContext, activeCellCode]);
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)("chat-input-container", { "editing": isEditing }) },
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_6__.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_6__["default"], { app: app }),
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AttachFileButton__WEBPACK_IMPORTED_MODULE_7__["default"], { onFileUploaded: handleFileUpload, notebookTracker: notebookTracker }),
+            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("button", { className: "context-button", onClick: () => {
                     var _a;
                     setDropdownVisible(true);
@@ -1316,9 +1360,9 @@ const ChatInput = ({ app, initialContent, onSave, onCancel, isEditing, contextMa
                     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_8__["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_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 })))),
         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_5__.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_6__.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') {
@@ -1345,7 +1389,7 @@ const ChatInput = ({ app, initialContent, onSave, onCancel, isEditing, contextMa
                         }
                     }
                 } }),
-            isDropdownVisible && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdown__WEBPACK_IMPORTED_MODULE_9__["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_10__["default"], { options: getExpandedVarialbes(), onSelect: handleOptionSelect, filterText: dropdownFilter, isDropdownFromButton: isDropdownFromButton, onFilterChange: setDropdownFilter, onClose: handleDropdownClose }))),
         isEditing &&
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "message-edit-buttons" },
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { onClick: () => onSave(input, undefined, mapAdditionalContext()) }, "Save"),
@@ -4255,6 +4299,10 @@ const deployStreamlitApp = async (notebookPanel, appDeployService, appManagerSer
         }
         catch (error) {
             console.log('Authentication cancelled or failed:', error);
+            const errorMsg = 'Authentication failed: ' + error;
+            _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__.Notification.emit(errorMsg, 'error', {
+                autoClose: false
+            });
             return; // Exit early if authentication was cancelled
         }
     }
@@ -4271,6 +4319,10 @@ const deployStreamlitApp = async (notebookPanel, appDeployService, appManagerSer
         selectedFiles = await (0,_FilesSelectorUtils__WEBPACK_IMPORTED_MODULE_6__.fileSelectorPopup)(notebookPath);
     }
     catch (error) {
+        const errorMsg = 'Files selection failed: ' + error;
+        _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__.Notification.emit(errorMsg, 'error', {
+            autoClose: false
+        });
         console.log('File selection failed:', error);
         return;
     }
@@ -8831,11 +8883,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 _IconButton__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ./IconButton */ "./lib/components/IconButton.js");
-/* harmony import */ var _icons_PaperClipIcon__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../icons/PaperClipIcon */ "./lib/icons/PaperClipIcon.js");
-/* harmony import */ var _restAPI_utils__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../restAPI/utils */ "./lib/restAPI/utils.js");
-/* harmony import */ var _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! @jupyterlab/apputils */ "webpack/sharing/consume/default/@jupyterlab/apputils");
-/* harmony import */ var _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__);
+/* harmony import */ var _IconButton__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ./IconButton */ "./lib/components/IconButton.js");
+/* harmony import */ var _icons_PaperClipIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../icons/PaperClipIcon */ "./lib/icons/PaperClipIcon.js");
+/* harmony import */ var _utils_fileUpload__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../utils/fileUpload */ "./lib/utils/fileUpload.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -8844,22 +8894,9 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-
-// Constants for file handling
-const CHUNKED_UPLOAD_SIZE_CUTOFF = 25 * 1024 * 1024; // 25MB cutoff for chunked uploads
-const CHUNK_SIZE = 10 * 1024 * 1024; // 10MB chunks
 const AttachFileButton = ({ onFileUploaded, notebookTracker }) => {
     const fileInputRef = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(null);
     const [isUploading, setIsUploading] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
-    // Helper function to get notebook directory from notebook path
-    const getNotebookDirectory = (notebookPath) => {
-        const lastSlashIndex = notebookPath.lastIndexOf('/');
-        if (lastSlashIndex === -1) {
-            // No directory, just filename (root directory)
-            return '.';
-        }
-        return notebookPath.substring(0, lastSlashIndex);
-    };
     const handleClick = () => {
         var _a;
         // Don't allow clicks if uploading
@@ -8880,21 +8917,11 @@ const AttachFileButton = ({ onFileUploaded, notebookTracker }) => {
             return;
         setIsUploading(true);
         try {
-            // Check file size and handle accordingly
-            if (file.size > CHUNKED_UPLOAD_SIZE_CUTOFF) {
-                console.log(`File ${file.name} is larger than 25MB (${(file.size / (1024 * 1024)).toFixed(2)}MB). Splitting into chunks...`);
-                await handleLargeFile(file);
-            }
-            else {
-                // Upload file directly for files <= 25MB
-                await uploadFile(file);
-            }
+            // Use the shared upload utility
+            await (0,_utils_fileUpload__WEBPACK_IMPORTED_MODULE_1__.uploadFileToBackend)(file, notebookTracker, onFileUploaded);
         }
         catch (error) {
-            _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__.Notification.emit(`Upload failed: ${error}`, "error", {
-                autoClose: 5 * 1000 // 5 seconds
-            });
-            console.error('Error during file upload:', error);
+            // Error handling is already done in the utility function
         }
         finally {
             setIsUploading(false);
@@ -8904,116 +8931,9 @@ const AttachFileButton = ({ onFileUploaded, notebookTracker }) => {
             }
         }
     };
-    const handleLargeFile = async (file) => {
-        const totalChunks = Math.ceil(file.size / CHUNK_SIZE);
-        console.log(`Splitting file into ${totalChunks} chunks of ${(CHUNK_SIZE / (1024 * 1024)).toFixed(2)}MB each`);
-        // Get notebook directory path
-        const notebookPanel = notebookTracker.currentWidget;
-        if (!notebookPanel) {
-            console.error('No notebook is currently active');
-            throw new Error('No notebook is currently active');
-        }
-        const notebookPath = notebookPanel.context.path;
-        const notebookDir = getNotebookDirectory(notebookPath);
-        for (let chunkIndex = 0; chunkIndex < totalChunks; chunkIndex++) {
-            const start = chunkIndex * CHUNK_SIZE;
-            const end = Math.min(start + CHUNK_SIZE, file.size);
-            const chunk = file.slice(start, end);
-            console.log(`Uploading chunk ${chunkIndex + 1}/${totalChunks}:`, {
-                chunkNumber: chunkIndex + 1,
-                totalChunks: totalChunks,
-                chunkSize: chunk.size,
-                chunkSizeMB: (chunk.size / (1024 * 1024)).toFixed(2),
-                startByte: start,
-                endByte: end,
-                fileName: file.name,
-                originalFileSize: file.size,
-                originalFileSizeMB: (file.size / (1024 * 1024)).toFixed(2)
-            });
-            // Upload chunk to backend
-            const success = await uploadChunk(chunk, file.name, chunkIndex + 1, totalChunks, notebookDir);
-            if (!success) {
-                console.error(`Failed to upload chunk ${chunkIndex + 1}`);
-                throw new Error(`Failed to upload chunk ${chunkIndex + 1}`);
-            }
-        }
-        console.log(`Successfully uploaded all ${totalChunks} chunks for file: ${file.name}`);
-        // Notify the parent component that the file was uploaded
-        onFileUploaded(file);
-    };
-    const uploadChunk = async (chunk, filename, chunkNumber, totalChunks, notebookDir) => {
-        try {
-            // Create FormData for chunk upload
-            const formData = new FormData();
-            formData.append('file', chunk, filename);
-            formData.append('chunk_number', chunkNumber.toString());
-            formData.append('total_chunks', totalChunks.toString());
-            formData.append('notebook_dir', notebookDir);
-            // Upload chunk to backend
-            const resp = await (0,_restAPI_utils__WEBPACK_IMPORTED_MODULE_2__.requestAPI)('upload', {
-                method: 'POST',
-                body: formData
-            });
-            if (resp.error) {
-                _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__.Notification.emit(`Upload failed: ${resp.error.message}`, "error", {
-                    autoClose: 5 * 1000 // 5 seconds
-                });
-                console.error(`Chunk ${chunkNumber} upload failed:`, resp.error.message);
-                return false;
-            }
-            else if (resp.data) {
-                if (resp.data.chunk_complete) {
-                    console.log(`All chunks uploaded successfully. File reconstructed: ${resp.data.filename}`);
-                }
-                else if (resp.data.chunk_received) {
-                    console.log(`Chunk ${resp.data.chunk_number}/${resp.data.total_chunks} uploaded successfully`);
-                }
-                return true;
-            }
-            return false;
-        }
-        catch (error) {
-            _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__.Notification.emit(`Upload failed: ${error}`, "error", {
-                autoClose: 5 * 1000 // 5 seconds
-            });
-            console.error(`Error uploading chunk ${chunkNumber}:`, error);
-            return false;
-        }
-    };
-    const uploadFile = async (file) => {
-        // Get notebook directory path
-        const notebookPanel = notebookTracker.currentWidget;
-        if (!notebookPanel) {
-            console.error('No notebook is currently active');
-            throw new Error('No notebook is currently active');
-        }
-        const notebookPath = notebookPanel.context.path;
-        const notebookDir = getNotebookDirectory(notebookPath);
-        // Create FormData for file upload
-        const formData = new FormData();
-        formData.append('file', file);
-        formData.append('notebook_dir', notebookDir);
-        // Upload file to backend using FormData
-        const resp = await (0,_restAPI_utils__WEBPACK_IMPORTED_MODULE_2__.requestAPI)('upload', {
-            method: 'POST',
-            body: formData
-        });
-        if (resp.error) {
-            _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__.Notification.emit(`Upload failed: ${resp.error.message}`, "error", {
-                autoClose: 5 * 1000 // 5 seconds
-            });
-            console.error('Upload failed:', resp.error.message);
-        }
-        else if (resp.data) {
-            console.log('File uploaded successfully:', resp.data);
-            // Notify the parent component that the file was uploaded, 
-            // which will update the context manager.
-            onFileUploaded(file);
-        }
-    };
     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { style: { position: 'relative' } },
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("input", { ref: fileInputRef, type: "file", style: { display: 'none' }, onChange: handleFileSelect, accept: "*" }),
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_IconButton__WEBPACK_IMPORTED_MODULE_3__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_PaperClipIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null), title: isUploading ? 'Uploading...' : 'Attach File', onClick: handleClick, className: 'icon-button-hover', disabled: isUploading, style: {
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_IconButton__WEBPACK_IMPORTED_MODULE_2__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_PaperClipIcon__WEBPACK_IMPORTED_MODULE_3__["default"], null), title: isUploading ? 'Uploading...' : 'Attach File', onClick: handleClick, className: 'icon-button-hover', disabled: isUploading, style: {
                 ...(isUploading && { opacity: 0.5 }),
                 cursor: isUploading ? 'not-allowed' : 'pointer'
             } })));
@@ -11525,6 +11445,166 @@ const stripFileEnding = (rule) => {
 };
 
 
+/***/ }),
+
+/***/ "./lib/utils/fileUpload.js":
+/*!*********************************!*\
+  !*** ./lib/utils/fileUpload.js ***!
+  \*********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   uploadFileToBackend: () => (/* binding */ uploadFileToBackend)
+/* harmony export */ });
+/* harmony import */ var _restAPI_utils__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../restAPI/utils */ "./lib/restAPI/utils.js");
+/* 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__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+const CHUNKED_UPLOAD_SIZE_CUTOFF = 25 * 1024 * 1024; // 25MB cutoff for chunked uploads
+const CHUNK_SIZE = 10 * 1024 * 1024; // 10MB chunks
+const getNotebookDirectory = (notebookPath) => {
+    const lastSlashIndex = notebookPath.lastIndexOf('/');
+    if (lastSlashIndex === -1) {
+        // No directory, just filename (root directory)
+        return '.';
+    }
+    return notebookPath.substring(0, lastSlashIndex);
+};
+const uploadChunk = async (chunk, filename, chunkNumber, totalChunks, notebookDir) => {
+    try {
+        // Create FormData for chunk upload
+        const formData = new FormData();
+        formData.append('file', chunk, filename);
+        formData.append('chunk_number', chunkNumber.toString());
+        formData.append('total_chunks', totalChunks.toString());
+        formData.append('notebook_dir', notebookDir);
+        // Upload chunk to backend
+        const resp = await (0,_restAPI_utils__WEBPACK_IMPORTED_MODULE_1__.requestAPI)('upload', {
+            method: 'POST',
+            body: formData
+        });
+        if (resp.error) {
+            _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__.Notification.emit(`Upload failed: ${resp.error.message}`, "error", {
+                autoClose: 5 * 1000 // 5 seconds
+            });
+            console.error(`Chunk ${chunkNumber} upload failed:`, resp.error.message);
+            return false;
+        }
+        else if (resp.data) {
+            if (resp.data.chunk_complete) {
+                console.log(`All chunks uploaded successfully. File reconstructed: ${resp.data.filename}`);
+            }
+            else if (resp.data.chunk_received) {
+                console.log(`Chunk ${resp.data.chunk_number}/${resp.data.total_chunks} uploaded successfully`);
+            }
+            return true;
+        }
+        return false;
+    }
+    catch (error) {
+        _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__.Notification.emit(`Upload failed: ${error}`, "error", {
+            autoClose: 5 * 1000 // 5 seconds
+        });
+        console.error(`Error uploading chunk ${chunkNumber}:`, error);
+        return false;
+    }
+};
+const handleLargeFile = async (file, notebookTracker, onFileUploaded) => {
+    const totalChunks = Math.ceil(file.size / CHUNK_SIZE);
+    console.log(`Splitting file into ${totalChunks} chunks of ${(CHUNK_SIZE / (1024 * 1024)).toFixed(2)}MB each`);
+    // Get notebook directory path
+    const notebookPanel = notebookTracker.currentWidget;
+    if (!notebookPanel) {
+        console.error('No notebook is currently active');
+        throw new Error('No notebook is currently active');
+    }
+    const notebookPath = notebookPanel.context.path;
+    const notebookDir = getNotebookDirectory(notebookPath);
+    for (let chunkIndex = 0; chunkIndex < totalChunks; chunkIndex++) {
+        const start = chunkIndex * CHUNK_SIZE;
+        const end = Math.min(start + CHUNK_SIZE, file.size);
+        const chunk = file.slice(start, end);
+        console.log(`Uploading chunk ${chunkIndex + 1}/${totalChunks}:`, {
+            chunkNumber: chunkIndex + 1,
+            totalChunks: totalChunks,
+            chunkSize: chunk.size,
+            chunkSizeMB: (chunk.size / (1024 * 1024)).toFixed(2),
+            startByte: start,
+            endByte: end,
+            fileName: file.name,
+            originalFileSize: file.size,
+            originalFileSizeMB: (file.size / (1024 * 1024)).toFixed(2)
+        });
+        // Upload chunk to backend
+        const success = await uploadChunk(chunk, file.name, chunkIndex + 1, totalChunks, notebookDir);
+        if (!success) {
+            console.error(`Failed to upload chunk ${chunkIndex + 1}`);
+            throw new Error(`Failed to upload chunk ${chunkIndex + 1}`);
+        }
+    }
+    console.log(`Successfully uploaded all ${totalChunks} chunks for file: ${file.name}`);
+    // Notify the parent component that the file was uploaded
+    onFileUploaded(file);
+};
+const uploadFile = async (file, notebookTracker, onFileUploaded) => {
+    // Get notebook directory path
+    const notebookPanel = notebookTracker.currentWidget;
+    if (!notebookPanel) {
+        console.error('No notebook is currently active');
+        throw new Error('No notebook is currently active');
+    }
+    const notebookPath = notebookPanel.context.path;
+    const notebookDir = getNotebookDirectory(notebookPath);
+    // Create FormData for file upload
+    const formData = new FormData();
+    formData.append('file', file);
+    formData.append('notebook_dir', notebookDir);
+    // Upload file to backend using FormData
+    const resp = await (0,_restAPI_utils__WEBPACK_IMPORTED_MODULE_1__.requestAPI)('upload', {
+        method: 'POST',
+        body: formData
+    });
+    if (resp.error) {
+        _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__.Notification.emit(`Upload failed: ${resp.error.message}`, "error", {
+            autoClose: 5 * 1000 // 5 seconds
+        });
+        console.error('Upload failed:', resp.error.message);
+    }
+    else if (resp.data) {
+        console.log('File uploaded successfully:', resp.data);
+        // Notify the parent component that the file was uploaded, 
+        // which will update the context manager.
+        onFileUploaded(file);
+    }
+};
+const uploadFileToBackend = async (file, notebookTracker, onFileUploaded) => {
+    try {
+        // Check file size and handle accordingly
+        if (file.size > CHUNKED_UPLOAD_SIZE_CUTOFF) {
+            console.log(`File ${file.name} is larger than 25MB (${(file.size / (1024 * 1024)).toFixed(2)}MB). Splitting into chunks...`);
+            await handleLargeFile(file, notebookTracker, onFileUploaded);
+        }
+        else {
+            // Upload file directly for files <= 25MB
+            await uploadFile(file, notebookTracker, onFileUploaded);
+        }
+    }
+    catch (error) {
+        _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_0__.Notification.emit(`Upload failed: ${error}`, "error", {
+            autoClose: 5 * 1000 // 5 seconds
+        });
+        console.error('Error during file upload:', error);
+        throw error;
+    }
+};
+
+
 /***/ }),
 
 /***/ "./lib/utils/models.js":
@@ -14517,6 +14597,11 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   margin-top: 0;
 }
 
+.chat-input-container.drag-over {
+  border: 2px dashed var(--purple-500);
+  transform: scale(1.02);
+}
+
 .chat-input-text-area-container {
   position: relative;
   height: min-content;
@@ -14597,7 +14682,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 
 .context-button:disabled:hover {
   background-color: var(--jp-layout-color2);
-}`, "",{"version":3,"sources":["webpack://./style/ChatInput.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,gBAAgB;EAChB,SAAS;EACT,cAAc;EACd,gBAAgB;EAChB,2DAA2D;EAC3D;;;iCAG+B;EAC/B,WAAW;EACX,mBAAmB;EACnB,iDAAiD;AACnD;;AAEA;EACE,aAAa;AACf;;AAEA;EACE,kBAAkB;EAClB,mBAAmB;AACrB;;AAEA;EACE,aAAa;EACb,YAAY;EACZ,YAAY;EACZ,WAAW;EACX,aAAa;EACb,gBAAgB;EAChB,sBAAsB;EACtB,yBAAyB;EACzB,2DAA2D;EAC3D,eAAe;EACf,gBAAgB;EAChB,qCAAqC;EACrC,gBAAgB;EAChB,iDAAiD;AACnD;;AAEA;EACE,aAAa;AACf;;AAEA;EACE,2BAA2B;EAC3B;;;iCAG+B;AACjC;;AAEA;EACE,+BAA+B;AACjC;;AAEA;EACE,+BAA+B;EAC/B,2BAA2B;AAC7B;;AAEA;EACE,+BAA+B;AACjC;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,yCAAyC;EACzC,yCAAyC;EACzC,kBAAkB;EAClB,gBAAgB;EAChB,eAAe;EACf,yCAAyC;EACzC,eAAe;EACf,oCAAoC;EACpC,sCAAsC;AACxC;;AAEA;EACE,yCAAyC;AAC3C;;AAEA,0BAA0B;AAC1B;EACE,YAAY;EACZ,mBAAmB;EACnB,yCAAyC;AAC3C;;AAEA;EACE,YAAY;EACZ,mBAAmB;EACnB,yCAAyC;AAC3C;;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.chat-input-container {\n  position: sticky;\n  bottom: 0;\n  margin: 10px 0;\n  margin-top: auto;\n  background-color: var(--chat-user-message-background-color);\n  box-shadow: \n    0 4px 12px rgba(0, 0, 0, 0.08),\n    0 2px 4px rgba(0, 0, 0, 0.06),\n    0 0 0 1px rgba(0, 0, 0, 0.02);\n  width: 100%;\n  border-radius: 12px;\n  transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);\n}\n\n.chat-input-container.editing {\n  margin-top: 0;\n}\n\n.chat-input-text-area-container {\n  position: relative;\n  height: min-content;\n}\n\n.chat-input {\n  outline: none;\n  border: none;\n  resize: none;\n  width: 100%;\n  padding: 10px;\n  overflow-y: auto;\n  box-sizing: border-box;\n  flex-shrink: 0 !important;\n  background-color: var(--chat-user-message-background-color);\n  font-size: 12px;\n  font-weight: 400;\n  font-family: var(--jp-ui-font-family);\n  line-height: 1.2;\n  transition: all 0.2s cubic-bezier(0.4, 0, 0.2, 1);\n}\n\n.active-cell-preview-container {\n  padding: 10px;\n}\n\n.chat-input-container:focus-within {\n  transform: translateY(-1px);\n  box-shadow: \n    0 2px 8px rgba(0, 0, 0, 0.12),\n    0 8px 16px rgba(0, 0, 0, 0.08),\n    0 0 0 1px rgba(0, 0, 0, 0.04);\n}\n\n.chat-input:focus {\n  color: var(--jp-ui-font-color1);\n}\n\n.chat-input::placeholder {\n  color: var(--jp-ui-font-color2);\n  transition: color 0.2s ease;\n}\n\n.chat-input:focus::placeholder {\n  color: var(--jp-ui-font-color3);\n}\n\n.context-button {\n  display: flex;\n  align-items: center;\n  background-color: var(--jp-layout-color2);\n  border: 1px solid var(--jp-border-color1);\n  border-radius: 3px;\n  padding: 4px 8px;\n  font-size: 12px;\n  height: var(--chat-context-button-height);\n  cursor: pointer;\n  color: var(--jp-content-font-color1);\n  transition: background-color 0.2s ease;\n}\n\n.context-button:hover {\n  background-color: var(--jp-layout-color3);\n}\n\n/* Disabled state styles */\n.chat-input:disabled {\n  opacity: 0.6;\n  cursor: not-allowed;\n  background-color: var(--jp-layout-color2);\n}\n\n.context-button:disabled {\n  opacity: 0.6;\n  cursor: not-allowed;\n  background-color: var(--jp-layout-color2);\n}\n\n.context-button:disabled:hover {\n  background-color: var(--jp-layout-color2);\n}"],"sourceRoot":""}]);
+}`, "",{"version":3,"sources":["webpack://./style/ChatInput.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,gBAAgB;EAChB,SAAS;EACT,cAAc;EACd,gBAAgB;EAChB,2DAA2D;EAC3D;;;iCAG+B;EAC/B,WAAW;EACX,mBAAmB;EACnB,iDAAiD;AACnD;;AAEA;EACE,aAAa;AACf;;AAEA;EACE,oCAAoC;EACpC,sBAAsB;AACxB;;AAEA;EACE,kBAAkB;EAClB,mBAAmB;AACrB;;AAEA;EACE,aAAa;EACb,YAAY;EACZ,YAAY;EACZ,WAAW;EACX,aAAa;EACb,gBAAgB;EAChB,sBAAsB;EACtB,yBAAyB;EACzB,2DAA2D;EAC3D,eAAe;EACf,gBAAgB;EAChB,qCAAqC;EACrC,gBAAgB;EAChB,iDAAiD;AACnD;;AAEA;EACE,aAAa;AACf;;AAEA;EACE,2BAA2B;EAC3B;;;iCAG+B;AACjC;;AAEA;EACE,+BAA+B;AACjC;;AAEA;EACE,+BAA+B;EAC/B,2BAA2B;AAC7B;;AAEA;EACE,+BAA+B;AACjC;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,yCAAyC;EACzC,yCAAyC;EACzC,kBAAkB;EAClB,gBAAgB;EAChB,eAAe;EACf,yCAAyC;EACzC,eAAe;EACf,oCAAoC;EACpC,sCAAsC;AACxC;;AAEA;EACE,yCAAyC;AAC3C;;AAEA,0BAA0B;AAC1B;EACE,YAAY;EACZ,mBAAmB;EACnB,yCAAyC;AAC3C;;AAEA;EACE,YAAY;EACZ,mBAAmB;EACnB,yCAAyC;AAC3C;;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.chat-input-container {\n  position: sticky;\n  bottom: 0;\n  margin: 10px 0;\n  margin-top: auto;\n  background-color: var(--chat-user-message-background-color);\n  box-shadow: \n    0 4px 12px rgba(0, 0, 0, 0.08),\n    0 2px 4px rgba(0, 0, 0, 0.06),\n    0 0 0 1px rgba(0, 0, 0, 0.02);\n  width: 100%;\n  border-radius: 12px;\n  transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);\n}\n\n.chat-input-container.editing {\n  margin-top: 0;\n}\n\n.chat-input-container.drag-over {\n  border: 2px dashed var(--purple-500);\n  transform: scale(1.02);\n}\n\n.chat-input-text-area-container {\n  position: relative;\n  height: min-content;\n}\n\n.chat-input {\n  outline: none;\n  border: none;\n  resize: none;\n  width: 100%;\n  padding: 10px;\n  overflow-y: auto;\n  box-sizing: border-box;\n  flex-shrink: 0 !important;\n  background-color: var(--chat-user-message-background-color);\n  font-size: 12px;\n  font-weight: 400;\n  font-family: var(--jp-ui-font-family);\n  line-height: 1.2;\n  transition: all 0.2s cubic-bezier(0.4, 0, 0.2, 1);\n}\n\n.active-cell-preview-container {\n  padding: 10px;\n}\n\n.chat-input-container:focus-within {\n  transform: translateY(-1px);\n  box-shadow: \n    0 2px 8px rgba(0, 0, 0, 0.12),\n    0 8px 16px rgba(0, 0, 0, 0.08),\n    0 0 0 1px rgba(0, 0, 0, 0.04);\n}\n\n.chat-input:focus {\n  color: var(--jp-ui-font-color1);\n}\n\n.chat-input::placeholder {\n  color: var(--jp-ui-font-color2);\n  transition: color 0.2s ease;\n}\n\n.chat-input:focus::placeholder {\n  color: var(--jp-ui-font-color3);\n}\n\n.context-button {\n  display: flex;\n  align-items: center;\n  background-color: var(--jp-layout-color2);\n  border: 1px solid var(--jp-border-color1);\n  border-radius: 3px;\n  padding: 4px 8px;\n  font-size: 12px;\n  height: var(--chat-context-button-height);\n  cursor: pointer;\n  color: var(--jp-content-font-color1);\n  transition: background-color 0.2s ease;\n}\n\n.context-button:hover {\n  background-color: var(--jp-layout-color3);\n}\n\n/* Disabled state styles */\n.chat-input:disabled {\n  opacity: 0.6;\n  cursor: not-allowed;\n  background-color: var(--jp-layout-color2);\n}\n\n.context-button:disabled {\n  opacity: 0.6;\n  cursor: not-allowed;\n  background-color: var(--jp-layout-color2);\n}\n\n.context-button:disabled:hover {\n  background-color: var(--jp-layout-color2);\n}"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -20451,4 +20536,4 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
 /***/ })
 
 }]);
-//# sourceMappingURL=lib_index_js.2db61d2b629817845901.js.map
+//# sourceMappingURL=lib_index_js.5c7d84a45ddeb5704b61.js.map