From ca66875180190db280c40e5cdbcea100a0ad9c73 Mon Sep 17 00:00:00 2001
From: saloui <saifeddine.aloui@cea.fr>
Date: Fri, 9 Jun 2023 16:51:27 +0200
Subject: [PATCH] upgraded unmount personalities

---
 app.py                                         | 8 ++++++--
 tests/end_point_tests/unmount_personality.http | 2 +-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/app.py b/app.py
index 67f31fa3..a053aab9 100644
--- a/app.py
+++ b/app.py
@@ -728,8 +728,12 @@ class LoLLMsWebUI(LoLLMsAPPI):
             self.config["personalities"].remove(f"{language}/{category}/{name}")
             if self.config["active_personality_id"]>=index:
                 self.config["active_personality_id"]=0
-            self.personalities = self.process.rebuild_personalities()
-            self.personality = self.personalities[self.config["active_personality_id"]]
+            if len(self.config["personalities"])>0:
+                self.personalities = self.process.rebuild_personalities()
+                self.personality = self.personalities[self.config["active_personality_id"]]
+            else:
+                self.personalities = []
+                self.personality = None
             self.apply_settings()
             return jsonify({
                         "status": True,
diff --git a/tests/end_point_tests/unmount_personality.http b/tests/end_point_tests/unmount_personality.http
index ca9c03aa..a69e8eb2 100644
--- a/tests/end_point_tests/unmount_personality.http
+++ b/tests/end_point_tests/unmount_personality.http
@@ -4,5 +4,5 @@ Content-Type: application/json
 {
     "language": "english",
     "category": "generic",
-    "name": "lollms"
+    "name": "gpt4all"
 }