Update static/script.js

static/script.js

@@ -74,22 +74,46 @@ function handleResponse(userInput) {
 }
 
 function displayOptions(options) {
-    // Function to display options on the UI
     const optionsContainer = document.getElementById('options-container');
     if (!optionsContainer) {
         console.error('Options container not found!');
         return;
     }
 
-    options.forEach(option => {
+    optionsContainer.innerHTML = ''; // Clear previous options if any
+
+    options.forEach(opt => {
         const button = document.createElement('button');
-        button.classList.add(option.class);
-        button.textContent = option.text;
-        button.onclick = () => handleResponse(option.text);
+        button.textContent = opt.text;
+        button.className = `option-button ${opt.class}`;
+        button.onclick = () => {
+            addMessage('user', opt.text);
+            conversation.push({ role: 'user', message: opt.text });
+            setTimeout(() => handleResponse(opt.text), 500);
+        };
         optionsContainer.appendChild(button);
     });
+
+    optionsContainer.appendChild(document.createElement('br')); // Line break after options
+
+    // Optional 'Go Back' button functionality
+    if (conversation.length > 1) {
+        const backButton = document.createElement('button');
+        backButton.textContent = 'Go Back';
+        backButton.className = 'option-button';
+        backButton.onclick = () => {
+            if (conversation.length > 1) {
+                conversation.pop(); // Remove last message
+                chatMessages.innerHTML = ''; // Clear current messages
+                conversation.forEach(msg => addMessage(msg.role, msg.message)); // Re-render
+                setTimeout(() => handleResponse(conversation[conversation.length - 1].message), 500);
+            }
+        };
+        optionsContainer.appendChild(backButton);
+    }
 }
 
+
 document.getElementById('userInput').addEventListener('keypress', function(e) {
     if (e.key === 'Enter') {
         sendMessage();