Merge branch 'feature/add-prompt-translate'

# Conflicts:
#	requirements_versions.txt
#	webui.py

modules/async_worker.py

@@ -44,6 +44,7 @@ def worker():
     from modules.util import remove_empty_str, HWC3, resize_image, \
         get_image_shape_ceil, set_image_shape_ceil, get_shape_ceil, resample_image
     from modules.upscaler import perform_upscale
+    from modules.translator import translate2en
 
     try:
         async_gradio_app = shared.gradio_root
@@ -128,6 +129,7 @@ def worker():
 
         prompt = args.pop()
         negative_prompt = args.pop()
+        translate_prompts = args.pop()
         style_selections = args.pop()
         performance_selection = args.pop()
         aspect_ratios_selection = args.pop()
@@ -203,6 +205,10 @@ def worker():
             modules.patch.adm_scaler_end = advanced_parameters.adm_scaler_end = 0.0
             steps = 8
 
+        if translate_prompts:
+            prompt = translate2en(prompt, 'prompt')
+            negative_prompt = translate2en(negative_prompt, 'negative prompt')
+
         modules.patch.adaptive_cfg = advanced_parameters.adaptive_cfg
         print(f'[Parameters] Adaptive CFG = {modules.patch.adaptive_cfg}')
 

modules/translator.py

@@ -0,0 +1,15 @@
+import translators
+from functools import lru_cache
+
+@lru_cache(maxsize=32, typed=False)
+def translate2en(text, element):
+    if not text:
+        return text
+
+    try:
+        result = translators.translate_text(text,to_language='en')
+        print(f'[Parameters] Translated {element}: {result}')
+        return result
+    except Exception as e:
+        print(f'[Parameters] Error during translation of {element}: {e}')
+        return text