mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
use separate module for translator
This commit is contained in:
@@ -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'Translated {element}: {result}')
|
||||||
|
return result
|
||||||
|
except Exception as e:
|
||||||
|
print(f'Error during translation of {element}: {e}')
|
||||||
|
return text
|
||||||
@@ -14,26 +14,12 @@ import modules.advanced_parameters as advanced_parameters
|
|||||||
import modules.style_sorter as style_sorter
|
import modules.style_sorter as style_sorter
|
||||||
import args_manager
|
import args_manager
|
||||||
import copy
|
import copy
|
||||||
import translators
|
|
||||||
|
|
||||||
from modules.sdxl_styles import legal_style_names
|
from modules.sdxl_styles import legal_style_names
|
||||||
from modules.private_logger import get_current_html_path
|
from modules.private_logger import get_current_html_path
|
||||||
from modules.ui_gradio_extensions import reload_javascript
|
from modules.ui_gradio_extensions import reload_javascript
|
||||||
from modules.auth import auth_enabled, check_auth
|
from modules.auth import auth_enabled, check_auth
|
||||||
from functools import lru_cache
|
from modules.translator import translate2en
|
||||||
|
|
||||||
@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'Translated {element}: {result}')
|
|
||||||
return result
|
|
||||||
except Exception as e:
|
|
||||||
print(f'Error during translation of {element}: {e}')
|
|
||||||
return text
|
|
||||||
|
|
||||||
def generate_clicked(*args):
|
def generate_clicked(*args):
|
||||||
import fcbh.model_management as model_management
|
import fcbh.model_management as model_management
|
||||||
@@ -47,6 +33,7 @@ def generate_clicked(*args):
|
|||||||
|
|
||||||
args = list(args)
|
args = list(args)
|
||||||
|
|
||||||
|
# translate prompts
|
||||||
if args[2]:
|
if args[2]:
|
||||||
args[0] = translate2en(args[0], 'prompt')
|
args[0] = translate2en(args[0], 'prompt')
|
||||||
args[1] = translate2en(args[1], 'negative prompt')
|
args[1] = translate2en(args[1], 'negative prompt')
|
||||||
|
|||||||
Reference in New Issue
Block a user