Restructure GUI folder & modify styles.css

2024-03-09 16:19:47 -05:00
parent bdf5715868
commit ee6f775b5b
8 changed files with 32 additions and 11 deletions
--- a/gui/js/renderer.js
+++ b/gui/js/renderer.js
@@ -0,0 +1,78 @@
+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;
+}