function displayMessage(message, sender) { const chatHistory = document.getElementById('chatHistory'); const messageDiv = document.createElement('div'); messageDiv.classList.add('message'); if (sender === 'user') { messageDiv.classList.add('userMessage'); } else if (sender === 'assistant') { messageDiv.classList.add('assistantMessage'); } messageDiv.textContent = message; chatHistory.appendChild(messageDiv); chatHistory.scrollTop = chatHistory.scrollHeight; } document.addEventListener('DOMContentLoaded', () => { const chatForm = document.getElementById('chatForm'); const promptInput = document.getElementById('promptInput'); const modelSelect = document.getElementById('modelSelect'); const sendButton = document.getElementById('sendButton'); const stopButton = document.getElementById('stopButton'); const modelSelectionContainer = document.getElementById('modelSelectionContainer'); if (chatForm && promptInput && modelSelect) { chatForm.addEventListener('submit', async (event) => { event.preventDefault(); const userInput = promptInput.value; const selectedModel = modelSelect.value; promptInput.value = ''; modelSelectionContainer.style.display = 'none'; sendButton.disabled = true; promptInput.disabled = true; stopButton.disabled = false; displayMessage(userInput, 'user'); const typingIndicator = displayTypingIndicator(); try { const response = await window.electronAPI.sendPrompt(userInput, selectedModel); if (response.error) { throw new Error(response.error); } const assistantMessage = response.choices[0].message.content.trim(); displayMessage(assistantMessage, 'assistant'); } catch (error) { console.error(`Error sending prompt: ${error.message}`); displayMessage(`Error: ${error.message}`, 'assistant'); } finally { typingIndicator.remove(); sendButton.disabled = false; promptInput.disabled = false; promptInput.focus(); stopButton.disabled = true; } }); } else { console.error('chatForm or promptInput elements not found!'); } }); document.getElementById('stopButton').addEventListener('click', () => { window.electronAPI.abortPrompt(); document.getElementById('stopButton').disabled = true; }); function displayTypingIndicator() { const chatHistory = document.getElementById('chatHistory'); const typingIndicator = document.createElement('div'); typingIndicator.classList.add('message', 'assistantMessage', 'ellipsis'); typingIndicator.textContent = 'Processing'; chatHistory.appendChild(typingIndicator); chatHistory.scrollTop = chatHistory.scrollHeight; return typingIndicator; }