add ability to create extensions that add localizations

4 files changed, 7 insertions, 5 deletions

diff --git a/modules/localization.py b/modules/localization.py
index b1810cda..f6a6f2fb 100644
--- a/modules/localization.py
+++ b/modules/localization.py
@@ -3,6 +3,7 @@ import os
 import sys

 import traceback

 

+

 localizations = {}

 

 

@@ -16,6 +17,11 @@ def list_localizations(dirname):
 

         localizations[fn] = os.path.join(dirname, file)

 

+    from modules import scripts

+    for file in scripts.list_scripts("localizations", ".json"):

+        fn, ext = os.path.splitext(file.filename)

+        localizations[fn] = file.path

+

 

 def localization_js(current_localization_name):

     fn = localizations.get(current_localization_name, None)

diff --git a/modules/scripts.py b/modules/scripts.py
index 366c90d7..637b2329 100644
--- a/modules/scripts.py
+++ b/modules/scripts.py
@@ -3,7 +3,6 @@ import sys
 import traceback

 from collections import namedtuple

 

-import modules.ui as ui

 import gradio as gr

 

 from modules.processing import StableDiffusionProcessing

diff --git a/modules/shared.py b/modules/shared.py
index 70b998ff..e8bacd3c 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -221,8 +221,6 @@ interrogator = modules.interrogate.InterrogateModels("interrogate")
 

 face_restorers = []

 

-localization.list_localizations(cmd_opts.localizations_dir)

-

 

 def realesrgan_models_names():

     import modules.realesrgan_model

diff --git a/modules/ui.py b/modules/ui.py
index 76ca9b07..23643c22 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1563,11 +1563,10 @@ def create_ui(wrap_gradio_gpu_call):
             shared.state.need_restart = True

 

         restart_gradio.click(

-

             fn=request_restart,

+            _js='restart_reload',

             inputs=[],

             outputs=[],

-            _js='restart_reload'

         )

 

         if column is not None: