undefined - Follow Up Deployment

index.html

@@ -620,6 +620,22 @@
                 defaultChatSettings = JSON.parse(savedDefaults);
             }
 
+            // Load custom models from localStorage
+            const savedCustomModels = localStorage.getItem('customModels');
+            if (savedCustomModels) {
+                customModels = JSON.parse(savedCustomModels);
+                customModels.forEach(model => {
+                    models.push({
+                        id: model.id,
+                        name: model.name,
+                        endpoint: model.endpoint,
+                        key: model.key,
+                        streaming: model.streaming || false,
+                        type: 'custom'
+                    });
+                });
+            }
+
             // Load custom providers
             customProviders.forEach(provider => {
                 models.push({
@@ -851,6 +867,12 @@
         function renderCustomModels() {
             customModelsList.innerHTML = '';
             
+            // Reload custom models from localStorage to ensure we have latest
+            const savedCustomModels = localStorage.getItem('customModels');
+            if (savedCustomModels) {
+                customModels = JSON.parse(savedCustomModels);
+            }
+            
             if (customModels.length === 0) {
                 customModelsList.innerHTML = `
                     <div class="text-center text-gray-500 dark:text-gray-400 py-2 text-sm">
@@ -1085,7 +1107,7 @@
             }
             
             // Get current chat
-            const chatIndex = chats.findIndex(c => c.id === currentChatId);
+            let chatIndex = chats.findIndex(c => c.id === currentChatId);
             if (chatIndex === -1) return;
             
             const chat = chats[chatIndex];
@@ -1104,17 +1126,27 @@
                 timestamp: new Date().toISOString()
             };
             
-            chat.messages.push(userMessage);
-            chat.lastMessage = message.length > 30 ? message.substring(0, 30) + '...' : message;
+            // Ensure we have the latest chat reference
+            chatIndex = chats.findIndex(c => c.id === currentChatId);
+            if (chatIndex === -1) return;
+            let currentChat = chats[chatIndex];
+            
+            currentChat.messages.push(userMessage);
+            currentChat.lastMessage = message.length > 30 ? message.substring(0, 30) + '...' : message;
             
-            // Update UI
+            // Save chat immediately after adding user message
+            chats[chatIndex] = currentChat;
+            saveChats();
+            
+            // Render all messages including the new user message
             renderMessages(chat.messages);
             messageInput.value = '';
+            chatArea.scrollTop = chatArea.scrollHeight;
             
             // Create assistant message element for streaming/response
-            const messageElement = document.createElement('div');
-            messageElement.className = 'flex gap-4 justify-start';
-            messageElement.innerHTML = `
+            const assistantMessageElement = document.createElement('div');
+            assistantMessageElement.className = 'flex gap-4 justify-start';
+            assistantMessageElement.innerHTML = `
                 <div class="flex-shrink-0 w-8 h-8 rounded-full bg-primary-500 text-white flex items-center justify-center">
                     <i class="fas fa-robot"></i>
                 </div>
@@ -1127,15 +1159,15 @@
                     <div class="prose dark:prose-invert hidden" id="streaming-content"></div>
                 </div>
             `;
-            chatArea.appendChild(messageElement);
-            const contentElement = messageElement.querySelector('#streaming-content');
-            const loadingIndicator = messageElement.querySelector('#loading-indicator');
+            chatArea.appendChild(assistantMessageElement);
+            const contentElement = assistantMessageElement.querySelector('#streaming-content');
+            const loadingIndicator = assistantMessageElement.querySelector('#loading-indicator');
 
             try {
                 // Ensure we have the latest chat reference first
-                let chatIndex = chats.findIndex(c => c.id === currentChatId);
+                chatIndex = chats.findIndex(c => c.id === currentChatId);
                 if (chatIndex === -1) throw new Error('Chat not found');
-                let currentChat = chats[chatIndex];
+                currentChat = chats[chatIndex];
                 
                 // Get the current model
                 const model = models.find(m => m.id === currentChat.model);
@@ -1184,7 +1216,7 @@
                 };
                 
                 // Replace the last message (which was the loading one) with the actual response
-                currentChat.messages[currentChat.messages.length - 1] = assistantMessage;
+                currentChat.messages.push(assistantMessage);
                 saveChats();
                 
                 // Update chat list
@@ -1476,6 +1508,24 @@
 
         // Select model
         function selectModel(modelId) {
+            // Ensure models array is up to date
+            const savedCustomModels = localStorage.getItem('customModels');
+            if (savedCustomModels) {
+                customModels = JSON.parse(savedCustomModels);
+                // Update models array with latest custom models
+                models = models.filter(m => m.type !== 'custom');
+                customModels.forEach(model => {
+                    models.push({
+                        id: model.id,
+                        name: model.name,
+                        endpoint: model.endpoint,
+                        key: model.key,
+                        streaming: model.streaming || false,
+                        type: 'custom'
+                    });
+                });
+            }
+            
             const model = models.find(m => m.id === modelId);
             if (!model) return;