From 7df7e44d070e02c6ce03201660d5d25d02326fbd Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Thu, 6 Jul 2023 01:01:17 +0200
Subject: [PATCH] upgraded ui

---
 app.py                                  | 16 ++++++++++++++-
 web/src/components/Message.vue          | 12 ++++++++---
 web/src/components/PersonalityEntry.vue |  2 +-
 web/src/views/DiscussionsView.vue       |  7 ++++++-
 web/src/views/SettingsView.vue          | 27 ++++++++++++++++++++++++-
 5 files changed, 57 insertions(+), 7 deletions(-)

diff --git a/app.py b/app.py
index ed228636..30d0a2e0 100644
--- a/app.py
+++ b/app.py
@@ -126,6 +126,7 @@ class LoLLMsWebUI(LoLLMsAPPI):
         self.add_endpoint("/install_model_from_path", "install_model_from_path", self.install_model_from_path, methods=["GET"])
         
         self.add_endpoint("/reinstall_binding", "reinstall_binding", self.reinstall_binding, methods=["POST"])
+        self.add_endpoint("/reinstall_personality", "reinstall_personality", self.reinstall_personality, methods=["POST"])
 
         self.add_endpoint("/switch_personal_path", "switch_personal_path", self.switch_personal_path, methods=["POST"])
 
@@ -864,14 +865,27 @@ class LoLLMsWebUI(LoLLMsAPPI):
                         "status": True
                     })
         
+    def reinstall_personality(self):
+        try:
+            data = request.get_json()
+            # Further processing of the data
+        except Exception as e:
+            print(f"Error occurred while parsing JSON: {e}")
+            return jsonify({"status":False, 'error':str(e)})
         
+        ASCIIColors.info(f"- Reinstalling personality {data['name']}...")
+        try:
+            ASCIIColors.info("Unmounting personality")
+        except Exception as e:
+            return jsonify({"status":False, 'error':str(e)})
+
     def reinstall_binding(self):
         try:
             data = request.get_json()
             # Further processing of the data
         except Exception as e:
             print(f"Error occurred while parsing JSON: {e}")
-            return jsonify({"status":False, 'error':str(ex)})
+            return jsonify({"status":False, 'error':str(e)})
         ASCIIColors.info(f"- Reinstalling binding {data['name']}...")
         try:
             ASCIIColors.info("Unmounting binding and model")
diff --git a/web/src/components/Message.vue b/web/src/components/Message.vue
index 9e856f40..a51f008f 100644
--- a/web/src/components/Message.vue
+++ b/web/src/components/Message.vue
@@ -117,6 +117,9 @@
 
                 <div class="overflow-x-auto w-full ">
                     <!-- MESSAGE CONTENT -->
+                    <div v-for="(step, index) in steps" :key="index" class="step">
+                        <Step :done="step.done" :message="step.message" />
+                    </div>
                     <MarkdownRenderer ref="mdRender" v-if="!editMsgMode" :markdown-text="message.content">
                     </MarkdownRenderer>
                     <textarea v-if="editMsgMode" ref="mdTextarea" :rows="4"
@@ -163,12 +166,14 @@ const bUrl = import.meta.env.VITE_GPT4ALL_API_BASEURL
 import { nextTick } from 'vue'
 import feather from 'feather-icons'
 import MarkdownRenderer from './MarkdownRenderer.vue';
