From c0de178c7d8fe55da4f09b1389cefa5d5c4e9f38 Mon Sep 17 00:00:00 2001
From: shadowcz007 <chizhiwei007@163.com>
Date: Sat, 11 May 2024 23:58:05 +0800
Subject: [PATCH] add re_start

---
 __init__.py                 | 15 ++++++++-------
 web/javascript/ui_mixlab.js | 17 +++++++++++++++--
 2 files changed, 23 insertions(+), 9 deletions(-)

diff --git a/__init__.py b/__init__.py
index e1b7a55..4be948c 100644
--- a/__init__.py
+++ b/__init__.py
@@ -737,13 +737,14 @@ async def my_hander_method(request):
 
     return web.json_response(result)
 
-# parser.add_argument("--llm-model", type=str, help="Path to Model file (gguf). Run the Local LLM by MixlabNodes")
-# if args.llm_model and os.path.exists(args.llm_model):
-#     start_local_llm({
-#         "model_path":args.llm_model
-#     })
-#     print("Local LLM Start")
-
+# 重启服务
+@routes.post('/mixlab/re_start')
+def re_start(request):
+    try:
+        sys.stdout.close_log()
+    except Exception as e:
+        pass
+    return os.execv(sys.executable, [sys.executable] + sys.argv)
 
 
 
diff --git a/web/javascript/ui_mixlab.js b/web/javascript/ui_mixlab.js
index 5a0dbb4..6911622 100644
--- a/web/javascript/ui_mixlab.js
+++ b/web/javascript/ui_mixlab.js
@@ -824,7 +824,6 @@ function createModelsModal (models) {
 
   
 
-
   const title = document.createElement('p')
   title.innerText='Models';
   title.style=`font-size: 18px;
@@ -841,7 +840,15 @@ function createModelsModal (models) {
 
   headTitleElement.appendChild(left_d) 
 
-  headTitleElement.appendChild(n_gpu_div) 
+  headTitleElement.appendChild(n_gpu_div)
+
+  //重启
+  const reStart = document.createElement('small')
+  reStart.textContent = '重启'
+  reStart.style.padding = '4px'
+
+  headTitleElement.appendChild(reStart)
+
   
   if (localStorage.getItem('_mixlab_auto_llama_open')) {
     linkIcon.style.backgroundColor = '#66ff6c'
@@ -860,6 +867,12 @@ function createModelsModal (models) {
     }
   })
 
+  reStart.addEventListener('click', e => {
+    e.stopPropagation()
+    fetch('mixlab/re_start')
+  })
+
+
   n_gpu.addEventListener('click', e => {
     e.stopPropagation()
     localStorage.setItem('_mixlab_llama_n_gpu', n_gpu.value)