Upload 4 files

mods/btn_com.py

@@ -101,8 +101,10 @@ def init(cfg):
                 if cfg['session_active'] != finish:
                     raise RuntimeError('任务中断！请稍等或Reset，如已Reset，请忽略。')
                 cfg['session_active'] = not cfg['session_active']
+                yield tmp
+                if finish and cfg['btn_stop_status']:
+                    raise RuntimeError('Stop或Reset被按下，任务已中断！如非您所为，可能他人正在使用中！')
                 cfg['btn_stop_status'] = finish
-                return tmp
 
         return _inner
 

mods/btn_submit.py

@@ -44,6 +44,8 @@ def init(cfg):
             model.eval_t(t_msg, _n_keep, _n_discard)
             yield history, model.venv_info
             # ========== 模型输出 ==========
+            if cfg['btn_stop_status']:
+                return
             model.venv_create('char')
             _tmp = btn_com(_n_keep, _n_discard,
                            _temperature, _repeat_penalty, _frequency_penalty,

mods/btn_suggest.py

@@ -16,6 +16,9 @@ def init(cfg):
         with lock:
             if not cfg['session_active']:
                 raise RuntimeError
+            if cfg['btn_stop_status']:
+                yield '', model.venv_info
+                return
             # ========== 模型输出建议 ==========
             model.venv_create('suggest')  # 创建隔离环境
             _tmp = btn_com(_n_keep, _n_discard,

mods/btn_vo.py

@@ -16,6 +16,9 @@ def init(cfg):
         with lock:
             if not cfg['session_active']:
                 raise RuntimeError
+            if cfg['btn_stop_status']:
+                yield '', model.venv_info
+                return
             # ========== 及时清理上一次生成的旁白 ==========
             model.venv_remove('vo')
             print('清理旁白', model.venv_info)