Replace with updated NoteGenie version

static/js/main.js

@@ -444,6 +444,7 @@ document.addEventListener('DOMContentLoaded', function() {
         // Close any existing event source
         if (eventSource) {
             eventSource.close();
+            eventSource = null;
         }
         
         // Keep track of when we should update the preview (not on every tiny chunk)
@@ -456,7 +457,12 @@ document.addEventListener('DOMContentLoaded', function() {
         // Make sure notebook is in proper format before sending
         const notebookToSend = typeof notebook === 'string' ? JSON.parse(notebook) : notebook;
         
-        console.log('Sending notebook for editing:', notebookToSend); // Debug log
+        // Debug logging to help diagnose issues
+        console.log('Notebook edit request:', {
+            cellCount: notebookToSend.cells.length,
+            promptLength: editPrompt.length,
+            model: modelName
+        });
         
         // First, send the notebook data to the server so it's available in the session
         fetch('/prepare_edit_notebook', {
@@ -472,7 +478,7 @@ document.addEventListener('DOMContentLoaded', function() {
         })
         .then(response => {
             if (!response.ok) {
-                throw new Error(`Server error: ${response.status}`);
+                throw new Error(`Server error (${response.status}): ${response.statusText}`);
             }
             return response.json();
         })
@@ -498,135 +504,161 @@ document.addEventListener('DOMContentLoaded', function() {
                     headers['X-API-Key'] = backupApiKey;
                 }
                 
-                // Now create a new event source for editing with the API key included in both URL and headers
-                eventSource = createEventSourceWithHeaders(`/edit_notebook?${urlParams.toString()}`, headers);
+                console.log('Starting EventSource for edit with headers:', Object.keys(headers).length > 0 ? 'Headers present' : 'No headers');
                 
-                // Track the last time we got a chunk - for timeout detection
-                let lastChunkTime = Date.now();
-                const CONNECTION_TIMEOUT = 30000; // 30 seconds without data = timeout
-                
-                // Start a monitoring timer to detect stalled connections
-                const connectionTimer = setInterval(() => {
-                    if (Date.now() - lastChunkTime > CONNECTION_TIMEOUT) {
-                        clearInterval(connectionTimer);
-                        if (eventSource) {
-                            console.log("Connection timed out - closing event source");
-                            eventSource.close();
-                            eventSource = null;
-                            
-                            // Process what we've got so far
-                            updateAiMessage(aiMessageId, "**NoteGenie:** Edit processing timed out, but I'll use what I've received so far.");
-                            processNotebookResponse(aiResponseText);
-                            setGeneratingState(false);
-                            
-                            // Switch back to generate mode
-                            setMode('generate');
-                        }
-                    }
-                }, 5000); // Check every 5 seconds
-                
-                eventSource.onmessage = function(event) {
-                    // Update our last-activity timestamp
-                    lastChunkTime = Date.now();
+                // Now create a new event source for editing with the API key included in both URL and headers
+                try {
+                    eventSource = createEventSourceWithHeaders(`/edit_notebook?${urlParams.toString()}`, headers);
                     
-                    try {
-                        const data = JSON.parse(event.data);
-                        
-                        // Handle errors sent from the server
-                        if (data.error) {
-                            console.error("Server error:", data.error);
-                            updateAiMessage(aiMessageId, `**Error:** ${data.error}`);
-                            
-                            // Try to salvage what we have so far
-                            if (aiResponseText && (aiResponseText.includes('MARKDOWN CELL') || aiResponseText.includes('CODE CELL'))) {
+                    // Track the last time we got a chunk - for timeout detection
+                    let lastChunkTime = Date.now();
+                    const CONNECTION_TIMEOUT = 30000; // 30 seconds without data = timeout
+                    
+                    // Start a monitoring timer to detect stalled connections
+                    const connectionTimer = setInterval(() => {
+                        if (Date.now() - lastChunkTime > CONNECTION_TIMEOUT) {
+                            clearInterval(connectionTimer);
+                            if (eventSource) {
+                                console.log("Connection timed out - closing event source");
+                                eventSource.close();
+                                eventSource = null;
+                                
+                                // Process what we've got so far
+                                updateAiMessage(aiMessageId, "**NoteGenie:** Edit processing timed out, but I'll use what I've received so far.");
                                 processNotebookResponse(aiResponseText);
+                                setGeneratingState(false);
+                                
+                                // Switch back to generate mode
+                                setMode('generate');
                             }
-                            
-                            eventSource.close();
-                            eventSource = null;
-                            setGeneratingState(false);
-                            clearInterval(connectionTimer);
-                            
-                            // Switch back to generate mode
-                            setMode('generate');
-                            return;
                         }
+                    }, 5000); // Check every 5 seconds
+                    
+                    eventSource.onopen = function(event) {
+                        console.log('EventSource connection opened for edit');
+                    };
+                    
+                    eventSource.onmessage = function(event) {
+                        // Update our last-activity timestamp
+                        lastChunkTime = Date.now();
                         
-                        if (data.chunk) {
-                            aiResponseText += data.chunk;
+                        try {
+                            const data = JSON.parse(event.data);
                             
-                            // Extract notebook info as soon as it's available, similar to generate function
-                            const nameMatch = aiResponseText.match(/NOTEBOOK_NAME:?\s*(.+?)(?:\n|$)/);
-                            const descMatch = aiResponseText.match(/NOTEBOOK_DESCRIPTION:?\s*(.+?)(?:\n|$)/);
-                            
-                            if (nameMatch && nameMatch[1].trim()) {
-                                // Update notebook title immediately when found
-                                notebookTitleEl.textContent = nameMatch[1].trim();
+                            // Handle errors sent from the server
+                            if (data.error) {
+                                console.error("Server error during edit:", data.error);
+                                updateAiMessage(aiMessageId, `**Error:** ${data.error}`);
+                                
+                                // Try to salvage what we have so far
+                                if (aiResponseText && (aiResponseText.includes('MARKDOWN CELL') || aiResponseText.includes('CODE CELL'))) {
+                                    processNotebookResponse(aiResponseText);
+                                }
+                                
+                                eventSource.close();
+                                eventSource = null;
+                                setGeneratingState(false);
+                                clearInterval(connectionTimer);
+                                
+                                // Switch back to generate mode
+                                setMode('generate');
+                                return;
                             }
                             
-                            if (descMatch && descMatch[1].trim()) {
-                                // Update AI message with the actual description from the edited notebook
-                                updateAiMessage(aiMessageId, `**NoteGenie:** ${descMatch[1].trim()}`);
-                            } else {
-                                // Show a simple editing message while waiting for description
-                                updateAiMessage(aiMessageId, "**NoteGenie:** Updating notebook based on your edit request...");
+                            if (data.chunk) {
+                                aiResponseText += data.chunk;
+                                
+                                // Extract notebook info as soon as it's available, similar to generate function
+                                const nameMatch = aiResponseText.match(/NOTEBOOK_NAME:?\s*(.+?)(?:\n|$)/);
+                                const descMatch = aiResponseText.match(/NOTEBOOK_DESCRIPTION:?\s*(.+?)(?:\n|$)/);
+                                
+                                if (nameMatch && nameMatch[1].trim()) {
+                                    // Update notebook title immediately when found
+                                    notebookTitleEl.textContent = nameMatch[1].trim();
+                                }
+                                
+                                if (descMatch && descMatch[1].trim()) {
+                                    // Update AI message with the actual description from the edited notebook
+                                    updateAiMessage(aiMessageId, `**NoteGenie:** ${descMatch[1].trim()}`);
+                                } else {
+                                    // Show a simple editing message while waiting for description
+                                    updateAiMessage(aiMessageId, "**NoteGenie:** Updating notebook based on your edit request...");
+                                }
+                                
+                                // Update preview periodically during streaming, just like in generate function
+                                const now = Date.now();
+                                if (now - lastPreviewUpdate > PREVIEW_UPDATE_INTERVAL) {
+                                    lastPreviewUpdate = now;
+                                    
+                                    // Only try to update preview if we have meaningful content
+                                    if (aiResponseText.includes('MARKDOWN CELL') || aiResponseText.includes('CODE CELL')) {
+                                        updateNotebookPreviewDuringStream(aiResponseText);
+                                    }
+                                }
                             }
                             
-                            // Update preview periodically during streaming, just like in generate function
-                            const now = Date.now();
-                            if (now - lastPreviewUpdate > PREVIEW_UPDATE_INTERVAL) {
-                                lastPreviewUpdate = now;
+                            if (data.done) {
+                                console.log('Edit completed successfully');
+                                eventSource.close();
+                                eventSource = null;
+                                clearInterval(connectionTimer);
                                 
-                                // Only try to update preview if we have meaningful content
-                                if (aiResponseText.includes('MARKDOWN CELL') || aiResponseText.includes('CODE CELL')) {
-                                    updateNotebookPreviewDuringStream(aiResponseText);
-                                }
+                                // Process the complete response for final rendering
+                                processNotebookResponse(aiResponseText);
+                                setGeneratingState(false);
+                                
+                                // Get the final description for the AI message
+                                const finalDescMatch = aiResponseText.match(/NOTEBOOK_DESCRIPTION:?\s*(.+?)(?=\n\s*---|\n\s*$|$)/);
+                                const finalDesc = finalDescMatch ? finalDescMatch[1].trim() : "I've updated the notebook based on your edit request.";
+                                
+                                // Update AI message with final description
+                                updateAiMessage(aiMessageId, `**NoteGenie:** ${finalDesc}`);
+                                
+                                // Switch back to generate mode
+                                setMode('generate');
                             }
+                        } catch (error) {
+                            console.error('Error parsing event data:', error, event.data);
                         }
+                    };
+                    
+                    eventSource.onerror = function(err) {
+                        // More detailed error logging
+                        console.error('EventSource error during edit:', err);
                         
-                        if (data.done) {
+                        // Get any status text or error message if available
+                        let errorMessage = 'Failed to connect to edit service';
+                        if (err instanceof Event && err.target && err.target.status) {
+                            errorMessage += ` (Status: ${err.target.status})`;
+                        }
+                        
+                        // Close and clean up event source
+                        if (eventSource) {
                             eventSource.close();
                             eventSource = null;
-                            clearInterval(connectionTimer);
-                            
-                            // Process the complete response for final rendering
+                        }
+                        
+                        // Clear any timers
+                        clearInterval(connectionTimer);
+                        
+                        // Try to salvage what we have so far
+                        if (aiResponseText && (aiResponseText.includes('MARKDOWN CELL') || aiResponseText.includes('CODE CELL'))) {
+                            updateAiMessage(aiMessageId, '**Warning:** Connection issue occurred but I\'ll try to process what I received so far.');
                             processNotebookResponse(aiResponseText);
-                            setGeneratingState(false);
-                            
-                            // Get the final description for the AI message
-                            const finalDescMatch = aiResponseText.match(/NOTEBOOK_DESCRIPTION:?\s*(.+?)(?=\n\s*---|\n\s*$|$)/);
-                            const finalDesc = finalDescMatch ? finalDescMatch[1].trim() : "I've updated the notebook based on your edit request.";
-                            
-                            // Update AI message with final description
-                            updateAiMessage(aiMessageId, `**NoteGenie:** ${finalDesc}`);
-                            
-                            // Switch back to generate mode
-                            setMode('generate');
+                        } else {
+                            updateAiMessage(aiMessageId, `**Error:** Failed to edit notebook. ${errorMessage}`);
                         }
-                    } catch (error) {
-                        console.error('Error parsing event data:', error, event.data);
-                    }
-                };
-                
-                eventSource.onerror = function(err) {
-                    console.error('EventSource error:', err);
-                    eventSource.close();
-                    eventSource = null;
-                    clearInterval(connectionTimer);
-                    
-                    // Try to salvage what we have so far
-                    if (aiResponseText && (aiResponseText.includes('MARKDOWN CELL') || aiResponseText.includes('CODE CELL'))) {
-                        updateAiMessage(aiMessageId, '**Warning:** Connection issue occurred but I\'ll try to process what I received so far.');
-                        processNotebookResponse(aiResponseText);
-                    } else {
-                        updateAiMessage(aiMessageId, '**Error:** Failed to edit notebook. ' + (err.message || 'Unknown error'));
-                    }
-                    
+                        
+                        setGeneratingState(false);
+                        
+                        // Switch back to generate mode
+                        setMode('generate');
+                    };
+                } catch (e) {
+                    console.error('Error creating EventSource:', e);
+                    updateAiMessage(aiMessageId, `**Error:** Could not establish connection for editing: ${e.message}`);
                     setGeneratingState(false);
-                    
-                    // Switch back to generate mode
-                    setMode('generate');
-                };
+                }
             } else {
                 throw new Error(data.message || 'Failed to prepare notebook for editing');
             }