+import Step from './Step.vue';
 export default {
     // eslint-disable-next-line vue/multi-word-component-names
     name: 'Message',
     emits: ['copy', 'delete', 'rankUp', 'rankDown', 'updateMessage', 'resendMessage'],
     components: {
-        MarkdownRenderer
+        MarkdownRenderer,
+        Step
     },
     props: {
         message: Object,
@@ -177,7 +182,7 @@ export default {
     data() {
         return {
             expanded: false,
-
+            steps: [],
             new_message_content: '',
             showConfirmation: false,
             editMsgMode: false,
@@ -186,7 +191,8 @@ export default {
 
         }
     }, mounted() {
-
+        console.log("Mounted message")
+        console.log(this.message)
         this.new_message_content = this.message.content
         nextTick(() => {
             feather.replace()
diff --git a/web/src/components/PersonalityEntry.vue b/web/src/components/PersonalityEntry.vue
index 1d53cd2a..068d1927 100644
--- a/web/src/components/PersonalityEntry.vue
+++ b/web/src/components/PersonalityEntry.vue
@@ -22,7 +22,7 @@
 
           <span class="sr-only">Settings</span>
         </button>
-        <button v-if="personality.installed" title="Click to Reinstall personality" type="button"  @click.stop="toggleReinstall"
+        <button v-if="selected_computed" title="Click to Reinstall personality" type="button"  @click.stop="toggleReinstall"
           class="inline-flex items-center gap-2 px-3 py-2 text-xs font-medium text-center focus:outline-none text-white bg-red-700 hover:bg-red-800 focus:ring-4 focus:ring-red-300  rounded-lg  dark:bg-red-600 dark:hover:bg-red-700 dark:focus:ring-red-900">
           Reinstall personality
 
diff --git a/web/src/views/DiscussionsView.vue b/web/src/views/DiscussionsView.vue
index 7b907ec7..f4ac16ca 100644
--- a/web/src/views/DiscussionsView.vue
+++ b/web/src/views/DiscussionsView.vue
@@ -730,9 +730,14 @@ export default {
                 this.isGenerating = true;
                 const index = this.discussionArr.findIndex((x) => x.parent == parent && x.id == msgObj.ai_message_id)
                 const messageItem = this.discussionArr[index]
-                if (messageItem) {
+                if (messageItem && msgObj.type<2) {
                     messageItem.content = msgObj.data
                 }
+                else{
+                    if (msgObj.type == 1){
+                        messageItem.steps
+                    }
+                }
                 // // Disables as per request
                 // nextTick(() => {
                 //     const msgList = document.getElementById('messages-list')
diff --git a/web/src/views/SettingsView.vue b/web/src/views/SettingsView.vue
index 0aeec6fa..3eb4ab24 100644
--- a/web/src/views/SettingsView.vue
+++ b/web/src/views/SettingsView.vue
@@ -1019,7 +1019,7 @@
                                         :key="'index-' + index + '-' + pers.name" :personality="pers"
                                         :full_path="pers.full_path"
                                         :selected="configFile.active_personality_id == configFile.personalities.findIndex(item => item === pers.full_path)"
-                                        :on-selected="onPersonalitySelected" :on-mounted="onPersonalityMounted"
+                                        :on-selected="onPersonalitySelected" :on-mounted="onPersonalityMounted" on-reinstall="onPersonalityReinstall"
                                         :on-settings="onSettingsPersonality" />
                                 </TransitionGroup>
                             </div>
@@ -2647,6 +2647,31 @@ export default {
             })
 
         },
+        onPersonalityReinstall(persItem){
+            console.log('on reinstall ', persItem)
+            this.isLoading = true
+            axios.post('/reinstall_binding', { name: binding_object.binding.folder }).then((res) => {
+
+                if (res) {
+                    this.isLoading = false
+                    console.log('reinstall_binding', res)
+                    if (res.data.status) {
+                        this.$refs.toast.showToast("Reinstalled binding successfully!", 4, true)
+                    } else {
+                        this.$refs.toast.showToast("Could not reinstall binding", 4, false)
+                    }
+                    return res.data;
+                }
+                this.isLoading = false
+            })
+                // eslint-disable-next-line no-unused-vars
+
+                .catch(error => {
+                    this.isLoading = false
+                    this.$refs.toast.showToast("Could not reinstall binding\n" + error.message, 4, false)
+                    return { 'status': false }
+                });
+        },
         onPersonalityMounted(persItem) {
             //this.isLoading = true
             console.log('on sel ', persItem)