78 lines
3.0 KiB
JavaScript
78 lines
3.0 KiB
JavaScript
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;
|
|
} |