cleanup

AIRI-AUBE · Mar 11, 2023 · 547cc76 · 547cc76
1 parent ed2b673
commit 547cc76
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 4 deletions.
diff --git a/javascript/ui.js b/javascript/ui.js
@@ -211,3 +211,21 @@ function restart_reload(){
 
     return []
 }
+
+function login(username) {
+    var txt2img=gradioApp().querySelector('#tabs').querySelectorAll('button')[0];
+    var img2img=gradioApp().querySelector('#tabs').querySelectorAll('button')[1];
+    var extras=gradioApp().querySelector('#tabs').querySelectorAll('button')[2];
+    var pnginfo=gradioApp().querySelector('#tabs').querySelectorAll('button')[3];
+    var train=gradioApp().querySelector('#tabs').querySelectorAll('button')[4];
+    var user=gradioApp().querySelector('#tabs').querySelectorAll('button')[5];
+    var extension=gradioApp().querySelector('#tabs').querySelectorAll('button')[6];
+
+    txt2img.style.display=(username=='admin'?'none':'block')
+    img2img.style.display=(username=='admin'?'none':'block')
+    extras.style.display=(username=='admin'?'none':'block')
+    pnginfo.style.display=(username=='admin'?'none':'block')
+    train.style.display=(username=='admin'?'none':'block')
+    user.style.display=(username=='admin'?'block':'none')
+    extension.style.display=(username=='admin'?'none':'block')
+}
diff --git a/localizations/zh_CN.json b/localizations/zh_CN.json
@@ -642,6 +642,8 @@
     "Instance count": "实例数量",
     "Submit training job sucessful": "训练任务提交成功",
     "Settings saved failed": "设置保存错误",
+    "Settings changed without save": "设置变更但未保存",
+    "Settings changed and saved": "设置变更并保存",
     "SageMaker endpoint": "SageMaker 端点",
     "User": "用户",
     "Model": "模型",

diff --git a/localizations/zh_TW.json b/localizations/zh_TW.json
@@ -636,6 +636,8 @@
     "Instance count": "實例數量",
     "Submit training job sucessful": "訓練任務提交成功",
     "Settings saved failed": "設置保存錯誤",
+    "Settings changed without save": "設置變更但未保存",
+    "Settings changed and saved": "設置變更並保存",
     "SageMaker endpoint": "SageMaker 端點",
     "User": "使用者",
     "Model": "模型",

diff --git a/modules/ui.py b/modules/ui.py
@@ -795,8 +795,8 @@ def run_settings(username, *args):
             if response.status_code != 200:
                 raise RuntimeError("Settings saved failed")
         except RuntimeError:
-            return opts.dumpjson(), f'{len(changed)} settings changed without save: {", ".join(changed)}.'
-        return opts.dumpjson(), f'{len(changed)} settings changed: {", ".join(changed)}.'
+            return opts.dumpjson(), 'Settings changed without save'
+        return opts.dumpjson(), 'Settings changed and saved'
 
     def run_settings_single(value, key, request : gr.Request):
         tokens = shared.demo.server_app.tokens
@@ -1978,7 +1978,12 @@ def sagemaker_train_hypernetwork(
 
     with gr.Blocks(analytics_enabled=False) as user_interface:
         username_state = gr.Textbox(value="", elem_id="username", visible=False)
-        username_state.change(fn=None, inputs=[username_state], outputs=None, _js="function(username){var x=gradioApp().querySelector('#tabs').querySelectorAll('button')[5];x.style.display=(username=='admin'?'block':'none')}")
+        username_state.change(
+            fn=None,
+            inputs=[username_state],
+            outputs=None,
+            _js="login"
+        )
 
         user_dataframe = gr.Dataframe(
             headers=["Username", "Password", "Options"],
@@ -2042,7 +2047,6 @@ def save_userdata(user_dataframe, request: gr.Request):
                 'action': 'save',
                 'items': items
             }
-            print(inputs)
             response = requests.post(url=f'{shared.api_endpoint}/sd/user', json=inputs)
             print(response.text)
             if response.status_code == 200: