kuzco-cli/gui/renderer.js

71 lines
2.7 KiB
JavaScript
Raw Normal View History

2024-03-09 02:15:01 -05:00
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 modelSelectionContainer = document.getElementById('modelSelectionContainer');
2024-03-09 02:15:01 -05:00
if (chatForm && promptInput && modelSelect) {
2024-03-09 02:15:01 -05:00
chatForm.addEventListener('submit', async (event) => {
event.preventDefault();
2024-03-09 02:15:01 -05:00
const userInput = promptInput.value;
const selectedModel = modelSelect.value;
2024-03-09 02:15:01 -05:00
promptInput.value = '';
modelSelectionContainer.style.display = 'none';
2024-03-09 02:15:01 -05:00
sendButton.disabled = true;
promptInput.disabled = true;
2024-03-09 02:15:01 -05:00
displayMessage(userInput, 'user');
2024-03-09 03:20:24 -05:00
const typingIndicator = displayTypingIndicator();
2024-03-09 02:15:01 -05:00
try {
const response = await window.electronAPI.sendPrompt(userInput, selectedModel);
2024-03-09 05:56:54 -05:00
if (response.error) {
throw new Error(response.error);
}
2024-03-09 02:15:01 -05:00
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 {
2024-03-09 03:20:24 -05:00
typingIndicator.remove();
sendButton.disabled = false;
promptInput.disabled = false;
promptInput.focus();
2024-03-09 02:15:01 -05:00
}
});
} else {
console.error('chatForm or promptInput elements not found!');
}
2024-03-09 03:20:24 -05:00
});
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;
}