From 11d193a9e2652c186e279a151c1a49e5a8513a74 Mon Sep 17 00:00:00 2001
From: Adrien <adrien.bouvais.pro@gmail.com>
Date: Wed, 22 May 2024 09:20:46 +0200
Subject: [PATCH] Better LLM selections

---
 Chat.go                            |  4 +-
 database.go                        |  1 +
 views/partials/message-bot.html    |  2 +-
 views/partials/popover-models.html | 83 ++++++++++++++++++++++++++----
 4 files changed, 78 insertions(+), 12 deletions(-)

diff --git a/Chat.go b/Chat.go
index 3cc0010..d944083 100644
--- a/Chat.go
+++ b/Chat.go
@@ -59,6 +59,7 @@ type TemplateMessage struct {
 	Hidden  bool
 	Id      string
 	Name    string
+	Model   string
 	ModelID string
 }
 
@@ -88,7 +89,8 @@ func generateChatHTML() string {
 				Content: markdownToHTML(message.Content),
 				Hidden:  !message.Selected, // Assuming Hidden is a field you want to include from Message
 				Id:      message.ID.String(),
-				Name:    message.LLM.Model.Name,
+				Name:    message.LLM.Name,
+				Model:   message.LLM.Model.Name,
 				ModelID: message.LLM.Model.ModelID,
 			}
 			templateMessages = append(templateMessages, templateMessage)
diff --git a/database.go b/database.go
index 3feb6b5..1eed4bf 100644
--- a/database.go
+++ b/database.go
@@ -271,6 +271,7 @@ func getAllMessages() []Message {
 		content,
 		date,
 		llm : {
+			name,
 			modelInfo : {
 				modelID,
 				name,
diff --git a/views/partials/message-bot.html b/views/partials/message-bot.html
index c46e095..3b80cc6 100644
--- a/views/partials/message-bot.html
+++ b/views/partials/message-bot.html
@@ -29,7 +29,7 @@
                     {% if not message.Hidden %}
                     <div class="message-header">
                         <p>
-                            {{ message.Name }}
+                            <strong>{{ message.Name }}</strong> <small>{{ message.Model }}</small>
                         </p>
                     </div>
                     <div class="message-body">
diff --git a/views/partials/popover-models.html b/views/partials/popover-models.html
index 78e1716..41c3a47 100644
--- a/views/partials/popover-models.html
+++ b/views/partials/popover-models.html
@@ -23,20 +23,73 @@
 </div>
 
 <script>
+    let lastSelectedIndex = null;
+
+    document.addEventListener('keydown', function (event) {
+        if (event.key === 'Shift') {
+            document.body.classList.add('shift-pressed');
+        }
+    });
+
+    document.addEventListener('keyup', function (event) {
+        if (event.key === 'Shift') {
+            document.body.classList.remove('shift-pressed');
+            lastSelectedIndex = null;
+        }
+
+        // Remove all "shiftselected" classes
+        const elements = Array.from(document.getElementsByClassName('icon-text'));
+        for (let i = 0; i < elements.length; i++) {
+            elements[i].classList.remove('shiftselected');
+        }
+
+        window.getSelection().removeAllRanges();
+    });
+
     function toggleSelection(element) {
-        // Check if the element currently has the 'has-text-info' class
-        if (element.classList.contains('selected')) {
-            // It has 'has-text-info', so remove it and add 'has-text'
-            element.classList.add('unselected');
-            element.classList.remove('selected');
+        const elements = Array.from(document.getElementsByClassName('icon-text'));
+        const index = elements.indexOf(element);
+
+        if (document.body.classList.contains('shift-pressed') && lastSelectedIndex !== null) {
+            const [start, end] = [lastSelectedIndex, index].sort((a, b) => a - b);
+            let allSelected = true;
+            for (let i = start; i <= end; i++) {
+                if (!elements[i].classList.contains('selected')) {
+                    allSelected = false;
+                    break;
+                }
+            }
+            for (let i = start; i <= end; i++) {
+                if (allSelected) {
+                    elements[i].classList.remove('selected');
+                    elements[i].classList.add('unselected');
+                } else {
+                    elements[i].classList.add('selected');
+                    elements[i].classList.remove('unselected');
+                }
+            }
+            lastSelectedIndex = null;
+
+            const elements2 = Array.from(document.getElementsByClassName('icon-text'));
+            for (let i = 0; i < elements2.length; i++) {
+                elements2[i].classList.remove('shiftselected');
+            }
+        } else if (document.body.classList.contains('shift-pressed') && lastSelectedIndex === null) {
+            lastSelectedIndex = index;
+            element.classList.toggle('shiftselected');
         } else {
-            // It doesn't have 'has-text-info', so add it and remove 'has-text'
-            element.classList.remove('unselected');
-            element.classList.add('selected');
+            element.classList.toggle('selected');
+            element.classList.toggle('unselected');
         }
         toggleSendButton();
     }
 
+    document.addEventListener('click', function (event) {
+        if (event.shiftKey) {
+            event.preventDefault();
+        }
+    });
+
     function getSelectedModelsIDs() {
         var selectedModelsIDs = [];
         var selectedModels = document.getElementsByClassName('selected');
@@ -49,8 +102,7 @@
 
 <style>
     .selected {
-        border: 2px solid #3273dc;
-        /* Bulma's primary color */
+        border: 2px solid #126d0f;
         border-radius: 4px;
         padding: 1px;
         margin: 2px;
@@ -60,4 +112,15 @@
         border-radius: 4px;
         padding: 3px;
     }
+
+    .shiftselected {
+        background: #126d0f;
+        border-radius: 4px;
+        padding: 1px;
+        margin: 2px;
+    }
+
+    .shift-pressed *::selection {
+        background: transparent;
+    }
 </style>
\ No newline at end of file